local_doc_qa.py

2023-05-22 10:39:55 +08:00 · 2023-05-22 10:39:55 +08:00 · 4634a578f4
parent 9140a5e70c
commit 4634a578f4
1 changed files with 1 additions and 0 deletions
--- a/chains/local_doc_qa.py
+++ b/chains/local_doc_qa.py
@ -21,6 +21,7 @@ import models.shared as shared
 from agent import bing_search
 from langchain.docstore.document import Document
 from sentence_transformers import SentenceTransformer, CrossEncoder, util
+from sklearn.neighbors import NearestNeighbors

 class SemanticSearch:
    def __init__(self):