Update CustomRetriever.py

CustomRetriever.py

@@ -12,14 +12,9 @@ class RetrieverWithScores(BaseRetriever):
     def __init__(self, vdb, thold=0.7):
         self.vdb=vdb
         self.thold=thold
-                #def __init__(self, retriever: BaseRetriever):  # Add an __init__ to store the existing retriever
-                    #super().__init__(retriever=retriever)
-    def _get_relevant_documents(self, query: str, *, run_manager: CallbackManagerForRetrieverRun)-> List[Document]:                   
-
-                    return self.retr_func.invoke(query) #(self.vdb, query)
 
     @chain
-    def retr_func(self, query: str)-> List[Document]: #(vdb, query: str)-> List[Document]:
+    def retr_func(query: str)-> List[Document]: #(vdb, query: str)-> List[Document]:
         docs, scores = zip(*self.vdb.similarity_search_with_relevance_scores(query))#similarity_search_with_score(query))
         result=[]
         for doc, score in zip(docs, scores):
@@ -29,4 +24,10 @@ class RetrieverWithScores(BaseRetriever):
         if len(result)==0:
             result.append(Document(metadata={}, page_content='No data'))
         
-        return result #docs
+        return result #docs
+                #def __init__(self, retriever: BaseRetriever):  # Add an __init__ to store the existing retriever
+                    #super().__init__(retriever=retriever)
+    def _get_relevant_documents(self, query: str, *, run_manager: CallbackManagerForRetrieverRun)-> List[Document]:                   
+
+        return self.retr_func.invoke(query) #(self.vdb, query)
+