puter/paseo
1
0
Fork 0
mirror of https://github.com/getpaseo/paseo.git synced 2026-07-29 12:01:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' of github.com:getpaseo/paseo

Browse Source
This commit is contained in:
Mohamed Boudra
2026-06-20 22:29:28 +07:00
parent 7f74853174 c2d7796b78
commit 5180708e26
2 changed files with 24 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
packages/server/src/services/quota-fetcher/providers/kimi.ts
View File

@@ -24,6 +24,7 @@ const KimiAuthSchema = z.object({
interface KimiQuotaProviderOptions {
logger: Logger;
fetch?: ProviderApiFetch;
homeDir?: string;
}
export class KimiQuotaProvider implements ProviderUsageFetcher {
@@ -32,10 +33,12 @@ export class KimiQuotaProvider implements ProviderUsageFetcher {
private readonly logger: Logger;
private readonly fetchApi: ProviderApiFetch;
private readonly homeDir?: string;
constructor(options: KimiQuotaProviderOptions) {
this.logger = options.logger;
this.fetchApi = options.fetch ?? fetch;
this.homeDir = options.homeDir;
}
async fetchUsage(): Promise<ProviderUsage> {
@@ -88,13 +91,14 @@ export class KimiQuotaProvider implements ProviderUsageFetcher {
}
private async readKimiToken(): Promise<string | null> {
const homeDir = this.homeDir ?? homedir();
const paths = [
join(
process.env["KIMI_CODE_HOME"] || join(homedir(), ".kimi-code"),
process.env["KIMI_CODE_HOME"] || join(homeDir, ".kimi-code"),
"credentials",
"kimi-code.json",
),
join(homedir(), ".kimi", "credentials", "kimi-code.json"),
join(homeDir, ".kimi", "credentials", "kimi-code.json"),
];
for (const path of paths) {

26
packages/server/src/services/quota-fetcher/service.test.ts
View File

@@ -335,7 +335,11 @@ describe("real provider usage fetchers", () => {
});
function service(
options: { platform?: typeof process.platform; keychain?: () => Promise<unknown | null> } = {},
options: {
platform?: typeof process.platform;
keychain?: () => Promise<unknown | null>;
kimiHomeDir?: string;
} = {},
) {
const logger = createLogger();
const fetchThroughTestDouble = ((url: RequestInfo | URL, init?: RequestInit) =>
@@ -356,7 +360,11 @@ describe("real provider usage fetchers", () => {
new CursorQuotaProvider({ logger, fetch: fetchThroughTestDouble }),
new ZaiQuotaProvider({ logger, fetch: fetchThroughTestDouble }),
new GrokQuotaProvider({ logger, fetch: fetchThroughTestDouble }),
new KimiQuotaProvider({ logger, fetch: fetchThroughTestDouble }),
new KimiQuotaProvider({
logger,
fetch: fetchThroughTestDouble,
homeDir: options.kimiHomeDir,
}),
],
cacheTtlMs: 0,
});
@@ -712,11 +720,11 @@ describe("real provider usage fetchers", () => {
it("fetches Kimi usage from the CLI credential home", async () => {
writeKimiCredentials(join(homeDir, ".kimi-code"), "kimi_cli_token");
fetchApi = vi.fn(async (url: RequestInfo | URL, init?: RequestInit) => {
expect(url.toString()).toBe("https://api.kimi.com/coding/v1/usages");
expect((init?.headers as Record<string, string> | undefined)?.Authorization).toBe(
"Bearer kimi_cli_token",
);
let requestedUrl: string | null = null;
let authorization: string | null = null;
fetchApi = (async (url: RequestInfo | URL, init?: RequestInit) => {
requestedUrl = url.toString();
authorization = (init?.headers as Record<string, string> | undefined)?.Authorization ?? null;
return jsonResponse({
usage: {
limit: "200",
@@ -726,8 +734,10 @@ describe("real provider usage fetchers", () => {
});
}) as unknown as typeof fetch;
const kimi = findProvider(await service().listUsage(), "kimi");
const kimi = findProvider(await service({ kimiHomeDir: homeDir }).listUsage(), "kimi");
expect(requestedUrl).toBe("https://api.kimi.com/coding/v1/usages");
expect(authorization).toBe("Bearer kimi_cli_token");
expect(kimi).toMatchObject({
status: "available",
windows: [