mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-10-10 03:29:50 +08:00
dayuan.jiang
0855b35ff2
Found by the PR review, each with a test that failed first: - Quota: any key header skipped it, even one the provider never reads (x-aws-access-key-id with OpenAI), so a request ran on the server's key without being counted. The check now runs after the model is resolved and uses usesServerCredentials. On main already. - usesServerCredentials read the raw base URL; "/" cleans up to none, so an Ollama request ran on the server's key past the server-model check. - SGLang's default 127.0.0.1:8000 only fills the settings form. Chat and the model list used it as a real address, so the server called its own machine even with private URLs blocked. Now a base URL is required. - With a user's OpenAI key and no base URL, the SDK read the server's OPENAI_BASE_URL. The official endpoint is now passed. On main already. - The Test button refused nothing on the server's keys (Ollama Cloud), and a 15 s timeout reported "connected, no tool call". - The model list for Ollama without a base URL came from ollama.com while chat went to the server's Ollama. - Bedrock's "Too many tokens, please wait" counted as context too long. - On the server's keys the provider's error text stays in the server log; it can name the server's AWS account, role or internal hosts. - Desktop app: the preset keys are the user's own (NEXT_AI_DRAWIO_DESKTOP), so Max Output Tokens can be raised and keyless models in settings work again. A launch that found the remembered port taken no longer replaces it, which hid the user's chats and settings for good.
39 lines
1.2 KiB
TypeScript
39 lines
1.2 KiB
TypeScript
// @vitest-environment node
|
|
import { mkdtempSync, readFileSync, writeFileSync } from "node:fs"
|
|
import { tmpdir } from "node:os"
|
|
import { join } from "node:path"
|
|
import { beforeEach, describe, expect, it, vi } from "vitest"
|
|
|
|
const userData = vi.hoisted(() => ({ dir: "" }))
|
|
vi.mock("electron", () => ({
|
|
app: { isPackaged: true, getPath: () => userData.dir },
|
|
}))
|
|
|
|
import { saveServerPort } from "@/electron/main/port-manager"
|
|
|
|
const savedPort = () =>
|
|
JSON.parse(readFileSync(join(userData.dir, "server-port.json"), "utf-8"))
|
|
.port
|
|
|
|
beforeEach(() => {
|
|
userData.dir = mkdtempSync(join(tmpdir(), "port-manager-"))
|
|
})
|
|
|
|
describe("saveServerPort", () => {
|
|
it("remembers the port of the first launch", () => {
|
|
saveServerPort(13370)
|
|
expect(savedPort()).toBe(13370)
|
|
})
|
|
|
|
it("keeps the remembered port when a launch had to use another", () => {
|
|
// The app's data lives under the remembered port's origin; going
|
|
// back to it once it is free brings the chats and settings back
|
|
writeFileSync(
|
|
join(userData.dir, "server-port.json"),
|
|
JSON.stringify({ port: 61337 }),
|
|
)
|
|
saveServerPort(13371)
|
|
expect(savedPort()).toBe(61337)
|
|
})
|
|
})
|