diff --git a/docs/_static/llama-stack-spec.html b/docs/_static/llama-stack-spec.html
index bf1aaedaf..9f798773a 100644
--- a/docs/_static/llama-stack-spec.html
+++ b/docs/_static/llama-stack-spec.html
@@ -11341,7 +11341,7 @@
"embedding_dimension": {
"type": "integer"
},
- "provider_vector_db_name": {
+ "vector_db_name": {
"type": "string"
}
},
@@ -15631,7 +15631,7 @@
"type": "string",
"description": "The identifier of the vector database in the provider."
},
- "provider_vector_db_name": {
+ "vector_db_name": {
"type": "string",
"description": "The name of the vector database."
}
diff --git a/docs/_static/llama-stack-spec.yaml b/docs/_static/llama-stack-spec.yaml
index e02f275bb..6d2c1efdd 100644
--- a/docs/_static/llama-stack-spec.yaml
+++ b/docs/_static/llama-stack-spec.yaml
@@ -7984,7 +7984,7 @@ components:
type: string
embedding_dimension:
type: integer
- provider_vector_db_name:
+ vector_db_name:
type: string
additionalProperties: false
required:
@@ -10940,7 +10940,7 @@ components:
type: string
description: >-
The identifier of the vector database in the provider.
- provider_vector_db_name:
+ vector_db_name:
type: string
description: The name of the vector database.
additionalProperties: false
diff --git a/llama_stack/apis/vector_dbs/vector_dbs.py b/llama_stack/apis/vector_dbs/vector_dbs.py
index 31bbfaa0a..ee0b6217e 100644
--- a/llama_stack/apis/vector_dbs/vector_dbs.py
+++ b/llama_stack/apis/vector_dbs/vector_dbs.py
@@ -19,7 +19,7 @@ class VectorDB(Resource):
embedding_model: str
embedding_dimension: int
- provider_vector_db_name: str | None = None
+ vector_db_name: str | None = None
@property
def vector_db_id(self) -> str:
@@ -72,7 +72,7 @@ class VectorDBs(Protocol):
embedding_dimension: int | None = 384,
provider_id: str | None = None,
provider_vector_db_id: str | None = None,
- provider_vector_db_name: str | None = None,
+ vector_db_name: str | None = None,
) -> VectorDB:
"""Register a vector database.
@@ -81,7 +81,7 @@ class VectorDBs(Protocol):
:param embedding_dimension: The dimension of the embedding model.
:param provider_id: The identifier of the provider.
:param provider_vector_db_id: The identifier of the vector database in the provider.
- :param provider_vector_db_name: The name of the vector database.
+ :param vector_db_name: The name of the vector database.
:returns: A VectorDB.
"""
...
diff --git a/llama_stack/distribution/routers/vector_io.py b/llama_stack/distribution/routers/vector_io.py
index 50336056e..4f1f1dd12 100644
--- a/llama_stack/distribution/routers/vector_io.py
+++ b/llama_stack/distribution/routers/vector_io.py
@@ -83,7 +83,7 @@ class VectorIORouter(VectorIO):
embedding_dimension: int | None = 384,
provider_id: str | None = None,
provider_vector_db_id: str | None = None,
- provider_vector_db_name: str | None = None,
+ vector_db_name: str | None = None,
) -> None:
logger.debug(f"VectorIORouter.register_vector_db: {vector_db_id}, {embedding_model}")
await self.routing_table.register_vector_db(
@@ -92,7 +92,7 @@ class VectorIORouter(VectorIO):
embedding_dimension,
provider_id,
provider_vector_db_id,
- provider_vector_db_name,
+ vector_db_name,
)
async def insert_chunks(
diff --git a/llama_stack/distribution/routing_tables/vector_dbs.py b/llama_stack/distribution/routing_tables/vector_dbs.py
index 3c7fd258a..a6c97b09c 100644
--- a/llama_stack/distribution/routing_tables/vector_dbs.py
+++ b/llama_stack/distribution/routing_tables/vector_dbs.py
@@ -36,7 +36,7 @@ class VectorDBsRoutingTable(CommonRoutingTableImpl, VectorDBs):
embedding_dimension: int | None = 384,
provider_id: str | None = None,
provider_vector_db_id: str = "",
- provider_vector_db_name: str | None = None,
+ vector_db_name: str | None = None,
) -> VectorDB:
if provider_vector_db_id == "":
provider_vector_db_id = vector_db_id
@@ -63,7 +63,7 @@ class VectorDBsRoutingTable(CommonRoutingTableImpl, VectorDBs):
"provider_resource_id": provider_vector_db_id,
"embedding_model": embedding_model,
"embedding_dimension": model.metadata["embedding_dimension"],
- "vector_db_name": provider_vector_db_name,
+ "vector_db_name": vector_db_name,
}
vector_db = TypeAdapter(VectorDBWithOwner).validate_python(vector_db_data)
await self.register_object(vector_db)
diff --git a/llama_stack/providers/utils/memory/openai_vector_store_mixin.py b/llama_stack/providers/utils/memory/openai_vector_store_mixin.py
index d1a01948e..bc13e9bf7 100644
--- a/llama_stack/providers/utils/memory/openai_vector_store_mixin.py
+++ b/llama_stack/providers/utils/memory/openai_vector_store_mixin.py
@@ -166,7 +166,7 @@ class OpenAIVectorStoreMixin(ABC):
embedding_model=embedding_model,
provider_id=provider_id,
provider_resource_id=provider_vector_db_id,
- provider_vector_db_name=name,
+ vector_db_name=name,
)
# Register the vector DB
await self.register_vector_db(vector_db)