Spaces:
Sleeping
Sleeping
Update llm/utils.py
Browse files- 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
|
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)
|
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 |
|