yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2026-01-08 13:06:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve flaky tool invocation integration test (#10187)

Browse Source
This commit is contained in:
Vladimir Blagojevic 2025-12-03 15:42:26 +01:00 committed by GitHub
parent 152a16c82b
commit dfd678eff4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
test/tools/test_component_tool.py
View File

@ -543,7 +543,7 @@ class TestComponentToolInPipeline:
# Connect components
pipeline.connect("llm.replies", "tool_invoker.messages")
message = ChatMessage.from_user(text="Vladimir")
message = ChatMessage.from_user(text="Using tools, greet Vladimir")
# Run pipeline
result = pipeline.run({"llm": {"messages": [message]}})
@ -569,7 +569,7 @@ class TestComponentToolInPipeline:
pipeline.add_component("tool_invoker", ToolInvoker(tools=[tool]))
pipeline.connect("llm.replies", "tool_invoker.messages")
message = ChatMessage.from_user(text="I am Alice and I'm 30 years old")
message = ChatMessage.from_user(text="Greet the user Alice who is 30 years old")
result = pipeline.run({"llm": {"messages": [message]}})
tool_messages = result["tool_invoker"]["tool_messages"]
@ -625,7 +625,9 @@ class TestComponentToolInPipeline:
pipeline.add_component("tool_invoker", ToolInvoker(tools=[tool]))
pipeline.connect("llm.replies", "tool_invoker.messages")
message = ChatMessage.from_user(text="Diana lives at 123 Elm Street in Metropolis")
message = ChatMessage.from_user(
text="Process information about the person Diana who lives at 123 Elm Street in Metropolis"
)
result = pipeline.run({"llm": {"messages": [message]}})
tool_messages = result["tool_invoker"]["tool_messages"]