Skip to content

Commit 2c9c022

Browse files
Update expose reranker
1 parent 82e1bb3 commit 2c9c022

3 files changed

Lines changed: 6 additions & 1 deletion

File tree

pyepsilla/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,4 @@
44
from . import vectordb
55
from . import cloud
66
from . import enterprise
7+
from . import utils

pyepsilla/utils/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
#!/usr/bin/env python
2+
# -*- coding:utf-8 -*-
3+
4+
from .search_engine import VectorRetriever, Reranker, RRFReRanker, RelativeScoreFusionReranker, DistributionBasedScoreFusionReranker, SearchEngine

pyepsilla/utils/search_engine.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def retrieve(self, query: str) -> list[dict]:
5656
return response["result"]
5757

5858
class Reranker:
59-
def rerank(self, candidates: list[list[any]]) -> list[any]:
59+
def rerank(self, candidates: list[list[any]], query: str = None) -> list[any]:
6060
pass
6161

6262
class RRFReRanker(Reranker):

0 commit comments

Comments
 (0)