feat(api): 模型接入 - 统一出网 User-Agent 并支持添加时测试模型

Fixes #39
2026-07-29 18:31:01 +08:00
parent 09f5ff03cc
commit 55875207c9
19 changed files with 348 additions and 36 deletions
--- a/web/src/lib/model-connection-test.ts
+++ b/web/src/lib/model-connection-test.ts
@@ -0,0 +1,67 @@
+import { requestAudioGeneration } from "@/services/api/audio";
+import { requestGeneration, requestImageQuestion } from "@/services/api/image";
+import { createVideoGenerationTask } from "@/services/api/video";
+import { defaultConfig, encodeChannelModel, type ModelCapability, type ModelChannel } from "@/stores/use-config-store";
+import type { ModelProtocol } from "@/lib/model-protocols";
+
+export async function testChannelModelConnection(channel: ModelChannel, model: string, capability: ModelCapability, protocol: ModelProtocol) {
+    if (!channel.baseUrl.trim()) throw new Error("请先填写 Base URL");
+    if (!channel.apiKey.trim()) throw new Error("请先填写 API Key");
+    const selectedModel = encodeChannelModel(channel.id, model);
+    const modelCost = channel.modelCosts?.find((item) => item.model === model);
+    const testProtocol = channel.apiFormat === "gemini" && !modelCost?.protocol ? undefined : protocol;
+    const testChannel: ModelChannel = {
+        ...channel,
+        models: channel.models.includes(model) ? channel.models : [...channel.models, model],
+        modelCosts: [
+            {
+                model,
+                displayName: modelCost?.displayName,
+                capability,
+                protocol: testProtocol,
+                billingMode: modelCost?.billingMode || "fixed_request",
+                unitPriceMicrocredits: modelCost?.unitPriceMicrocredits || 0,
+            },
+            ...(channel.modelCosts || []).filter((item) => item.model !== model),
+        ],
+    };
+    const config = {
+        ...defaultConfig,
+        channelMode: "local" as const,
+        baseUrl: channel.baseUrl,
+        apiKey: channel.apiKey,
+        apiFormat: channel.apiFormat,
+        channels: [testChannel],
+        model: selectedModel,
+        imageModel: selectedModel,
+        videoModel: selectedModel,
+        textModel: selectedModel,
+        audioModel: selectedModel,
+        models: [selectedModel],
+        imageModels: capability === "image" ? [selectedModel] : [],
+        videoModels: capability === "video" ? [selectedModel] : [],
+        textModels: capability === "text" ? [selectedModel] : [],
+        audioModels: capability === "audio" ? [selectedModel] : [],
+        count: "1",
+        size: capability === "image" ? "1024x1024" : "16:9",
+        videoSeconds: "6",
+        vquality: "720",
+        videoGenerateAudio: "false",
+    };
+
+    switch (capability) {
+        case "text":
+            await requestImageQuestion(config, [{ role: "user", content: "Reply with OK." }], () => undefined);
+            return "文本响应正常";
+        case "image":
+            await requestGeneration(config, "A simple gray circle on a white background.");
+            return "图片生成正常";
+        case "audio":
+            await requestAudioGeneration(config, "Model test.");
+            return "音频生成正常";
+        case "video": {
+            const task = await createVideoGenerationTask(config, "A static gray circle on a white background.");
+            return `视频任务已创建（${task.id}）`;
+        }
+    }
+}