Swap to langchain-experimental for SQLDatabaseChain in MRKL demo (lan…

…gchain-ai#27)
alistairwalsh · Aug 22, 2023 · 910156b · 910156b
1 parent 64e64e4
commit 910156b
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 2 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,6 +18,7 @@ sentence-transformers = "^2.2.2"
 torch = ">=2.0.0, !=2.0.1"
 tabulate = "^0.9.0"
 streamlit-feedback = "^0.0.9"
+langchain-experimental = "^0.0.10"
 
 [tool.poetry.group.dev.dependencies]
 black = "^23.3.0"

diff --git a/streamlit_agent/mrkl_demo.py b/streamlit_agent/mrkl_demo.py
@@ -6,9 +6,10 @@
 from langchain.agents import AgentType
 from langchain.agents import initialize_agent, Tool
 from langchain.callbacks import StreamlitCallbackHandler
-from langchain.chains import LLMMathChain, SQLDatabaseChain
+from langchain.chains import LLMMathChain
 from langchain.llms import OpenAI
 from langchain.utilities import DuckDuckGoSearchAPIWrapper
+from langchain_experimental.sql import SQLDatabaseChain
 
 from streamlit_agent.callbacks.capturing_callback_handler import playback_callbacks
 from streamlit_agent.clear_results import with_clear_container