tool.py

from smolagents import Tool
from typing import Any, Optional

class SimpleTool(Tool):
    name = "divisors"
    description = "This is a tool that returns a list of the divisors for an integer number, including itself."
    inputs = {'numb': {'type': 'integer', 'description': 'the number for which to get the list of divisors.'}}
    output_type = "string"

    def forward(self, numb: int) -> str:
        """
        This is a tool that returns a list of the forward for an integer number, including itself.

        Args:
            numb: the number for which to get the list of forward.
        """
        i = 2
        divlist = []
        while i <= numb:
            if numb % i == 0:
                print(i)
                divlist.append(i)
            i += 1

        res = ', '.join(map(str, divlist))
        return res