fix: Cohere - fix chat message creation (#1289)

Author: anakin87

integrations/cohere/src/haystack_integrations/components/generators/cohere/chat/chat_generator.py

@@ -172,7 +172,7 @@ def run(self, messages: List[ChatMessage], generation_kwargs: Optional[Dict[str,
                     response_text += event.text
                 elif event.event_type == "stream-end":
                     finish_response = event.response
-            chat_message = ChatMessage.from_assistant(content=response_text)
+            chat_message = ChatMessage.from_assistant(response_text)
 
             if finish_response and finish_response.meta:
                 if finish_response.meta.billed_units:
@@ -219,7 +219,7 @@ def _build_message(self, cohere_response):
             # TODO revisit to see if we need to handle multiple tool calls
             message = ChatMessage.from_assistant(cohere_response.tool_calls[0].json())
         elif cohere_response.text:
-            message = ChatMessage.from_assistant(content=cohere_response.text)
+            message = ChatMessage.from_assistant(cohere_response.text)
         message.meta.update(
             {
                 "model": self.model,

integrations/cohere/tests/test_cohere_chat_generator.py

@@ -144,7 +144,7 @@ def test_message_to_dict(self, chat_messages):
     )
     @pytest.mark.integration
     def test_live_run(self):
-        chat_messages = [ChatMessage.from_user(content="What's the capital of France")]
+        chat_messages = [ChatMessage.from_user("What's the capital of France")]
         component = CohereChatGenerator(generation_kwargs={"temperature": 0.8})
         results = component.run(chat_messages)
         assert len(results["replies"]) == 1
@@ -181,7 +181,7 @@ def __call__(self, chunk: StreamingChunk) -> None:
 
         callback = Callback()
         component = CohereChatGenerator(streaming_callback=callback)
-        results = component.run([ChatMessage.from_user(content="What's the capital of France? answer in a word")])
+        results = component.run([ChatMessage.from_user("What's the capital of France? answer in a word")])
 
         assert len(results["replies"]) == 1
         message: ChatMessage = results["replies"][0]
@@ -202,7 +202,7 @@ def __call__(self, chunk: StreamingChunk) -> None:
     )
     @pytest.mark.integration
     def test_live_run_with_connector(self):
-        chat_messages = [ChatMessage.from_user(content="What's the capital of France")]
+        chat_messages = [ChatMessage.from_user("What's the capital of France")]
         component = CohereChatGenerator(generation_kwargs={"temperature": 0.8})
         results = component.run(chat_messages, generation_kwargs={"connectors": [{"id": "web-search"}]})
         assert len(results["replies"]) == 1
@@ -227,7 +227,7 @@ def __call__(self, chunk: StreamingChunk) -> None:
                 self.responses += chunk.content if chunk.content else ""
 
         callback = Callback()
-        chat_messages = [ChatMessage.from_user(content="What's the capital of France? answer in a word")]
+        chat_messages = [ChatMessage.from_user("What's the capital of France? answer in a word")]
         component = CohereChatGenerator(streaming_callback=callback)
         results = component.run(chat_messages, generation_kwargs={"connectors": [{"id": "web-search"}]})