From 26035dfa5966422d9ce6e1b05f7eb4ab8eb8be27 Mon Sep 17 00:00:00 2001 From: Cao Hoang <65607230+cnhhoang850@users.noreply.github.com> Date: Tue, 25 Apr 2023 06:27:38 +0700 Subject: [PATCH] remove default usage of openai model in SQLDatabaseToolkit (#2884) #2866 This toolkit used openai LLM as the default, which could incurr unwanted cost. --- langchain/agents/agent_toolkits/sql/toolkit.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/langchain/agents/agent_toolkits/sql/toolkit.py b/langchain/agents/agent_toolkits/sql/toolkit.py index 1e32cb64..491d2460 100644 --- a/langchain/agents/agent_toolkits/sql/toolkit.py +++ b/langchain/agents/agent_toolkits/sql/toolkit.py @@ -5,7 +5,6 @@ from pydantic import Field from langchain.agents.agent_toolkits.base import BaseToolkit from langchain.llms.base import BaseLLM -from langchain.llms.openai import OpenAI from langchain.sql_database import SQLDatabase from langchain.tools import BaseTool from langchain.tools.sql_database.tool import ( @@ -20,7 +19,7 @@ class SQLDatabaseToolkit(BaseToolkit): """Toolkit for interacting with SQL databases.""" db: SQLDatabase = Field(exclude=True) - llm: BaseLLM = Field(default_factory=lambda: OpenAI(temperature=0)) + llm: BaseLLM = Field(exclude=True) @property def dialect(self) -> str: