SQLエージェントのサンプル

SQLiteのサンプル．
まず，データベースに接続を行う

from langchain_community.utilities import SQLDatabase
from sqlalchemy import create_engine

engine = create_engine("sqlite:///データベース.db")
db = SQLDatabase(engine=engine)

ここで，llm = ChatOpenAI(model = model, temperature = temperature)が与えられているとして，

from langchain_community.agent_toolkits import create_sql_agent

agent_executor = create_sql_agent(llm=llm, db=db, agent_type="tool-calling", verbose=True)
agent_executor.invoke({"input": "質問文"})

とすると，SQLを生成して推論を行う．