Fix DB schema (#135)

src/components/CreateCustomModelDialog.tsx

@@ -27,8 +27,8 @@ export function CreateCustomModelDialog({
   onSuccess,
   providerId,
 }: CreateCustomModelDialogProps) {
-  const [id, setId] = useState("");
-  const [name, setName] = useState("");
+  const [apiName, setApiName] = useState("");
+  const [displayName, setDisplayName] = useState("");
   const [description, setDescription] = useState("");
   const [maxOutputTokens, setMaxOutputTokens] = useState<string>("");
   const [contextWindow, setContextWindow] = useState<string>("");
@@ -38,8 +38,8 @@ export function CreateCustomModelDialog({
   const mutation = useMutation({
     mutationFn: async () => {
       const params = {
-        id,
-        name,
+        apiName,
+        displayName,
         providerId,
         description: description || undefined,
         maxOutputTokens: maxOutputTokens
@@ -48,8 +48,9 @@ export function CreateCustomModelDialog({
         contextWindow: contextWindow ? parseInt(contextWindow, 10) : undefined,
       };
 
-      if (!params.id) throw new Error("Model ID is required");
-      if (!params.name) throw new Error("Model Name is required");
+      if (!params.apiName) throw new Error("Model API name is required");
+      if (!params.displayName)
+        throw new Error("Model display name is required");
       if (maxOutputTokens && isNaN(params.maxOutputTokens ?? NaN))
         throw new Error("Max Output Tokens must be a valid number");
       if (contextWindow && isNaN(params.contextWindow ?? NaN))
@@ -69,8 +70,8 @@ export function CreateCustomModelDialog({
   });
 
   const resetForm = () => {
-    setId("");
-    setName("");
+    setApiName("");
+    setDisplayName("");
     setDescription("");
     setMaxOutputTokens("");
     setContextWindow("");
@@ -105,9 +106,9 @@ export function CreateCustomModelDialog({
               </Label>
               <Input
                 id="model-id"
-                value={id}
+                value={apiName}
                 onChange={(e: React.ChangeEvent<HTMLInputElement>) =>
-                  setId(e.target.value)
+                  setApiName(e.target.value)
                 }
                 className="col-span-3"
                 placeholder="This must match the model expected by the API"
@@ -121,9 +122,9 @@ export function CreateCustomModelDialog({
               </Label>
               <Input
                 id="model-name"
-                value={name}
+                value={displayName}
                 onChange={(e: React.ChangeEvent<HTMLInputElement>) =>
-                  setName(e.target.value)
+                  setDisplayName(e.target.value)
                 }
                 className="col-span-3"
                 placeholder="Human-friendly name for the model"

src/components/settings/ModelsSection.tsx

@@ -46,7 +46,7 @@ export function ModelsSection({ providerId }: ModelsSectionProps) {
         <div className="mt-4 space-y-3">
           {models.map((model) => (
             <div
-              key={model.name}
+              key={model.apiName}
               className="p-4 bg-white dark:bg-gray-800 border border-gray-200 dark:border-gray-700 rounded-lg shadow-sm"
             >
               <div className="flex justify-between items-center">
@@ -56,7 +56,7 @@ export function ModelsSection({ providerId }: ModelsSectionProps) {
                 {/* Optional: Add an edit/delete button here later */}
               </div>
               <p className="text-sm text-gray-500 dark:text-gray-400 italic">
-                {model.name}
+                {model.apiName}
               </p>
               {model.description && (
                 <p className="text-sm text-gray-600 dark:text-gray-300 mt-1">

src/db/schema.ts

@@ -82,8 +82,9 @@ export const language_model_providers = sqliteTable(
 );
 
 export const language_models = sqliteTable("language_models", {
-  id: text("id").primaryKey(),
-  name: text("name").notNull(),
+  id: integer("id").primaryKey({ autoIncrement: true }),
+  displayName: text("display_name").notNull(),
+  apiName: text("api_name").notNull(),
   provider_id: text("provider_id")
     .notNull()
     .references(() => language_model_providers.id, { onDelete: "cascade" }),

src/ipc/handlers/language_model_handlers.ts

@@ -87,8 +87,8 @@ export function registerLanguageModelHandlers() {
       params: CreateCustomLanguageModelParams,
     ): Promise<void> => {
       const {
-        id,
-        name,
+        apiName,
+        displayName,
         providerId,
         description,
         maxOutputTokens,
@@ -96,11 +96,11 @@ export function registerLanguageModelHandlers() {
       } = params;
 
       // Validation
-      if (!id) {
-        throw new Error("Model ID is required");
+      if (!apiName) {
+        throw new Error("Model API name is required");
       }
-      if (!name) {
-        throw new Error("Model name is required");
+      if (!displayName) {
+        throw new Error("Model display name is required");
       }
       if (!providerId) {
         throw new Error("Provider ID is required");
@@ -117,21 +117,10 @@ export function registerLanguageModelHandlers() {
         throw new Error(`Provider with ID "${providerId}" not found`);
       }
 
-      // Check if model ID already exists
-      const existingModel = db
-        .select()
-        .from(languageModelsSchema)
-        .where(eq(languageModelsSchema.id, id))
-        .get();
-
-      if (existingModel) {
-        throw new Error(`A model with ID "${id}" already exists`);
-      }
-
       // Insert the new model
       await db.insert(languageModelsSchema).values({
-        id,
-        name,
+        displayName,
+        apiName,
         provider_id: providerId,
         description: description || null,
         max_output_tokens: maxOutputTokens || null,

src/ipc/ipc_types.ts

@@ -144,16 +144,26 @@ export interface LanguageModelProvider {
   type: "custom" | "local" | "cloud";
 }
 
-export interface LanguageModel {
-  id: string;
-  name: string;
-  displayName: string;
-  description: string;
-  tag?: string;
-  maxOutputTokens?: number;
-  contextWindow?: number;
-  type: "local" | "cloud" | "custom";
-}
+export type LanguageModel =
+  | {
+      id: number;
+      apiName: string;
+      displayName: string;
+      description: string;
+      tag?: string;
+      maxOutputTokens?: number;
+      contextWindow?: number;
+      type: "custom";
+    }
+  | {
+      apiName: string;
+      displayName: string;
+      description: string;
+      tag?: string;
+      maxOutputTokens?: number;
+      contextWindow?: number;
+      type: "local" | "cloud";
+    };
 
 export interface CreateCustomLanguageModelProviderParams {
   id: string;
@@ -163,8 +173,8 @@ export interface CreateCustomLanguageModelProviderParams {
 }
 
 export interface CreateCustomLanguageModelParams {
-  id: string;
-  name: string;
+  apiName: string;
+  displayName: string;
   providerId: string;
   description?: string;
   maxOutputTokens?: number;

src/ipc/shared/language_model_helpers.ts

@@ -157,7 +157,7 @@ export async function getLanguageModels(obj: {
       const models = MODEL_OPTIONS[providerId as RegularModelProvider] || [];
       return models.map((model) => ({
         ...model,
-        id: model.name,
+        apiName: model.name,
         type: "cloud",
       }));
     } else {
@@ -174,7 +174,8 @@ export async function getLanguageModels(obj: {
         .select({
           id: languageModelsSchema.id,
           // Map DB columns to LanguageModel fields
-          name: languageModelsSchema.name,
+          displayName: languageModelsSchema.displayName,
+          apiName: languageModelsSchema.apiName,
           // No display_name in DB, use name instead
           description: languageModelsSchema.description,
           // No tag in DB
@@ -186,7 +187,6 @@ export async function getLanguageModels(obj: {
 
       return customModelsDb.map((model) => ({
         ...model,
-        displayName: model.name, // Use name as displayName for custom models
         // Ensure possibly null fields are handled, provide defaults or undefined if needed
         description: model.description ?? "",
         tag: undefined, // No tag for custom models from DB