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

minor fixes

Browse source
This commit is contained in:
Dinesh Yeduguru 2025-01-07 16:37:39 -08:00
parent 6632d7e410
commit c3865faf37
6 changed files with 19 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llama_stack/providers/tests/agents/test_agents.py
View file

@ -8,6 +8,7 @@ import os
from typing import Dict, List
import pytest
from llama_models.llama3.api.datatypes import BuiltinTool
from llama_stack.apis.agents import (
AgentConfig,
@ -335,7 +336,7 @@ class TestAgents:
assert isinstance(tool_execution, ToolExecutionStep)
assert len(tool_execution.tool_calls) > 0
actual_tool_name = tool_execution.tool_calls[0].tool_name
assert actual_tool_name == "web_search"
assert actual_tool_name == BuiltinTool.brave_search
assert len(tool_execution.tool_responses) > 0
check_turn_complete_event(turn_response, session_id, search_query_messages)