Adding two default arguments to cache function

This commit is contained in:
João Moura
2024-03-27 15:04:35 -03:00
parent f17057898d
commit 80f9613959

View File

@@ -20,7 +20,7 @@ class BaseTool(BaseModel, ABC):
"""The schema for the arguments that the tool accepts."""
description_updated: bool = False
"""Flag to check if the description has been updated."""
cache_function: Optional[Callable] = lambda: True
cache_function: Optional[Callable] = lambda _args, _result: True
"""Function that will be used to determine if the tool should be cached, should return a boolean. If None, the tool will be cached."""
@validator("args_schema", always=True, pre=True)