From 4634a578f41856ee89a12410c8bd7fb0dd62af30 Mon Sep 17 00:00:00 2001
From: yawudede <35065151+yawudede@users.noreply.github.com>
Date: Mon, 22 May 2023 10:39:55 +0800
Subject: [PATCH] local_doc_qa.py

---
 chains/local_doc_qa.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/chains/local_doc_qa.py b/chains/local_doc_qa.py
index 8eafaf7..ef76466 100644
--- 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):