yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-12-24 21:48:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: updating HuggingFaceAPIChatGenerator tests

Browse Source
This commit is contained in:
David S. Batista 2025-01-14 16:47:29 +01:00 committed by GitHub
parent 34bd31ef32
commit 425ce9b98f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/components/generators/chat/test_hugging_face_api.py
View File

@ -540,7 +540,7 @@ class TestHuggingFaceAPIChatGenerator:
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})