使用LangChain+Claude操作PostgreSQL时SQL结果异常及语法错误排查

1. 调整结果返回限制 #

SQLDatabaseChain默认会限制返回的结果行数（默认top_k=1），这是导致仅显示1条记录的直接原因。初始化时显式设置更大的top_k值：

db_chain = SQLDatabaseChain(llm=llm, database=db, verbose=True, top_k=10)

该参数控制LLM处理时最多使用的查询结果行数，设为大于6的值即可完整获取所有返回记录。

2. 修正QUERY模板的格式冲突 #

当前QUERY模板中的双引号嵌套可能导致Claude输出的SQL或结果出现语法解析错误。将模板内的双引号替换为单引号，避免格式混淆：

QUERY = """
Given an input question, first create a syntactically correct postgresql query to run, then look at the results of the query and return the answer.
Use the following format:

'Question': 'Question here'
'SQLQuery': 'SQL Query to run'
'SQLResult': 'Result of the SQLQuery'
'Answer': 'Final answer here'

'{question}'
"""

或者直接简化模板，去掉强制键值对格式，让LLM自然返回结果，减少格式解析异常。

3. 替换为SQL Agent工具链（更稳定方案） #

SQLDatabaseChain在处理多结果场景时局限性较强，改用SQLDatabaseToolkit+Agent组合能更智能地处理查询和结果返回：

# 替换原db_chain初始化代码
toolkit = SQLDatabaseToolkit(db=db, llm=llm)
agent = create_sql_agent(
    llm=llm,
    toolkit=toolkit,
    agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
    verbose=True
)

# 修改get_prompt函数中的调用逻辑
def get_prompt():
    print("Type 'exit' to quit")
    while True:
        prompt = input("Enter a prompt: ")
        if prompt.lower() == 'exit':
            print('Exiting...')
            break
        else:
            try:
                # 直接传入用户提问，无需套入QUERY模板
                result = agent.run(prompt)
                print(result)
            except Exception as e:
                print("Error:", e)
                pass

Agent会自动处理SQL生成、执行和结果整理，大幅降低格式错误概率。

4. 验证数据库查询完整性 #

直接在PostgreSQL客户端（如psql）执行LLM生成的SQL语句，确认确实返回6条记录，排除数据库权限、数据过滤等底层问题。同时确保连接数据库的用户postgres拥有delhi_cime_2数据库的完整查询权限。

5. 调整LLM输出长度限制 #

增加ChatAnthropic的max_tokens参数，避免结果因长度限制被截断：

llm = ChatAnthropic(temperature=0, max_tokens=1024)

内容的提问来源于stack exchange，提问作者Umang Bisht