Update app.py

app.py

@@ -128,18 +128,30 @@ def get_ai_response(prompt, history):
         model = st.session_state.selected_model
         model_config = AVAILABLE_MODELS.get(model, AVAILABLE_MODELS["gpt-3.5-turbo"])
         
-        response = client.chat.completions.create(
-            model=model,
-            messages=messages,
-            temperature=model_config["temperature"],
-            max_tokens=model_config["output_tokens"],
-            stream=False
-        )
+        # Check if the model requires max_completion_tokens instead of max_tokens
+        if model.startswith("o3-mini") or model.startswith("o1") or model.startswith("gpt-4o"):
+            response = client.chat.completions.create(
+                model=model,
+                messages=messages,
+                temperature=model_config["temperature"],
+                max_completion_tokens=model_config["output_tokens"],
+                stream=False
+            )
+        else:
+            response = client.chat.completions.create(
+                model=model,
+                messages=messages,
+                temperature=model_config["temperature"],
+                max_tokens=model_config["output_tokens"],
+                stream=False
+            )
+        
         return response.choices[0].message.content
     
     except Exception as e:
         return f"An error occurred: {str(e)}."
 
+
 # Function to create a new conversation
 def create_new_chat():
     new_id = str(uuid.uuid4())