albertvillanova's picture
Upload tool
ca00577 verified
raw
history blame contribute delete
433 Bytes
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