diff --git a/ai/providers/anthropic.py b/ai/providers/anthropic.py index d961e69..7d0df54 100644 --- a/ai/providers/anthropic.py +++ b/ai/providers/anthropic.py @@ -43,7 +43,7 @@ def set_model(self, model_name: str): self.current_model = model_name def get_models(self) -> dict: - if self.api_key is not None: + if self.api_key: return self.MODELS else: return {} diff --git a/ai/providers/openai.py b/ai/providers/openai.py index 927c90e..fba886a 100644 --- a/ai/providers/openai.py +++ b/ai/providers/openai.py @@ -35,7 +35,7 @@ def set_model(self, model_name: str): self.current_model = model_name def get_models(self) -> dict: - if self.api_key is not None: + if self.api_key: return self.MODELS else: return {}