John Graham Reynolds commited on
Commit
7c23bb3
·
1 Parent(s): 2946731

update to use objects from databricks_langchain

Browse files
Files changed (2) hide show
  1. chain.py +3 -2
  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