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

reverting to original call order for a simpler change

Browse source 
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
Francisco Javier Arceo 2025-07-11 16:10:10 -04:00
parent 56811a1483
commit d361154102
6 changed files with 25 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -79,20 +79,20 @@ class VectorIORouter(VectorIO):
async def register_vector_db(
self,
vector_db_id: str,
provider_vector_db_id: str,
embedding_model: str,
embedding_dimension: int | None = 384,
provider_id: str | None = None,
vector_db_name: str | None = None,
provider_vector_db_id: str | None = None,
) -> None:
logger.debug(f"VectorIORouter.register_vector_db: {vector_db_id}, {embedding_model}")
await self.routing_table.register_vector_db(
vector_db_id,
provider_vector_db_id,
embedding_model,
embedding_dimension,
provider_id,
vector_db_name,
provider_vector_db_id,
)
async def insert_chunks(
@ -126,6 +126,7 @@ class VectorIORouter(VectorIO):
embedding_model: str | None = None,
embedding_dimension: int | None = None,
provider_id: str | None = None,
provider_vector_db_id: str | None = None,
) -> VectorStoreObject:
logger.debug(f"VectorIORouter.openai_create_vector_store: name={name}, provider_id={provider_id}")
@ -139,11 +140,11 @@ class VectorIORouter(VectorIO):
vector_db_id = f"vs_{uuid.uuid4()}"
registered_vector_db = await self.routing_table.register_vector_db(
vector_db_id,
vector_db_id,
embedding_model,
embedding_dimension,
provider_id,
provider_vector_db_id,
name,
)
return await self.routing_table.get_provider_impl(registered_vector_db.identifier).openai_create_vector_store(