phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

Small fix to library client

Browse source
This commit is contained in:
Ashwin Bharambe 2024-12-16 14:08:30 -08:00
parent 5e08812bcb
commit eb37fba9da
2 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/distribution/library_client.py
View file

@ -257,6 +257,8 @@ class AsyncLlamaStackAsLibraryClient(AsyncLlamaStackClient):
endpoints = get_all_api_endpoints()
endpoint_impls = {}
for api, api_endpoints in endpoints.items():
if api not in self.impls:
continue
for endpoint in api_endpoints:
impl = self.impls[api]
func = getattr(impl, endpoint.name)