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

remove logs from test

Browse source
This commit is contained in:
Hardik Shah 2025-01-30 11:53:01 -08:00
parent 80348cbe94
commit 66956277a0
1 changed files with 2 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -357,7 +357,7 @@ def test_rag_and_code_agent(llama_stack_client, agent_config):
user_prompts = [ user_prompts = [
( (
"What are the top 5 topics that were explained? Only list succinct bullet points.", "What are the top 5 topics that were explained? Only list succinct bullet points.",
documents, [],
"query_from_memory", "query_from_memory",
), ),
("What is the average yearly inflation?", [inflation_doc], "code_interpreter"), ("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, session_id=session_id,
documents=docs, 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) logs_str = "".join(logs)
assert f"Tool:{tool_name}" in logs_str assert f"Tool:{tool_name}" in logs_str