From 0cf4099985a0c9c3fddc310e9713b9d64237b238 Mon Sep 17 00:00:00 2001 From: James Braza Date: Tue, 7 Jan 2025 14:24:31 -0800 Subject: [PATCH] Removed niche gpt-3.5-turbo case --- llmclient/llms.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/llmclient/llms.py b/llmclient/llms.py index 8a3317d..0607fc6 100644 --- 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"]