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

memory bank registration fixes

Browse source
This commit is contained in:
Ashwin Bharambe 2024-10-06 22:00:54 -07:00 • committed by Ashwin Bharambe
parent 099a95b614
commit 3725e74906
8 changed files with 108 additions and 62 deletions
Download patch file Download diff file Expand all files Collapse all files

7
llama_stack/distribution/configure.py
View file

@ -64,8 +64,8 @@ def configure_api_providers(
) -> StackRunConfig:
is_nux = len(config.providers) == 0
apis = set((config.apis or list(build_spec.providers.keys())))
config.apis = [a for a in apis if a != "telemetry"]
# keep this default so all APIs are served
config.apis = []
if is_nux:
print(
@ -79,7 +79,8 @@ def configure_api_providers(
provider_registry = get_provider_registry()
builtin_apis = [a.routing_table_api for a in builtin_automatically_routed_apis()]
for api_str in config.apis:
apis_to_serve = [a.value for a in Api if a not in (Api.telemetry, Api.inspect)]
for api_str in apis_to_serve:
api = Api(api_str)
if api in builtin_apis:
continue