|
|
@ -5,7 +5,6 @@ from pydantic import Field
|
|
|
|
|
|
|
|
|
|
|
|
from langchain.agents.agent_toolkits.base import BaseToolkit
|
|
|
|
from langchain.agents.agent_toolkits.base import BaseToolkit
|
|
|
|
from langchain.llms.base import BaseLLM
|
|
|
|
from langchain.llms.base import BaseLLM
|
|
|
|
from langchain.llms.openai import OpenAI
|
|
|
|
|
|
|
|
from langchain.sql_database import SQLDatabase
|
|
|
|
from langchain.sql_database import SQLDatabase
|
|
|
|
from langchain.tools import BaseTool
|
|
|
|
from langchain.tools import BaseTool
|
|
|
|
from langchain.tools.sql_database.tool import (
|
|
|
|
from langchain.tools.sql_database.tool import (
|
|
|
@ -20,7 +19,7 @@ class SQLDatabaseToolkit(BaseToolkit):
|
|
|
|
"""Toolkit for interacting with SQL databases."""
|
|
|
|
"""Toolkit for interacting with SQL databases."""
|
|
|
|
|
|
|
|
|
|
|
|
db: SQLDatabase = Field(exclude=True)
|
|
|
|
db: SQLDatabase = Field(exclude=True)
|
|
|
|
llm: BaseLLM = Field(default_factory=lambda: OpenAI(temperature=0))
|
|
|
|
llm: BaseLLM = Field(exclude=True)
|
|
|
|
|
|
|
|
|
|
|
|
@property
|
|
|
|
@property
|
|
|
|
def dialect(self) -> str:
|
|
|
|
def dialect(self) -> str:
|
|
|
|