KeyArgo
/
smolagents
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Bugfix: Groq via LiteLLM 🚅 (#605) 

Co-authored-by: Aymeric Roucher <69208727+aymeric-roucher@users.noreply.github.com>
This commit is contained in:
João Galego 2025-02-14 18:32:12 +00:00 committed by GitHub
parent a940f42c04
commit a6cc506d09
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/smolagents/models.py
View File

@ -863,9 +863,8 @@ class LiteLLMModel(Model):
import litellm import litellm
model_info: dict = litellm.get_model_info(self.model_id) 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 model_info["key"] != "llava"
return False return False
def __call__( def __call__(