fix: make Ollama Chat Generator compatible with new ChatMessage (#1256)

deepset-ai · Dec 18, 2024 · aff6748 · aff6748
1 parent 7f62ca8
commit aff6748
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...ions/ollama/src/haystack_integrations/components/generators/ollama/chat/chat_generator.py b/...ions/ollama/src/haystack_integrations/components/generators/ollama/chat/chat_generator.py
@@ -116,7 +116,7 @@ def _build_message_from_ollama_response(self, ollama_response: ChatResponse) ->
         Converts the non-streaming response from the Ollama API to a ChatMessage.
         """
         response_dict = ollama_response.model_dump()
-        message = ChatMessage.from_assistant(content=response_dict["message"]["content"])
+        message = ChatMessage.from_assistant(response_dict["message"]["content"])
         message.meta.update({key: value for key, value in response_dict.items() if key != "message"})
         return message