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:
2026-05-29 12:40:10 +08:00
parent 2ea4bd4410
commit 933c2133f0
56 changed files with 4706 additions and 9 deletions

View File

@@ -0,0 +1,115 @@
import { fireEvent, screen, waitFor } from "@testing-library/react";
import { describe, expect, mock, test } from "bun:test";
import { createElement } from "react";
import type { Model, Provider } from "../../../src/shared/api";
import { ModelTable } from "../../../src/web/pages/models/components/ModelTable";
import { renderWithProviders } from "../test-utils";
const ENABLED_PROVIDER: Provider = {
apiKey: "sk-test",
baseUrl: "https://api.openai.com/v1",
createdAt: "2024-01-01T00:00:00.000Z",
enabled: true,
id: "pv1",
name: "OpenAI",
type: "openai",
updatedAt: "2024-01-01T00:00:00.000Z",
};
const DISABLED_PROVIDER: Provider = {
apiKey: "sk-off",
baseUrl: "https://api.deepseek.com/v1",
createdAt: "2024-01-01T00:00:00.000Z",
enabled: false,
id: "pv2",
name: "DeepSeek",
type: "openai-compatible",
updatedAt: "2024-01-01T00:00:00.000Z",
};
const ENABLED_MODEL: Model = {
capabilities: ["text", "reasoning"],
contextLength: 128000,
createdAt: "2024-01-01T00:00:00.000Z",
enabled: true,
id: "m1",
maxOutputTokens: 4096,
modelId: "gpt-4o",
name: "GPT-4o",
providerId: "pv1",
updatedAt: "2024-01-01T00:00:00.000Z",
};
const DISABLED_MODEL: Model = {
capabilities: ["text"],
contextLength: null,
createdAt: "2024-01-01T00:00:00.000Z",
enabled: false,
id: "m2",
maxOutputTokens: null,
modelId: "deepseek-chat",
name: "DeepSeek Chat",
providerId: "pv2",
updatedAt: "2024-01-01T00:00:00.000Z",
};
function clickLatestConfirmButton() {
const buttons = screen.getAllByRole("button", { name: /OK|确定/ });
fireEvent.click(buttons[buttons.length - 1]!);
}
describe("ModelTable", () => {
test("渲染模型表格数据", () => {
renderWithProviders(
createElement(ModelTable, {
data: { items: [ENABLED_MODEL, DISABLED_MODEL], page: 1, pageSize: 20, total: 2 },
loading: false,
onDelete: () => Promise.resolve(),
onDisable: () => Promise.resolve(),
onEdit: () => undefined,
onEnable: () => Promise.resolve(),
onPageChange: () => undefined,
page: 1,
pageSize: 20,
providers: [ENABLED_PROVIDER, DISABLED_PROVIDER],
}),
);
expect(screen.getByText("GPT-4o")).not.toBeNull();
expect(screen.getByText("gpt-4o")).not.toBeNull();
expect(screen.getByText("DeepSeek Chat")).not.toBeNull();
});
test("模型表格操作触发 enable/disable/delete", async () => {
const onDisable = mock(() => Promise.resolve());
const onEnable = mock(() => Promise.resolve());
const onDelete = mock(() => Promise.resolve());
renderWithProviders(
createElement(ModelTable, {
data: { items: [ENABLED_MODEL, DISABLED_MODEL], page: 1, pageSize: 20, total: 2 },
loading: false,
onDelete,
onDisable,
onEdit: () => undefined,
onEnable,
onPageChange: () => undefined,
page: 1,
pageSize: 20,
providers: [ENABLED_PROVIDER, DISABLED_PROVIDER],
}),
);
const disableButtons = screen.getAllByRole("button", { name: /禁用/ });
fireEvent.click(disableButtons[0]!);
await waitFor(() => expect(screen.getByText("确认禁用此模型?")).not.toBeNull());
clickLatestConfirmButton();
await waitFor(() => expect(onDisable).toHaveBeenCalledWith("m1"));
const enableButtons = screen.getAllByRole("button", { name: /启用/ });
fireEvent.click(enableButtons[0]!);
await waitFor(() => expect(onEnable).toHaveBeenCalledWith("m2"));
});
});

View File

@@ -0,0 +1,89 @@
import { fireEvent, screen, waitFor } from "@testing-library/react";
import { describe, expect, mock, test } from "bun:test";
import { createElement } from "react";
import type { Provider } from "../../../src/shared/api";
import { ProviderTable } from "../../../src/web/pages/models/components/ProviderTable";
import { renderWithProviders } from "../test-utils";
const ENABLED_PROVIDER: Provider = {
apiKey: "sk-test",
baseUrl: "https://api.openai.com/v1",
createdAt: "2024-01-01T00:00:00.000Z",
enabled: true,
id: "pv1",
name: "OpenAI",
type: "openai",
updatedAt: "2024-01-01T00:00:00.000Z",
};
const DISABLED_PROVIDER: Provider = {
apiKey: "sk-off",
baseUrl: "https://api.deepseek.com/v1",
createdAt: "2024-01-01T00:00:00.000Z",
enabled: false,
id: "pv2",
name: "DeepSeek",
type: "openai-compatible",
updatedAt: "2024-01-01T00:00:00.000Z",
};
function clickLatestConfirmButton() {
const buttons = screen.getAllByRole("button", { name: /OK|确定/ });
fireEvent.click(buttons[buttons.length - 1]!);
}
describe("ProviderTable", () => {
test("渲染供应商表格数据", () => {
renderWithProviders(
createElement(ProviderTable, {
data: { items: [ENABLED_PROVIDER, DISABLED_PROVIDER], page: 1, pageSize: 20, total: 2 },
loading: false,
onDelete: () => Promise.resolve(),
onDisable: () => Promise.resolve(),
onEdit: () => undefined,
onEnable: () => Promise.resolve(),
onPageChange: () => undefined,
onTest: () => Promise.resolve({ message: "ok", ok: true }),
page: 1,
pageSize: 20,
}),
);
expect(screen.getAllByText("OpenAI").length).toBeGreaterThan(0);
expect(screen.getByText("DeepSeek")).not.toBeNull();
expect(screen.getByText("https://api.openai.com/v1")).not.toBeNull();
});
test("供应商表格操作触发 enable/disable/delete", async () => {
const onDisable = mock(() => Promise.resolve());
const onEnable = mock(() => Promise.resolve());
const onDelete = mock(() => Promise.resolve());
renderWithProviders(
createElement(ProviderTable, {
data: { items: [ENABLED_PROVIDER, DISABLED_PROVIDER], page: 1, pageSize: 20, total: 2 },
loading: false,
onDelete,
onDisable,
onEdit: () => undefined,
onEnable,
onPageChange: () => undefined,
onTest: () => Promise.resolve({ message: "ok", ok: true }),
page: 1,
pageSize: 20,
}),
);
const disableButtons = screen.getAllByRole("button", { name: /禁用/ });
fireEvent.click(disableButtons[0]!);
await waitFor(() => expect(screen.getByText("确认禁用此供应商?")).not.toBeNull());
clickLatestConfirmButton();
await waitFor(() => expect(onDisable).toHaveBeenCalledWith("pv1"));
const enableButtons = screen.getAllByRole("button", { name: /启用/ });
fireEvent.click(enableButtons[0]!);
await waitFor(() => expect(onEnable).toHaveBeenCalledWith("pv2"));
});
});

View File

@@ -0,0 +1,105 @@
import { describe, expect, test } from "bun:test";
import {
createModel,
deleteModel,
disableModel,
enableModel,
fetchModel,
fetchModelList,
updateModel,
} from "../../../src/web/hooks/use-models";
import { installFetchMock, jsonResponse } from "../test-utils";
const MODEL = {
capabilities: ["text"] as Array<"text">,
contextLength: null,
createdAt: "2024-01-01T00:00:00.000Z",
enabled: true,
id: "m1",
maxOutputTokens: null,
modelId: "gpt-4o",
name: "GPT-4o",
providerId: "pv1",
updatedAt: "2024-01-01T00:00:00.000Z",
};
async function expectRejectsWithMessage(action: () => Promise<unknown>, message: string) {
try {
await action();
throw new Error("expected rejection");
} catch (error) {
expect(error).toBeInstanceOf(Error);
expect((error as Error).message).toBe(message);
}
}
function jsonBody(body: BodyInit | null | undefined): unknown {
return JSON.parse(typeof body === "string" ? body : "{}");
}
describe("use-models request helpers", () => {
test("fetchModelList 按协议拼接 query 参数(含 providerId", async () => {
const calls = installFetchMock(() => jsonResponse({ items: [MODEL], page: 1, pageSize: 20, total: 1 }));
const result = await fetchModelList({ keyword: "GPT", page: 1, pageSize: 20, providerId: "pv1" });
expect(result.items).toHaveLength(1);
expect(calls[0]?.method).toBe("GET");
expect(calls[0]?.url).toContain("providerId=pv1");
expect(calls[0]?.url).toContain("keyword=GPT");
});
test("模型 CRUD 与 enable/disable 使用正确 method、URL 与 body", async () => {
const calls = installFetchMock((call) => {
if (call.method === "DELETE") return new Response(null, { status: 204 });
return jsonResponse(
{ model: MODEL },
{ status: call.method === "POST" && call.url === "/api/models" ? 201 : 200 },
);
});
await createModel({
capabilities: ["text"],
modelId: "gpt-4o",
name: "GPT-4o",
providerId: "pv1",
});
await updateModel("m1", { name: "GPT-4o Mini" });
await enableModel("m1");
await disableModel("m1");
await deleteModel("m1");
await fetchModel("m1");
expect(calls.map((call) => `${call.method} ${call.url}`)).toEqual([
"POST /api/models",
"PATCH /api/models/m1",
"POST /api/models/m1/enable",
"POST /api/models/m1/disable",
"DELETE /api/models/m1",
"GET /api/models/m1",
]);
expect(jsonBody(calls[0]?.body)).toEqual({
capabilities: ["text"],
modelId: "gpt-4o",
name: "GPT-4o",
providerId: "pv1",
});
expect(jsonBody(calls[1]?.body)).toEqual({ name: "GPT-4o Mini" });
});
test("错误响应优先使用后端 error 字段", async () => {
installFetchMock(() => jsonResponse({ error: "模型名称已存在" }, { status: 409 }));
await expectRejectsWithMessage(
() => createModel({ capabilities: ["text"], modelId: "gpt-4o", name: "重复", providerId: "pv1" }),
"模型名称已存在",
);
});
test("非 JSON 错误响应回退到 HTTP 状态", async () => {
installFetchMock(() => new Response("broken", { status: 500 }));
await expectRejectsWithMessage(() => fetchModel("m-missing"), "HTTP 500");
});
});

View File

@@ -0,0 +1,106 @@
import { describe, expect, test } from "bun:test";
import {
createProvider,
deleteProvider,
disableProvider,
enableProvider,
fetchProvider,
fetchProviderList,
testProviderConnection,
updateProvider,
} from "../../../src/web/hooks/use-providers";
import { installFetchMock, jsonResponse } from "../test-utils";
const PROVIDER = {
apiKey: "sk-test",
baseUrl: "https://api.openai.com/v1",
createdAt: "2024-01-01T00:00:00.000Z",
enabled: true,
id: "pv1",
name: "OpenAI",
type: "openai" as const,
updatedAt: "2024-01-01T00:00:00.000Z",
};
async function expectRejectsWithMessage(action: () => Promise<unknown>, message: string) {
try {
await action();
throw new Error("expected rejection");
} catch (error) {
expect(error).toBeInstanceOf(Error);
expect((error as Error).message).toBe(message);
}
}
function jsonBody(body: BodyInit | null | undefined): unknown {
return JSON.parse(typeof body === "string" ? body : "{}");
}
describe("use-providers request helpers", () => {
test("fetchProviderList builds correct query params", async () => {
const calls = installFetchMock(() => jsonResponse({ items: [PROVIDER], page: 1, pageSize: 20, total: 1 }));
const result = await fetchProviderList({ keyword: "OpenAI", page: 1, pageSize: 20 });
expect(result.items).toHaveLength(1);
expect(calls[0]?.method).toBe("GET");
expect(calls[0]?.url).toBe("/api/providers?page=1&pageSize=20&keyword=OpenAI");
});
test("CRUD and enable/disable use correct method, URL and body", async () => {
const calls = installFetchMock((call) => {
if (call.method === "DELETE") return new Response(null, { status: 204 });
return jsonResponse(
{ provider: PROVIDER },
{ status: call.method === "POST" && call.url === "/api/providers" ? 201 : 200 },
);
});
await createProvider({ apiKey: "sk-test", baseUrl: "https://api.openai.com/v1", name: "OpenAI", type: "openai" });
await updateProvider("pv1", { name: "New OpenAI" });
await enableProvider("pv1");
await disableProvider("pv1");
await deleteProvider("pv1");
await fetchProvider("pv1");
expect(calls.map((c) => c.method + " " + c.url)).toEqual([
"POST /api/providers",
"PATCH /api/providers/pv1",
"POST /api/providers/pv1/enable",
"POST /api/providers/pv1/disable",
"DELETE /api/providers/pv1",
"GET /api/providers/pv1",
]);
expect(jsonBody(calls[0]?.body)).toEqual({
apiKey: "sk-test",
baseUrl: "https://api.openai.com/v1",
name: "OpenAI",
type: "openai",
});
expect(jsonBody(calls[1]?.body)).toEqual({ name: "New OpenAI" });
});
test("testProviderConnection uses correct URL and parses response", async () => {
installFetchMock(() => jsonResponse({ providerTestResponse: { message: "ok", ok: true } }));
const result = await testProviderConnection("pv1");
expect(result).toEqual({ message: "ok", ok: true });
});
test("error response uses backend error field", async () => {
installFetchMock(() => jsonResponse({ error: "dup" }, { status: 409 }));
await expectRejectsWithMessage(
() => createProvider({ apiKey: "sk", baseUrl: "https://x.com", name: "dup", type: "openai-compatible" }),
"dup",
);
});
test("non-JSON error falls back to HTTP status", async () => {
installFetchMock(() => new Response("broken", { status: 500 }));
await expectRejectsWithMessage(() => fetchProvider("missing"), "HTTP 500");
});
});

View File

@@ -0,0 +1,127 @@
import { fireEvent, screen, waitFor } from "@testing-library/react";
import { describe, expect, mock, test } from "bun:test";
import { createElement } from "react";
import type { Model, Provider } from "../../../src/shared/api";
import { ModelFormModal } from "../../../src/web/pages/models/components/ModelFormModal";
import { ProviderFormModal } from "../../../src/web/pages/models/components/ProviderFormModal";
import { renderWithProviders } from "../test-utils";
const ENABLED_PROVIDER: Provider = {
apiKey: "sk-test",
baseUrl: "https://api.openai.com/v1",
createdAt: "2024-01-01T00:00:00.000Z",
enabled: true,
id: "pv1",
name: "OpenAI",
type: "openai",
updatedAt: "2024-01-01T00:00:00.000Z",
};
const DISABLED_PROVIDER: Provider = {
apiKey: "sk-off",
baseUrl: "https://api.deepseek.com/v1",
createdAt: "2024-01-01T00:00:00.000Z",
enabled: false,
id: "pv2",
name: "DeepSeek",
type: "openai-compatible",
updatedAt: "2024-01-01T00:00:00.000Z",
};
const ENABLED_MODEL: Model = {
capabilities: ["text", "reasoning"],
contextLength: 128000,
createdAt: "2024-01-01T00:00:00.000Z",
enabled: true,
id: "m1",
maxOutputTokens: 4096,
modelId: "gpt-4o",
name: "GPT-4o",
providerId: "pv1",
updatedAt: "2024-01-01T00:00:00.000Z",
};
function clickLatestConfirmButton() {
const buttons = screen.getAllByRole("button", { name: /OK|确\s*定/ });
fireEvent.click(buttons[buttons.length - 1]!);
}
describe("ProviderFormModal", () => {
test("编辑供应商表单只提交变更字段", async () => {
const updateCalls: unknown[] = [];
renderWithProviders(
createElement(ProviderFormModal, {
editingProvider: ENABLED_PROVIDER,
onCancel: () => undefined,
onCreate: () => Promise.resolve(),
onOpenChange: () => undefined,
onUpdate: (args: unknown) => {
updateCalls.push(args);
return Promise.resolve();
},
open: true,
submitting: false,
}),
);
await waitFor(() => expect(screen.getByPlaceholderText("请输入供应商名称")).not.toBeNull());
fireEvent.change(screen.getByPlaceholderText("请输入供应商名称"), { target: { value: "New OpenAI" } });
clickLatestConfirmButton();
await waitFor(() => expect(updateCalls.length).toBe(1));
expect(updateCalls[0]).toEqual({ data: { name: "New OpenAI" }, id: "pv1" });
});
});
describe("ModelFormModal", () => {
test("编辑模型表单只提交变更字段", async () => {
const updateCalls: unknown[] = [];
renderWithProviders(
createElement(ModelFormModal, {
editingModel: ENABLED_MODEL,
onCancel: () => undefined,
onCreate: () => Promise.resolve(),
onOpenChange: () => undefined,
onUpdate: (args: unknown) => {
updateCalls.push(args);
return Promise.resolve();
},
open: true,
providers: [ENABLED_PROVIDER, DISABLED_PROVIDER],
submitting: false,
}),
);
await waitFor(() => expect(screen.getByPlaceholderText("请输入模型名称")).not.toBeNull());
fireEvent.change(screen.getByPlaceholderText("请输入模型名称"), { target: { value: "GPT-4o Mini" } });
clickLatestConfirmButton();
await waitFor(() => expect(updateCalls.length).toBe(1));
expect(updateCalls[0]).toEqual({ data: { name: "GPT-4o Mini" }, id: "m1" });
});
test("模型表单校验失败不会提交", async () => {
const onCreate = mock(() => Promise.resolve());
renderWithProviders(
createElement(ModelFormModal, {
editingModel: null,
onCancel: () => undefined,
onCreate,
onOpenChange: () => undefined,
onUpdate: () => Promise.resolve(),
open: true,
providers: [ENABLED_PROVIDER],
submitting: false,
}),
);
await waitFor(() => expect(screen.getByPlaceholderText("请输入模型名称")).not.toBeNull());
clickLatestConfirmButton();
expect(onCreate).not.toHaveBeenCalled();
});
});