mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-10-13 04:59:51 +08:00
dayuan.jiang
8ca2d03e7f
- lib/system-prompts.ts: delete the extended prompt kept only to reach the 4096-token cache threshold of two old Claude models; describe the current canvas-first UI in the App Features section - app/api/chat/route.ts: remove the server-side example cache (the browser answers examples itself), the ENABLE_HISTORY_XML_REPLACE option and its helper, the unread messageMetadata fields; the DEBUG_LLM_PAYLOAD blocks become one describeMessages summary - app/api/log-feedback and log-save share scoreLatestTrace in lib/langfuse.ts - app/api/admin/test-model and validate-model call a shared testModelConnection() instead of one route faking a request to the other and passing trust through the body - lib/dynamo-quota-manager.ts reads the failing item from the conditional update instead of a second GetItem - lib/ai-providers.ts: geminiThinkingOptions helper for the google and vertexai cases, drop the unused fourth resolveBaseURL parameter - edge-functions: always stream, shrink messageSchema, drop model aliases nothing sends - delete app/api/chat/xml_guide.md, parseOutputTokenLimit, unused type exports and export keywords on file-internal symbols - env.example: drop variables nothing reads, document the quota, validation and analytics variables, mention the apiKeyEnv array form (also in docs/*/ai-providers.md)
78 lines
2.5 KiB
TypeScript
78 lines
2.5 KiB
TypeScript
// @vitest-environment node
|
|
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest"
|
|
import { onRequest } from "@/edge-functions/api/edgeai/chat/completions"
|
|
|
|
// EdgeOne's global AI object, answering with an empty stream
|
|
beforeEach(() => {
|
|
vi.stubGlobal("AI", {
|
|
chatCompletions: async () => new ReadableStream(),
|
|
})
|
|
})
|
|
afterEach(() => {
|
|
vi.unstubAllGlobals()
|
|
})
|
|
|
|
function request(headers: Record<string, string>): Request {
|
|
return new Request("http://localhost/api/edgeai/chat/completions", {
|
|
method: "POST",
|
|
headers,
|
|
body: JSON.stringify({ messages: [{ role: "user", content: "hi" }] }),
|
|
})
|
|
}
|
|
|
|
const json = { "Content-Type": "application/json" }
|
|
|
|
// An accepted request streams the model's reply
|
|
function expectStream(res: Response) {
|
|
expect(res.status).toBe(200)
|
|
expect(res.headers.get("content-type")).toContain("text/event-stream")
|
|
}
|
|
|
|
describe("EdgeOne chat completions function", () => {
|
|
it("sends no CORS headers", async () => {
|
|
const res = await onRequest({ request: request(json), env: {} })
|
|
expectStream(res)
|
|
expect(res.headers.get("access-control-allow-origin")).toBeNull()
|
|
})
|
|
|
|
it("rejects non-JSON requests", async () => {
|
|
const res = await onRequest({
|
|
request: request({ "Content-Type": "text/plain" }),
|
|
env: {},
|
|
})
|
|
expect(res.status).toBe(400)
|
|
// A plain-text type that only mentions JSON needs no CORS preflight
|
|
const disguised = await onRequest({
|
|
request: request({
|
|
"Content-Type": "text/plain; x=application/json",
|
|
}),
|
|
env: {},
|
|
})
|
|
expect(disguised.status).toBe(400)
|
|
const withCharset = await onRequest({
|
|
request: request({
|
|
"Content-Type": "application/json; charset=utf-8",
|
|
}),
|
|
env: {},
|
|
})
|
|
expectStream(withCharset)
|
|
})
|
|
|
|
it("checks the access code when ACCESS_CODE_LIST is set", async () => {
|
|
const env = { ACCESS_CODE_LIST: "secret" }
|
|
const missing = await onRequest({ request: request(json), env })
|
|
expect(missing.status).toBe(401)
|
|
|
|
const ok = await onRequest({
|
|
request: request({ ...json, "x-access-code": "secret" }),
|
|
env,
|
|
})
|
|
expectStream(ok)
|
|
})
|
|
|
|
it("lets requests through when env is unavailable", async () => {
|
|
const res = await onRequest({ request: request(json) })
|
|
expectStream(res)
|
|
})
|
|
})
|