make sure we are using the langchain_databricks implementation

chain.py

@@ -4,8 +4,9 @@ import streamlit as st
 from operator import itemgetter
 from langchain_huggingface import HuggingFaceEmbeddings
 from langchain_databricks.vectorstores import DatabricksVectorSearch
-from langchain_community.chat_models import ChatDatabricks
-from langchain_community.vectorstores import DatabricksVectorSearch
+# from langchain_community.chat_models import ChatDatabricks # lets be consistent with the packages were using
+from langchain_databricks import ChatDatabricks
+# from langchain_community.vectorstores import DatabricksVectorSearch # is this causing an issue?
 from langchain_core.runnables import RunnableLambda
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.prompts import PromptTemplate, ChatPromptTemplate, MessagesPlaceholder