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

update vector_stores.search

Browse source
This commit is contained in:
Hardik Shah 2025-06-12 15:19:30 -07:00
parent de37a04c3e
commit f8b85c2176
8 changed files with 180 additions and 60 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -17,7 +17,7 @@ from llama_stack.apis.vector_io import (
VectorStoreDeleteResponse,
VectorStoreListResponse,
VectorStoreObject,
VectorStoreSearchResponse,
VectorStoreSearchResponsePage,
)
from llama_stack.log import get_logger
from llama_stack.providers.datatypes import RoutingTable
@ -242,7 +242,7 @@ class VectorIORouter(VectorIO):
max_num_results: int | None = 10,
ranking_options: dict[str, Any] | None = None,
rewrite_query: bool | None = False,
) -> VectorStoreSearchResponse:
) -> VectorStoreSearchResponsePage:
logger.debug(f"VectorIORouter.openai_search_vector_store: {vector_store_id}")
# Route based on vector store ID
provider = self.routing_table.get_provider_impl(vector_store_id)