SQLDatabaseChain invoke()返回的Answer值异常求助

我之前也踩过类似的坑，看着verbose里SQL查询结果明明是正确的59，但最终Answer却跑偏成了SQL语句，确实挺闹心的。咱们一步步来排查解决：

首先排查版本问题 #

LangChain的SQLDatabaseChain属于实验性模块，不同版本的行为可能有差异，先把相关包升级到最新版试试：

pip install --upgrade langchain langchain-experimental langchain-community

自定义Prompt模板强制输出格式 #

默认的Prompt可能让LLM误解了输出要求，咱们直接给它一个明确的格式模板，让它清楚Answer部分该返回什么：

from langchain_experimental.sql import SQLDatabaseChain
from langchain.prompts import PromptTemplate

# 自定义模板，严格规定各部分输出内容
template = """给定一个问题，请先生成语法正确的{dialect}查询语句，执行后根据查询结果返回问题的答案。
请严格按照以下格式输出：

Question: {question}
SQLQuery: {query}
SQLResult: {result}
Answer: {answer}

只能使用下面列出的表：
{table_info}

Question: {question}"""

prompt = PromptTemplate(
    input_variables=["question", "dialect", "table_info", "query", "result"],
    template=template,
)

# 用自定义prompt初始化链
db_chain = SQLDatabaseChain.from_llm(llm, db, verbose=True, prompt=prompt)
qns1 = db_chain.invoke("How many t-shirts do we have left for nike in extra small size and red color?")

手动提取中间结果（应急方案） #

如果上面的方法暂时没生效，咱们可以直接从中间步骤里拿到正确的查询结果，绕过LLM的格式化问题：

# 初始化时开启返回中间步骤
db_chain = SQLDatabaseChain.from_llm(llm, db, verbose=True, return_intermediate_steps=True)
qns1 = db_chain.invoke("How many t-shirts do we have left for nike in extra small size and red color?")

# 提取SQL查询结果（根据你的数据库返回格式调整，比如这里是嵌套元组的情况）
sql_result = qns1["intermediate_steps"][-1]["sql_result"]
final_answer = sql_result[0][0]
print(f"正确答案：{final_answer}")

额外检查点 #

• 确认数据库返回的结果格式：比如执行目标SQL后，返回的是单个数值还是嵌套的元组/列表？LLM可能对嵌套结构的解析有偏差，可以手动打印sql_result看看具体结构。
• 试试开启use_query_checker=True参数，让链额外检查SQL语句的正确性，有时候隐性的SQL问题也会导致后续输出异常。

备注：内容来源于stack exchange，提问作者punsoca