langchian,agent连接数据库并操作
·
使用SQLDatabaseChain构建的agent,根据数据库回答问题
注意langchain高版本不支持写入数据库,示例langchain==0.1.16
from langchain_experimental.sql import SQLDatabaseChain
from langchain.utilities import SQLDatabase
from langchain_community.llms.tongyi import Tongyi
from langchain.agents import create_sql_agent
from langchain.agents.agent_toolkits import SQLDatabaseToolkit
from langchain.agents.agent_types import AgentType
# 初始化语言模型
llm = Tongyi(
model='qwen-turbo',
temperature=0.2,
top_p=0.7,
streaming=True
)
# 连接到数据库
db = SQLDatabase.from_uri('mysql+pymysql://root:123456@localhost:3306/test')
# db = SQLDatabase.from_uri("sqlite:///db/Chinook.db")
# 创建 SQLDatabaseToolkit 实例
toolkit = SQLDatabaseToolkit(db=db, llm=llm)
# 创建 SQL 代理
agent_executor = create_sql_agent(
llm=llm,
toolkit=toolkit,
verbose=True,
agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
)
agent_executor.run('创建stu表,包括用户名name和密码password两个列')
DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。
更多推荐



所有评论(0)