diff --git a/app/api/admin/ai-providers/route.ts b/app/api/admin/ai-providers/route.ts index 237b329..10b4346 100644 --- a/app/api/admin/ai-providers/route.ts +++ b/app/api/admin/ai-providers/route.ts @@ -11,7 +11,7 @@ import { ChatSDKError } from "@/lib/errors"; const createProviderSchema = z.object({ name: z.string().min(1).max(128), - type: z.enum(["alibaba", "openai", "custom", "local", "gemini", "qianwen-video", "volcano-vision", "volcano-video", "alibaba-image"]), + type: z.enum(["alibaba", "openai", "custom", "local", "gemini", "huanke-gemini", "qianwen-video", "volcano-vision", "volcano-video", "alibaba-image"]), baseUrl: z.string().min(1), // 改为简单的字符串验证,支持 localhost apiKey: z.string().min(1), isEnabled: z.boolean().optional(), @@ -20,7 +20,7 @@ const createProviderSchema = z.object({ const updateProviderSchema = z.object({ id: z.string().uuid(), name: z.string().min(1).max(128).optional(), - type: z.enum(["alibaba", "openai", "custom", "local", "gemini", "qianwen-video", "volcano-vision", "volcano-video", "alibaba-image"]).optional(), + type: z.enum(["alibaba", "openai", "custom", "local", "gemini", "huanke-gemini", "qianwen-video", "volcano-vision", "volcano-video", "alibaba-image"]).optional(), baseUrl: z.string().min(1).optional(), // 改为简单的字符串验证,支持 localhost apiKey: z.string().min(1).optional(), isEnabled: z.boolean().optional(), diff --git a/lib/db/schema.ts b/lib/db/schema.ts index 9d69c7d..4673eaf 100644 --- a/lib/db/schema.ts +++ b/lib/db/schema.ts @@ -215,7 +215,7 @@ export type ToolConfig = InferSelectModel; export const aiProvider = pgTable("AiProvider", { id: uuid("id").primaryKey().notNull().defaultRandom(), name: varchar("name", { length: 128 }).notNull(), // 提供商名称,如 "阿里云通义千问" - type: varchar("type", { enum: ["alibaba", "openai", "custom", "local", "gemini", "qianwen-video", "volcano-vision", "volcano-video", "alibaba-image"] }) + type: varchar("type", { enum: ["alibaba", "openai", "custom", "local", "gemini", "huanke-gemini", "qianwen-video", "volcano-vision", "volcano-video", "alibaba-image"] }) .notNull() .default("custom"), // 提供商类型 baseUrl: text("baseUrl").notNull(), // API 基础 URL