phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-23 02:32:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix agents to run custom tools

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-12-20 22:02:00 -08:00
parent 9192a9bbb4
commit 2ad67529ef
4 changed files with 6 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_stack/providers/tests/agents/test_agents.py
View file

@ -149,7 +149,10 @@ async def create_agent_turn_with_search_tool(
tool_execution = tool_execution_events[0].event.payload.step_details
assert isinstance(tool_execution, ToolExecutionStep)
assert len(tool_execution.tool_calls) > 0
assert tool_execution.tool_calls[0].tool_name == tool_name
actual_tool_name = tool_execution.tool_calls[0].tool_name
if isinstance(actual_tool_name, BuiltinTool):
actual_tool_name = actual_tool_name.value
assert actual_tool_name == tool_name
assert len(tool_execution.tool_responses) > 0
check_turn_complete_event(turn_response, session_id, search_query_messages)