mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-10-11 03:59:58 +08:00
fix: older defects (batch C) and the second batch's review
Chats: - New Chat right after an answer saves that chat once. Saves run one at a time and read the chat on screen when their turn comes; a save scheduled for a chat that is no longer on screen is dropped. A chat whose id was still on its way to the URL no longer comes back after New Chat (the next answer went into it). - Crossing the 768 px breakpoint keeps the chat panel: a streaming answer, unsaved messages and attachments stay. The panel gets the sizes of each side, and a panel collapsed on desktop opens on mobile. - The chat's export waits for its own reply: an edit's history export still on its way no longer answers it with the older diagram, and two file saves at once no longer swap results. - A second edit in one answer is previewed on the first edit's result. - Stop also ends a running screenshot check; a chat that cannot be saved (storage full) can be left with "Continue without saving". - Small diagrams with shapes count as diagrams; the tool card no longer crashes on malformed operations. Quota and providers: - Requests that reach the server's own endpoints count toward the quota: EdgeOne (always its own endpoint now), a private base URL whatever key header is sent, keyless Ollama without a URL. With the quota on, a redirect is followed only to a public address. The output cap applies to these requests too. - Stop records the tokens of the steps that finished; the screenshot check counts its tokens without counting a request. - EdgeOne configured only by AI_PROVIDER works, also in the admin Test, which forwards the access code. Azure set up only in the admin panel works in chat. The Test sends a Bedrock session token. - The admin panel's Test of an entry without a URL uses the server's URL as the server does (no private address check for it); the admin panel no longer writes an Ollama URL. MCP server: - Write tools and start_session run one at a time, so two at once never drop each other's change; a cancelled call waiting its turn is skipped. get_diagram and export_diagram keep the session they started with. - Export to .drawio first gets the user's latest edits from the browser. - History thumbnails: one that arrives after the next AI write is dropped; a sync reply keeps the image; a version that changed only page settings is its own entry. - A diagram over the 10 MB limit is saved without its image, or the user is told to download it (the server now answers 413 instead of cutting the connection). - Labels holding text like id='1' or parent='1' are no longer read as attributes (a layer or a parent was deleted). A broken bare <mxGraphModel> file is refused. - After a sync reply the tab no longer sends its autosave copy again. Desktop and files: - A newer switch of the same preset is not rolled back by an older one that failed. .env values with escaped quotes are read whole. - MCP saved files: a file that could not be read stays protected while a folder without permission hides it, and is saved again once deleted. - The desktop app reports "no chats" only when the count was read and no model settings are stored.
This commit is contained in:
dayuan.jiang
@@ -89,6 +89,24 @@ describe("provider error texts in the stream", () => {
|
||||
// The SDK retries a refused connection twice, waiting between
|
||||
}, 20_000)
|
||||
|
||||
it("shows the server's keyless Ollama error on the web too", async () => {
|
||||
// No key, no money involved; round three hid this text
|
||||
process.env.AI_PROVIDER = "ollama"
|
||||
process.env.AI_MODEL = "llama3"
|
||||
process.env.OLLAMA_BASE_URL = "http://ollama.internal:11434/api"
|
||||
vi.stubGlobal(
|
||||
"fetch",
|
||||
vi.fn(async () => {
|
||||
throw Object.assign(new TypeError("fetch failed"), {
|
||||
cause: new Error("connect ECONNREFUSED 10.0.0.9:11434"),
|
||||
})
|
||||
}),
|
||||
)
|
||||
expect(await streamedError({})).not.toBe(
|
||||
"The provider returned an error.",
|
||||
)
|
||||
}, 20_000)
|
||||
|
||||
it("shows EdgeOne's own daily quota explanation", async () => {
|
||||
// The function answers 429, which the SDK retries with a wait;
|
||||
// the status does not decide whether the text is shown
|
||||
@@ -124,3 +142,107 @@ describe("provider error texts in the stream", () => {
|
||||
expect(message).toBe("The provider returned an error.")
|
||||
})
|
||||
})
|
||||
|
||||
describe("the output cap", () => {
|
||||
it("holds for the server's own keyless endpoints too", async () => {
|
||||
process.env.MAX_OUTPUT_TOKENS = "8000"
|
||||
const sent: string[] = []
|
||||
vi.stubGlobal(
|
||||
"fetch",
|
||||
vi.fn(async (_url: string, init?: RequestInit) => {
|
||||
sent.push(String(init?.body ?? ""))
|
||||
return new Response("{}", { status: 400 })
|
||||
}),
|
||||
)
|
||||
try {
|
||||
const res = await chat(
|
||||
new Request("http://localhost/api/chat", {
|
||||
method: "POST",
|
||||
headers: {
|
||||
"Content-Type": "application/json",
|
||||
"x-ai-provider": "ollama",
|
||||
"x-ai-base-url": "http://127.0.0.1:11434/api",
|
||||
"x-ai-model": "llama3",
|
||||
"x-max-output-tokens": "200000",
|
||||
},
|
||||
body: JSON.stringify({
|
||||
messages: [
|
||||
{
|
||||
id: "u1",
|
||||
role: "user",
|
||||
parts: [{ type: "text", text: "Draw" }],
|
||||
},
|
||||
],
|
||||
xml: "",
|
||||
}),
|
||||
}),
|
||||
)
|
||||
await res.text()
|
||||
expect(sent[0]).toContain('"max_output_tokens":8000')
|
||||
} finally {
|
||||
delete process.env.MAX_OUTPUT_TOKENS
|
||||
}
|
||||
})
|
||||
})
|
||||
|
||||
describe("a tool call that never got its result", () => {
|
||||
it("is left out of the prompt instead of failing every later message", async () => {
|
||||
// Stop while the screenshot check ran left display_diagram without
|
||||
// a result, and the chat was saved like that
|
||||
process.env.AI_PROVIDER = "openai"
|
||||
process.env.AI_MODEL = "gpt-5.5"
|
||||
process.env.OPENAI_API_KEY = "server-key"
|
||||
const sent: string[] = []
|
||||
vi.stubGlobal(
|
||||
"fetch",
|
||||
vi.fn(async (_url: string, init?: RequestInit) => {
|
||||
sent.push(String(init?.body ?? ""))
|
||||
return new Response(
|
||||
JSON.stringify({ error: { message: "x" } }),
|
||||
{
|
||||
status: 400,
|
||||
headers: { "Content-Type": "application/json" },
|
||||
},
|
||||
)
|
||||
}),
|
||||
)
|
||||
const res = await chat(
|
||||
new Request("http://localhost/api/chat", {
|
||||
method: "POST",
|
||||
headers: { "Content-Type": "application/json" },
|
||||
body: JSON.stringify({
|
||||
messages: [
|
||||
{
|
||||
id: "u1",
|
||||
role: "user",
|
||||
parts: [{ type: "text", text: "Draw a box" }],
|
||||
},
|
||||
{
|
||||
id: "a1",
|
||||
role: "assistant",
|
||||
parts: [
|
||||
{
|
||||
type: "tool-display_diagram",
|
||||
toolCallId: "call-without-result",
|
||||
state: "input-available",
|
||||
input: { xml: "<mxCell/>" },
|
||||
},
|
||||
],
|
||||
},
|
||||
{
|
||||
id: "u2",
|
||||
role: "user",
|
||||
parts: [{ type: "text", text: "Make it red" }],
|
||||
},
|
||||
],
|
||||
xml: "",
|
||||
}),
|
||||
}),
|
||||
)
|
||||
await res.text()
|
||||
// The request reached the model, without the unanswered call
|
||||
expect(sent).toHaveLength(1)
|
||||
expect(sent[0]).toContain("Make it red")
|
||||
expect(sent[0]).not.toContain("call-without-result")
|
||||
})
|
||||
})
|
||||
|
||||
Reference in New Issue
Block a user