LangChain-结合GLM+SQL+函数调用实现数据库查询(三)
针对 LangChain-结合GLM+SQL+函数调用实现数据库查询(二)-CSDN博客 进一步简化
通过 LangChain 和大语言模型(GLM-4)实现了一个 AI 代理,能够根据自然语言提问自动生成 SQL 查询语句,并连接 MySQL 数据库执行查询,最终返回结果。
整个流程如下:
用户提问 → AI 生成 SQL → 执行查询 → 返回结果
一 .导入依赖模块
import os
from urllib.parse import quote
from dotenv import load_dotenv
from langchain_community.utilities import SQLDatabase
from langchain_community.agent_toolkits import SQLDatabaseToolkit
from langchain_core.messages import SystemMessage, HumanMessage
from langchain_openai import ChatOpenAI
from langgraph.prebuilt import chat_agent_executor, create_react_agent
import logging
二.配置加载与日志设置
从 .env 文件中读取数据库和 API 密钥等敏感信息。
设置日志级别为 INFO,方便调试。
load