Spaces:
Sleeping
Sleeping
John Graham Reynolds
commited on
Commit
·
7c23bb3
1
Parent(s):
2946731
update to use objects from databricks_langchain
Browse files- chain.py +3 -2
- requirements.txt +1 -0
chain.py
CHANGED
@@ -5,9 +5,10 @@ import streamlit as st
|
|
5 |
from functools import partial
|
6 |
from operator import itemgetter
|
7 |
from langchain_huggingface import HuggingFaceEmbeddings
|
8 |
-
from langchain_databricks.vectorstores import DatabricksVectorSearch
|
|
|
9 |
# from langchain_community.chat_models import ChatDatabricks # lets be consistent with the packages were using
|
10 |
-
from langchain_databricks import ChatDatabricks
|
11 |
# from langchain_community.vectorstores import DatabricksVectorSearch # is this causing an issue?
|
12 |
from langchain_core.runnables import RunnableLambda
|
13 |
from langchain_core.output_parsers import StrOutputParser
|
|
|
5 |
from functools import partial
|
6 |
from operator import itemgetter
|
7 |
from langchain_huggingface import HuggingFaceEmbeddings
|
8 |
+
# from langchain_databricks.vectorstores import DatabricksVectorSearch # this is the version we were previously using
|
9 |
+
from databricks_langchain import DatabricksVectorSearch, ChatDatabricks # new version resolving deprecation warning
|
10 |
# from langchain_community.chat_models import ChatDatabricks # lets be consistent with the packages were using
|
11 |
+
# from langchain_databricks import ChatDatabricks # this is the version we were previously using
|
12 |
# from langchain_community.vectorstores import DatabricksVectorSearch # is this causing an issue?
|
13 |
from langchain_core.runnables import RunnableLambda
|
14 |
from langchain_core.output_parsers import StrOutputParser
|
requirements.txt
CHANGED
@@ -3,6 +3,7 @@ transformers
|
|
3 |
databricks
|
4 |
langchain
|
5 |
langchain_core
|
|
|
6 |
langchain-databricks
|
7 |
langchain-community
|
8 |
langchain-huggingface
|
|
|
3 |
databricks
|
4 |
langchain
|
5 |
langchain_core
|
6 |
+
databricks_langchain
|
7 |
langchain-databricks
|
8 |
langchain-community
|
9 |
langchain-huggingface
|