github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-09 13:00:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

Getting SQL cmd directly from SQLDatabase Chain. (#7940)

Browse Source 
- Description: Get SQL Cmd directly generated by SQL-Database Chain
without executing it in the DB engine.
- Issue: #4853 
- Tag maintainer: @hinthornw,@baskaryan

---------

Co-authored-by: Harrison Chase <hw.chase.17@gmail.com>
This commit is contained in:
Mohammad Mohtashim 2023-07-21 10:36:55 +05:00 committed by GitHub
parent 5a084e1b20
commit 355b7d8b86
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
langchain/chains/sql_database/base.py
View File

@ -41,6 +41,8 @@ class SQLDatabaseChain(Chain):
"""Number of results to return from the query""" """Number of results to return from the query"""
input_key: str = "query" #: :meta private: input_key: str = "query" #: :meta private:
output_key: str = "result" #: :meta private: output_key: str = "result" #: :meta private:
return_sql: bool = False
"""Will return sql-command directly without executing it"""
return_intermediate_steps: bool = False return_intermediate_steps: bool = False
"""Whether or not to return the intermediate steps along with the final answer.""" """Whether or not to return the intermediate steps along with the final answer."""
return_direct: bool = False return_direct: bool = False
@ -117,6 +119,8 @@ class SQLDatabaseChain(Chain):
callbacks=_run_manager.get_child(), callbacks=_run_manager.get_child(),
**llm_inputs, **llm_inputs,
).strip() ).strip()
if self.return_sql:
return {self.output_key: sql_cmd}
if not self.use_query_checker: if not self.use_query_checker:
_run_manager.on_text(sql_cmd, color="green", verbose=self.verbose) _run_manager.on_text(sql_cmd, color="green", verbose=self.verbose)
intermediate_steps.append( intermediate_steps.append(