MJ-Prod commited on
Commit
1818d80
·
1 Parent(s): dee0373

fix moved imports for langchain 0.5

Browse files
Files changed (1) hide show
  1. fiscal.py +10 -2
fiscal.py CHANGED
@@ -6,14 +6,22 @@ from langchain_community.vectorstores import Chroma
6
  from langchain_community.retrievers import BM25Retriever
7
  from langchain_core.prompts import PromptTemplate
8
  from langchain_core.documents import Document
9
- from langchain.chains import ConversationalRetrievalChain
10
- from langchain.memory import ConversationBufferMemory
11
 
12
  try:
13
  from langchain.retrievers import EnsembleRetriever
14
  except ImportError:
15
  from langchain_community.retrievers import EnsembleRetriever
16
 
 
 
 
 
 
 
 
 
 
 
17
  # ---------- Embeddings ----------
18
  embeddings = HuggingFaceEmbeddings(
19
  model_name="nomic-ai/nomic-embed-text-v1",
 
6
  from langchain_community.retrievers import BM25Retriever
7
  from langchain_core.prompts import PromptTemplate
8
  from langchain_core.documents import Document
 
 
9
 
10
  try:
11
  from langchain.retrievers import EnsembleRetriever
12
  except ImportError:
13
  from langchain_community.retrievers import EnsembleRetriever
14
 
15
+ try:
16
+ from langchain.chains import ConversationalRetrievalChain
17
+ except ImportError:
18
+ from langchain_community.chains import ConversationalRetrievalChain
19
+
20
+ try:
21
+ from langchain.memory import ConversationBufferMemory
22
+ except ImportError:
23
+ from langchain_community.memory import ConversationBufferMemory
24
+
25
  # ---------- Embeddings ----------
26
  embeddings = HuggingFaceEmbeddings(
27
  model_name="nomic-ai/nomic-embed-text-v1",