Files
next-ai-draw-io/tests/unit/short-model-name.test.ts
T
dayuan.jiang 366d9ee4ff feat(ui): canvas-first redesign with a self-hosted draw.io
The canvas now fills the window and the chat floats on its right. draw.io is
served from our own origin (public/drawio, downloaded at build time and
checked against the release's SHA-256), so the app can drive the editor
directly:

- every AI change becomes a version card with a thumbnail; compare, restore
  or undo it, and Ctrl+Z on the canvas takes an AI change back in one step
- shapes the AI just changed are highlighted on the canvas
- selected shapes appear as a chip and are sent to the model
- the app's own canvas toolbar, shape library and page tabs replace
  draw.io's chrome
- new start screen, session rename and switcher, open .drawio files
- settings split into Models & keys, Appearance, Advanced and About
- phones switch between canvas and chat
- dark mode keeps the diagram on a white sheet and no longer flashes

With an external draw.io (NEXT_PUBLIC_DRAWIO_BASE_URL) the app falls back to
draw.io's own toolbar and the features above that need same origin are off.

The chat logic moved from the 1500-line chat panel into a chat engine plus
small UI components; zustand holds canvas, versions, settings and UI state.
2026-10-06 22:01:09 +09:00

38 lines
1.3 KiB
TypeScript

import { describe, expect, it } from "vitest"
import { shortModelName } from "@/components/model-selector"
describe("shortModelName", () => {
it("drops the provider or region prefix the logo already shows", () => {
expect(shortModelName("nvidia/nemotron-3-ultra-550b-a55b:free")).toBe(
"nemotron-3-ultra-550b-a55b:free",
)
expect(shortModelName("global.anthropic.claude-opus-5-5")).toBe(
"claude-opus-5-5",
)
expect(shortModelName("us.amazon.nova-2-lite-v1:0")).toBe(
"nova-2-lite-v1:0",
)
expect(shortModelName("openai.gpt-6-luna")).toBe("gpt-6-luna")
expect(shortModelName("@tx/deepseek-ai/deepseek-v32")).toBe(
"deepseek-v32",
)
})
it("keeps the vendor when the rest does not name it", () => {
expect(shortModelName("deepseek.r1-v1:0")).toBe("deepseek.r1-v1:0")
expect(shortModelName("us.deepseek.r1-v1:0")).toBe("deepseek.r1-v1:0")
expect(shortModelName("deepseek.v3-v1:0")).toBe("deepseek.v3-v1:0")
})
it("keeps version numbers with dots", () => {
for (const id of [
"gpt-5.5",
"glm-4.7",
"gemini-2.5-flash",
"kimi-k2.6",
]) {
expect(shortModelName(id)).toBe(id)
}
})
})