Merge pull request #48 from epsilla-cloud/expose_search_engine

0.3.6 Expose search engine

Author: richard-epsilla

pyepsilla/__init__.py

@@ -4,3 +4,4 @@
 from . import vectordb
 from . import cloud
 from . import enterprise
+from . import utils

pyepsilla/utils/__init__.py

@@ -0,0 +1,4 @@
+#!/usr/bin/env python
+# -*- coding:utf-8 -*-
+
+from .search_engine import VectorRetriever, Reranker, RRFReRanker, RelativeScoreFusionReranker, DistributionBasedScoreFusionReranker, SearchEngine

pyepsilla/utils/search_engine.py

@@ -56,7 +56,7 @@ def retrieve(self, query: str) -> list[dict]:
         return response["result"]
 
 class Reranker:
-    def rerank(self, candidates: list[list[any]]) -> list[any]:
+    def rerank(self, candidates: list[list[any]], query: str = None) -> list[any]:
         pass
 
 class RRFReRanker(Reranker):

pyepsilla/vectordb/version.py

@@ -1 +1 @@
-__version__ = "0.3.5"
+__version__ = "0.3.6"