File size: 574 Bytes
044c01f
8fe992b
 
 
 
 
044c01f
 
8fe992b
 
044c01f
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from typing import Any
from smolagents.tools import Tool

class FinalAnswerTool(Tool):
    name = "final_answer"
    description = "Provides a final answer to the given problem."
    inputs = {"answer": {"type": "string", "description": "The final answer to the problem"}}
    output_type = "string"

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)  # Initialize the parent class
        self.is_initialized = True  # Correctly initialized

    def forward(self, answer: Any) -> str:
        return str(answer)  # Ensures a string output