From 11bd202f18d8018c4e092828b77984c5c13b4902 Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Mon, 20 Oct 2025 10:48:37 -0700 Subject: [PATCH] rename key for chroma db --- llama_stack/distributions/ci-tests/run.yaml | 2 +- llama_stack/distributions/dell/run-with-safety.yaml | 2 +- llama_stack/distributions/dell/run.yaml | 2 +- llama_stack/distributions/open-benchmark/run.yaml | 2 +- llama_stack/distributions/postgres-demo/run.yaml | 2 +- llama_stack/distributions/starter-gpu/run.yaml | 2 +- llama_stack/distributions/starter/run.yaml | 2 +- llama_stack/providers/inline/vector_io/chroma/config.py | 4 ++-- llama_stack/providers/remote/vector_io/chroma/chroma.py | 2 +- llama_stack/providers/remote/vector_io/chroma/config.py | 4 ++-- 10 files changed, 12 insertions(+), 12 deletions(-) diff --git a/llama_stack/distributions/ci-tests/run.yaml b/llama_stack/distributions/ci-tests/run.yaml index 3e6f20900..48d486298 100644 --- a/llama_stack/distributions/ci-tests/run.yaml +++ b/llama_stack/distributions/ci-tests/run.yaml @@ -114,7 +114,7 @@ providers: provider_type: remote::chromadb config: url: ${env.CHROMADB_URL:=} - kvstore: + persistence: namespace: vector_io::chroma_remote backend: kv_default - provider_id: ${env.PGVECTOR_DB:+pgvector} diff --git a/llama_stack/distributions/dell/run-with-safety.yaml b/llama_stack/distributions/dell/run-with-safety.yaml index a246755cc..3130285b9 100644 --- a/llama_stack/distributions/dell/run-with-safety.yaml +++ b/llama_stack/distributions/dell/run-with-safety.yaml @@ -26,7 +26,7 @@ providers: provider_type: remote::chromadb config: url: ${env.CHROMADB_URL:=} - kvstore: + persistence: namespace: vector_io::chroma_remote backend: kv_default safety: diff --git a/llama_stack/distributions/dell/run.yaml b/llama_stack/distributions/dell/run.yaml index aa911c92f..af1a96a21 100644 --- a/llama_stack/distributions/dell/run.yaml +++ b/llama_stack/distributions/dell/run.yaml @@ -22,7 +22,7 @@ providers: provider_type: remote::chromadb config: url: ${env.CHROMADB_URL:=} - kvstore: + persistence: namespace: vector_io::chroma_remote backend: kv_default safety: diff --git a/llama_stack/distributions/open-benchmark/run.yaml b/llama_stack/distributions/open-benchmark/run.yaml index ddaf3688d..39e975cdb 100644 --- a/llama_stack/distributions/open-benchmark/run.yaml +++ b/llama_stack/distributions/open-benchmark/run.yaml @@ -46,7 +46,7 @@ providers: provider_type: remote::chromadb config: url: ${env.CHROMADB_URL:=} - kvstore: + persistence: namespace: vector_io::chroma_remote backend: kv_default - provider_id: ${env.ENABLE_PGVECTOR:+pgvector} diff --git a/llama_stack/distributions/postgres-demo/run.yaml b/llama_stack/distributions/postgres-demo/run.yaml index 7831b403d..9556b1287 100644 --- a/llama_stack/distributions/postgres-demo/run.yaml +++ b/llama_stack/distributions/postgres-demo/run.yaml @@ -22,7 +22,7 @@ providers: provider_type: remote::chromadb config: url: ${env.CHROMADB_URL:=} - kvstore: + persistence: namespace: vector_io::chroma_remote backend: kv_default safety: diff --git a/llama_stack/distributions/starter-gpu/run.yaml b/llama_stack/distributions/starter-gpu/run.yaml index f69ae2733..a120f0a48 100644 --- a/llama_stack/distributions/starter-gpu/run.yaml +++ b/llama_stack/distributions/starter-gpu/run.yaml @@ -114,7 +114,7 @@ providers: provider_type: remote::chromadb config: url: ${env.CHROMADB_URL:=} - kvstore: + persistence: namespace: vector_io::chroma_remote backend: kv_default - provider_id: ${env.PGVECTOR_DB:+pgvector} diff --git a/llama_stack/distributions/starter/run.yaml b/llama_stack/distributions/starter/run.yaml index 99c425e5f..e0f00f3e9 100644 --- a/llama_stack/distributions/starter/run.yaml +++ b/llama_stack/distributions/starter/run.yaml @@ -114,7 +114,7 @@ providers: provider_type: remote::chromadb config: url: ${env.CHROMADB_URL:=} - kvstore: + persistence: namespace: vector_io::chroma_remote backend: kv_default - provider_id: ${env.PGVECTOR_DB:+pgvector} diff --git a/llama_stack/providers/inline/vector_io/chroma/config.py b/llama_stack/providers/inline/vector_io/chroma/config.py index 9d61e3ba7..1798f10de 100644 --- a/llama_stack/providers/inline/vector_io/chroma/config.py +++ b/llama_stack/providers/inline/vector_io/chroma/config.py @@ -15,7 +15,7 @@ from llama_stack.schema_utils import json_schema_type @json_schema_type class ChromaVectorIOConfig(BaseModel): db_path: str - kvstore: KVStoreReference = Field(description="Config for KV store backend") + persistence: KVStoreReference = Field(description="Config for KV store backend") @classmethod def sample_run_config( @@ -23,7 +23,7 @@ class ChromaVectorIOConfig(BaseModel): ) -> dict[str, Any]: return { "db_path": db_path, - "kvstore": KVStoreReference( + "persistence": KVStoreReference( backend="kv_default", namespace="vector_io::chroma", ).model_dump(exclude_none=True), diff --git a/llama_stack/providers/remote/vector_io/chroma/chroma.py b/llama_stack/providers/remote/vector_io/chroma/chroma.py index 5792a83c6..b07207cc6 100644 --- a/llama_stack/providers/remote/vector_io/chroma/chroma.py +++ b/llama_stack/providers/remote/vector_io/chroma/chroma.py @@ -151,7 +151,7 @@ class ChromaVectorIOAdapter(OpenAIVectorStoreMixin, VectorIO, VectorDBsProtocolP self.vector_db_store = None async def initialize(self) -> None: - self.kvstore = await kvstore_impl(self.config.kvstore) + self.kvstore = await kvstore_impl(self.config.persistence) self.vector_db_store = self.kvstore if isinstance(self.config, RemoteChromaVectorIOConfig): diff --git a/llama_stack/providers/remote/vector_io/chroma/config.py b/llama_stack/providers/remote/vector_io/chroma/config.py index f356232ad..209ba90bb 100644 --- a/llama_stack/providers/remote/vector_io/chroma/config.py +++ b/llama_stack/providers/remote/vector_io/chroma/config.py @@ -15,13 +15,13 @@ from llama_stack.schema_utils import json_schema_type @json_schema_type class ChromaVectorIOConfig(BaseModel): url: str | None - kvstore: KVStoreReference = Field(description="Config for KV store backend") + persistence: KVStoreReference = Field(description="Config for KV store backend") @classmethod def sample_run_config(cls, __distro_dir__: str, url: str = "${env.CHROMADB_URL}", **kwargs: Any) -> dict[str, Any]: return { "url": url, - "kvstore": KVStoreReference( + "persistence": KVStoreReference( backend="kv_default", namespace="vector_io::chroma_remote", ).model_dump(exclude_none=True),