feat: add model selector dropdown to AI Schema Creator chat

Author: yash-pouranik

apps/dashboard-api/src/controllers/ai.controller.js

@@ -169,7 +169,7 @@ const collectionCreator = async (req, res, next) => {
 
     try {
         const { projectId } = req.params;
-        const { userMessage } = req.body;
+        const { userMessage, model } = req.body;
 
         if (typeof userMessage !== 'string' || userMessage.trim().length === 0) {
             throw new AppError(400, "User message is required");
@@ -245,7 +245,8 @@ const collectionCreator = async (req, res, next) => {
             messages: session.messages,
             developer_id: req.user._id.toString(),
             plan: effectivePlan,
-            encrypted_byok: encryptedByok
+            encrypted_byok: encryptedByok,
+            model: model || 'llama-3.3-70b-versatile'
         });
 
         const allowedTypes = new Set(['String', 'Number', 'Boolean', 'Date', 'Object', 'Array', 'Ref']);

apps/python-service/routers/ai.py

@@ -80,6 +80,7 @@ class CollectionCreatorRequest(BaseModel):
     developer_id: str
     plan: str = "free"
     encrypted_byok: EncryptedByok | None = None
+    model: str = "llama-3.3-70b-versatile"
 
 class CollectionCreatorResponse(BaseModel):
     type: Literal["clarify", "schema", "complete"]
@@ -189,6 +190,7 @@ async def collection_creator(request: CollectionCreatorRequest, req: Request):
             encrypted_byok=request.encrypted_byok.model_dump() if request.encrypted_byok else None,
             feature="collection-creator",
             increment=(len(request.messages) == 1),
+            model=request.model,
         )
 
         structured_llm = llm.with_structured_output(CollectionCreatorResponse)

apps/python-service/services/byok.py

@@ -41,6 +41,7 @@ async def resolve_ai_client(
     encrypted_byok: dict | None,
     feature: str = "query-builder",
     increment: bool = True,
+    model: str = "llama-3.3-70b-versatile",
 ) -> ChatGroq:
     """Resolve the appropriate AI client based on BYOK and plan.
 
@@ -69,10 +70,10 @@ async def resolve_ai_client(
         logger.info("🔑 Attempting to decrypt BYOK Groq key for developer_id=%s...", developer_id)
         try:
             decrypted_key = decrypt_transit(encrypted_byok["groqKey"])
-            logger.info("✅ BYOK Groq key decrypted successfully. Initializing ChatGroq client (model=llama-3.3-70b-versatile)...")
+            logger.info("✅ BYOK Groq key decrypted successfully. Initializing ChatGroq client (model=%s)...", model)
             return ChatGroq(
                 api_key=decrypted_key,
-                model_name="llama-3.3-70b-versatile",
+                model_name=model,
                 temperature=0,
             )
         except Exception as e:
@@ -135,10 +136,10 @@ async def resolve_ai_client(
         )
 
     # ── 4. Return platform client ──
-    logger.info("✅ Rate limit check passed for developer_id=%s. Initializing Platform ChatGroq client (model=llama-3.3-70b-versatile)...", developer_id)
+    logger.info("✅ Rate limit check passed for developer_id=%s. Initializing Platform ChatGroq client (model=%s)...", developer_id, model)
     return ChatGroq(
         api_key=settings.GROQ_API_KEY,
-        model_name="llama-3.3-70b-versatile",
+        model_name=model,
         temperature=0,
     )
 

apps/web-dashboard/src/components/CollectionCreatorAgent.jsx

@@ -11,6 +11,19 @@ const SUGGESTIONS = [
   { label: 'Content Management (CMS)', prompt: 'Design a headless CMS with articles, authors, categories, media assets, and publishing revisions' },
 ];
 
+const MODELS = [
+  'llama-3.3-70b-versatile',
+  'llama-3.1-8b-instant',
+  'qwen/qwen3.6-27b',
+  'openai/gpt-oss-120b',
+  'openai/gpt-oss-20b',
+  'openai/gpt-oss-safeguard-20b',
+  'canopylabs/orpheus-v1-english',
+  'canopylabs/orpheus-arabic-saudi',
+  'groq/compound',
+  'groq/compound-mini'
+];
+
 const createMsg = (role, content) => ({
   id: `${Date.now()}_${Math.random().toString(36).slice(2, 7)}`,
   role,
@@ -28,6 +41,7 @@ export default function CollectionCreatorAgent({ projectId, onInsertAll }) {
   const [inputValue, setInputValue] = useState('');
   const [isInserting, setIsInserting] = useState(false);
   const [insertResults, setInsertResults] = useState(null);
+  const [selectedModel, setSelectedModel] = useState(MODELS[0]);
 
   const messagesEndRef = useRef(null);
   const timersRef = useRef([]);
@@ -71,7 +85,8 @@ export default function CollectionCreatorAgent({ projectId, onInsertAll }) {
 
     try {
       const res = await api.post(`/api/projects/${projectId}/ai/collection-creator`, {
-        userMessage: textToSend
+        userMessage: textToSend,
+        model: selectedModel
       });
 
       const { message, schema: newSchema, iterationsLeft: left, iterationLimit: limit } = res.data.data;
@@ -228,6 +243,15 @@ export default function CollectionCreatorAgent({ projectId, onInsertAll }) {
             <Bot size={16} className="text-[var(--color-primary)]" />
             <span>AI Schema Assistant</span>
           </h3>
+          <select 
+            value={selectedModel}
+            onChange={(e) => setSelectedModel(e.target.value)}
+            className="text-xs py-1 px-2 rounded-md border border-[var(--color-border)] bg-[var(--color-bg-input)] text-[var(--color-text-main)] focus:outline-none focus:border-[var(--color-primary)] max-w-[180px] overflow-hidden text-ellipsis whitespace-nowrap"
+          >
+            {MODELS.map(m => (
+              <option key={m} value={m}>{m}</option>
+            ))}
+          </select>
         </div>
 
         {/* Messages Stream (Only this scrolls) */}