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

fix list tools method name

Browse source
This commit is contained in:
Dinesh Yeduguru 2025-01-07 16:17:38 -08:00
parent 87068278ac
commit 6632d7e410
12 changed files with 84 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/apis/tools/tools.py
View file

@ -130,8 +130,8 @@ class ToolGroups(Protocol):
class ToolRuntime(Protocol):
tool_store: ToolStore
@webmethod(route="/tool-runtime/list-tools", method="POST")
async def list_tools(
@webmethod(route="/tool-runtime/list-tools", method="GET")
async def list_runtime_tools(
self, tool_group_id: Optional[str] = None, mcp_endpoint: Optional[URL] = None
) -> List[ToolDef]: ...

2
llama_stack/distribution/routers/routers.py
View file

@ -417,7 +417,7 @@ class ToolRuntimeRouter(ToolRuntime):
args=args,
)
async def list_tools(
async def list_runtime_tools(
self, tool_group_id: Optional[str] = None, mcp_endpoint: Optional[URL] = None
) -> List[ToolDef]:
return await self.routing_table.get_provider_impl(tool_group_id).list_tools(

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

@ -508,7 +508,7 @@ class ToolGroupsRoutingTable(CommonRoutingTableImpl, ToolGroups):
args: Optional[Dict[str, Any]] = None,
) -> None:
tools = []
tool_defs = await self.impls_by_provider_id[provider_id].list_tools(
tool_defs = await self.impls_by_provider_id[provider_id].list_runtime_tools(
toolgroup_id, mcp_endpoint
)
tool_host = (

2
llama_stack/providers/inline/tool_runtime/code_interpreter/code_interpreter.py
View file

@ -44,7 +44,7 @@ class CodeInterpreterToolRuntimeImpl(ToolsProtocolPrivate, ToolRuntime):
async def unregister_tool(self, tool_id: str) -> None:
return
async def list_tools(
async def list_runtime_tools(
self, tool_group_id: Optional[str] = None, mcp_endpoint: Optional[URL] = None
) -> List[ToolDef]:
return [

2
llama_stack/providers/inline/tool_runtime/memory/memory.py
View file

@ -51,7 +51,7 @@ class MemoryToolRuntimeImpl(ToolsProtocolPrivate, ToolRuntime):
async def initialize(self):
pass
async def list_tools(
async def list_runtime_tools(
self, tool_group_id: Optional[str] = None, mcp_endpoint: Optional[URL] = None
) -> List[ToolDef]:
return [

2
llama_stack/providers/remote/tool_runtime/bing_search/bing_search.py
View file

@ -51,7 +51,7 @@ class BingSearchToolRuntimeImpl(
)
return provider_data.api_key
async def list_tools(
async def list_runtime_tools(
self, tool_group_id: Optional[str] = None, mcp_endpoint: Optional[URL] = None
) -> List[ToolDef]:
return [

2
llama_stack/providers/remote/tool_runtime/brave_search/brave_search.py
View file

@ -49,7 +49,7 @@ class BraveSearchToolRuntimeImpl(
)
return provider_data.api_key
async def list_tools(
async def list_runtime_tools(
self, tool_group_id: Optional[str] = None, mcp_endpoint: Optional[URL] = None
) -> List[ToolDef]:
return [

2
llama_stack/providers/remote/tool_runtime/model_context_protocol/model_context_protocol.py
View file

@ -29,7 +29,7 @@ class ModelContextProtocolToolRuntimeImpl(ToolsProtocolPrivate, ToolRuntime):
async def initialize(self):
pass
async def list_tools(
async def list_runtime_tools(
self, tool_group_id: Optional[str] = None, mcp_endpoint: Optional[URL] = None
) -> List[ToolDef]:
if mcp_endpoint is None:

2
llama_stack/providers/remote/tool_runtime/tavily_search/tavily_search.py
View file

@ -50,7 +50,7 @@ class TavilySearchToolRuntimeImpl(
)
return provider_data.api_key
async def list_tools(
async def list_runtime_tools(
self, tool_group_id: Optional[str] = None, mcp_endpoint: Optional[URL] = None
) -> List[ToolDef]:
return [

2
llama_stack/providers/remote/tool_runtime/wolfram_alpha/wolfram_alpha.py
View file

@ -51,7 +51,7 @@ class WolframAlphaToolRuntimeImpl(
)
return provider_data.api_key
async def list_tools(
async def list_runtime_tools(
self, tool_group_id: Optional[str] = None, mcp_endpoint: Optional[URL] = None
) -> List[ToolDef]:
return [