diff --git a/src/app/api/translator/translate/route.js b/src/app/api/translator/translate/route.js index d5fb2687..c0f3c12b 100644 --- a/src/app/api/translator/translate/route.js +++ b/src/app/api/translator/translate/route.js @@ -2,7 +2,7 @@ import { NextResponse } from "next/server"; import { detectFormat, getTargetFormat } from "open-sse/services/provider.js"; import { translateRequest } from "open-sse/translator/index.js"; import { FORMATS } from "open-sse/translator/formats.js"; -import { parseModel } from "open-sse/services/model.js"; +import { getModelInfo } from "@/sse/services/model.js"; import { getProviderConnections } from "@/lib/localDb.js"; import { getExecutor } from "open-sse/executors/index.js"; @@ -18,7 +18,7 @@ export async function POST(request) { case 1: { // Detect provider + formats from 1_req_client.json const clientBody = body.body || body; - const { provider, model } = parseModel(clientBody.model); + const { provider, model } = await getModelInfo(clientBody.model); const sourceFormat = detectFormat(clientBody); const targetFormat = getTargetFormat(provider); return NextResponse.json({ success: true, result: { provider, model, sourceFormat, targetFormat } }); @@ -28,7 +28,7 @@ export async function POST(request) { // source → OpenAI intermediate (mirrors 3_req_openai.json) // Translate source→openai only (half of the pipeline) const clientBody = body.body || body; - const { provider, model } = parseModel(clientBody.model); + const { provider, model } = await getModelInfo(clientBody.model); const sourceFormat = detectFormat(clientBody); const stream = clientBody.stream !== false; diff --git a/tests/unit/translator-custom-prefix.test.js b/tests/unit/translator-custom-prefix.test.js new file mode 100644 index 00000000..b38a620f --- /dev/null +++ b/tests/unit/translator-custom-prefix.test.js @@ -0,0 +1,55 @@ +import { describe, it, expect, vi, beforeEach } from "vitest"; + +// Mock getModelInfo to simulate custom provider prefix resolution +vi.mock("@/sse/services/model.js", () => ({ + getModelInfo: vi.fn(async (modelId) => { + // Simulate prefix resolution for custom providers + if (modelId.startsWith("my-prefix/")) { + return { + provider: "openai-compatible-chat-abc123", + model: modelId.replace("my-prefix/", "") + }; + } + if (modelId.startsWith("acme/")) { + return { + provider: "anthropic-compatible-claude-xyz", + model: modelId.replace("acme/", "") + }; + } + // Default: no prefix + return { + provider: modelId.split("/")[1] || "unknown", + model: modelId.split("/")[1] || modelId + }; + }) +})); + +describe("Translator custom provider prefix resolution", () => { + beforeEach(() => { + vi.clearAllMocks(); + }); + + it("should resolve OpenAI-compatible provider prefix via getModelInfo", async () => { + const { getModelInfo } = await import("@/sse/services/model.js"); + const result = await getModelInfo("my-prefix/gpt-4"); + + expect(result.provider).toBe("openai-compatible-chat-abc123"); + expect(result.model).toBe("gpt-4"); + }); + + it("should resolve Anthropic-compatible provider prefix via getModelInfo", async () => { + const { getModelInfo } = await import("@/sse/services/model.js"); + const result = await getModelInfo("acme/claude-3"); + + expect(result.provider).toBe("anthropic-compatible-claude-xyz"); + expect(result.model).toBe("claude-3"); + }); + + it("should handle model ID without prefix", async () => { + const { getModelInfo } = await import("@/sse/services/model.js"); + const result = await getModelInfo("openai/gpt-4"); + + expect(result.provider).toBeDefined(); + expect(result.model).toBe("gpt-4"); + }); +});