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

more better

Browse source
This commit is contained in:
Ashwin Bharambe 2025-10-05 21:55:44 -07:00
parent ea233c2134
commit a385e0d95e
15 changed files with 230 additions and 208 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/core/datatypes.py
View file

@ -664,13 +664,11 @@ reference them from multiple stores. If not specified, default SQLite stores wil
providers_dict[api] = [p.model_dump() for p in provider_list]
# Resolve backend references
resolved_providers = resolve_provider_kvstore_references(
providers_dict, self.persistence.backends
)
resolved_providers = resolve_provider_kvstore_references(providers_dict, self.persistence.backends)
# Convert back to Provider objects
for api, provider_list in resolved_providers.items():
self.providers[api] = [Provider(**p) for p in provider_list]
for api, provider_dicts in resolved_providers.items():
self.providers[api] = [Provider(**provider_dict) for provider_dict in provider_dicts]
return self

6
llama_stack/core/provider_config_resolver.py
View file

@ -29,7 +29,7 @@ def resolve_provider_kvstore_references(
db_path: /path/to/kvstore.db
namespace: faiss
"""
for api, provider_list in providers.items():
for _api, provider_list in providers.items():
for provider in provider_list:
config = provider.get("config", {})
_resolve_kvstore_in_dict(config, persistence_backends)
@ -53,7 +53,9 @@ def _resolve_kvstore_in_dict(config: dict[str, Any], backends: dict[str, Any]) -
# Clone the backend config and apply namespace
backend_config = backends[backend_name]
resolved_config = backend_config.model_dump() if hasattr(backend_config, "model_dump") else dict(backend_config)
resolved_config = (
backend_config.model_dump() if hasattr(backend_config, "model_dump") else dict(backend_config)
)
if namespace:
resolved_config["namespace"] = namespace