feat: 新增模型管理功能（供应商 + 模型 CRUD）

- 新增 providers/models 数据库表、迁移和数据访问层 - 新增 15 个后端 API 路由（供应商/模型 CRUD + 连通性测试） - 新增 AI 服务层（registry.ts: buildProviderRegistry + testProviderConnection） - 新增前端模型管理页面（Tabs: 供应商/模型，含表格、表单、工具栏） - 新增前端 hooks（use-providers, use-models） - 新增共享类型和 MODEL_CAPABILITIES 常量 - 新增 10 个测试文件（66 个测试用例，4 个因 bun test ESM 兼容问题待修复） - 更新开发文档（architecture, backend, frontend） - 附带 apply-review 修复：统一错误响应、提取共享常量、清理重复测试注意：registry.test.ts 中 4 个测试因 bun test 无法解析 createProviderRegistry ESM 导出而失败，详情见 context.md
2026-05-29 12:40:10 +08:00
parent 2ea4bd4410
commit 933c2133f0
56 changed files with 4706 additions and 9 deletions
--- a/src/server/db/models.ts
+++ b/src/server/db/models.ts
@@ -0,0 +1,237 @@
+import type Database from "bun:sqlite";
+
+import { and, desc, eq, like, or, sql } from "drizzle-orm";
+import { drizzle } from "drizzle-orm/bun-sqlite";
+
+import type { CreateModelRequest, Model, ModelCapability, UpdateModelRequest } from "../../shared/api";
+
+import { models, providers } from "./schema";
+
+export function createModel(
+  raw: Database,
+  request: CreateModelRequest,
+): { error: string; status: number } | { model: Model } {
+  const db = wrap(raw);
+
+  const provider = db.select().from(providers).where(eq(providers.id, request.providerId)).get();
+  if (!provider) return { error: "供应商不存在", status: 400 };
+
+  const name = request.name.trim();
+  if (!name) return { error: "模型名称不能为空", status: 400 };
+
+  const modelId = request.modelId.trim();
+  if (!modelId) return { error: "模型 ID 不能为空", status: 400 };
+
+  const capabilities = request.capabilities;
+  if (!capabilities || capabilities.length === 0) {
+    return { error: "至少选择一个能力标签", status: 400 };
+  }
+
+  const id = crypto.randomUUID();
+  const now = new Date().toISOString();
+
+  try {
+    db.insert(models)
+      .values({
+        capabilities: JSON.stringify(capabilities),
+        contextLength: request.contextLength ?? null,
+        createdAt: now,
+        enabled: true,
+        id,
+        maxOutputTokens: request.maxOutputTokens ?? null,
+        modelId,
+        name,
+        providerId: request.providerId,
+        updatedAt: now,
+      })
+      .run();
+  } catch (e: unknown) {
+    const msg = e instanceof Error ? e.message : String(e);
+    if (msg.includes("UNIQUE constraint")) {
+      return { error: "该供应商下模型 ID 已存在", status: 409 };
+    }
+    throw e;
+  }
+
+  const row = db.select().from(models).where(eq(models.id, id)).get();
+  return { model: toModel(row!) };
+}
+
+export function deleteModel(raw: Database, id: string): { error: string; status: number } | { success: true } {
+  const db = wrap(raw);
+  const existing = db.select().from(models).where(eq(models.id, id)).get();
+  if (!existing) return { error: "模型不存在", status: 404 };
+
+  db.delete(models).where(eq(models.id, id)).run();
+  return { success: true };
+}
+
+export function disableModel(raw: Database, id: string): { error: string; status: number } | { model: Model } {
+  const db = wrap(raw);
+  const existing = db.select().from(models).where(eq(models.id, id)).get();
+  if (!existing) return { error: "模型不存在", status: 404 };
+  if (!existing.enabled) return { error: "模型已禁用", status: 409 };
+
+  const now = new Date().toISOString();
+  db.update(models).set({ enabled: false, updatedAt: now }).where(eq(models.id, id)).run();
+
+  const updated = db.select().from(models).where(eq(models.id, id)).get();
+  return { model: toModel(updated!) };
+}
+
+export function enableModel(raw: Database, id: string): { error: string; status: number } | { model: Model } {
+  const db = wrap(raw);
+  const existing = db.select().from(models).where(eq(models.id, id)).get();
+  if (!existing) return { error: "模型不存在", status: 404 };
+  if (existing.enabled) return { error: "模型已启用", status: 409 };
+
+  const now = new Date().toISOString();
+  db.update(models).set({ enabled: true, updatedAt: now }).where(eq(models.id, id)).run();
+
+  const updated = db.select().from(models).where(eq(models.id, id)).get();
+  return { model: toModel(updated!) };
+}
+
+export function getModel(raw: Database, id: string): { error: string; status: number } | { model: Model } {
+  const db = wrap(raw);
+  const row = db.select().from(models).where(eq(models.id, id)).get();
+
+  if (!row) return { error: "模型不存在", status: 404 };
+  return { model: toModel(row) };
+}
+
+export function getModelsByProviderId(raw: Database, providerId: string): number {
+  const db = wrap(raw);
+  const result = db
+    .select({ count: sql<number>`count(*)` })
+    .from(models)
+    .where(eq(models.providerId, providerId))
+    .get();
+  return Number(result?.count ?? 0);
+}
+
+export function listModels(
+  raw: Database,
+  options: { keyword?: string; page: number; pageSize: number; providerId?: string },
+): { items: Model[]; page: number; pageSize: number; total: number } {
+  const db = wrap(raw);
+  const conditions = [];
+
+  if (options.providerId) {
+    conditions.push(eq(models.providerId, options.providerId));
+  }
+
+  if (options.keyword) {
+    const pattern = `%${options.keyword}%`;
+    conditions.push(or(like(models.name, pattern), like(models.modelId, pattern))!);
+  }
+
+  const where = conditions.length > 0 ? and(...conditions) : undefined;
+
+  const countResult = db
+    .select({ count: sql<number>`count(*)` })
+    .from(models)
+    .where(where)
+    .get();
+
+  const total = Number(countResult?.count ?? 0);
+
+  const rows = db
+    .select()
+    .from(models)
+    .where(where)
+    .orderBy(desc(models.createdAt))
+    .limit(options.pageSize)
+    .offset((options.page - 1) * options.pageSize)
+    .all();
+
+  return {
+    items: rows.map(toModel),
+    page: options.page,
+    pageSize: options.pageSize,
+    total,
+  };
+}
+
+export function updateModel(
+  raw: Database,
+  id: string,
+  request: UpdateModelRequest,
+): { error: string; status: number } | { model: Model } {
+  const db = wrap(raw);
+  const existing = db.select().from(models).where(eq(models.id, id)).get();
+  if (!existing) return { error: "模型不存在", status: 404 };
+
+  const updates: Partial<typeof models.$inferInsert> = {
+    updatedAt: new Date().toISOString(),
+  };
+
+  const name = request.name?.trim();
+  if (name === "") return { error: "模型名称不能为空", status: 400 };
+  if (name !== undefined && name !== existing.name) {
+    updates.name = name;
+  }
+
+  const modelId = request.modelId?.trim();
+  if (modelId === "") return { error: "模型 ID 不能为空", status: 400 };
+  if (modelId !== undefined) {
+    updates.modelId = modelId;
+  }
+
+  if (request.providerId !== undefined) {
+    const provider = db.select().from(providers).where(eq(providers.id, request.providerId)).get();
+    if (!provider) return { error: "供应商不存在", status: 400 };
+    updates.providerId = request.providerId;
+  }
+
+  if (request.capabilities !== undefined) {
+    if (request.capabilities.length === 0) {
+      return { error: "至少选择一个能力标签", status: 400 };
+    }
+    updates.capabilities = JSON.stringify(request.capabilities);
+  }
+
+  if (request.contextLength !== undefined) {
+    updates.contextLength = request.contextLength;
+  }
+
+  if (request.maxOutputTokens !== undefined) {
+    updates.maxOutputTokens = request.maxOutputTokens;
+  }
+
+  if (Object.keys(updates).length === 1 && updates.updatedAt) {
+    return { model: toModel(existing) };
+  }
+
+  try {
+    db.update(models).set(updates).where(eq(models.id, id)).run();
+  } catch (e: unknown) {
+    const msg = e instanceof Error ? e.message : String(e);
+    if (msg.includes("UNIQUE constraint")) {
+      return { error: "该供应商下模型 ID 已存在", status: 409 };
+    }
+    throw e;
+  }
+
+  const updated = db.select().from(models).where(eq(models.id, id)).get();
+  return { model: toModel(updated!) };
+}
+
+function toModel(row: typeof models.$inferSelect): Model {
+  return {
+    capabilities: JSON.parse(row.capabilities) as ModelCapability[],
+    contextLength: row.contextLength,
+    createdAt: row.createdAt,
+    enabled: row.enabled,
+    id: row.id,
+    maxOutputTokens: row.maxOutputTokens,
+    modelId: row.modelId,
+    name: row.name,
+    providerId: row.providerId,
+    updatedAt: row.updatedAt,
+  };
+}
+
+function wrap(raw: Database) {
+  return drizzle(raw);
+}
--- a/src/server/db/providers.ts
+++ b/src/server/db/providers.ts
@@ -0,0 +1,202 @@
+import type Database from "bun:sqlite";
+
+import { and, desc, eq, like, sql } from "drizzle-orm";
+import { drizzle } from "drizzle-orm/bun-sqlite";
+
+import type { CreateProviderRequest, Provider, UpdateProviderRequest } from "../../shared/api";
+
+import { providers } from "./schema";
+
+export function createProvider(
+  raw: Database,
+  request: CreateProviderRequest,
+): { error: string; status: number } | { provider: Provider } {
+  const db = wrap(raw);
+  const name = request.name.trim();
+  if (!name) return { error: "供应商名称不能为空", status: 400 };
+
+  const baseUrl = request.baseUrl.trim();
+  if (!baseUrl) return { error: "Base URL 不能为空", status: 400 };
+
+  const apiKey = request.apiKey.trim();
+  if (!apiKey) return { error: "API Key 不能为空", status: 400 };
+
+  const id = crypto.randomUUID();
+  const now = new Date().toISOString();
+
+  try {
+    db.insert(providers)
+      .values({
+        apiKey,
+        baseUrl,
+        createdAt: now,
+        enabled: true,
+        id,
+        name,
+        type: request.type,
+        updatedAt: now,
+      })
+      .run();
+  } catch (e: unknown) {
+    const msg = e instanceof Error ? e.message : String(e);
+    if (msg.includes("UNIQUE constraint")) {
+      return { error: "供应商名称已存在", status: 409 };
+    }
+    throw e;
+  }
+
+  const row = db.select().from(providers).where(eq(providers.id, id)).get();
+  return { provider: toProvider(row!) };
+}
+
+export function deleteProvider(raw: Database, id: string): { error: string; status: number } | { success: true } {
+  const db = wrap(raw);
+  const existing = db.select().from(providers).where(eq(providers.id, id)).get();
+  if (!existing) return { error: "供应商不存在", status: 404 };
+
+  db.delete(providers).where(eq(providers.id, id)).run();
+  return { success: true };
+}
+
+export function disableProvider(raw: Database, id: string): { error: string; status: number } | { provider: Provider } {
+  const db = wrap(raw);
+  const existing = db.select().from(providers).where(eq(providers.id, id)).get();
+  if (!existing) return { error: "供应商不存在", status: 404 };
+  if (!existing.enabled) return { error: "供应商已禁用", status: 409 };
+
+  const now = new Date().toISOString();
+  db.update(providers).set({ enabled: false, updatedAt: now }).where(eq(providers.id, id)).run();
+
+  const updated = db.select().from(providers).where(eq(providers.id, id)).get();
+  return { provider: toProvider(updated!) };
+}
+
+export function enableProvider(raw: Database, id: string): { error: string; status: number } | { provider: Provider } {
+  const db = wrap(raw);
+  const existing = db.select().from(providers).where(eq(providers.id, id)).get();
+  if (!existing) return { error: "供应商不存在", status: 404 };
+  if (existing.enabled) return { error: "供应商已启用", status: 409 };
+
+  const now = new Date().toISOString();
+  db.update(providers).set({ enabled: true, updatedAt: now }).where(eq(providers.id, id)).run();
+
+  const updated = db.select().from(providers).where(eq(providers.id, id)).get();
+  return { provider: toProvider(updated!) };
+}
+
+export function getProvider(raw: Database, id: string): { error: string; status: number } | { provider: Provider } {
+  const db = wrap(raw);
+  const row = db.select().from(providers).where(eq(providers.id, id)).get();
+
+  if (!row) return { error: "供应商不存在", status: 404 };
+  return { provider: toProvider(row) };
+}
+
+export function listProviders(
+  raw: Database,
+  options: { keyword?: string; page: number; pageSize: number },
+): { items: Provider[]; page: number; pageSize: number; total: number } {
+  const db = wrap(raw);
+  const conditions = [];
+
+  if (options.keyword) {
+    const pattern = `%${options.keyword}%`;
+    conditions.push(like(providers.name, pattern));
+  }
+
+  const where = conditions.length > 0 ? and(...conditions) : undefined;
+
+  const countResult = db
+    .select({ count: sql<number>`count(*)` })
+    .from(providers)
+    .where(where)
+    .get();
+
+  const total = Number(countResult?.count ?? 0);
+
+  const rows = db
+    .select()
+    .from(providers)
+    .where(where)
+    .orderBy(desc(providers.createdAt))
+    .limit(options.pageSize)
+    .offset((options.page - 1) * options.pageSize)
+    .all();
+
+  return {
+    items: rows.map(toProvider),
+    page: options.page,
+    pageSize: options.pageSize,
+    total,
+  };
+}
+
+export function updateProvider(
+  raw: Database,
+  id: string,
+  request: UpdateProviderRequest,
+): { error: string; status: number } | { provider: Provider } {
+  const db = wrap(raw);
+  const existing = db.select().from(providers).where(eq(providers.id, id)).get();
+  if (!existing) return { error: "供应商不存在", status: 404 };
+
+  const updates: Partial<typeof providers.$inferInsert> = {
+    updatedAt: new Date().toISOString(),
+  };
+
+  const name = request.name?.trim();
+  if (name === "") return { error: "供应商名称不能为空", status: 400 };
+  if (name !== undefined && name !== existing.name) {
+    updates.name = name;
+  }
+
+  const baseUrl = request.baseUrl?.trim();
+  if (baseUrl === "") return { error: "Base URL 不能为空", status: 400 };
+  if (baseUrl !== undefined) {
+    updates.baseUrl = baseUrl;
+  }
+
+  const apiKey = request.apiKey?.trim();
+  if (apiKey === "") return { error: "API Key 不能为空", status: 400 };
+  if (apiKey !== undefined) {
+    updates.apiKey = apiKey;
+  }
+
+  if (request.type !== undefined) {
+    updates.type = request.type;
+  }
+
+  if (Object.keys(updates).length === 1 && updates.updatedAt) {
+    return { provider: toProvider(existing) };
+  }
+
+  try {
+    db.update(providers).set(updates).where(eq(providers.id, id)).run();
+  } catch (e: unknown) {
+    const msg = e instanceof Error ? e.message : String(e);
+    if (msg.includes("UNIQUE constraint")) {
+      return { error: "供应商名称已存在", status: 409 };
+    }
+    throw e;
+  }
+
+  const updated = db.select().from(providers).where(eq(providers.id, id)).get();
+  return { provider: toProvider(updated!) };
+}
+
+function toProvider(row: typeof providers.$inferSelect): Provider {
+  return {
+    apiKey: row.apiKey,
+    baseUrl: row.baseUrl,
+    createdAt: row.createdAt,
+    enabled: row.enabled,
+    id: row.id,
+    name: row.name,
+    type: row.type,
+    updatedAt: row.updatedAt,
+  };
+}
+
+function wrap(raw: Database) {
+  return drizzle(raw);
+}
--- a/src/server/db/schema.ts
+++ b/src/server/db/schema.ts
@@ -1,4 +1,4 @@
-import { sqliteTable, text } from "drizzle-orm/sqlite-core";
+import { index, integer, sqliteTable, text, uniqueIndex } from "drizzle-orm/sqlite-core";

 export const projects = sqliteTable("projects", {
  archivedAt: text("archived_at"),
@@ -12,6 +12,41 @@ export const projects = sqliteTable("projects", {
  updatedAt: text("updated_at").notNull(),
 });

+export const providers = sqliteTable("providers", {
+  apiKey: text("api_key").notNull(),
+  baseUrl: text("base_url").notNull(),
+  createdAt: text("created_at").notNull(),
+  enabled: integer("enabled", { mode: "boolean" }).notNull().default(true),
+  id: text("id").primaryKey(),
+  name: text("name").notNull().unique(),
+  type: text("type", { enum: ["anthropic", "openai", "openai-compatible"] })
+    .notNull()
+    .default("openai-compatible"),
+  updatedAt: text("updated_at").notNull(),
+});
+
+export const models = sqliteTable(
+  "models",
+  {
+    capabilities: text("capabilities").notNull(),
+    contextLength: integer("context_length"),
+    createdAt: text("created_at").notNull(),
+    enabled: integer("enabled", { mode: "boolean" }).notNull().default(true),
+    id: text("id").primaryKey(),
+    maxOutputTokens: integer("max_output_tokens"),
+    modelId: text("model_id").notNull(),
+    name: text("name").notNull(),
+    providerId: text("provider_id")
+      .notNull()
+      .references(() => providers.id),
+    updatedAt: text("updated_at").notNull(),
+  },
+  (table) => [
+    uniqueIndex("models_provider_id_model_id_unique").on(table.providerId, table.modelId),
+    index("models_provider_id_idx").on(table.providerId),
+  ],
+);
+
 export const schemaMigrations = sqliteTable("schema_migrations", {
  appliedAt: text("applied_at").notNull(),
  checksum: text("checksum").notNull(),