Allow user to select model from the assistant setup page

Author: chriscarrollsmith

routers/setup.py

@@ -52,9 +52,25 @@ async def read_setup(
     status: Optional[str] = None,
     message_text: Optional[str] = None
 ) -> Response:
-    # Check if assistant ID is missing
-    current_tools = []
-    setup_message = "" # Message specific to setup state (e.g., API key missing)
+    # Variable initializations
+    current_tools: List[str] = []
+    current_model: Optional[str] = None
+    # Populate with all models extracted from user-provided HTML, sorted
+    available_models: List[str] = sorted([
+        "gpt-3.5-turbo", "gpt-3.5-turbo-0125", "gpt-3.5-turbo-1106", "gpt-3.5-turbo-16k", 
+        "gpt-4", "gpt-4-0125-preview", "gpt-4-0613", "gpt-4-1106-preview", 
+        "gpt-4-turbo", "gpt-4-turbo-2024-04-09", "gpt-4-turbo-preview", 
+        "gpt-4.1", "gpt-4.1-2025-04-14", "gpt-4.1-mini", "gpt-4.1-mini-2025-04-14", 
+        "gpt-4.1-nano", "gpt-4.1-nano-2025-04-14", 
+        "gpt-4.5-preview", "gpt-4.5-preview-2025-02-27", 
+        "gpt-4o", "gpt-4o-2024-05-13", "gpt-4o-2024-08-06", "gpt-4o-2024-11-20", 
+        "gpt-4o-mini", "gpt-4o-mini-2024-07-18", 
+        "o1", "o1-2024-12-17", 
+        "o3-mini", "o3-mini-2025-01-31"
+    ])
+    setup_message: str = ""
+
+    # Check if env variables are set
     load_dotenv(override=True)
     openai_api_key = os.getenv("OPENAI_API_KEY")
     assistant_id = os.getenv("ASSISTANT_ID")
@@ -67,6 +83,7 @@ async def read_setup(
             try:
                 assistant = await client.beta.assistants.retrieve(assistant_id)
                 current_tools = [tool.type for tool in assistant.tools]
+                current_model = assistant.model  # Get the model from the assistant
             except Exception as e:
                 logger.error(f"Failed to retrieve assistant {assistant_id}: {e}")
                 # If we can't retrieve the assistant, proceed as if it doesn't exist
@@ -81,14 +98,17 @@ async def read_setup(
             "status": status, # Pass status from query params
             "status_message": message_text, # Pass message from query params
             "assistant_id": assistant_id,
-            "current_tools": current_tools
+            "current_tools": current_tools,
+            "current_model": current_model,
+            "available_models": available_models # Pass available models to template
         }
     )
 
 
 @router.post("/assistant")
 async def create_update_assistant(
     tool_types: List[ToolTypes] = Form(...),
+    model: str = Form(...),
     client: AsyncOpenAI = Depends(lambda: AsyncOpenAI())
 ) -> RedirectResponse:
     """
@@ -101,6 +121,7 @@ async def create_update_assistant(
         client=client,
         assistant_id=current_assistant_id,
         tool_types=tool_types,
+        model=model,
         logger=logger
     )
 

templates/setup.html

@@ -32,8 +32,28 @@ <h3>Setup Required</h3>
           {% else %} {# Covers missing assistant ID or user wants to update #}
               <div class="setupSection">
                 {% if assistant_id %}<h3>Update Assistant</h3>{% else %}<h3>Create Assistant</h3>{% endif %}
-                <p class="setupMessage">Select tools for your assistant:</p>
+                
                 <form action="{{ url_for('create_update_assistant') }}" method="POST">
+                  <div>
+                    <label for="model-select">Select Model:</label>
+                    <select name="model" id="model-select" class="input" required>
+                      {% if available_models %}
+                        {% for model_id in available_models %}
+                          <option value="{{ model_id }}" {% if model_id == current_model %}selected{% endif %}>{{ model_id }}</option>
+                        {% endfor %}
+                      {% else %}
+                        <option value="" disabled>Could not load models</option>
+                        {# Optionally include a default fallback if loading fails #}
+                        {% if current_model %}
+                          <option value="{{ current_model }}" selected>{{ current_model }} (current)</option>
+                        {% else %}
+                           <option value="gpt-4o" selected>gpt-4o (default)</option> {# Default fallback #}
+                        {% endif %}
+                      {% endif %}
+                    </select>
+                  </div>
+                  
+                  <p class="setupMessage" style="margin-top: 15px;">Select tools for your assistant:</p>
                   <div>
                     <label>
                       <input type="checkbox" name="tool_types" value="code_interpreter" {% if "code_interpreter" in current_tools %}checked{% endif %}>

utils/create_assistant.py

@@ -69,6 +69,7 @@ async def create_or_update_assistant(
         client: AsyncOpenAI,
         assistant_id: str | None,
         tool_types: List[ToolTypes],
+        model: str,
         logger: logging.Logger
 ) -> str | None:
     """
@@ -83,7 +84,7 @@ async def create_or_update_assistant(
                 assistant_id,
                 instructions="You are a helpful assistant.",
                 name="Quickstart Assistant",
-                model="gpt-4o",
+                model=model,
                 tools=tool_params
             )
             logger.info(f"Updated assistant with ID: {assistant_id}")
@@ -92,7 +93,7 @@ async def create_or_update_assistant(
             assistant = await client.beta.assistants.create(
                 instructions="You are a helpful assistant.",
                 name="Quickstart Assistant",
-                model="gpt-4o",
+                model=model,
                 tools=tool_params
             )
             logger.info(f"Created new assistant: {assistant.id}")