set

refuel-ai · Oct 1, 2024 · a0b7923 · a0b7923
1 parent a3e66c9
commit a0b7923
Showing 1 changed file with 12 additions and 10 deletions.
diff --git a/src/autolabel/models/openai.py b/src/autolabel/models/openai.py
@@ -60,16 +60,18 @@ class OpenAILLM(BaseModel):
         ]
     )
 
-    SUPPORTS_JSON_OUTPUTS = [
-        "gpt-3.5-turbo-0125",
-        "gpt-3.5-turbo",
-        "gpt-4-0125-preview",
-        "gpt-4-1106-preview",
-        "gpt-4-turbo-preview",
-        "gpt-4o",
-        "gpt-4o-2024-08-06",
-        "gpt-4o-mini",
-    ]
+    SUPPORTS_JSON_OUTPUTS = set(
+        [
+            "gpt-3.5-turbo-0125",
+            "gpt-3.5-turbo",
+            "gpt-4-0125-preview",
+            "gpt-4-1106-preview",
+            "gpt-4-turbo-preview",
+            "gpt-4o",
+            "gpt-4o-2024-08-06",
+            "gpt-4o-mini",
+        ]
+    )
 
     SUPPORTS_STRUCTURED_OUTPUTS = set(
         [