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

test: fix test_rag_agent test

Browse source 
Summary:

Test Plan:
This commit is contained in:
Eric Huang 2025-02-21 15:19:26 -08:00
parent e7d261ef4a
commit 01aef1382c
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/client-sdk/agents/test_agents.py
View file

@ -496,10 +496,11 @@ def test_rag_agent(llama_stack_client, agent_config):
stream=False,
)
# rag is called
assert response.steps[0].tool_calls[0].tool_name == "query_from_memory"
tool_execution_step = next(step for step in response.steps if step.step_type == "tool_execution")
assert tool_execution_step.tool_calls[0].tool_name == "query_from_memory"
# document ids are present in metadata
assert "num-0" in response.steps[0].tool_responses[0].metadata["document_ids"]
assert expected_kw in response.output_message.content
assert "num-0" in tool_execution_step.tool_responses[0].metadata["document_ids"]
assert expected_kw in response.output_message.content.lower()
def test_rag_and_code_agent(llama_stack_client, agent_config):