Update app.py

app.py

@@ -53,8 +53,8 @@ MODEL_CONFIGS = {
     "gpt-4.1-mini": {"max_tokens": 33000, "param_name": "max_tokens", "api_version": None, "category": "OpenAI", "warning": None},
     "gpt-4.1-nano": {"max_tokens": 33000, "param_name": "max_tokens", "api_version": None, "category": "OpenAI", "warning": None},
     "o3-mini": {"max_completion_tokens": 100000, "param_name": "max_completion_tokens", "api_version": "2024-12-01-preview", "category": "OpenAI", "warning": None},
-    "o1": {"ax_completion_tokens": 100000, "param_name": "max_completion_tokens", "api_version": "2024-12-01-preview", "category": "OpenAI", "warning": None},
-    "o1-mini": {"ax_completion_tokens": 66000, "param_name": "max_completion_tokens", "api_version": "2024-12-01-preview", "category": "OpenAI", "warning": None},
+    "o1": {"max_completion_tokens": 100000, "param_name": "max_completion_tokens", "api_version": "2024-12-01-preview", "category": "OpenAI", "warning": None},
+    "o1-mini": {"max_completion_tokens": 66000, "param_name": "max_completion_tokens", "api_version": "2024-12-01-preview", "category": "OpenAI", "warning": None},
     "o1-preview": {"max_tokens": 33000, "param_name": "max_tokens", "api_version": None, "category": "OpenAI", "warning": None},
     "Phi-4-multimodal-instruct": {"max_tokens": 4000, "param_name": "max_tokens", "api_version": None, "category": "Microsoft", "warning": None},
     "Mistral-large-2407": {"max_tokens": 4000, "param_name": "max_tokens", "api_version": None, "category": "Mistral", "warning": None},
@@ -365,6 +365,7 @@ Here's the complete Manim code:
                     messages=messages,
                     model=model_name,
                     max_tokens=config["max_tokens"]
+                    max_completion_tokens=config["max_completion_tokens"]
                 )
             else:
                 # Use the existing client with normal max_tokens
@@ -372,6 +373,7 @@ Here's the complete Manim code:
                     messages=messages,
                     model=model_name,
                     max_tokens=config["max_tokens"]
+                    max_completion_tokens=config["max_completion_tokens"]
                 )
             
             # Process the response
@@ -2159,6 +2161,7 @@ class MyScene(Scene):
                                 messages=[UserMessage("Hello, this is a connection test.")],
                                 model=model_name,
                                 max_tokens=50  # Small value for quick testing
+                                max_completion_tokens=50
                             )
                             
                             # Check if response is valid