Gainward777 commited on
Commit
8550f2a
·
verified ·
1 Parent(s): f424c90

Update llm/utils.py

Browse files
Files changed (1) hide show
  1. llm/utils.py +2 -2
llm/utils.py CHANGED
@@ -29,7 +29,7 @@ def retr_func(query: str)-> List[Document]:
29
  docs, scores = zip(*VDB.similarity_search_with_relevance_scores(query))#similarity_search_with_score(query))
30
  result=[]
31
  for doc, score in zip(docs, scores):
32
- if score>THOLD:#thold:
33
  doc.metadata["score"] = score
34
  result.append(doc)
35
  if len(result)==0:
@@ -40,7 +40,7 @@ def retr_func(query: str)-> List[Document]:
40
 
41
  class RetrieverWithScores(BaseRetriever):
42
  def _get_relevant_documents(self, query: str, *, run_manager: CallbackManagerForRetrieverRun)-> List[Document]:
43
- return retr_func.invoke(query) #(self.vdb, query)
44
 
45
 
46
 
 
29
  docs, scores = zip(*VDB.similarity_search_with_relevance_scores(query))#similarity_search_with_score(query))
30
  result=[]
31
  for doc, score in zip(docs, scores):
32
+ if score>THOLD:
33
  doc.metadata["score"] = score
34
  result.append(doc)
35
  if len(result)==0:
 
40
 
41
  class RetrieverWithScores(BaseRetriever):
42
  def _get_relevant_documents(self, query: str, *, run_manager: CallbackManagerForRetrieverRun)-> List[Document]:
43
+ return retr_func.invoke(query)
44
 
45
 
46