SerAcero
diff --git a/‎.gitignore
Copy file name to clipboardExpand all lines: .gitignore
+2Lines changed: 2 additions & 0 deletions b/‎.gitignore
Copy file name to clipboardExpand all lines: .gitignore
+2Lines changed: 2 additions & 0 deletions
diff --git a/‎tutorial/tests/testsuite/ai_helpers.py
Copy file name to clipboardExpand all lines: tutorial/tests/testsuite/ai_helpers.py
+10-4Lines changed: 10 additions & 4 deletions b/‎tutorial/tests/testsuite/ai_helpers.py
Copy file name to clipboardExpand all lines: tutorial/tests/testsuite/ai_helpers.py
+10-4Lines changed: 10 additions & 4 deletions
@@ -135,3 +135,5 @@ dmypy.json
 *.html
 .idea/
 drafts/
+.virtual_documents
+.zed
@@ -64,8 +64,14 @@ class OpenAIWrapper:
     """A simple API wrapper adapted for IPython environments"""
 
     # These are the models we can use: they must support structured responses
-    GPT_STABLE_MODELS = ("gpt-3.5-turbo", "gpt-4", "gpt-4o", "gpt-4o-mini")
-    GPT_ALL_MODELS = GPT_STABLE_MODELS
+    GPT_MODELS = (
+        "gpt-4o",
+        "gpt-4o-mini",
+        "gpt-4.1",
+        "gpt-4.1-mini",
+        "gpt-4.1-nano",
+        "o4-mini",
+    )
 
     DEFAULT_MODEL = "gpt-4o-mini"
     DEFAULT_LANGUAGE = "English"
@@ -134,11 +140,11 @@ def validate_api_key(cls, api_key: t.Optional[str]) -> ValidationResult:
     def validate_model(self, model: t.Optional[str]) -> ValidationResult:
         """Validate the model selection"""
         try:
-            if model not in self.GPT_ALL_MODELS:
+            if model not in self.GPT_MODELS:
                 return ValidationResult(
                     is_valid=False,
                     error=InvalidModelError(),
-                    message=f"Invalid model: {model}. Available models: {' '.join(self.GPT_ALL_MODELS)}",
+                    message=f"Invalid model: {model}. Available models: {' '.join(self.GPT_MODELS)}",
                 )
         except Exception as e:
             return ValidationResult(