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

address feedback

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-12-30 15:47:01 -08:00
parent 8bf3f8ea56
commit ac46bd5eb4
8 changed files with 24 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -104,7 +104,7 @@ async def create_agent_turn_with_search_tool(
agent_config = AgentConfig(
**{
**common_params,
"available_tools": [tool_name],
"tool_names": [tool_name],
}
)

4
llama_stack/providers/tests/tools/fixtures.py
View file

@ -13,9 +13,9 @@ from llama_models.llama3.api.datatypes import BuiltinTool
from llama_stack.apis.models import ModelInput, ModelType
from llama_stack.apis.tools import (
BuiltInToolDef,
CustomToolDef,
ToolGroupInput,
ToolParameter,
UserDefinedToolDef,
UserDefinedToolGroupDef,
)
from llama_stack.distribution.datatypes import Api, Provider
@ -50,7 +50,7 @@ def tool_group_input_memory() -> ToolGroupInput:
tool_group_id="memory_group",
tool_group=UserDefinedToolGroupDef(
tools=[
CustomToolDef(
UserDefinedToolDef(
name="memory",
description="Query the memory bank",
parameters=[