Skip to content

Commit 820ec43

Browse files
HabebNawathaleseb
authored andcommitted
fix: make provider_id a required field in vector DB registration
1 parent 9e5ef1a commit 820ec43

File tree

2 files changed

+2
-11
lines changed

2 files changed

+2
-11
lines changed

llama_stack/apis/vector_dbs/vector_dbs.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,8 +90,8 @@ async def register_vector_db(
9090
self,
9191
vector_db_id: str,
9292
embedding_model: str,
93+
provider_id: str,
9394
embedding_dimension: int | None = 384,
94-
provider_id: str | None = None,
9595
vector_db_name: str | None = None,
9696
provider_vector_db_id: str | None = None,
9797
) -> VectorDB:

llama_stack/core/routing_tables/vector_dbs.py

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -47,21 +47,12 @@ async def register_vector_db(
4747
self,
4848
vector_db_id: str,
4949
embedding_model: str,
50+
provider_id: str,
5051
embedding_dimension: int | None = 384,
51-
provider_id: str | None = None,
5252
provider_vector_db_id: str | None = None,
5353
vector_db_name: str | None = None,
5454
) -> VectorDB:
5555
provider_vector_db_id = provider_vector_db_id or vector_db_id
56-
if provider_id is None:
57-
if len(self.impls_by_provider_id) > 0:
58-
provider_id = list(self.impls_by_provider_id.keys())[0]
59-
if len(self.impls_by_provider_id) > 1:
60-
logger.warning(
61-
f"No provider specified and multiple providers available. Arbitrarily selected the first provider {provider_id}."
62-
)
63-
else:
64-
raise ValueError("No provider available. Please configure a vector_io provider.")
6556
model = await lookup_model(self, embedding_model)
6657
if model is None:
6758
raise ModelNotFoundError(embedding_model)

0 commit comments

Comments
 (0)