from typing import Any, Optional | |
from smolagents.tools import Tool | |
class CustomTool(Tool): | |
name = "custom_tool" | |
description = "A custom test tool" | |
inputs = {'input_text': {'type': 'string', 'description': 'Some input text'}} | |
output_type = "string" | |
def forward(self, input_text: str) -> str: | |
return f"Processed: {input_text}" | |
def __init__(self, *args, **kwargs): | |
self.is_initialized = False | |