from smolagents import Tool
from typing import Any, Optional

class SimpleTool(Tool):
    name = "save_notes_to_file"
    description = "Save generated notes to a file."
    inputs = {"unit_number":{"type":"integer","description":"The unit number"},"unit_name":{"type":"string","description":"Name of the unit"},"content":{"type":"string","description":"Content to save"}}
    output_type = "string"

    def forward(self, unit_number: int,
        unit_name: str,
        content: str) -> str:
        """
        Save generated notes to a file.

        Args:
            unit_number: The unit number
            unit_name: Name of the unit
            content: Content to save

        Returns:
            Confirmation message
        """
        filename = f"unit_{unit_number}_{unit_name.replace(' ', '_').lower()}.md"
        with open(filename, "w") as f:
            f.write(content)
        return f"Notes for Unit {unit_number}: {unit_name} saved to {filename}"