feat: 新增模型管理功能（供应商 + 模型 CRUD）
- 新增 providers/models 数据库表、迁移和数据访问层 - 新增 15 个后端 API 路由（供应商/模型 CRUD + 连通性测试） - 新增 AI 服务层（registry.ts: buildProviderRegistry + testProviderConnection） - 新增前端模型管理页面（Tabs: 供应商/模型，含表格、表单、工具栏） - 新增前端 hooks（use-providers, use-models） - 新增共享类型和 MODEL_CAPABILITIES 常量 - 新增 10 个测试文件（66 个测试用例，4 个因 bun test ESM 兼容问题待修复） - 更新开发文档（architecture, backend, frontend） - 附带 apply-review 修复：统一错误响应、提取共享常量、清理重复测试 注意：registry.test.ts 中 4 个测试因 bun test 无法解析 createProviderRegistry ESM 导出而失败，详情见 context.md
This commit is contained in:
78
tests/server/ai/registry.test.ts
Normal file
78
tests/server/ai/registry.test.ts
Normal file
@@ -0,0 +1,78 @@
|
||||
import { describe, expect, mock, test } from "bun:test";
|
||||
|
||||
import { createMigratedTestDatabase } from "../../helpers";
|
||||
|
||||
describe("AI registry", () => {
|
||||
test("testProviderConnection rejects invalid config", async () => {
|
||||
void mock.module("ai", () => ({
|
||||
generateText: mock(() => {
|
||||
throw new Error("Connection failed");
|
||||
}),
|
||||
}));
|
||||
|
||||
const { testProviderConnection } = await import("../../../src/server/ai/registry");
|
||||
|
||||
const result = await testProviderConnection({
|
||||
apiKey: "bad-key",
|
||||
baseUrl: "https://0.0.0.0:1",
|
||||
name: "Bad",
|
||||
type: "openai-compatible",
|
||||
});
|
||||
|
||||
expect(result.ok).toBe(false);
|
||||
expect(result.message).toContain("连接失败");
|
||||
expect(typeof result.message).toBe("string");
|
||||
});
|
||||
|
||||
test("testProviderConnection return shape is correct", async () => {
|
||||
void mock.module("ai", () => ({
|
||||
generateText: mock((_opts: unknown) => ({})),
|
||||
}));
|
||||
|
||||
const { testProviderConnection } = await import("../../../src/server/ai/registry");
|
||||
|
||||
const result = await testProviderConnection({
|
||||
apiKey: "sk-test",
|
||||
baseUrl: "https://api.openai.com/v1",
|
||||
name: "Test",
|
||||
type: "openai",
|
||||
});
|
||||
|
||||
expect(result.ok).toBe(true);
|
||||
expect(result.message).toBe("连接成功");
|
||||
});
|
||||
|
||||
test("buildProviderRegistry 从 DB 构建包含启用供应商的注册表", async () => {
|
||||
const handle = createMigratedTestDatabase("registry-build-test");
|
||||
const now = new Date().toISOString();
|
||||
|
||||
handle.db
|
||||
.prepare(
|
||||
"INSERT INTO providers (id, name, type, base_url, api_key, enabled, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?)",
|
||||
)
|
||||
.run("pv1", "OpenAI", "openai", "https://api.openai.com/v1", "sk-test", 1, now, now);
|
||||
handle.db
|
||||
.prepare(
|
||||
"INSERT INTO providers (id, name, type, base_url, api_key, enabled, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?)",
|
||||
)
|
||||
.run("pv2", "Disabled", "anthropic", "https://api.anthropic.com", "sk-off", 0, now, now);
|
||||
|
||||
const { buildProviderRegistry } = await import("../../../src/server/ai/registry");
|
||||
const registry = buildProviderRegistry(handle.db);
|
||||
|
||||
expect(() => registry.languageModel("pv1:gpt-4o")).not.toThrow();
|
||||
|
||||
handle.cleanup();
|
||||
});
|
||||
|
||||
test("buildProviderRegistry 无启用供应商时返回空注册表", async () => {
|
||||
const handle = createMigratedTestDatabase("registry-empty-test");
|
||||
|
||||
const { buildProviderRegistry } = await import("../../../src/server/ai/registry");
|
||||
const registry = buildProviderRegistry(handle.db);
|
||||
|
||||
expect(typeof registry.languageModel).toBe("function");
|
||||
|
||||
handle.cleanup();
|
||||
});
|
||||
});
|
||||
226
tests/server/db/models.test.ts
Normal file
226
tests/server/db/models.test.ts
Normal file
@@ -0,0 +1,226 @@
|
||||
import type Database from "bun:sqlite";
|
||||
|
||||
import { describe, expect, test } from "bun:test";
|
||||
|
||||
import {
|
||||
createModel,
|
||||
deleteModel,
|
||||
disableModel,
|
||||
enableModel,
|
||||
getModel,
|
||||
getModelsByProviderId,
|
||||
listModels,
|
||||
updateModel,
|
||||
} from "../../../src/server/db/models";
|
||||
import { createProvider } from "../../../src/server/db/providers";
|
||||
import { createMigratedTestDatabase } from "../../helpers";
|
||||
|
||||
function seedProvider(db: Database, name = "TestProvider"): string {
|
||||
const result = createProvider(db, { apiKey: "sk-test", baseUrl: "https://api.test.com/v1", name, type: "openai" });
|
||||
return (result as { provider: { id: string } }).provider.id;
|
||||
}
|
||||
|
||||
function withDb(callback: (db: Database) => void): void {
|
||||
const handle = createMigratedTestDatabase("models-test");
|
||||
try {
|
||||
callback(handle.db);
|
||||
handle.close();
|
||||
} finally {
|
||||
handle.cleanup();
|
||||
}
|
||||
}
|
||||
|
||||
describe("模型数据访问层", () => {
|
||||
test("创建模型", () => {
|
||||
withDb((db) => {
|
||||
const providerId = seedProvider(db);
|
||||
const result = createModel(db, {
|
||||
capabilities: ["text", "reasoning"],
|
||||
modelId: "gpt-4o",
|
||||
name: "GPT-4o",
|
||||
providerId,
|
||||
});
|
||||
expect("error" in result).toBe(false);
|
||||
const model = (
|
||||
result as {
|
||||
model: { capabilities: string[]; enabled: boolean; modelId: string; name: string; providerId: string };
|
||||
}
|
||||
).model;
|
||||
expect(model.name).toBe("GPT-4o");
|
||||
expect(model.modelId).toBe("gpt-4o");
|
||||
expect(model.providerId).toBe(providerId);
|
||||
expect(model.capabilities).toEqual(["text", "reasoning"]);
|
||||
expect(model.enabled).toBe(true);
|
||||
});
|
||||
});
|
||||
|
||||
test("供应商不存在时创建失败", () => {
|
||||
withDb((db) => {
|
||||
const result = createModel(db, {
|
||||
capabilities: ["text"],
|
||||
modelId: "test",
|
||||
name: "Test",
|
||||
providerId: "nonexistent",
|
||||
});
|
||||
expect("error" in result).toBe(true);
|
||||
expect((result as unknown as { status: number }).status).toBe(400);
|
||||
});
|
||||
});
|
||||
|
||||
test("同一供应商下模型 ID 唯一", () => {
|
||||
withDb((db) => {
|
||||
const providerId = seedProvider(db);
|
||||
createModel(db, { capabilities: ["text"], modelId: "gpt-4o", name: "Model1", providerId });
|
||||
const result = createModel(db, { capabilities: ["text"], modelId: "gpt-4o", name: "Model2", providerId });
|
||||
expect("error" in result).toBe(true);
|
||||
expect((result as unknown as { error: string }).error).toContain("已存在");
|
||||
});
|
||||
});
|
||||
|
||||
test("不同供应商下模型 ID 可重复", () => {
|
||||
withDb((db) => {
|
||||
const p1 = seedProvider(db, "P1");
|
||||
const p2 = seedProvider(db, "P2");
|
||||
const r1 = createModel(db, { capabilities: ["text"], modelId: "same-id", name: "M1", providerId: p1 });
|
||||
const r2 = createModel(db, { capabilities: ["text"], modelId: "same-id", name: "M2", providerId: p2 });
|
||||
expect("error" in r1).toBe(false);
|
||||
expect("error" in r2).toBe(false);
|
||||
});
|
||||
});
|
||||
|
||||
test("能力标签为空时创建失败", () => {
|
||||
withDb((db) => {
|
||||
const providerId = seedProvider(db);
|
||||
const result = createModel(db, { capabilities: [], modelId: "test", name: "Test", providerId });
|
||||
expect("error" in result).toBe(true);
|
||||
expect((result as unknown as { error: string }).error).toContain("能力标签");
|
||||
});
|
||||
});
|
||||
|
||||
test("列表查询（分页、关键字、供应商过滤）", () => {
|
||||
withDb((db) => {
|
||||
const p1 = seedProvider(db, "P1");
|
||||
const p2 = seedProvider(db, "P2");
|
||||
createModel(db, { capabilities: ["text"], modelId: "m1", name: "Alpha", providerId: p1 });
|
||||
createModel(db, { capabilities: ["text"], modelId: "m2", name: "Beta", providerId: p1 });
|
||||
createModel(db, { capabilities: ["text"], modelId: "m3", name: "Gamma", providerId: p2 });
|
||||
|
||||
const all = listModels(db, { page: 1, pageSize: 20 });
|
||||
expect(all.total).toBe(3);
|
||||
|
||||
const filtered = listModels(db, { page: 1, pageSize: 20, providerId: p1 });
|
||||
expect(filtered.total).toBe(2);
|
||||
|
||||
const searched = listModels(db, { keyword: "Alpha", page: 1, pageSize: 20 });
|
||||
expect(searched.total).toBe(1);
|
||||
});
|
||||
});
|
||||
|
||||
test("获取模型详情", () => {
|
||||
withDb((db) => {
|
||||
const providerId = seedProvider(db);
|
||||
const created = createModel(db, { capabilities: ["text"], modelId: "gpt-4o", name: "GPT-4o", providerId });
|
||||
const id = (created as { model: { id: string } }).model.id;
|
||||
|
||||
const result = getModel(db, id);
|
||||
expect("error" in result).toBe(false);
|
||||
expect((result as { model: { name: string } }).model.name).toBe("GPT-4o");
|
||||
});
|
||||
});
|
||||
|
||||
test("获取不存在的模型返回 404", () => {
|
||||
withDb((db) => {
|
||||
const result = getModel(db, "nonexistent");
|
||||
expect("error" in result).toBe(true);
|
||||
expect((result as unknown as { status: number }).status).toBe(404);
|
||||
});
|
||||
});
|
||||
|
||||
test("更新模型", () => {
|
||||
withDb((db) => {
|
||||
const providerId = seedProvider(db);
|
||||
const created = createModel(db, { capabilities: ["text"], modelId: "gpt-4o", name: "原名", providerId });
|
||||
const id = (created as { model: { id: string } }).model.id;
|
||||
|
||||
const result = updateModel(db, id, { capabilities: ["text", "reasoning"], name: "新名" });
|
||||
expect("error" in result).toBe(false);
|
||||
const updated = (result as { model: { capabilities: string[]; name: string } }).model;
|
||||
expect(updated.name).toBe("新名");
|
||||
expect(updated.capabilities).toEqual(["text", "reasoning"]);
|
||||
});
|
||||
});
|
||||
|
||||
test("启用/禁用模型", () => {
|
||||
withDb((db) => {
|
||||
const providerId = seedProvider(db);
|
||||
const created = createModel(db, { capabilities: ["text"], modelId: "gpt-4o", name: "测试", providerId });
|
||||
const id = (created as { model: { id: string } }).model.id;
|
||||
|
||||
const disabled = disableModel(db, id);
|
||||
expect("error" in disabled).toBe(false);
|
||||
expect((disabled as { model: { enabled: boolean } }).model.enabled).toBe(false);
|
||||
|
||||
const enabled = enableModel(db, id);
|
||||
expect("error" in enabled).toBe(false);
|
||||
expect((enabled as { model: { enabled: boolean } }).model.enabled).toBe(true);
|
||||
});
|
||||
});
|
||||
|
||||
test("重复禁用失败", () => {
|
||||
withDb((db) => {
|
||||
const providerId = seedProvider(db);
|
||||
const created = createModel(db, { capabilities: ["text"], modelId: "gpt-4o", name: "测试", providerId });
|
||||
const id = (created as { model: { id: string } }).model.id;
|
||||
disableModel(db, id);
|
||||
|
||||
const result = disableModel(db, id);
|
||||
expect("error" in result).toBe(true);
|
||||
expect((result as unknown as { status: number }).status).toBe(409);
|
||||
});
|
||||
});
|
||||
|
||||
test("删除模型", () => {
|
||||
withDb((db) => {
|
||||
const providerId = seedProvider(db);
|
||||
const created = createModel(db, { capabilities: ["text"], modelId: "gpt-4o", name: "删除测试", providerId });
|
||||
const id = (created as { model: { id: string } }).model.id;
|
||||
|
||||
const result = deleteModel(db, id);
|
||||
expect("error" in result).toBe(false);
|
||||
|
||||
const after = getModel(db, id);
|
||||
expect("error" in after).toBe(true);
|
||||
});
|
||||
});
|
||||
|
||||
test("getModelsByProviderId 返回正确数量", () => {
|
||||
withDb((db) => {
|
||||
const p1 = seedProvider(db, "P1");
|
||||
const p2 = seedProvider(db, "P2");
|
||||
createModel(db, { capabilities: ["text"], modelId: "m1", name: "M1", providerId: p1 });
|
||||
createModel(db, { capabilities: ["text"], modelId: "m2", name: "M2", providerId: p1 });
|
||||
createModel(db, { capabilities: ["text"], modelId: "m3", name: "M3", providerId: p2 });
|
||||
|
||||
expect(getModelsByProviderId(db, p1)).toBe(2);
|
||||
expect(getModelsByProviderId(db, p2)).toBe(1);
|
||||
});
|
||||
});
|
||||
|
||||
test("可选字段 contextLength 和 maxOutputTokens", () => {
|
||||
withDb((db) => {
|
||||
const providerId = seedProvider(db);
|
||||
const result = createModel(db, {
|
||||
capabilities: ["text"],
|
||||
contextLength: 128000,
|
||||
maxOutputTokens: 4096,
|
||||
modelId: "gpt-4o",
|
||||
name: "GPT-4o",
|
||||
providerId,
|
||||
});
|
||||
expect("error" in result).toBe(false);
|
||||
const model = (result as { model: { contextLength: null | number; maxOutputTokens: null | number } }).model;
|
||||
expect(model.contextLength).toBe(128000);
|
||||
expect(model.maxOutputTokens).toBe(4096);
|
||||
});
|
||||
});
|
||||
});
|
||||
195
tests/server/db/providers.test.ts
Normal file
195
tests/server/db/providers.test.ts
Normal file
@@ -0,0 +1,195 @@
|
||||
import type Database from "bun:sqlite";
|
||||
|
||||
import { describe, expect, test } from "bun:test";
|
||||
|
||||
import {
|
||||
createProvider,
|
||||
deleteProvider,
|
||||
disableProvider,
|
||||
enableProvider,
|
||||
getProvider,
|
||||
listProviders,
|
||||
updateProvider,
|
||||
} from "../../../src/server/db/providers";
|
||||
import { createMigratedTestDatabase } from "../../helpers";
|
||||
|
||||
function withDb(callback: (db: Database) => void): void {
|
||||
const handle = createMigratedTestDatabase("providers-test");
|
||||
try {
|
||||
callback(handle.db);
|
||||
handle.close();
|
||||
} finally {
|
||||
handle.cleanup();
|
||||
}
|
||||
}
|
||||
|
||||
describe("供应商数据访问层", () => {
|
||||
test("创建供应商", () => {
|
||||
withDb((db) => {
|
||||
const result = createProvider(db, {
|
||||
apiKey: "sk-test",
|
||||
baseUrl: "https://api.openai.com/v1",
|
||||
name: "OpenAI",
|
||||
type: "openai",
|
||||
});
|
||||
expect("error" in result).toBe(false);
|
||||
const provider = (
|
||||
result as { provider: { apiKey: string; baseUrl: string; enabled: boolean; name: string; type: string } }
|
||||
).provider;
|
||||
expect(provider.name).toBe("OpenAI");
|
||||
expect(provider.type).toBe("openai");
|
||||
expect(provider.baseUrl).toBe("https://api.openai.com/v1");
|
||||
expect(provider.apiKey).toBe("sk-test");
|
||||
expect(provider.enabled).toBe(true);
|
||||
});
|
||||
});
|
||||
|
||||
test("供应商名称唯一", () => {
|
||||
withDb((db) => {
|
||||
createProvider(db, { apiKey: "sk-1", baseUrl: "https://a.com", name: "唯一", type: "openai" });
|
||||
const result = createProvider(db, { apiKey: "sk-2", baseUrl: "https://b.com", name: "唯一", type: "openai" });
|
||||
expect("error" in result).toBe(true);
|
||||
expect((result as unknown as { error: string }).error).toContain("已存在");
|
||||
});
|
||||
});
|
||||
|
||||
test("名称为空时创建失败", () => {
|
||||
withDb((db) => {
|
||||
const result = createProvider(db, { apiKey: "sk", baseUrl: "https://a.com", name: " ", type: "openai" });
|
||||
expect("error" in result).toBe(true);
|
||||
expect((result as unknown as { error: string }).error).toContain("不能为空");
|
||||
});
|
||||
});
|
||||
|
||||
test("列表查询（分页和关键字）", () => {
|
||||
withDb((db) => {
|
||||
createProvider(db, { apiKey: "sk-1", baseUrl: "https://a.com", name: "Alpha", type: "openai" });
|
||||
createProvider(db, { apiKey: "sk-2", baseUrl: "https://b.com", name: "Beta", type: "anthropic" });
|
||||
createProvider(db, { apiKey: "sk-3", baseUrl: "https://c.com", name: "Gamma", type: "openai-compatible" });
|
||||
|
||||
const result1 = listProviders(db, { page: 1, pageSize: 20 });
|
||||
expect(result1.total).toBe(3);
|
||||
|
||||
const result2 = listProviders(db, { keyword: "Alpha", page: 1, pageSize: 20 });
|
||||
expect(result2.total).toBe(1);
|
||||
expect(result2.items[0]!.name).toBe("Alpha");
|
||||
|
||||
const result3 = listProviders(db, { page: 1, pageSize: 2 });
|
||||
expect(result3.items.length).toBe(2);
|
||||
});
|
||||
});
|
||||
|
||||
test("获取供应商详情", () => {
|
||||
withDb((db) => {
|
||||
const created = createProvider(db, { apiKey: "sk", baseUrl: "https://a.com", name: "详情", type: "openai" });
|
||||
const id = (created as { provider: { id: string } }).provider.id;
|
||||
|
||||
const result = getProvider(db, id);
|
||||
expect("error" in result).toBe(false);
|
||||
expect((result as { provider: { name: string } }).provider.name).toBe("详情");
|
||||
});
|
||||
});
|
||||
|
||||
test("获取不存在的供应商返回 404", () => {
|
||||
withDb((db) => {
|
||||
const result = getProvider(db, "nonexistent");
|
||||
expect("error" in result).toBe(true);
|
||||
expect((result as unknown as { status: number }).status).toBe(404);
|
||||
});
|
||||
});
|
||||
|
||||
test("更新供应商", () => {
|
||||
withDb((db) => {
|
||||
const created = createProvider(db, { apiKey: "sk", baseUrl: "https://a.com", name: "原名", type: "openai" });
|
||||
const id = (created as { provider: { id: string } }).provider.id;
|
||||
|
||||
const result = updateProvider(db, id, { name: "新名" });
|
||||
expect("error" in result).toBe(false);
|
||||
expect((result as { provider: { name: string } }).provider.name).toBe("新名");
|
||||
});
|
||||
});
|
||||
|
||||
test("更新供应商名称重复失败", () => {
|
||||
withDb((db) => {
|
||||
createProvider(db, { apiKey: "sk-1", baseUrl: "https://a.com", name: "已存在", type: "openai" });
|
||||
const created = createProvider(db, { apiKey: "sk-2", baseUrl: "https://b.com", name: "原名", type: "openai" });
|
||||
const id = (created as { provider: { id: string } }).provider.id;
|
||||
|
||||
const result = updateProvider(db, id, { name: "已存在" });
|
||||
expect("error" in result).toBe(true);
|
||||
expect((result as unknown as { error: string }).error).toContain("已存在");
|
||||
});
|
||||
});
|
||||
|
||||
test("启用/禁用供应商", () => {
|
||||
withDb((db) => {
|
||||
const created = createProvider(db, { apiKey: "sk", baseUrl: "https://a.com", name: "测试", type: "openai" });
|
||||
const id = (created as { provider: { id: string } }).provider.id;
|
||||
|
||||
const disabled = disableProvider(db, id);
|
||||
expect("error" in disabled).toBe(false);
|
||||
expect((disabled as { provider: { enabled: boolean } }).provider.enabled).toBe(false);
|
||||
|
||||
const enabled = enableProvider(db, id);
|
||||
expect("error" in enabled).toBe(false);
|
||||
expect((enabled as { provider: { enabled: boolean } }).provider.enabled).toBe(true);
|
||||
});
|
||||
});
|
||||
|
||||
test("重复禁用失败", () => {
|
||||
withDb((db) => {
|
||||
const created = createProvider(db, { apiKey: "sk", baseUrl: "https://a.com", name: "测试", type: "openai" });
|
||||
const id = (created as { provider: { id: string } }).provider.id;
|
||||
disableProvider(db, id);
|
||||
|
||||
const result = disableProvider(db, id);
|
||||
expect("error" in result).toBe(true);
|
||||
expect((result as unknown as { status: number }).status).toBe(409);
|
||||
});
|
||||
});
|
||||
|
||||
test("重复启用失败", () => {
|
||||
withDb((db) => {
|
||||
const created = createProvider(db, { apiKey: "sk", baseUrl: "https://a.com", name: "测试", type: "openai" });
|
||||
const id = (created as { provider: { id: string } }).provider.id;
|
||||
|
||||
const result = enableProvider(db, id);
|
||||
expect("error" in result).toBe(true);
|
||||
expect((result as unknown as { status: number }).status).toBe(409);
|
||||
});
|
||||
});
|
||||
|
||||
test("删除供应商", () => {
|
||||
withDb((db) => {
|
||||
const created = createProvider(db, { apiKey: "sk", baseUrl: "https://a.com", name: "删除测试", type: "openai" });
|
||||
const id = (created as { provider: { id: string } }).provider.id;
|
||||
|
||||
const result = deleteProvider(db, id);
|
||||
expect("error" in result).toBe(false);
|
||||
|
||||
const after = getProvider(db, id);
|
||||
expect("error" in after).toBe(true);
|
||||
});
|
||||
});
|
||||
|
||||
test("删除不存在的供应商返回 404", () => {
|
||||
withDb((db) => {
|
||||
const result = deleteProvider(db, "nonexistent");
|
||||
expect("error" in result).toBe(true);
|
||||
expect((result as unknown as { status: number }).status).toBe(404);
|
||||
});
|
||||
});
|
||||
|
||||
test("默认类型为 openai-compatible", () => {
|
||||
withDb((db) => {
|
||||
createProvider(db, { apiKey: "sk", baseUrl: "https://a.com", name: "默认类型", type: "openai-compatible" });
|
||||
const result = createProvider(db, {
|
||||
apiKey: "sk2",
|
||||
baseUrl: "https://b.com",
|
||||
name: "显式默认",
|
||||
type: "openai-compatible",
|
||||
});
|
||||
expect((result as { provider: { type: string } }).provider.type).toBe("openai-compatible");
|
||||
});
|
||||
});
|
||||
});
|
||||
222
tests/server/routes/models.test.ts
Normal file
222
tests/server/routes/models.test.ts
Normal file
@@ -0,0 +1,222 @@
|
||||
import type Database from "bun:sqlite";
|
||||
|
||||
import { describe, expect, test } from "bun:test";
|
||||
|
||||
import type { Model, RuntimeMode } from "../../../src/shared/api";
|
||||
|
||||
import { createMigratedMemoryTestDatabase } from "../../helpers";
|
||||
|
||||
const MODE: RuntimeMode = "test";
|
||||
|
||||
async function createModelViaHandler(req: Request, db: Database): Promise<Response> {
|
||||
const { handleCreateModel: h } = await import("../../../src/server/routes/models/create");
|
||||
return h(req, db, MODE);
|
||||
}
|
||||
|
||||
function createTestModel(db: Database, pName: string, providerId?: string): Model {
|
||||
const pid = providerId ?? seedProvider(db);
|
||||
const result = createModel(db, {
|
||||
capabilities: ["text"],
|
||||
modelId: pName.toLowerCase().replace(/[^a-z0-9-]/g, "-"),
|
||||
name: pName,
|
||||
providerId: pid,
|
||||
});
|
||||
if ("error" in result) throw new Error(result.error);
|
||||
return result.model;
|
||||
}
|
||||
|
||||
async function deleteModelViaHandler(req: Request, db: Database): Promise<Response> {
|
||||
const { handleDeleteModel: h } = await import("../../../src/server/routes/models/delete");
|
||||
return h(req, db, MODE);
|
||||
}
|
||||
|
||||
async function disableModelViaHandler(req: Request, db: Database): Promise<Response> {
|
||||
const { handleDisableModel: h } = await import("../../../src/server/routes/models/disable");
|
||||
return h(req, db, MODE);
|
||||
}
|
||||
|
||||
async function enableModelViaHandler(req: Request, db: Database): Promise<Response> {
|
||||
const { handleEnableModel: h } = await import("../../../src/server/routes/models/enable");
|
||||
return h(req, db, MODE);
|
||||
}
|
||||
|
||||
async function getModelViaHandler(req: Request, db: Database): Promise<Response> {
|
||||
const { handleGetModel: h } = await import("../../../src/server/routes/models/get");
|
||||
return h(req, db, MODE);
|
||||
}
|
||||
|
||||
async function listModelsViaHandler(req: Request, db: Database): Promise<Response> {
|
||||
const { handleListModels: h } = await import("../../../src/server/routes/models/list");
|
||||
return h(req, db, MODE);
|
||||
}
|
||||
|
||||
import { createModel } from "../../../src/server/db/models";
|
||||
import { createProvider } from "../../../src/server/db/providers";
|
||||
|
||||
function seedProvider(db: Database, name?: string): string {
|
||||
const result = createProvider(db, {
|
||||
apiKey: "sk-test",
|
||||
baseUrl: "https://api.test.com/v1",
|
||||
name: name ?? "TestProvider",
|
||||
type: "openai",
|
||||
});
|
||||
if ("error" in result) throw new Error(result.error);
|
||||
return result.provider.id;
|
||||
}
|
||||
|
||||
async function updateModelViaHandler(req: Request, db: Database): Promise<Response> {
|
||||
const { handleUpdateModel: h } = await import("../../../src/server/routes/models/update");
|
||||
return h(req, db, MODE);
|
||||
}
|
||||
|
||||
async function withRouteDb(callback: (db: Database) => Promise<void>): Promise<void> {
|
||||
const handle = createMigratedMemoryTestDatabase("route-model-test");
|
||||
try {
|
||||
await callback(handle.db);
|
||||
handle.close();
|
||||
} finally {
|
||||
handle.cleanup();
|
||||
}
|
||||
}
|
||||
|
||||
describe("models API routes", () => {
|
||||
test("POST /api/models create", async () => {
|
||||
await withRouteDb(async (db) => {
|
||||
const providerId = seedProvider(db);
|
||||
|
||||
const req = new Request("http://localhost/api/models", {
|
||||
body: JSON.stringify({
|
||||
capabilities: ["text", "reasoning"],
|
||||
modelId: "gpt-4o",
|
||||
name: "GPT-4o",
|
||||
providerId,
|
||||
}),
|
||||
headers: { "Content-Type": "application/json" },
|
||||
method: "POST",
|
||||
});
|
||||
const res = await createModelViaHandler(req, db);
|
||||
expect(res.status).toBe(201);
|
||||
const body = (await res.json()) as { model: Model };
|
||||
expect(body.model.name).toBe("GPT-4o");
|
||||
expect(body.model.modelId).toBe("gpt-4o");
|
||||
});
|
||||
});
|
||||
|
||||
test("GET /api/models list", async () => {
|
||||
await withRouteDb(async (db) => {
|
||||
const p1 = seedProvider(db, "ListP1");
|
||||
const p2 = seedProvider(db, "ListP2");
|
||||
createTestModel(db, "A-Model", p1);
|
||||
createTestModel(db, "B-Model", p2);
|
||||
|
||||
const req = new Request("http://localhost/api/models?page=1&pageSize=20");
|
||||
const res = await listModelsViaHandler(req, db);
|
||||
expect(res.status).toBe(200);
|
||||
const body = (await res.json()) as { items: Model[]; total: number };
|
||||
expect(body.total).toBe(2);
|
||||
expect(body.items.length).toBe(2);
|
||||
});
|
||||
});
|
||||
|
||||
test("GET /api/models filter by providerId", async () => {
|
||||
await withRouteDb(async (db) => {
|
||||
const p1 = seedProvider(db, "P1");
|
||||
const p2 = seedProvider(db, "P2");
|
||||
createTestModel(db, "M1", p1);
|
||||
createTestModel(db, "M2", p2);
|
||||
|
||||
const res = await listModelsViaHandler(
|
||||
new Request("http://localhost/api/models?page=1&pageSize=20&providerId=" + p1),
|
||||
db,
|
||||
);
|
||||
expect(res.status).toBe(200);
|
||||
const body = (await res.json()) as { items: Model[]; total: number };
|
||||
expect(body.total).toBe(1);
|
||||
});
|
||||
});
|
||||
|
||||
test("GET /api/models/:id get detail", async () => {
|
||||
await withRouteDb(async (db) => {
|
||||
const model = createTestModel(db, "Detail");
|
||||
|
||||
const req = new Request("http://localhost/api/models/" + model.id);
|
||||
const res = await getModelViaHandler(req, db);
|
||||
expect(res.status).toBe(200);
|
||||
const body = (await res.json()) as { model: Model };
|
||||
expect(body.model.name).toBe("Detail");
|
||||
});
|
||||
});
|
||||
|
||||
test("PATCH /api/models/:id update", async () => {
|
||||
await withRouteDb(async (db) => {
|
||||
const model = createTestModel(db, "OldName");
|
||||
|
||||
const req = new Request("http://localhost/api/models/" + model.id, {
|
||||
body: JSON.stringify({ name: "Updated" }),
|
||||
headers: { "Content-Type": "application/json" },
|
||||
method: "PATCH",
|
||||
});
|
||||
const res = await updateModelViaHandler(req, db);
|
||||
expect(res.status).toBe(200);
|
||||
const body = (await res.json()) as { model: Model };
|
||||
expect(body.model.name).toBe("Updated");
|
||||
});
|
||||
});
|
||||
|
||||
test("POST /api/models/:id/enable", async () => {
|
||||
await withRouteDb(async (db) => {
|
||||
const model = createTestModel(db, "EnableTest");
|
||||
await disableModelViaHandler(
|
||||
new Request("http://localhost/api/models/" + model.id + "/disable", { method: "POST" }),
|
||||
db,
|
||||
);
|
||||
|
||||
const req = new Request("http://localhost/api/models/" + model.id + "/enable", { method: "POST" });
|
||||
const res = await enableModelViaHandler(req, db);
|
||||
expect(res.status).toBe(200);
|
||||
const body = (await res.json()) as { model: Model };
|
||||
expect(body.model.enabled).toBe(true);
|
||||
});
|
||||
});
|
||||
|
||||
test("POST /api/models/:id/disable", async () => {
|
||||
await withRouteDb(async (db) => {
|
||||
const model = createTestModel(db, "DisableTest");
|
||||
|
||||
const req = new Request("http://localhost/api/models/" + model.id + "/disable", { method: "POST" });
|
||||
const res = await disableModelViaHandler(req, db);
|
||||
expect(res.status).toBe(200);
|
||||
const body = (await res.json()) as { model: Model };
|
||||
expect(body.model.enabled).toBe(false);
|
||||
});
|
||||
});
|
||||
|
||||
test("DELETE /api/models/:id", async () => {
|
||||
await withRouteDb(async (db) => {
|
||||
const model = createTestModel(db, "DeleteTest");
|
||||
|
||||
const req = new Request("http://localhost/api/models/" + model.id, { method: "DELETE" });
|
||||
const res = await deleteModelViaHandler(req, db);
|
||||
expect(res.status).toBe(204);
|
||||
});
|
||||
});
|
||||
|
||||
test("invalid capabilities returns 400", async () => {
|
||||
await withRouteDb(async (db) => {
|
||||
const providerId = seedProvider(db);
|
||||
|
||||
const req = new Request("http://localhost/api/models", {
|
||||
body: JSON.stringify({
|
||||
capabilities: ["invalid-cap"],
|
||||
modelId: "test",
|
||||
name: "Test",
|
||||
providerId,
|
||||
}),
|
||||
headers: { "Content-Type": "application/json" },
|
||||
method: "POST",
|
||||
});
|
||||
const res = await createModelViaHandler(req, db);
|
||||
expect(res.status).toBe(400);
|
||||
});
|
||||
});
|
||||
});
|
||||
198
tests/server/routes/providers.test.ts
Normal file
198
tests/server/routes/providers.test.ts
Normal file
@@ -0,0 +1,198 @@
|
||||
import type Database from "bun:sqlite";
|
||||
|
||||
import { describe, expect, test } from "bun:test";
|
||||
|
||||
import type { Provider, RuntimeMode } from "../../../src/shared/api";
|
||||
|
||||
import { createMigratedMemoryTestDatabase } from "../../helpers";
|
||||
|
||||
const MODE: RuntimeMode = "test";
|
||||
|
||||
async function createProviderViaHandler(req: Request, db: Database): Promise<Response> {
|
||||
const { handleCreateProvider: h } = await import("../../../src/server/routes/providers/create");
|
||||
return h(req, db, MODE);
|
||||
}
|
||||
|
||||
function createTestProvider(db: Database, name = "测试供应商"): Provider {
|
||||
const result = createProvider(db, {
|
||||
apiKey: "sk-test",
|
||||
baseUrl: "https://api.test.com/v1",
|
||||
name,
|
||||
type: "openai",
|
||||
});
|
||||
if ("error" in result) throw new Error(result.error);
|
||||
return result.provider;
|
||||
}
|
||||
|
||||
async function deleteProviderViaHandler(req: Request, db: Database): Promise<Response> {
|
||||
const { handleDeleteProvider: h } = await import("../../../src/server/routes/providers/delete");
|
||||
return h(req, db, MODE);
|
||||
}
|
||||
|
||||
async function disableProviderViaHandler(req: Request, db: Database): Promise<Response> {
|
||||
const { handleDisableProvider: h } = await import("../../../src/server/routes/providers/disable");
|
||||
return h(req, db, MODE);
|
||||
}
|
||||
|
||||
async function enableProviderViaHandler(req: Request, db: Database): Promise<Response> {
|
||||
const { handleEnableProvider: h } = await import("../../../src/server/routes/providers/enable");
|
||||
return h(req, db, MODE);
|
||||
}
|
||||
|
||||
async function getProviderViaHandler(req: Request, db: Database): Promise<Response> {
|
||||
const { handleGetProvider: h } = await import("../../../src/server/routes/providers/get");
|
||||
return h(req, db, MODE);
|
||||
}
|
||||
|
||||
async function listProvidersViaHandler(req: Request, db: Database): Promise<Response> {
|
||||
const { handleListProviders: h } = await import("../../../src/server/routes/providers/list");
|
||||
return h(req, db, MODE);
|
||||
}
|
||||
|
||||
import { createProvider } from "../../../src/server/db/providers";
|
||||
|
||||
async function updateProviderViaHandler(req: Request, db: Database): Promise<Response> {
|
||||
const { handleUpdateProvider: h } = await import("../../../src/server/routes/providers/update");
|
||||
return h(req, db, MODE);
|
||||
}
|
||||
|
||||
async function withRouteDb(callback: (db: Database) => Promise<void>): Promise<void> {
|
||||
const handle = createMigratedMemoryTestDatabase("route-provider-test");
|
||||
try {
|
||||
await callback(handle.db);
|
||||
handle.close();
|
||||
} finally {
|
||||
handle.cleanup();
|
||||
}
|
||||
}
|
||||
|
||||
describe("供应商 API 路由", () => {
|
||||
test("POST /api/providers 创建供应商", async () => {
|
||||
await withRouteDb(async (db) => {
|
||||
const req = new Request("http://localhost/api/providers", {
|
||||
body: JSON.stringify({
|
||||
apiKey: "sk-test",
|
||||
baseUrl: "https://api.openai.com/v1",
|
||||
name: "OpenAI",
|
||||
type: "openai",
|
||||
}),
|
||||
headers: { "Content-Type": "application/json" },
|
||||
method: "POST",
|
||||
});
|
||||
const res = await createProviderViaHandler(req, db);
|
||||
expect(res.status).toBe(201);
|
||||
const body = (await res.json()) as { provider: Provider };
|
||||
expect(body.provider.name).toBe("OpenAI");
|
||||
expect(body.provider.type).toBe("openai");
|
||||
});
|
||||
});
|
||||
|
||||
test("GET /api/providers 列表查询", async () => {
|
||||
await withRouteDb(async (db) => {
|
||||
createTestProvider(db, "A供应商");
|
||||
createTestProvider(db, "B供应商");
|
||||
|
||||
const req = new Request("http://localhost/api/providers?page=1&pageSize=20");
|
||||
const res = await listProvidersViaHandler(req, db);
|
||||
expect(res.status).toBe(200);
|
||||
const body = (await res.json()) as { items: Provider[]; total: number };
|
||||
expect(body.total).toBe(2);
|
||||
expect(body.items.length).toBe(2);
|
||||
});
|
||||
});
|
||||
|
||||
test("GET /api/providers/:id 获取详情", async () => {
|
||||
await withRouteDb(async (db) => {
|
||||
const provider = createTestProvider(db, "详情路由");
|
||||
|
||||
const req = new Request(`http://localhost/api/providers/${provider.id}`);
|
||||
const res = await getProviderViaHandler(req, db);
|
||||
expect(res.status).toBe(200);
|
||||
const body = (await res.json()) as { provider: Provider };
|
||||
expect(body.provider.name).toBe("详情路由");
|
||||
});
|
||||
});
|
||||
|
||||
test("PATCH /api/providers/:id 更新供应商", async () => {
|
||||
await withRouteDb(async (db) => {
|
||||
const provider = createTestProvider(db, "更新路由");
|
||||
|
||||
const req = new Request(`http://localhost/api/providers/${provider.id}`, {
|
||||
body: JSON.stringify({ name: "已更新" }),
|
||||
headers: { "Content-Type": "application/json" },
|
||||
method: "PATCH",
|
||||
});
|
||||
const res = await updateProviderViaHandler(req, db);
|
||||
expect(res.status).toBe(200);
|
||||
const body = (await res.json()) as { provider: Provider };
|
||||
expect(body.provider.name).toBe("已更新");
|
||||
});
|
||||
});
|
||||
|
||||
test("POST /api/providers/:id/enable 启用", async () => {
|
||||
await withRouteDb(async (db) => {
|
||||
const provider = createTestProvider(db, "启用测试");
|
||||
await disableProviderViaHandler(
|
||||
new Request(`http://localhost/api/providers/${provider.id}/disable`, { method: "POST" }),
|
||||
db,
|
||||
);
|
||||
|
||||
const req = new Request(`http://localhost/api/providers/${provider.id}/enable`, { method: "POST" });
|
||||
const res = await enableProviderViaHandler(req, db);
|
||||
expect(res.status).toBe(200);
|
||||
const body = (await res.json()) as { provider: Provider };
|
||||
expect(body.provider.enabled).toBe(true);
|
||||
});
|
||||
});
|
||||
|
||||
test("POST /api/providers/:id/disable 禁用", async () => {
|
||||
await withRouteDb(async (db) => {
|
||||
const provider = createTestProvider(db, "禁用测试");
|
||||
|
||||
const req = new Request(`http://localhost/api/providers/${provider.id}/disable`, { method: "POST" });
|
||||
const res = await disableProviderViaHandler(req, db);
|
||||
expect(res.status).toBe(200);
|
||||
const body = (await res.json()) as { provider: Provider };
|
||||
expect(body.provider.enabled).toBe(false);
|
||||
});
|
||||
});
|
||||
|
||||
test("DELETE /api/providers/:id 删除供应商", async () => {
|
||||
await withRouteDb(async (db) => {
|
||||
const provider = createTestProvider(db, "删除路由");
|
||||
|
||||
const req = new Request(`http://localhost/api/providers/${provider.id}`, { method: "DELETE" });
|
||||
const res = await deleteProviderViaHandler(req, db);
|
||||
expect(res.status).toBe(204);
|
||||
});
|
||||
});
|
||||
|
||||
test("创建同名供应商返回 409", async () => {
|
||||
await withRouteDb(async (db) => {
|
||||
const req1 = new Request("http://localhost/api/providers", {
|
||||
body: JSON.stringify({
|
||||
apiKey: "sk-a",
|
||||
baseUrl: "https://a.com",
|
||||
name: "重复名",
|
||||
type: "openai",
|
||||
}),
|
||||
headers: { "Content-Type": "application/json" },
|
||||
method: "POST",
|
||||
});
|
||||
await createProviderViaHandler(req1, db);
|
||||
|
||||
const req2 = new Request("http://localhost/api/providers", {
|
||||
body: JSON.stringify({
|
||||
apiKey: "sk-b",
|
||||
baseUrl: "https://b.com",
|
||||
name: "重复名",
|
||||
type: "openai",
|
||||
}),
|
||||
headers: { "Content-Type": "application/json" },
|
||||
method: "POST",
|
||||
});
|
||||
const res = await createProviderViaHandler(req2, db);
|
||||
expect(res.status).toBe(409);
|
||||
});
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user