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

cleanup

Browse source
This commit is contained in:
Xi Yan 2024-10-23 15:19:41 -07:00
parent 07d45f2af3
commit 47d9030542
3 changed files with 8 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

1
llama_stack/distribution/distribution.py
View file

@ -35,7 +35,6 @@ def builtin_automatically_routed_apis() -> List[AutoRoutedApiInfo]:
routing_table_api=Api.memory_banks,
router_api=Api.memory,
),
# TODO: re-enable once we have proper checking on builtin routing: Issue #297
AutoRoutedApiInfo(
routing_table_api=Api.datasets,
router_api=Api.datasetio,

1
llama_stack/distribution/server/server.py
View file

@ -288,6 +288,7 @@ def main(
apis_to_serve = set(impls.keys())
for inf in builtin_automatically_routed_apis():
# if we do not serve the corresponding router API, we should not serve the routing table API
if inf.router_api.value not in apis_to_serve:
continue
apis_to_serve.add(inf.routing_table_api.value)