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

fix ollama integration by forcing flat messages (#406)

This commit is contained in:
Alex 2025-02-03 16:54:56 +01:00 committed by GitHub
parent 183869de04
commit 6d72ea7283
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/smolagents/models.py
View File

@ -690,6 +690,7 @@ class LiteLLMModel(Model):
api_base=self.api_base, api_base=self.api_base,
api_key=self.api_key, api_key=self.api_key,
convert_images_to_image_urls=True, convert_images_to_image_urls=True,
flatten_messages_as_text=self.model_id.startswith("ollama"),
custom_role_conversions=self.custom_role_conversions, custom_role_conversions=self.custom_role_conversions,
**kwargs, **kwargs,
) )