feat: 设置页新增模型卡片,支持为7种能力配置默认模型

This commit is contained in:
2026-06-07 09:51:04 +08:00
parent 43b14a94a3
commit 074ea0bb1a
8 changed files with 338 additions and 25 deletions

View File

@@ -0,0 +1,67 @@
import { describe, expect, test } from "bun:test";
import { screen, waitFor } from "@testing-library/react";
import { createElement } from "react";
import { ModelSettingsCard } from "../../../../../src/web/features/settings/components/ModelSettingsCard";
import { installFetchMock, jsonResponse, renderWithProviders } from "../../../test-utils";
function mockSettingsResponse(defaultModels?: Record<string, string | null>): Response {
return jsonResponse({ compact: false, defaultModels, theme: "system" });
}
function mockModelsResponse(items: Array<{ id: string; name: string }>): Response {
return jsonResponse({
items: items.map((m) => ({ ...m, capabilities: [], createdAt: "", externalId: "", providerId: "", updatedAt: "" })),
page: 1,
pageSize: 200,
total: items.length,
});
}
function mockEmptyModelsResponse(): Response {
return mockModelsResponse([]);
}
describe("ModelSettingsCard", () => {
test("渲染 7 个模型能力配置项", async () => {
installFetchMock((call) => {
if (call.url.includes("/api/settings")) return mockSettingsResponse();
if (call.url.includes("/api/models")) return mockEmptyModelsResponse();
return jsonResponse({});
});
renderWithProviders(createElement(ModelSettingsCard));
await waitFor(() => {
expect(screen.getByText("文本")).not.toBeNull();
expect(screen.getByText("图片识别")).not.toBeNull();
expect(screen.getByText("音频识别")).not.toBeNull();
expect(screen.getByText("视频识别")).not.toBeNull();
expect(screen.getByText("图片生成")).not.toBeNull();
expect(screen.getByText("音频生成")).not.toBeNull();
expect(screen.getByText("视频生成")).not.toBeNull();
});
});
test("回显已保存的默认模型值", async () => {
installFetchMock((call) => {
if (call.url.includes("/api/settings")) {
return mockSettingsResponse({ imageRecognition: "model-b", text: "model-a" });
}
if (call.url.includes("/api/models")) {
return mockModelsResponse([
{ id: "model-a", name: "GPT-4" },
{ id: "model-b", name: "Claude Vision" },
]);
}
return jsonResponse({});
});
renderWithProviders(createElement(ModelSettingsCard));
await waitFor(() => {
expect(screen.getByText("GPT-4")).not.toBeNull();
expect(screen.getByText("Claude Vision")).not.toBeNull();
});
});
});