github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-06 13:33:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove default usage of openai model in SQLDatabaseToolkit (#2884)

Browse Source 
#2866

This toolkit used openai LLM as the default, which could incurr unwanted
cost.
This commit is contained in:
Cao Hoang
2023-04-25 06:27:38 +07:00
committed by GitHub
parent 675d86aa11
commit 26035dfa59
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
langchain/agents/agent_toolkits/sql/toolkit.py
View File

@@ -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: