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"]