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

removing provider_ prefix

Browse source 
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
Francisco Javier Arceo 2025-07-10 09:19:34 -04:00
parent 36ca9543a5
commit 648db8408c
6 changed files with 12 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/distribution/routers/vector_io.py
View file

@ -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(

4
llama_stack/distribution/routing_tables/vector_dbs.py
View file

@ -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)