前言

这里用的是通义千文系列模型做示例，先看结果：

数据库内容

代码

from langchain_community.utilities import SQLDatabase
from langchain_experimental.sql import SQLDatabaseChain
from langchain_openai import ChatOpenAI

# 连接到 MySQL 数据库
db = SQLDatabase.from_uri("mysql+mysqlconnector://root:123456@localhost/itcast")
# 获取大语言模型实例
llm = ChatOpenAI(
    base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
    model="qwen-turbo",
    temperature=0,
    max_retries=5,
    api_key=""  # 请填入有效的 API 密钥
)
# 使用 from_llm 类方法实例化 SQLDatabaseChain
db_chain = SQLDatabaseChain.from_llm(llm, db, verbose=True)

# 提出自然语言问题
question = "查询所有学生的信息"

# 使用 invoke 方法替代 run 方法
result = db_chain.invoke({"query": question})
print("结果："+str(result['result']))