From d72ed6cb37abebaefda8da39e3e7a5d8fffb2065 Mon Sep 17 00:00:00 2001 From: "David S. Batista" Date: Tue, 14 Jan 2025 16:47:29 +0100 Subject: [PATCH] test: updating HuggingFaceAPIChatGenerator tests --- test/components/generators/chat/test_hugging_face_api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/components/generators/chat/test_hugging_face_api.py b/test/components/generators/chat/test_hugging_face_api.py index e2158ad6e9..fa83b98db7 100644 --- a/test/components/generators/chat/test_hugging_face_api.py +++ b/test/components/generators/chat/test_hugging_face_api.py @@ -540,7 +540,7 @@ def test_live_run_with_tools(self, tools): assert "Paris" in tool_call.arguments["city"] assert message.meta["finish_reason"] == "stop" - new_messages = chat_messages + [message, ChatMessage.from_tool(tool_result="22° C", origin=tool_call)] + new_messages = chat_messages + [message, ChatMessage.from_tool(tool_result="22°", origin=tool_call)] # the model tends to make tool calls if provided with tools, so we don't pass them here results = generator.run(new_messages, generation_kwargs={"max_tokens": 50})