方法一:使用SQLDatabaseToolKit和Agent 创建MySQL数据库连接: 使用SQLDatabase.from_uri方法从数据库URI创建MySQL数据库连接。 python from langchain.chains.database import SQLDatabase MYSQL_URI = "mysql+pymysql://username:password
流程:[user] --查询--> [SQLDatabaseChain] --生成sql--> --执行SQL--> --结果-->[user] llm=ChatGLM(endpoint_url='http://127.0.0.1:8000',max_token=80000,top_p=0.9)db=SQLDatabase.from_uri("sqlite:///data/instance",include_tables=['instance'],sample_rows_in_table_info=2)db_cha...
导入langchain模块 OpenAI, SQLDatabase, 和 SQLDatabaseChain 从环境变量文件中获取 OPENAI_API_KEY 使用SQLDatabase.from_uri() 方法设置数据库连接,指定连接URL 使用OpenAI() 创建llm对象,指定温度和openai_api_key 使用SQLDatabaseChain() 创建数据库链对象 db_chain,指定llm和数据库对象 get_prompt() 从控制...
fromlangchain_community.utilitiesimportSQLDatabasefromlangchain.chains.sql_database.queryimportcreate_sql_query_chain# 连接 MySQL 数据库db_user="root"db_password="12345678"db_host="IP"db_port="3306"db_name="demo"db=SQLDatabase.from_uri(f"mysql+pymysql://{db_user}:{db_password}@{db_host...
db=SQLDatabase.from_uri(f"sqlite:///{db_file_path}") 1. 2. 测试数据库 print(db.dialect)print(db.get_usable_table_names())print(db.run("SELECT * FROM Artist LIMIT 1;")) 1. 2. 3. sqlite ['Album', 'Artist', 'Customer', 'Employee', 'Genre', 'Invoice', 'InvoiceLine', 'Me...
节省令牌(tokens):通过仅考虑相关表的模式,SQL Agent能够更有效地使用令牌,减少不必要的令牌消耗。使用create_sql_agent函数可以初始化这个代理。这个代理包含了SQLDatabaseToolkit,它提供了一系列工具来:创建和执行查询 检查查询语法 检索表描述 ...等等这
db= SQLDatabase.from_uri(f"mysql+pymysql://{db_user}:{db_password}@{db_host}/{db_name}") 连接大模型LLM,目前智谱AI提供两种模型GLM-3-Turbo和GLM-4,这里我们用的是GLM-4: llm = ChatOpenAI( model_name="glm-4", openai_api_base="https://open.bigmodel.cn/api/paas/v4", ...
db = SQLDatabase.from_uri("sqlite:///Chinook.db") # 通过LLM 获取查询语句 from langchain.chains import create_sql_query_chain from langchain_openai import ChatOpenAI llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0) from langchain_community.tools.sql_database.tool import QuerySQLData...
SQLDatabase 支持连接sqlite、MySQL、PostgreSQL等常见数据库。比如:db = SQLDatabase.from_uri("sqlite://./notebooks/Chinook.db")。 代码语言:python 代码运行次数:0 运行 AI代码解释 class SQLDatabase: """SQLAlchemy wrapper around a database.""" def __init__( self, engine: Engine, schema: Opti...