Skip to content

Commit

Permalink
test: updating HuggingFaceAPIChatGenerator tests
Browse files Browse the repository at this point in the history
  • Loading branch information
davidsbatista committed Jan 14, 2025
1 parent f53119c commit d72ed6c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test/components/generators/chat/test_hugging_face_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -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})
Expand Down

0 comments on commit d72ed6c

Please sign in to comment.