mirror of
https://github.com/Nezumi-2711/9router.git
synced 2026-09-22 13:38:31 +00:00
feat(usage): add Kimi and DeepSeek usage handlers
Wire /v1/usages for Kimi (OAuth + API key) and balance API for DeepSeek, flag both providers with usage/usageApikey, and normalize their quotas in the dashboard ProviderLimits parser. Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
This commit is contained in:
decolua
co-authored by
Claude Fable 5
Claude Fable 5
parent
65ac9b3cec
commit
6eaa9f8369
@@ -0,0 +1,149 @@
|
||||
import { describe, it, expect, vi, beforeEach } from "vitest";
|
||||
|
||||
vi.mock("../../open-sse/utils/proxyFetch.js", () => ({
|
||||
proxyAwareFetch: vi.fn(),
|
||||
}));
|
||||
|
||||
import { proxyAwareFetch } from "../../open-sse/utils/proxyFetch.js";
|
||||
import { getUsageForProvider } from "../../open-sse/services/usage.js";
|
||||
import {
|
||||
USAGE_SUPPORTED_PROVIDERS,
|
||||
USAGE_APIKEY_PROVIDERS,
|
||||
} from "../../src/shared/constants/providers.js";
|
||||
import { parseQuotaData } from "../../src/app/(dashboard)/dashboard/usage/components/ProviderLimits/utils.js";
|
||||
|
||||
const BALANCE_URL = "https://api.deepseek.com/user/balance";
|
||||
|
||||
function jsonResponse(body, status = 200) {
|
||||
return new Response(JSON.stringify(body), {
|
||||
status,
|
||||
headers: { "Content-Type": "application/json" },
|
||||
});
|
||||
}
|
||||
|
||||
const ACTIVE_BALANCE = {
|
||||
is_available: true,
|
||||
balance_infos: [
|
||||
{
|
||||
currency: "USD",
|
||||
total_balance: "12.50",
|
||||
granted_balance: "2.50",
|
||||
topped_up_balance: "10.00",
|
||||
},
|
||||
{
|
||||
currency: "CNY",
|
||||
total_balance: "0.00",
|
||||
granted_balance: "0.00",
|
||||
topped_up_balance: "0.00",
|
||||
},
|
||||
],
|
||||
};
|
||||
|
||||
describe("deepseek registry usage flags", () => {
|
||||
it("is listed for apikey quota dashboard", () => {
|
||||
expect(USAGE_SUPPORTED_PROVIDERS).toContain("deepseek");
|
||||
expect(USAGE_APIKEY_PROVIDERS).toContain("deepseek");
|
||||
});
|
||||
});
|
||||
|
||||
describe("getUsageForProvider(deepseek)", () => {
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
it("GETs /user/balance with Bearer apiKey", async () => {
|
||||
proxyAwareFetch.mockResolvedValueOnce(jsonResponse(ACTIVE_BALANCE));
|
||||
|
||||
const usage = await getUsageForProvider({
|
||||
provider: "deepseek",
|
||||
apiKey: "sk-ds-test",
|
||||
});
|
||||
|
||||
expect(usage.message).toBeUndefined();
|
||||
expect(usage.plan).toBe("DeepSeek");
|
||||
expect(proxyAwareFetch).toHaveBeenCalledTimes(1);
|
||||
const [url, opts] = proxyAwareFetch.mock.calls[0];
|
||||
expect(url).toBe(BALANCE_URL);
|
||||
expect(opts.method).toBe("GET");
|
||||
expect(opts.headers.Authorization).toBe("Bearer sk-ds-test");
|
||||
});
|
||||
|
||||
it("maps balances without absolute remaining (UI treats remaining as %)", async () => {
|
||||
proxyAwareFetch.mockResolvedValueOnce(jsonResponse(ACTIVE_BALANCE));
|
||||
|
||||
const usage = await getUsageForProvider({
|
||||
provider: "deepseek",
|
||||
apiKey: "sk-ds-test",
|
||||
});
|
||||
|
||||
expect(usage.quotas["Balance (USD)"]).toMatchObject({
|
||||
used: 0,
|
||||
total: 12.5,
|
||||
remainingPercentage: 100,
|
||||
});
|
||||
expect(usage.quotas["Balance (USD)"].remaining).toBeUndefined();
|
||||
// Zero CNY still listed so user sees currency row
|
||||
expect(usage.quotas["Balance (CNY)"]).toMatchObject({
|
||||
used: 0,
|
||||
total: 0,
|
||||
remainingPercentage: 0,
|
||||
});
|
||||
});
|
||||
|
||||
it("marks plan unavailable when is_available false", async () => {
|
||||
proxyAwareFetch.mockResolvedValueOnce(
|
||||
jsonResponse({
|
||||
is_available: false,
|
||||
balance_infos: [
|
||||
{
|
||||
currency: "USD",
|
||||
total_balance: "0",
|
||||
granted_balance: "0",
|
||||
topped_up_balance: "0",
|
||||
},
|
||||
],
|
||||
}),
|
||||
);
|
||||
|
||||
const usage = await getUsageForProvider({
|
||||
provider: "deepseek",
|
||||
apiKey: "sk-ds-test",
|
||||
});
|
||||
|
||||
expect(usage.plan).toMatch(/insufficient|unavailable/i);
|
||||
expect(usage.quotas["Balance (USD)"].remainingPercentage).toBe(0);
|
||||
});
|
||||
|
||||
it("returns message on missing key / 401", async () => {
|
||||
const missing = await getUsageForProvider({ provider: "deepseek" });
|
||||
expect(missing.message).toMatch(/api key/i);
|
||||
expect(proxyAwareFetch).not.toHaveBeenCalled();
|
||||
|
||||
proxyAwareFetch.mockResolvedValueOnce(jsonResponse({ error: "no" }, 401));
|
||||
const auth = await getUsageForProvider({
|
||||
provider: "deepseek",
|
||||
apiKey: "bad",
|
||||
});
|
||||
expect(auth.message).toMatch(/auth|key|401/i);
|
||||
});
|
||||
});
|
||||
|
||||
describe("parseQuotaData(deepseek)", () => {
|
||||
it("forwards remainingPercentage for balance rows", () => {
|
||||
const rows = parseQuotaData("deepseek", {
|
||||
plan: "DeepSeek",
|
||||
quotas: {
|
||||
"Balance (USD)": {
|
||||
used: 0,
|
||||
total: 12.5,
|
||||
remainingPercentage: 100,
|
||||
},
|
||||
},
|
||||
});
|
||||
expect(rows[0]).toMatchObject({
|
||||
name: "Balance (USD)",
|
||||
total: 12.5,
|
||||
remainingPercentage: 100,
|
||||
});
|
||||
});
|
||||
});
|
||||
@@ -0,0 +1,299 @@
|
||||
import { describe, it, expect, vi, beforeEach } from "vitest";
|
||||
|
||||
vi.mock("../../open-sse/utils/proxyFetch.js", () => ({
|
||||
proxyAwareFetch: vi.fn(),
|
||||
}));
|
||||
|
||||
import { proxyAwareFetch } from "../../open-sse/utils/proxyFetch.js";
|
||||
import { getUsageForProvider } from "../../open-sse/services/usage.js";
|
||||
import { USAGE_SUPPORTED_PROVIDERS, USAGE_APIKEY_PROVIDERS } from "../../src/shared/constants/providers.js";
|
||||
import { PROVIDERS } from "../../open-sse/providers/index.js";
|
||||
import { parseQuotaData } from "../../src/app/(dashboard)/dashboard/usage/components/ProviderLimits/utils.js";
|
||||
|
||||
const KIMI_USAGE_URL = "https://api.kimi.com/coding/v1/usages";
|
||||
|
||||
function jsonResponse(body, status = 200) {
|
||||
return new Response(JSON.stringify(body), {
|
||||
status,
|
||||
headers: { "Content-Type": "application/json" },
|
||||
});
|
||||
}
|
||||
|
||||
const ACTIVE_USAGE = {
|
||||
user: {
|
||||
membership: { level: "LEVEL_ADVANCED" },
|
||||
},
|
||||
usage: {
|
||||
limit: "100",
|
||||
used: "35",
|
||||
remaining: "65",
|
||||
resetTime: "2026-08-01T00:00:00Z",
|
||||
},
|
||||
limits: [
|
||||
{
|
||||
window: { type: "rate" },
|
||||
detail: {
|
||||
limit: "60",
|
||||
remaining: "40",
|
||||
resetTime: "2026-07-29T12:00:00Z",
|
||||
},
|
||||
},
|
||||
],
|
||||
};
|
||||
|
||||
describe("kimi registry usage flags", () => {
|
||||
it("exposes usage + usageApikey so OAuth and apikey cards appear on /quota", () => {
|
||||
expect(USAGE_SUPPORTED_PROVIDERS).toContain("kimi");
|
||||
expect(USAGE_APIKEY_PROVIDERS).toContain("kimi");
|
||||
});
|
||||
|
||||
it("registers transport.usage url when present (optional)", () => {
|
||||
// Provider may or may not put usage url on transport; handler has its own constant.
|
||||
expect(PROVIDERS.kimi).toBeTruthy();
|
||||
});
|
||||
});
|
||||
|
||||
describe("getUsageForProvider(kimi) auth selection", () => {
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
it("OAuth path: Bearer + X-Msh-* (OmniRoute /usages parity; not chat x-api-key)", async () => {
|
||||
proxyAwareFetch.mockResolvedValueOnce(jsonResponse(ACTIVE_USAGE));
|
||||
|
||||
const usage = await getUsageForProvider({
|
||||
provider: "kimi",
|
||||
accessToken: "tok-abc",
|
||||
providerSpecificData: { deviceId: "stable-device-1" },
|
||||
});
|
||||
|
||||
expect(usage.message).toBeUndefined();
|
||||
expect(usage.plan).toBe("Allegro");
|
||||
expect(usage.quotas.Weekly).toMatchObject({
|
||||
used: 35,
|
||||
total: 100,
|
||||
remainingPercentage: 65,
|
||||
});
|
||||
|
||||
expect(proxyAwareFetch).toHaveBeenCalledTimes(1);
|
||||
const [url, opts] = proxyAwareFetch.mock.calls[0];
|
||||
expect(url).toBe(KIMI_USAGE_URL);
|
||||
expect(opts.method).toBe("GET");
|
||||
expect(opts.headers.Authorization).toBe("Bearer tok-abc");
|
||||
expect(opts.headers["x-api-key"]).toBeUndefined();
|
||||
expect(opts.headers["X-Msh-Platform"]).toBe("9router");
|
||||
expect(opts.headers["X-Msh-Device-Id"]).toBe("stable-device-1");
|
||||
expect(opts.headers["X-Msh-Version"]).toBeTruthy();
|
||||
});
|
||||
|
||||
it("apikey path: x-api-key only (no Bearer / X-Msh)", async () => {
|
||||
proxyAwareFetch.mockResolvedValueOnce(jsonResponse(ACTIVE_USAGE));
|
||||
|
||||
const usage = await getUsageForProvider({
|
||||
provider: "kimi",
|
||||
apiKey: "sk-test-123",
|
||||
});
|
||||
|
||||
expect(usage.message).toBeUndefined();
|
||||
expect(usage.quotas.Weekly.used).toBe(35);
|
||||
|
||||
const [, opts] = proxyAwareFetch.mock.calls[0];
|
||||
expect(opts.headers["x-api-key"]).toBe("sk-test-123");
|
||||
expect(opts.headers.Authorization).toBeUndefined();
|
||||
expect(opts.headers["X-Msh-Platform"]).toBeUndefined();
|
||||
});
|
||||
|
||||
it("prefers apiKey over accessToken when both present", async () => {
|
||||
proxyAwareFetch.mockResolvedValueOnce(jsonResponse(ACTIVE_USAGE));
|
||||
|
||||
await getUsageForProvider({
|
||||
provider: "kimi",
|
||||
accessToken: "tok-abc",
|
||||
apiKey: "sk-prefer-me",
|
||||
});
|
||||
|
||||
const [, opts] = proxyAwareFetch.mock.calls[0];
|
||||
expect(opts.headers["x-api-key"]).toBe("sk-prefer-me");
|
||||
expect(opts.headers.Authorization).toBeUndefined();
|
||||
expect(opts.headers["X-Msh-Platform"]).toBeUndefined();
|
||||
});
|
||||
|
||||
it("maps membership levels to plan display names", async () => {
|
||||
for (const [level, plan] of [
|
||||
["LEVEL_BASIC", "Moderato"],
|
||||
["LEVEL_INTERMEDIATE", "Allegretto"],
|
||||
["LEVEL_ADVANCED", "Allegro"],
|
||||
["LEVEL_STANDARD", "Vivace"],
|
||||
]) {
|
||||
proxyAwareFetch.mockResolvedValueOnce(
|
||||
jsonResponse({
|
||||
user: { membership: { level } },
|
||||
usage: { limit: "10", used: "1", remaining: "9" },
|
||||
}),
|
||||
);
|
||||
const usage = await getUsageForProvider({
|
||||
provider: "kimi",
|
||||
accessToken: "t",
|
||||
});
|
||||
expect(usage.plan).toBe(plan);
|
||||
}
|
||||
});
|
||||
|
||||
it("parses Weekly + Ratelimit; does not put absolute remaining on quota rows", async () => {
|
||||
proxyAwareFetch.mockResolvedValueOnce(jsonResponse(ACTIVE_USAGE));
|
||||
|
||||
const usage = await getUsageForProvider({
|
||||
provider: "kimi",
|
||||
accessToken: "tok",
|
||||
});
|
||||
|
||||
// Absolute remaining would break getRemainingPercentage (treats it as 0-100 %)
|
||||
expect(usage.quotas.Weekly.remaining).toBeUndefined();
|
||||
expect(usage.quotas.Weekly.remainingPercentage).toBe(65);
|
||||
expect(usage.quotas.Ratelimit).toMatchObject({
|
||||
used: 20,
|
||||
total: 60,
|
||||
remainingPercentage: expect.closeTo(40 / 60 * 100, 5),
|
||||
});
|
||||
expect(usage.quotas.Ratelimit.remaining).toBeUndefined();
|
||||
});
|
||||
|
||||
it("surfaces re-authorize message only on 401 unauthenticated", async () => {
|
||||
proxyAwareFetch.mockResolvedValueOnce(
|
||||
jsonResponse(
|
||||
{
|
||||
code: "unauthenticated",
|
||||
details: [
|
||||
{
|
||||
debug: {
|
||||
reason: "REASON_INVALID_AUTH_TOKEN",
|
||||
localizedMessage: { message: "Invalid auth token" },
|
||||
},
|
||||
},
|
||||
],
|
||||
},
|
||||
401,
|
||||
),
|
||||
);
|
||||
|
||||
const usage = await getUsageForProvider({
|
||||
provider: "kimi",
|
||||
accessToken: "expired",
|
||||
});
|
||||
|
||||
expect(usage.message).toMatch(/expired|re-authorize/i);
|
||||
expect(usage.message).not.toMatch(/subscribe|permission/i);
|
||||
expect(usage.quotas).toBeUndefined();
|
||||
});
|
||||
|
||||
it("maps 403 REASON_FEATURE_NO_PERMISSION to subscribe message (not expired)", async () => {
|
||||
// Live capture: valid OAuth JWT still returns 403 permission_denied when
|
||||
// the account has no Kimi Code usage entitlement.
|
||||
proxyAwareFetch.mockResolvedValueOnce(
|
||||
jsonResponse(
|
||||
{
|
||||
code: "permission_denied",
|
||||
details: [
|
||||
{
|
||||
type: "common.error.v1.ErrorDetail",
|
||||
debug: {
|
||||
reason: "REASON_FEATURE_NO_PERMISSION",
|
||||
localizedMessage: {
|
||||
locale: "en-US",
|
||||
message:
|
||||
"You do not have permission to use this feature. Please subscribe to access.",
|
||||
},
|
||||
},
|
||||
},
|
||||
],
|
||||
},
|
||||
403,
|
||||
),
|
||||
);
|
||||
|
||||
const usage = await getUsageForProvider({
|
||||
provider: "kimi",
|
||||
accessToken: "valid-but-no-sub",
|
||||
providerSpecificData: { deviceId: "stable-device-1" },
|
||||
});
|
||||
|
||||
expect(usage.message).toMatch(/permission|subscribe/i);
|
||||
expect(usage.message).not.toMatch(/expired|re-authorize/i);
|
||||
// Must not trip usage-route AUTH_EXPIRED_PATTERNS force-refresh loop
|
||||
expect(usage.message.toLowerCase()).not.toMatch(/expired|re-authorize|unauthorized|401/);
|
||||
});
|
||||
|
||||
it("formatKimiUsageError distinguishes 401 vs 403 feature gate", async () => {
|
||||
const { formatKimiUsageError } = await import(
|
||||
"../../open-sse/services/usage/kimi.js"
|
||||
);
|
||||
expect(formatKimiUsageError(401, '{"code":"unauthenticated"}')).toMatch(
|
||||
/expired|re-authorize/i,
|
||||
);
|
||||
expect(
|
||||
formatKimiUsageError(
|
||||
403,
|
||||
JSON.stringify({
|
||||
code: "permission_denied",
|
||||
details: [
|
||||
{
|
||||
debug: {
|
||||
reason: "REASON_FEATURE_NO_PERMISSION",
|
||||
localizedMessage: {
|
||||
message: "You do not have permission to use this feature.",
|
||||
},
|
||||
},
|
||||
},
|
||||
],
|
||||
}),
|
||||
),
|
||||
).toMatch(/permission|subscribe/i);
|
||||
});
|
||||
|
||||
it("returns tracked-per-request message when usage limit missing", async () => {
|
||||
proxyAwareFetch.mockResolvedValueOnce(
|
||||
jsonResponse({
|
||||
user: { membership: { level: "LEVEL_BASIC" } },
|
||||
usage: {},
|
||||
}),
|
||||
);
|
||||
|
||||
const usage = await getUsageForProvider({
|
||||
provider: "kimi",
|
||||
accessToken: "tok",
|
||||
});
|
||||
|
||||
expect(usage.plan).toBe("Moderato");
|
||||
expect(usage.message).toMatch(/tracked per request/i);
|
||||
});
|
||||
|
||||
it("returns missing-credentials message when neither token nor key", async () => {
|
||||
const usage = await getUsageForProvider({ provider: "kimi" });
|
||||
expect(usage.message).toMatch(/token|key|credential/i);
|
||||
expect(proxyAwareFetch).not.toHaveBeenCalled();
|
||||
});
|
||||
});
|
||||
|
||||
describe("parseQuotaData(kimi)", () => {
|
||||
it("forwards remainingPercentage for dashboard bars", () => {
|
||||
const rows = parseQuotaData("kimi", {
|
||||
plan: "Allegro",
|
||||
quotas: {
|
||||
Weekly: {
|
||||
used: 35,
|
||||
total: 100,
|
||||
remainingPercentage: 65,
|
||||
resetAt: "2026-08-01T00:00:00.000Z",
|
||||
},
|
||||
},
|
||||
});
|
||||
|
||||
expect(rows).toHaveLength(1);
|
||||
expect(rows[0]).toMatchObject({
|
||||
name: "Weekly",
|
||||
used: 35,
|
||||
total: 100,
|
||||
remainingPercentage: 65,
|
||||
});
|
||||
});
|
||||
});
|
||||
@@ -15,7 +15,8 @@ const load = () => import("../../open-sse/services/usage.js");
|
||||
const SUPPORTED = [
|
||||
"github", "gemini-cli", "antigravity", "claude", "codex", "kiro",
|
||||
"qoder", "qwen", "iflow", "ollama", "glm", "glm-cn",
|
||||
"minimax", "minimax-cn", "vercel-ai-gateway", "grok-cli",
|
||||
"minimax", "minimax-cn", "vercel-ai-gateway", "grok-cli", "kimi",
|
||||
"deepseek",
|
||||
];
|
||||
|
||||
describe("usage dispatch", () => {
|
||||
|
||||
Reference in New Issue
Block a user