Removed niche gpt-3.5-turbo case

Future-House · Jan 7, 2025 · 0cf4099 · 0cf4099
1 parent af732af
commit 0cf4099
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/llmclient/llms.py b/llmclient/llms.py
@@ -687,10 +687,9 @@ def __str__(self) -> str:
 
     @model_validator(mode="after")
     def set_model_name(self) -> Self:
-        if (
-            self.config.get("model") in {"gpt-3.5-turbo", None}
-            and self.name != "unknown"
-        ) or (self.name != "unknown" and "model" not in self.config):
+        if (self.config.get("model") is None and self.name != "unknown") or (
+            self.name != "unknown" and "model" not in self.config
+        ):
             self.config["model"] = self.name
         elif "model" in self.config and self.name == "unknown":
             self.name = self.config["model"]