mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-10-08 02:37:46 +08:00
dayuan.jiang
6236124338
The "Fetch models" button asks the provider for its models (OpenAI-style /models, Anthropic, Google, Ollama, OpenRouter, Vercel Gateway, AIHubMix) and shows them in a searchable picker. This replaces the route that only worked for AIHubMix. A snapshot of models.dev (MIT) says which models support tool calls. Models without them get a "no tool calls" badge in the picker and a hint in the model list, since drawing needs tool calls. Refresh the snapshot with scripts/update-model-catalog.mjs.
86 lines
2.8 KiB
TypeScript
86 lines
2.8 KiB
TypeScript
import { expect, type Page, test } from "@playwright/test"
|
|
import { getIframe } from "./lib/fixtures"
|
|
|
|
// qwen-mt-plus is a translation model; models.dev lists no tool calls for it
|
|
const CONFIG = {
|
|
version: 1,
|
|
providers: [
|
|
{
|
|
id: "p1",
|
|
provider: "qwen",
|
|
apiKey: "test-key",
|
|
models: [{ id: "m1", modelId: "qwen-mt-plus" }],
|
|
},
|
|
],
|
|
}
|
|
|
|
async function openQwenSettings(page: Page) {
|
|
await page.addInitScript((config) => {
|
|
localStorage.setItem(
|
|
"next-ai-draw-io-model-configs",
|
|
JSON.stringify(config),
|
|
)
|
|
}, CONFIG)
|
|
await page.goto("/", { waitUntil: "networkidle" })
|
|
await getIframe(page).waitFor({ state: "visible", timeout: 30000 })
|
|
await page.locator("button:has(svg.lucide-bot)").first().click()
|
|
await page.getByText("Configure Models...").click()
|
|
const dialog = page.locator('[role="dialog"]')
|
|
await dialog.getByText("Qwen (Alibaba)").first().click()
|
|
return dialog
|
|
}
|
|
|
|
test("fetches the provider's models and adds one from the picker", async ({
|
|
page,
|
|
}) => {
|
|
let request: Record<string, unknown> | undefined
|
|
await page.route("**/api/provider-models", async (route) => {
|
|
request = route.request().postDataJSON()
|
|
await route.fulfill({
|
|
json: {
|
|
models: [
|
|
{ id: "qwen-new-max", tools: true },
|
|
{ id: "qwen-text-only", tools: false },
|
|
],
|
|
},
|
|
})
|
|
})
|
|
const dialog = await openQwenSettings(page)
|
|
|
|
await expect(
|
|
dialog.getByText("may not be able to draw").first(),
|
|
).toBeVisible()
|
|
|
|
await dialog
|
|
.getByRole("button", { name: "Fetch models from the provider" })
|
|
.click()
|
|
const picker = page.locator('[role="listbox"]')
|
|
await expect(picker.getByText("qwen-new-max")).toBeVisible()
|
|
await expect(
|
|
picker.getByRole("option", { name: /qwen-text-only/ }),
|
|
).toContainText("no tool calls")
|
|
expect(request).toMatchObject({ provider: "qwen", apiKey: "test-key" })
|
|
|
|
await page.getByPlaceholder("Search models...").fill("new-max")
|
|
await picker.getByText("qwen-new-max").click()
|
|
await expect(dialog.locator('input[title="qwen-new-max"]')).toBeVisible()
|
|
})
|
|
|
|
test("shows a hint when the provider rejects the key", async ({ page }) => {
|
|
await page.route("**/api/provider-models", (route) =>
|
|
route.fulfill({
|
|
status: 401,
|
|
json: { code: "invalid_api_key", error: "Incorrect API key" },
|
|
}),
|
|
)
|
|
const dialog = await openQwenSettings(page)
|
|
await dialog
|
|
.getByRole("button", { name: "Fetch models from the provider" })
|
|
.click()
|
|
await expect(
|
|
dialog.getByText(
|
|
"The provider rejected the API key. Check it in model settings. Incorrect API key",
|
|
),
|
|
).toBeVisible()
|
|
})
|