方法一:使用SQLDatabaseToolKit和Agent 创建MySQL数据库连接: 使用SQLDatabase.from_uri方法从数据库URI创建MySQL数据库连接。 python from langchain.chains.database import SQLDatabase MYSQL_URI = "mysql+pymysql://username:password@localhost:3306/dbname" db = SQLDatabase.from_uri(MYSQL_URI) 创建SQLDat...
导入langchain模块 OpenAI, SQLDatabase, 和 SQLDatabaseChain 从环境变量文件中获取 OPENAI_API_KEY 使用SQLDatabase.from_uri() 方法设置数据库连接,指定连接URL 使用OpenAI() 创建llm对象,指定温度和openai_api_key 使用SQLDatabaseChain() 创建数据库链对象 db_chain,指定llm和数据库对象 get_prompt() 从控制...
节省令牌(tokens):通过仅考虑相关表的模式,SQL Agent能够更有效地使用令牌,减少不必要的令牌消耗。使用create_sql_agent函数可以初始化这个代理。这个代理包含了SQLDatabaseToolkit,它提供了一系列工具来:创建和执行查询 检查查询语法 检索表描述 ...等等这
fromlangchain.utilitiesimportSQLDatabasefromlangchain.chat_modelsimportChatOpenAIfromlangchain.schema.output_parserimportStrOutputParserfromlangchainimporthubfromkeysimportOPENAI_API_KEYimportosos.environ["OPENAI_API_KEY"]=OPENAI_API_KEY# Initialize databasedb=SQLDatabase.from_uri("sqlite:///Chinook.db")...
fromlangchain_community.utilitiesimportSQLDatabaseimportsqlalchemyassa# 连接到demo数据库db=SQLDatabase.from_uri("sqlite:///demo.db") 1. 2. 3. 4. 5. 光标查询 查询模式:cursor,将结果作为SQLAlchemy的CursorResult实例返回 result=db.run("SELECT * FROM User LIMIT 10;",fetch="cursor")print(...
from langchain.sql_database import SQLDatabase db = SQLDatabase.from_uri("sqlite:///example.db") llm = OpenAI(model_name="text-davinci-003", api_key="your_openai_api_key") db_chain = SQLDatabaseChain(llm=llm, database=db)
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...
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", ...
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...