From a6cc506d099e5bf19b2fd9f9899394b211d369f8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Galego?= Date: Fri, 14 Feb 2025 18:32:12 +0000 Subject: [PATCH] =?UTF-8?q?Bugfix:=20Groq=20via=20LiteLLM=20=F0=9F=9A=85?= =?UTF-8?q?=20(#605)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Aymeric Roucher <69208727+aymeric-roucher@users.noreply.github.com> --- src/smolagents/models.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/smolagents/models.py b/src/smolagents/models.py index 172757e..7b35cb1 100644 --- a/src/smolagents/models.py +++ b/src/smolagents/models.py @@ -863,9 +863,8 @@ class LiteLLMModel(Model): import litellm model_info: dict = litellm.get_model_info(self.model_id) - if model_info["litellm_provider"] == "ollama": + if model_info["litellm_provider"] in ["ollama", "groq"]: return model_info["key"] != "llava" - return False def __call__(