phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-22 22:12:29 +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

4
llama_stack/distribution/routers/routing_tables.py
View file

@ -28,7 +28,6 @@ from llama_stack.apis.scoring_functions import (
from llama_stack.apis.shields import Shield, Shields
from llama_stack.apis.tools import (
BuiltInToolDef,
CustomToolDef,
MCPToolGroupDef,
Tool,
ToolGroup,
@ -36,6 +35,7 @@ from llama_stack.apis.tools import (
ToolGroups,
ToolHost,
ToolPromptFormat,
UserDefinedToolDef,
UserDefinedToolGroupDef,
)
from llama_stack.distribution.datatypes import (
@ -540,7 +540,7 @@ class ToolGroupsRoutingTable(CommonRoutingTableImpl, ToolGroups):
raise ValueError(f"Unknown tool group: {tool_group}")
for tool_def in tool_defs:
if isinstance(tool_def, CustomToolDef):
if isinstance(tool_def, UserDefinedToolDef):
tools.append(
Tool(
identifier=tool_def.name,