Update app.py

app.py

@@ -59,28 +59,15 @@ def stream_assistant_response(client: OpenAI, model: str, messages: List[Dict[st
     full_response = ""
     message_placeholder = st.empty()
     try:
-        # Different parameter name for o-series models
-        if model.startswith("o"):
-            stream = client.chat.completions.create(
-                model=model,
-                messages=messages,
-                stream=True,
-                temperature=temperature,
-                max_completion_tokens=max_tokens,  # Use max_completion_tokens for o-series models
-                frequency_penalty=frequency_penalty,
-                presence_penalty=presence_penalty,
-            )
-        else:
-            # For other OpenAI models
-            stream = client.chat.completions.create(
-                model=model,
-                messages=messages,
-                stream=True,
-                temperature=temperature,
-                max_tokens=max_tokens,  # Use max_tokens for other models
-                frequency_penalty=frequency_penalty,
-                presence_penalty=presence_penalty,
-            )
+        stream = client.chat.completions.create(
+            model=model,
+            messages=messages,
+            stream=True,
+            temperature=temperature,
+            max_tokens=max_tokens,  # Use max_tokens for other models
+            frequency_penalty=frequency_penalty,
+            presence_penalty=presence_penalty,
+        )
             
         # Iterate through the stream to collect the response.
         for chunk in stream:
@@ -109,7 +96,7 @@ def main() -> None:
         st.header("Configuration")
 
         # Dropdown for model selection.
-        model_options = ["gpt-4o", "gpt-4o-mini", "o1", "o3-mini"]
+        model_options = ["gpt-4o", "gpt-4o-mini"]
         default_model = config.ai_model if config.ai_model in model_options else model_options[0]
         selected_model = st.selectbox(
             "Model",