mirror of
https://github.com/hwchase17/langchain.git
synced 2025-07-09 14:35:50 +00:00
Lint
This commit is contained in:
parent
52e5a8b43e
commit
040bb2983d
@ -17,6 +17,7 @@ from langchain.callbacks.manager import (
|
||||
CallbackManagerForToolRun,
|
||||
Callbacks,
|
||||
)
|
||||
from langchain.load.serializable import Serializable
|
||||
from langchain.pydantic_v1 import (
|
||||
BaseModel,
|
||||
Extra,
|
||||
@ -165,7 +166,7 @@ class ChildTool(BaseTool):
|
||||
] = False
|
||||
"""Handle the content of the ToolException thrown."""
|
||||
|
||||
class Config:
|
||||
class Config(Serializable.Config):
|
||||
"""Configuration for this pydantic object."""
|
||||
|
||||
arbitrary_types_allowed = True
|
||||
|
@ -2,7 +2,7 @@
|
||||
"""Tools for interacting with Spark SQL."""
|
||||
from typing import Any, Dict, Optional
|
||||
|
||||
from langchain.pydantic_v1 import BaseModel, Extra, Field, root_validator
|
||||
from langchain.pydantic_v1 import BaseModel, Field, root_validator
|
||||
|
||||
from langchain.schema.language_model import BaseLanguageModel
|
||||
from langchain.callbacks.manager import (
|
||||
@ -21,6 +21,9 @@ class BaseSparkSQLTool(BaseModel):
|
||||
|
||||
db: SparkSQL = Field(exclude=True)
|
||||
|
||||
class Config(BaseTool.Config):
|
||||
pass
|
||||
|
||||
|
||||
class QuerySparkSQLTool(BaseSparkSQLTool, BaseTool):
|
||||
"""Tool for querying a Spark SQL."""
|
||||
|
Loading…
Reference in New Issue
Block a user