github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-09 14:35:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Lint

Browse Source
This commit is contained in:
Nuno Campos 2023-10-02 10:11:26 +01:00
parent 52e5a8b43e
commit 040bb2983d
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
libs/langchain/langchain/tools/base.py
View File

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

5
libs/langchain/langchain/tools/spark_sql/tool.py
View File

@ -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."""