test-push-to-hub / boolean_default_tool.py
albertvillanova's picture
Upload tool
5c02d2d verified
raw
history blame contribute delete
574 Bytes
from typing import Any, Optional
from smolagents.tools import Tool
class BooleanDefaultTool(Tool):
name = "boolean_default_tool"
description = "A tool with a boolean default parameter"
inputs = {'text': {'type': 'string', 'description': 'Input text'}, 'flag': {'type': 'boolean', 'description': 'Boolean flag with default value', 'nullable': True}}
output_type = "string"
def forward(self, text: str, flag: bool = False) -> str:
return f"Text: {text}, Flag: {flag}"
def __init__(self, *args, **kwargs):
self.is_initialized = False