From 4301d2a9c2753c84b86a620e638bb25d2a0c4838 Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Mon, 4 Nov 2024 17:23:49 -0800 Subject: [PATCH] rename distribution_registry_store --- llama_stack/distribution/datatypes.py | 2 +- llama_stack/distribution/server/server.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/llama_stack/distribution/datatypes.py b/llama_stack/distribution/datatypes.py index 8139895ef..3a4806e27 100644 --- a/llama_stack/distribution/datatypes.py +++ b/llama_stack/distribution/datatypes.py @@ -139,7 +139,7 @@ One or more providers to use for each API. The same provider_type (e.g., meta-re can be instantiated multiple times (with different configs) if necessary. """, ) - distribution_registry_store: Optional[KVStoreConfig] = Field( + metadata_store: Optional[KVStoreConfig] = Field( default=None, description=""" Configuration for the persistence store used by the distribution registry. If not specified, diff --git a/llama_stack/distribution/server/server.py b/llama_stack/distribution/server/server.py index 636da3916..2560f4070 100644 --- a/llama_stack/distribution/server/server.py +++ b/llama_stack/distribution/server/server.py @@ -282,8 +282,8 @@ def main( app = FastAPI() # instantiate kvstore for storing and retrieving distribution metadata - if config.distribution_registry_store: - dist_kvstore = asyncio.run(kvstore_impl(config.distribution_registry_store)) + if config.metadata_store: + dist_kvstore = asyncio.run(kvstore_impl(config.metadata_store)) else: dist_kvstore = asyncio.run( kvstore_impl(