fix moved imports for langchain 0.4
Browse files
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(
|