ZacharyZhang-NY
07d0b4587d
The legacy application - its src, tests, landing site, promo assets, docs, release tooling, CI workflows, and root configs - leaves the tree; it stays reachable in history. Everything under AgentCompany-New moves up unchanged: same package name, so the production userData/database path is untouched, and e2e/build scripts already resolved paths relative to the project. The developer cred.txt stays gitignored at its new location. Full verify:ci passes from the new root.
59 lines
3.0 KiB
TypeScript
59 lines
3.0 KiB
TypeScript
import { randomUUID } from "node:crypto";
|
|
import { tmpdir } from "node:os";
|
|
import { join } from "node:path";
|
|
import { describe, expect, test } from "vitest";
|
|
import { mapAnthropicModels, mapOpenAiModels } from "../../src/main/services/provider-models-fetch";
|
|
import { DatabaseConnection } from "../../src/main/database/connection";
|
|
import { ProviderService } from "../../src/main/services/provider-service";
|
|
import { providerAdapters } from "../../src/main/services/provider-adapters";
|
|
|
|
const testVault = {
|
|
encryptSecret: (secret: string) => Buffer.from(secret, "utf8"),
|
|
decryptSecret: (ciphertext: Buffer) => ciphertext.toString("utf8"),
|
|
};
|
|
|
|
describe("provider model-list mapping", () => {
|
|
test("maps the OpenAI /models shape, deduping by id", () => {
|
|
const models = mapOpenAiModels({ data: [{ id: "gpt-x" }, { id: "gpt-y" }, { id: "gpt-x" }, { id: "" }, { other: 1 }] });
|
|
expect(models).toEqual([{ id: "gpt-x", name: "gpt-x" }, { id: "gpt-y", name: "gpt-y" }]);
|
|
});
|
|
|
|
test("maps the Anthropic /models shape, preferring display_name", () => {
|
|
const models = mapAnthropicModels({ data: [{ id: "claude-x", display_name: "Claude X" }, { id: "claude-y" }] });
|
|
expect(models).toEqual([{ id: "claude-x", name: "Claude X" }, { id: "claude-y", name: "claude-y" }]);
|
|
});
|
|
|
|
test("throws on a malformed response instead of returning a static list", () => {
|
|
expect(() => mapOpenAiModels({})).toThrow(/expected shape/);
|
|
expect(() => mapOpenAiModels({ data: [] })).toThrow(/no models/);
|
|
expect(() => mapAnthropicModels({ data: [{ id: "" }] })).toThrow(/no models/);
|
|
});
|
|
});
|
|
|
|
describe("ProviderService.listModels routing", () => {
|
|
test("routes a typed credential to the matching adapter and returns its live models", async () => {
|
|
const connection = new DatabaseConnection(join(tmpdir(), `ac-models-${randomUUID()}.db`));
|
|
let parsedSecret = "";
|
|
const fakeAdapter = {
|
|
...providerAdapters.openai,
|
|
authType: "api_key" as const,
|
|
parseAuthProfile: (secret: string) => { parsedSecret = secret; return { type: "api_key" as const, apiKey: secret }; },
|
|
listModels: async () => [{ id: "live-1", name: "live-1" }, { id: "live-2", name: "live-2" }],
|
|
};
|
|
const registry = { ...providerAdapters, openai: fakeAdapter };
|
|
const service = new ProviderService(connection.db, registry, testVault);
|
|
|
|
const models = await service.listModels({ providerKey: "openai", secret: "sk-live-key" });
|
|
expect(parsedSecret).toBe("sk-live-key");
|
|
expect(models).toEqual([{ id: "live-1", name: "live-1" }, { id: "live-2", name: "live-2" }]);
|
|
connection.close();
|
|
});
|
|
|
|
test("requires a credential reference when no secret is provided", async () => {
|
|
const connection = new DatabaseConnection(join(tmpdir(), `ac-models-${randomUUID()}.db`));
|
|
const service = new ProviderService(connection.db, providerAdapters, testVault);
|
|
await expect(service.listModels({ providerKey: "openai" })).rejects.toThrow(/credential is required/);
|
|
connection.close();
|
|
});
|
|
});
|