mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-10-08 10:47:48 +08:00
dayuan.jiang
c24aae6de0
- getAIModel resolves credentials (client key, server env vars, the existing SSRF rules) and createModel builds the model by SDK. The provider-by-provider switch shrinks from 24 cases to the few that differ (lib/ai-providers.ts 1531 -> 1106 lines) - /api/validate-model calls getAIModel instead of its own 24-case switch (503 -> 175 lines), which had drifted from the chat: it built Azure with createOpenAI, Kimi and MiMo with createOpenAI instead of createDeepSeek, and the official OpenAI endpoint with Chat Completions. A passing test now means the chat works - Plain OpenAI-compatible providers (SiliconFlow, SGLang, ModelScope, GLM, Qwen, Qiniu, Novita, Atlas Cloud, EdgeOne, Doubao, MiniMax in OpenAI mode, AIHubMix on a custom URL) use @ai-sdk/openai-compatible, which reads reasoning_content, so their reasoning shows, and accepts SGLang's stream as is (its 95-line stream rewrite is gone). includeUsage keeps token usage for quotas. <think> tags in their text become reasoning (extractReasoningMiddleware) - SGLang without a base URL used OpenAI's endpoint; it now defaults to http://127.0.0.1:8000/v1 like the Test button did - Chat requests to a client base URL refuse redirects, as the Test button already did (redirectGuardedFetch moves to lib/ssrf-protection) - The Test button streams like the chat (the ModelScope special case is gone), times out after 15 s, does not retry, asks the model to call a ping tool and warns when it answers without one, and tests all models at once. The time each test took shows on its check mark - Unknown provider names are rejected with Object.hasOwn, and the error texts list providers from PROVIDER_INFO instead of hand-kept lists
79 lines
2.6 KiB
TypeScript
79 lines
2.6 KiB
TypeScript
// @vitest-environment node
|
|
import { simulateReadableStream } from "ai"
|
|
import { MockLanguageModelV3 } from "ai/test"
|
|
import { afterEach, describe, expect, it, vi } from "vitest"
|
|
import { POST as validateDiagram } from "@/app/api/validate-diagram/route"
|
|
|
|
const RESULT = {
|
|
valid: false,
|
|
issues: [
|
|
{
|
|
type: "overlap",
|
|
severity: "critical",
|
|
description: "Box A covers box B",
|
|
},
|
|
],
|
|
suggestions: ["Move box B to the right"],
|
|
}
|
|
|
|
// A vision model that answers with the JSON in a few text chunks
|
|
vi.mock("@/lib/ai-providers", () => ({
|
|
getValidationModel: () =>
|
|
new MockLanguageModelV3({
|
|
doStream: (async () => {
|
|
const json = JSON.stringify(RESULT)
|
|
return {
|
|
stream: simulateReadableStream({
|
|
chunks: [
|
|
{ type: "text-start", id: "t" },
|
|
...[json.slice(0, 20), json.slice(20)].map(
|
|
(delta) => ({
|
|
type: "text-delta",
|
|
id: "t",
|
|
delta,
|
|
}),
|
|
),
|
|
{ type: "text-end", id: "t" },
|
|
{
|
|
type: "finish",
|
|
finishReason: { unified: "stop", raw: "stop" },
|
|
usage: {
|
|
inputTokens: { total: 1 },
|
|
outputTokens: { total: 1 },
|
|
},
|
|
},
|
|
],
|
|
}),
|
|
}
|
|
}) as any,
|
|
}),
|
|
}))
|
|
|
|
const post = () =>
|
|
new Request("http://localhost/api/validate-diagram", {
|
|
method: "POST",
|
|
headers: { "Content-Type": "application/json" },
|
|
body: JSON.stringify({ imageData: "data:image/png;base64,AAAA" }),
|
|
})
|
|
|
|
afterEach(() => {
|
|
delete process.env.ENABLE_VLM_VALIDATION
|
|
})
|
|
|
|
describe("POST /api/validate-diagram", () => {
|
|
it("streams the model's result as JSON text for useObject", async () => {
|
|
const res = await validateDiagram(post())
|
|
expect(JSON.parse(await res.text())).toEqual(RESULT)
|
|
})
|
|
|
|
it("answers valid when the check is turned off", async () => {
|
|
process.env.ENABLE_VLM_VALIDATION = "false"
|
|
const res = await validateDiagram(post())
|
|
expect(JSON.parse(await res.text())).toEqual({
|
|
valid: true,
|
|
issues: [],
|
|
suggestions: [],
|
|
})
|
|
})
|
|
})
|