MJ-Prod commited on
Commit
dee0373
·
1 Parent(s): 40452c2

fix moved imports for langchain 0.4

Browse files
Files changed (1) hide show
  1. fiscal.py +8 -3
fiscal.py CHANGED
@@ -1,13 +1,18 @@
1
  import os
2
  from typing import Dict
 
3
  from langchain_huggingface import HuggingFaceEmbeddings, ChatHuggingFace, HuggingFaceEndpoint
4
  from langchain_community.vectorstores import Chroma
5
  from langchain_community.retrievers import BM25Retriever
6
- from langchain_community.retrievers import EnsembleRetriever
7
- from langchain.chains import ConversationalRetrievalChain
8
- from langchain.memory import ConversationBufferMemory
9
  from langchain_core.prompts import PromptTemplate
10
  from langchain_core.documents import Document
 
 
 
 
 
 
 
11
 
12
  # ---------- Embeddings ----------
13
  embeddings = HuggingFaceEmbeddings(
 
1
  import os
2
  from typing import Dict
3
+
4
  from langchain_huggingface import HuggingFaceEmbeddings, ChatHuggingFace, HuggingFaceEndpoint
5
  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(