From 6d72ea7283425ec3ba7cac1a0092ebac628fd41c Mon Sep 17 00:00:00 2001
From: Alex <sysradium@users.noreply.github.com>
Date: Mon, 3 Feb 2025 16:54:56 +0100
Subject: [PATCH] fix ollama integration by forcing flat messages (#406)

---
 src/smolagents/models.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/smolagents/models.py b/src/smolagents/models.py
index d2fdd47..dc1c7c6 100644
--- a/src/smolagents/models.py
+++ b/src/smolagents/models.py
@@ -690,6 +690,7 @@ class LiteLLMModel(Model):
             api_base=self.api_base,
             api_key=self.api_key,
             convert_images_to_image_urls=True,
+            flatten_messages_as_text=self.model_id.startswith("ollama"),
             custom_role_conversions=self.custom_role_conversions,
             **kwargs,
         )