phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-27 18:50:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Kill noise from test output

Browse source
This commit is contained in:
Ashwin Bharambe 2025-02-21 15:37:23 -08:00
parent bf38d0aba0
commit 45ffe87d7c
3 changed files with 4 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -90,7 +90,6 @@ class TestClientTool(ClientTool):
def agent_config(llama_stack_client, text_model_id):
available_shields = [shield.identifier for shield in llama_stack_client.shields.list()]
available_shields = available_shields[:1]
print(f"Using shield: {available_shields}")
agent_config = AgentConfig(
model=text_model_id,
instructions="You are a helpful assistant",
@ -489,7 +488,6 @@ def test_rag_agent(llama_stack_client, agent_config):
),
]
for prompt, expected_kw in user_prompts:
print(f"User> {prompt}")
response = rag_agent.create_turn(
messages=[{"role": "user", "content": prompt}],
session_id=session_id,