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

feat: Introduce weighted and rrf reranker implementations

Browse source 
Signed-off-by: Varsha Prasad Narsing <varshaprasad96@gmail.com>
This commit is contained in:
Varsha Prasad Narsing 2025-06-04 15:59:44 -07:00
parent eab85a7121
commit 6ea5c10d48
14 changed files with 637 additions and 75 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/remote/vector_io/chroma/chroma.py
View file

@ -111,6 +111,8 @@ class ChromaIndex(EmbeddingIndex):
query_string: str,
k: int,
score_threshold: float,
reranker_type: str,
reranker_params: dict[str, Any] | None = None,
) -> QueryChunksResponse:
raise NotImplementedError("Hybrid search is not supported in Chroma")

2
llama_stack/providers/remote/vector_io/milvus/milvus.py
View file

@ -109,6 +109,8 @@ class MilvusIndex(EmbeddingIndex):
query_string: str,
k: int,
score_threshold: float,
reranker_type: str,
reranker_params: dict[str, Any] | None = None,
) -> QueryChunksResponse:
raise NotImplementedError("Hybrid search is not supported in Milvus")

2
llama_stack/providers/remote/vector_io/pgvector/pgvector.py
View file

@ -134,6 +134,8 @@ class PGVectorIndex(EmbeddingIndex):
query_string: str,
k: int,
score_threshold: float,
reranker_type: str,
reranker_params: dict[str, Any] | None = None,
) -> QueryChunksResponse:
raise NotImplementedError("Hybrid search is not supported in PGVector")

2
llama_stack/providers/remote/vector_io/qdrant/qdrant.py
View file

@ -118,6 +118,8 @@ class QdrantIndex(EmbeddingIndex):
query_string: str,
k: int,
score_threshold: float,
reranker_type: str,
reranker_params: dict[str, Any] | None = None,
) -> QueryChunksResponse:
raise NotImplementedError("Hybrid search is not supported in Qdrant")

2
llama_stack/providers/remote/vector_io/weaviate/weaviate.py
View file

@ -98,6 +98,8 @@ class WeaviateIndex(EmbeddingIndex):
query_string: str,
k: int,
score_threshold: float,
reranker_type: str,
reranker_params: dict[str, Any] | None = None,
) -> QueryChunksResponse:
raise NotImplementedError("Hybrid search is not supported in Weaviate")