CR

jerpint · Nov 21, 2023 · 64211c3 · 64211c3
1 parent 31fa41a
commit 64211c3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/buster/retriever/service.py b/buster/retriever/service.py
@@ -77,8 +77,8 @@ def get_topk_documents(self, query: str, sources: Optional[list[str]], top_k: in
                 logger.warning(f"Sources {sources} do not exist. Returning empty dataframe.")
                 return pd.DataFrame()
 
-        query_embedding = self.get_embedding(query)
-        sparse_query_embedding = self.get_sparse_embedding(query) if self.get_sparse_embedding is not None else None
+        query_embedding = self.embedding_fn(query)
+        sparse_query_embedding = self.sparse_embedding_fn(query) if self.sparse_embedding_fn is not None else None
 
         if isinstance(query_embedding, np.ndarray):
             # pinecone expects a list of floats, so convert from ndarray if necessary