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

more robust agent test

Browse source
This commit is contained in:
Xi Yan 2024-12-27 11:20:56 -08:00
parent 52d1e4f85e
commit db49fc8ad0
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -207,6 +207,10 @@ def test_builtin_tool_code_execution(llama_stack_client, agent_config):
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)
if "Tool:code_interpreter Response" not in logs_str:
assert len(logs_str) > 0
pytest.skip("code_interpreter not called by model")
assert "Tool:code_interpreter Response" in logs_str assert "Tool:code_interpreter Response" in logs_str
if "No such file or directory: 'bwrap'" in logs_str: if "No such file or directory: 'bwrap'" in logs_str:
assert "prime" in logs_str assert "prime" in logs_str