From a762629dee7d404b7144e207b5a168d170ee477e Mon Sep 17 00:00:00 2001
From: Logan Yang <logancyang@gmail.com>
Date: Tue, 25 Mar 2025 16:29:31 -0700
Subject: [PATCH] Add projectEnabled flag to CustomModel interface

---
 src/aiParams.ts                                | 1 +
 src/components/chat-components/ProjectList.tsx | 2 +-
 src/components/modals/AddProjectModal.tsx      | 6 +-----
 src/constants.ts                               | 3 +++
 4 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/aiParams.ts b/src/aiParams.ts
index 77331e8e..00d4b1ee 100644
--- a/src/aiParams.ts
+++ b/src/aiParams.ts
@@ -104,6 +104,7 @@ export interface CustomModel {
   maxTokens?: number;
 
   context?: number;
+  projectEnabled?: boolean;
   plusExclusive?: boolean;
   believerExclusive?: boolean;
   capabilities?: ModelCapability[];
diff --git a/src/components/chat-components/ProjectList.tsx b/src/components/chat-components/ProjectList.tsx
index c6df91c5..5b8616f4 100644
--- a/src/components/chat-components/ProjectList.tsx
+++ b/src/components/chat-components/ProjectList.tsx
@@ -312,7 +312,7 @@ export const ProjectList = memo(
                   <div className="flex items-center gap-2">
                     <Tooltip>
                       <TooltipTrigger asChild>
-                        <Button size="fit" onClick={handleAddProject}>
+                        <Button size="fit" className="px-2" onClick={handleAddProject}>
                           Create
                           <Plus className="size-3" />
                         </Button>
diff --git a/src/components/modals/AddProjectModal.tsx b/src/components/modals/AddProjectModal.tsx
index 12dcbe62..e93716c9 100644
--- a/src/components/modals/AddProjectModal.tsx
+++ b/src/components/modals/AddProjectModal.tsx
@@ -190,11 +190,7 @@ function AddProjectModalContent({ initialProject, onSave, onCancel }: AddProject
             onBlur={() => setTouched((prev) => ({ ...prev, projectModelKey: true }))}
             placeholder="Select a model"
             options={settings.activeModels
-              .filter(
-                (m) =>
-                  m.provider === "google" &&
-                  (m.name === "gemini-1.5-flash-001" || m.name === "gemini-1.5-pro-001")
-              )
+              .filter((m) => m.enabled && m.projectEnabled)
               .map((model) => ({
                 label: getModelDisplayWithIcons(model),
                 value: getModelKeyFromModel(model),
diff --git a/src/constants.ts b/src/constants.ts
index 8ec6cc16..53be541c 100644
--- a/src/constants.ts
+++ b/src/constants.ts
@@ -112,6 +112,7 @@ export const BUILTIN_CHAT_MODELS: CustomModel[] = [
     enabled: true,
     isBuiltIn: true,
     core: true,
+    projectEnabled: true,
     capabilities: [ModelCapability.VISION],
   },
   {
@@ -174,6 +175,7 @@ export const BUILTIN_CHAT_MODELS: CustomModel[] = [
     enabled: true,
     isBuiltIn: true,
     core: true,
+    projectEnabled: true,
     capabilities: [ModelCapability.VISION],
   },
   {
@@ -182,6 +184,7 @@ export const BUILTIN_CHAT_MODELS: CustomModel[] = [
     enabled: true,
     isBuiltIn: true,
     core: true,
+    projectEnabled: true,
     capabilities: [ModelCapability.VISION],
   },
   {