Removed duplicate BaseModel dependencies (#2471)

Removed duplicate BaseModel dependencies in class inheritances.
Also, sorted imports by `isort`.
This commit is contained in:
leo-gan
2023-04-06 12:45:16 -07:00
committed by GitHub
parent b6a101d121
commit fd69cc7e42
99 changed files with 187 additions and 257 deletions
+3 -3
View File
@@ -3,7 +3,7 @@ from __future__ import annotations
from typing import Any, Dict, List
from pydantic import BaseModel, Extra, Field
from pydantic import Extra, Field
from langchain.chains.base import Chain
from langchain.chains.llm import LLMChain
@@ -13,7 +13,7 @@ from langchain.schema import BaseLanguageModel
from langchain.sql_database import SQLDatabase
class SQLDatabaseChain(Chain, BaseModel):
class SQLDatabaseChain(Chain):
"""Chain for interacting with SQL Database.
Example:
@@ -107,7 +107,7 @@ class SQLDatabaseChain(Chain, BaseModel):
return "sql_database_chain"
class SQLDatabaseSequentialChain(Chain, BaseModel):
class SQLDatabaseSequentialChain(Chain):
"""Chain for querying SQL database that is a sequential chain.
The chain is as follows: