akryldigital commited on
Commit
82338f0
·
verified ·
1 Parent(s): 383ecc7

handle the same langchain Document issue

Browse files
Files changed (1) hide show
  1. src/vectorstore.py +5 -1
src/vectorstore.py CHANGED
@@ -16,11 +16,15 @@ if hasattr(torch.backends, 'mps'):
16
  torch.backends.mps.is_available = lambda: False
17
 
18
  from langchain_qdrant import QdrantVectorStore
19
- from langchain.docstore.document import Document
20
  from langchain_core.embeddings import Embeddings
21
  from sentence_transformers import SentenceTransformer
22
  from langchain_huggingface import HuggingFaceEmbeddings
23
 
 
 
 
 
 
24
 
25
  class MatryoshkaEmbeddings(Embeddings):
26
  """Custom embeddings class that supports Matryoshka dimension truncation."""
 
16
  torch.backends.mps.is_available = lambda: False
17
 
18
  from langchain_qdrant import QdrantVectorStore
 
19
  from langchain_core.embeddings import Embeddings
20
  from sentence_transformers import SentenceTransformer
21
  from langchain_huggingface import HuggingFaceEmbeddings
22
 
23
+ try:
24
+ from langchain.docstore.document import Document
25
+ except:
26
+ from langchain_core.documents import Document
27
+
28
 
29
  class MatryoshkaEmbeddings(Embeddings):
30
  """Custom embeddings class that supports Matryoshka dimension truncation."""