From 66956277a0a82c07a672c7159eec05f9bc8a945d Mon Sep 17 00:00:00 2001 From: Hardik Shah Date: Thu, 30 Jan 2025 11:53:01 -0800 Subject: [PATCH] remove logs from test --- tests/client-sdk/agents/test_agents.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/tests/client-sdk/agents/test_agents.py b/tests/client-sdk/agents/test_agents.py index 016c1f5e2..374c7c74c 100644 --- a/tests/client-sdk/agents/test_agents.py +++ b/tests/client-sdk/agents/test_agents.py @@ -357,7 +357,7 @@ def test_rag_and_code_agent(llama_stack_client, agent_config): user_prompts = [ ( "What are the top 5 topics that were explained? Only list succinct bullet points.", - documents, + [], "query_from_memory", ), ("What is the average yearly inflation?", [inflation_doc], "code_interpreter"), @@ -370,11 +370,7 @@ def test_rag_and_code_agent(llama_stack_client, agent_config): session_id=session_id, documents=docs, ) - logs = [] - for log in EventLogger().log(response): - logs.append(str(log)) - log.print() - # logs = [str(log) for log in EventLogger().log(response) if log is not None] + logs = [str(log) for log in EventLogger().log(response) if log is not None] logs_str = "".join(logs) assert f"Tool:{tool_name}" in logs_str