Files
next-ai-draw-io/tests/unit/short-model-name.test.ts
T
Dayuan Jiang c513d1a0c2 feat(ui): canvas-first redesign with a self-hosted draw.io (#957)
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
- draw.io's own simple UI: its toolbar, menus (auto layout, Mermaid, CSV
  import) and page tabs; the shape library starts closed
- new start screen, session rename and switcher, open .drawio files
- settings split into Models & keys, Appearance, Advanced and About
- Bedrock in the model settings also takes an API key instead of access keys
- phones switch between canvas and chat
- dark mode uses draw.io's own dark mode, with neutral greys around it

With an external draw.io (NEXT_PUBLIC_DRAWIO_BASE_URL) 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-08 16:20:15 +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)
}
})
})