github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-23 23:29:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

[retrievers][knn] Replace loop appends with list comprehension. (#5529)

Browse Source 
# Replace loop appends with list comprehension.

It's much faster, more idiomatic and slightly more readable.
This commit is contained in:
Taras Tsugrii 2023-05-31 18:57:24 -05:00 committed by GitHub
parent 0ad76c3380
commit b61f50665e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
langchain/retrievers/knn.py
View File

@ -51,13 +51,14 @@ class KNNRetriever(BaseRetriever, BaseModel):
denominator = np.max(similarities) - np.min(similarities) + 1e-6
normalized_similarities = (similarities - np.min(similarities)) / denominator
top_k_results = []
for row in sorted_ix[0 : self.k]:
top_k_results = [
Document(page_content=self.texts[row])
for row in sorted_ix[0 : self.k]
if (
self.relevancy_threshold is None
or normalized_similarities[row] >= self.relevancy_threshold
):
top_k_results.append(Document(page_content=self.texts[row]))
)
]
return top_k_results
async def aget_relevant_documents(self, query: str) -> List[Document]: