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

rename key for chroma db

Browse source
This commit is contained in:
Ashwin Bharambe 2025-10-20 10:48:37 -07:00
parent 2bba56a0a8
commit 11bd202f18
10 changed files with 12 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/distributions/ci-tests/run.yaml
View file

@ -114,7 +114,7 @@ providers:
provider_type: remote::chromadb provider_type: remote::chromadb
config: config:
url: ${env.CHROMADB_URL:=} url: ${env.CHROMADB_URL:=}
kvstore: persistence:
namespace: vector_io::chroma_remote namespace: vector_io::chroma_remote
backend: kv_default backend: kv_default
- provider_id: ${env.PGVECTOR_DB:+pgvector} - provider_id: ${env.PGVECTOR_DB:+pgvector}

2
llama_stack/distributions/dell/run-with-safety.yaml
View file

@ -26,7 +26,7 @@ providers:
provider_type: remote::chromadb provider_type: remote::chromadb
config: config:
url: ${env.CHROMADB_URL:=} url: ${env.CHROMADB_URL:=}
kvstore: persistence:
namespace: vector_io::chroma_remote namespace: vector_io::chroma_remote
backend: kv_default backend: kv_default
safety: safety:

2
llama_stack/distributions/dell/run.yaml
View file

@ -22,7 +22,7 @@ providers:
provider_type: remote::chromadb provider_type: remote::chromadb
config: config:
url: ${env.CHROMADB_URL:=} url: ${env.CHROMADB_URL:=}
kvstore: persistence:
namespace: vector_io::chroma_remote namespace: vector_io::chroma_remote
backend: kv_default backend: kv_default
safety: safety:

2
llama_stack/distributions/open-benchmark/run.yaml
View file

@ -46,7 +46,7 @@ providers:
provider_type: remote::chromadb provider_type: remote::chromadb
config: config:
url: ${env.CHROMADB_URL:=} url: ${env.CHROMADB_URL:=}
kvstore: persistence:
namespace: vector_io::chroma_remote namespace: vector_io::chroma_remote
backend: kv_default backend: kv_default
- provider_id: ${env.ENABLE_PGVECTOR:+pgvector} - provider_id: ${env.ENABLE_PGVECTOR:+pgvector}

2
llama_stack/distributions/postgres-demo/run.yaml
View file

@ -22,7 +22,7 @@ providers:
provider_type: remote::chromadb provider_type: remote::chromadb
config: config:
url: ${env.CHROMADB_URL:=} url: ${env.CHROMADB_URL:=}
kvstore: persistence:
namespace: vector_io::chroma_remote namespace: vector_io::chroma_remote
backend: kv_default backend: kv_default
safety: safety:

2
llama_stack/distributions/starter-gpu/run.yaml
View file

@ -114,7 +114,7 @@ providers:
provider_type: remote::chromadb provider_type: remote::chromadb
config: config:
url: ${env.CHROMADB_URL:=} url: ${env.CHROMADB_URL:=}
kvstore: persistence:
namespace: vector_io::chroma_remote namespace: vector_io::chroma_remote
backend: kv_default backend: kv_default
- provider_id: ${env.PGVECTOR_DB:+pgvector} - provider_id: ${env.PGVECTOR_DB:+pgvector}

2
llama_stack/distributions/starter/run.yaml
View file

@ -114,7 +114,7 @@ providers:
provider_type: remote::chromadb provider_type: remote::chromadb
config: config:
url: ${env.CHROMADB_URL:=} url: ${env.CHROMADB_URL:=}
kvstore: persistence:
namespace: vector_io::chroma_remote namespace: vector_io::chroma_remote
backend: kv_default backend: kv_default
- provider_id: ${env.PGVECTOR_DB:+pgvector} - provider_id: ${env.PGVECTOR_DB:+pgvector}

4
llama_stack/providers/inline/vector_io/chroma/config.py
View file

@ -15,7 +15,7 @@ from llama_stack.schema_utils import json_schema_type
@json_schema_type @json_schema_type
class ChromaVectorIOConfig(BaseModel): class ChromaVectorIOConfig(BaseModel):
db_path: str db_path: str
kvstore: KVStoreReference = Field(description="Config for KV store backend") persistence: KVStoreReference = Field(description="Config for KV store backend")
@classmethod @classmethod
def sample_run_config( def sample_run_config(
@ -23,7 +23,7 @@ class ChromaVectorIOConfig(BaseModel):
) -> dict[str, Any]: ) -> dict[str, Any]:
return { return {
"db_path": db_path, "db_path": db_path,
"kvstore": KVStoreReference( "persistence": KVStoreReference(
backend="kv_default", backend="kv_default",
namespace="vector_io::chroma", namespace="vector_io::chroma",
).model_dump(exclude_none=True), ).model_dump(exclude_none=True),

2
llama_stack/providers/remote/vector_io/chroma/chroma.py
View file

@ -151,7 +151,7 @@ class ChromaVectorIOAdapter(OpenAIVectorStoreMixin, VectorIO, VectorDBsProtocolP
self.vector_db_store = None self.vector_db_store = None
async def initialize(self) -> 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 self.vector_db_store = self.kvstore
if isinstance(self.config, RemoteChromaVectorIOConfig): if isinstance(self.config, RemoteChromaVectorIOConfig):

4
llama_stack/providers/remote/vector_io/chroma/config.py
View file

@ -15,13 +15,13 @@ from llama_stack.schema_utils import json_schema_type
@json_schema_type @json_schema_type
class ChromaVectorIOConfig(BaseModel): class ChromaVectorIOConfig(BaseModel):
url: str | None url: str | None
kvstore: KVStoreReference = Field(description="Config for KV store backend") persistence: KVStoreReference = Field(description="Config for KV store backend")
@classmethod @classmethod
def sample_run_config(cls, __distro_dir__: str, url: str = "${env.CHROMADB_URL}", **kwargs: Any) -> dict[str, Any]: def sample_run_config(cls, __distro_dir__: str, url: str = "${env.CHROMADB_URL}", **kwargs: Any) -> dict[str, Any]:
return { return {
"url": url, "url": url,
"kvstore": KVStoreReference( "persistence": KVStoreReference(
backend="kv_default", backend="kv_default",
namespace="vector_io::chroma_remote", namespace="vector_io::chroma_remote",
).model_dump(exclude_none=True), ).model_dump(exclude_none=True),