adding new result_as_answer options

This commit is contained in:
João Moura
2024-07-04 00:09:18 -04:00
parent a801f1cd24
commit b4d91d1ce0

View File

@@ -22,6 +22,8 @@ class BaseTool(BaseModel, ABC):
"""Flag to check if the description has been updated."""
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."""
result_as_answer: bool = False
"""Flag to check if the tool should be the final agent answer."""
@validator("args_schema", always=True, pre=True)
def _default_args_schema(cls, v: Type[V1BaseModel]) -> Type[V1BaseModel]: