Mintplex-Labs/langchain-python
1
0
Fork 0
mirror of https://github.com/Mintplex-Labs/langchain-python.git synced 2026-07-19 21:33:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

Harrison/sql db chain (#641)

Browse Source 
Co-authored-by: Bruno Bornsztein <bruno.bornsztein@gmail.com>
This commit is contained in:
Harrison Chase
2023-01-17 22:32:28 -08:00
committed by GitHub
parent b550f57912
commit 6be5f4e4c4
3 changed files with 19 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
langchain/chains/sql_database/base.py
+3 -1
View File
@@ -109,7 +109,9 @@ class SQLDatabaseSequentialChain(Chain, BaseModel):
**kwargs: Any,
) -> SQLDatabaseSequentialChain:
"""Load the necessary chains."""
sql_chain = SQLDatabaseChain(llm=llm, database=database, prompt=query_prompt)
sql_chain = SQLDatabaseChain(
llm=llm, database=database, prompt=query_prompt, **kwargs
)
decider_chain = LLMChain(
llm=llm, prompt=decider_prompt, output_key="table_names"
)