ZacharyZhang-NYandClaude Opus 5.5
5cd413ffcf
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
41 lines
3.0 KiB
TypeScript
41 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 { DatabaseConnection } from "../../src/main/database/connection";
|
|
import { OrganizationService } from "../../src/main/services/organization-service";
|
|
import { ProviderService } from "../../src/main/services/provider-service";
|
|
import { providerAdapters } from "../../src/main/services/provider-adapters";
|
|
import { setProviderModel } from "../../src/main/services/provider-default-model";
|
|
|
|
const vault = { encryptSecret: (secret: string) => Buffer.from(secret, "utf8"), decryptSecret: (ciphertext: Buffer) => ciphertext.toString("utf8") };
|
|
|
|
describe("changing a connected provider's default model", () => {
|
|
test("changes only the model: the credential, its sign-in and its health stay as they were", () => {
|
|
const connection = new DatabaseConnection(join(tmpdir(), `ac-default-model-${randomUUID()}.db`));
|
|
const db = connection.db;
|
|
const providers = new ProviderService(db, providerAdapters, vault);
|
|
let orgId = "";
|
|
connection.transaction(() => {
|
|
orgId = new OrganizationService(db).createOrganization({ organizationName: "Acme", employeeName: "Dana", mission: "m", goal: "g" });
|
|
});
|
|
const idOf = (key: string) => (db.prepare("select id from providers where org_id = ? and provider_key = ?").get(orgId, key) as { id: string }).id;
|
|
const openai = idOf("openai");
|
|
providers.saveCredential({ orgId, providerId: openai, secret: "sk-test", model: "gpt-5.4" });
|
|
db.prepare("update providers set health_state = 'healthy' where id = ?").run(openai);
|
|
const before = db.prepare("select credential_id, auth_state, health_state from providers where id = ?").get(openai);
|
|
|
|
setProviderModel(db, { orgId, providerId: openai, model: " gpt-6 " });
|
|
expect(db.prepare("select model from providers where id = ?").get(openai)).toEqual({ model: "gpt-6" });
|
|
expect(db.prepare("select credential_id, auth_state, health_state from providers where id = ?").get(openai)).toEqual(before);
|
|
expect(db.prepare("select action, payload_json from audit_events where entity_id = ? and action = 'default_model_changed'").get(openai))
|
|
.toEqual({ action: "default_model_changed", payload_json: JSON.stringify({ from: "gpt-5.4", to: "gpt-6" }) });
|
|
|
|
expect(() => setProviderModel(db, { orgId, providerId: openai, model: " " })).toThrow("Name the default model.");
|
|
expect(() => setProviderModel(db, { orgId, providerId: idOf("anthropic"), model: "claude" })).toThrow("Connect this provider first; its default model is set when it connects.");
|
|
db.prepare("update providers set credential_id = (select credential_id from providers where id = ?), auth_state = 'configured' where id = ?").run(openai, idOf("azure-openai"));
|
|
expect(() => setProviderModel(db, { orgId, providerId: idOf("azure-openai"), model: "gpt-6" }))
|
|
.toThrow("Azure OpenAI calls the deployment saved with its credential; change the deployment under Manage.");
|
|
});
|
|
});
|