local_doc_qa.py
This commit is contained in:
parent
9140a5e70c
commit
4634a578f4
|
|
@ -21,6 +21,7 @@ import models.shared as shared
|
||||||
from agent import bing_search
|
from agent import bing_search
|
||||||
from langchain.docstore.document import Document
|
from langchain.docstore.document import Document
|
||||||
from sentence_transformers import SentenceTransformer, CrossEncoder, util
|
from sentence_transformers import SentenceTransformer, CrossEncoder, util
|
||||||
|
from sklearn.neighbors import NearestNeighbors
|
||||||
|
|
||||||
class SemanticSearch:
|
class SemanticSearch:
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue