Zach Jones d7942a9f19 Fix type annotation for QueryCheckerTool.llm (#3237) ...

Currently `langchain.tools.sql_database.tool.QueryCheckerTool` has a
field `llm` with type `BaseLLM`. This breaks initialization for some
LLMs. For example, trying to use it with GPT4:

```python
from langchain.sql_database import SQLDatabase
from langchain.chat_models import ChatOpenAI
from langchain.tools.sql_database.tool import QueryCheckerTool


db = SQLDatabase.from_uri("some_db_uri")
llm = ChatOpenAI(model_name="gpt-4")
tool = QueryCheckerTool(db=db, llm=llm)

# pydantic.error_wrappers.ValidationError: 1 validation error for QueryCheckerTool
# llm
#   Can't instantiate abstract class BaseLLM with abstract methods _agenerate, _generate, _llm_type (type=type_error)
```

Seems like much of the rest of the codebase has switched from `BaseLLM`
to `BaseLanguageModel`. This PR makes the change for QueryCheckerTool as
well

Co-authored-by: Zachary Jones <zjones@zetaglobal.com>

2023-04-20 18:50:59 -07:00

..

__init__.py

Add a SQL agent for interacting with SQL Databases and JSON Agent for interacting with large JSON blobs (#1150)

2023-02-28 19:44:39 -08:00

prompt.py

Add a SQL agent for interacting with SQL Databases and JSON Agent for interacting with large JSON blobs (#1150)

2023-02-28 19:44:39 -08:00

tool.py

Fix type annotation for QueryCheckerTool.llm (#3237)

2023-04-20 18:50:59 -07:00