phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-06 10:42:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

bug fix, generate openapi spec

Browse source
This commit is contained in:
Ashwin Bharambe 2025-01-21 15:56:48 -08:00
parent 039982004d
commit 460dc8a72a
5 changed files with 680 additions and 1137 deletions
Download patch file Download diff file Expand all files Collapse all files

999
docs/resources/llama-stack-spec.html
View file

File diff suppressed because it is too large Load diff

812
docs/resources/llama-stack-spec.yaml
View file

File diff suppressed because it is too large Load diff

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

@ -440,7 +440,7 @@ class ToolRuntimeRouter(ToolRuntime):
) -> None: ) -> None:
self.routing_table = routing_table self.routing_table = routing_table
# TODO: this should be in sync with "get_all_api_endpoints()" # HACK ALERT this should be in sync with "get_all_api_endpoints()"
# TODO: make sure rag_tool vs builtin::memory is correct everywhere # TODO: make sure rag_tool vs builtin::memory is correct everywhere
self.rag_tool = self.RagToolImpl(routing_table) self.rag_tool = self.RagToolImpl(routing_table)
setattr(self, "rag_tool.query_context", self.rag_tool.query_context) setattr(self, "rag_tool.query_context", self.rag_tool.query_context)

2
llama_stack/distribution/server/endpoints.py
View file

@ -38,6 +38,8 @@ def get_all_api_endpoints() -> Dict[Api, List[ApiEndpoint]]:
for api, protocol in protocols.items(): for api, protocol in protocols.items():
endpoints = [] endpoints = []
protocol_methods = inspect.getmembers(protocol, predicate=inspect.isfunction) protocol_methods = inspect.getmembers(protocol, predicate=inspect.isfunction)
# HACK ALERT
if api == Api.tool_runtime: if api == Api.tool_runtime:
for tool_group in SpecialToolGroup: for tool_group in SpecialToolGroup:
sub_protocol = toolgroup_protocols[tool_group] sub_protocol = toolgroup_protocols[tool_group]

2
llama_stack/providers/inline/agents/meta_reference/agent_instance.py
View file

@ -970,7 +970,7 @@ async def execute_tool_call_maybe(
result = await tool_runtime_api.invoke_tool( result = await tool_runtime_api.invoke_tool(
tool_name=name, tool_name=name,
args=dict( kwargs=dict(
session_id=session_id, session_id=session_id,
**tool_call_args, **tool_call_args,
), ),