Skip to content

Commit

Permalink
langchain[patch]: Running SQLDatabaseChain adds prefix "SQLQuery:\n" (#…
Browse files Browse the repository at this point in the history
…14058)

- **Issue:** #12077

---------

Co-authored-by: Abdul Kader Maliyakkal <maliyakk@amazon.com>
  • Loading branch information
ridha and Abdul Kader Maliyakkal committed Dec 2, 2023
1 parent fd781c8 commit 82102c9
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion libs/experimental/langchain_experimental/sql/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@
from langchain_experimental.pydantic_v1 import Extra, Field, root_validator

INTERMEDIATE_STEPS_KEY = "intermediate_steps"
SQL_QUERY = "SQLQuery:"


class SQLDatabaseChain(Chain):
Expand Down Expand Up @@ -110,7 +111,7 @@ def _call(
run_manager: Optional[CallbackManagerForChainRun] = None,
) -> Dict[str, Any]:
_run_manager = run_manager or CallbackManagerForChainRun.get_noop_manager()
input_text = f"{inputs[self.input_key]}\nSQLQuery:"
input_text = f"{inputs[self.input_key]}\n{SQL_QUERY}"
_run_manager.on_text(input_text, verbose=self.verbose)
# If not present, then defaults to None which is all tables.
table_names_to_use = inputs.get("table_names_to_use")
Expand Down Expand Up @@ -140,6 +141,8 @@ def _call(
sql_cmd
) # output: sql generation (no checker)
intermediate_steps.append({"sql_cmd": sql_cmd}) # input: sql exec
if SQL_QUERY in sql_cmd:
sql_cmd = sql_cmd.split(SQL_QUERY)[1].strip()
result = self.database.run(sql_cmd)
intermediate_steps.append(str(result)) # output: sql exec
else:
Expand Down

0 comments on commit 82102c9

Please sign in to comment.