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
|