UDAYAARKA commited on
Commit
cb377d7
·
verified ·
1 Parent(s): da0aac6

Update tools/final_answer.py

Browse files
Files changed (1) hide show
  1. tools/final_answer.py +3 -3
tools/final_answer.py CHANGED
@@ -4,10 +4,10 @@ from smolagents.tools import Tool
4
  class FinalAnswerTool(Tool):
5
  name = "final_answer"
6
  description = "Provides a final answer to the given problem."
7
- inputs = {'answer': {'type': 'string', 'description': 'The final answer to the problem'}}
8
- output_type = "string"
9
 
10
- def forward(self, answer: Any) -> any:
11
  return answer
12
 
13
  def __init__(self, *args, **kwargs):
 
4
  class FinalAnswerTool(Tool):
5
  name = "final_answer"
6
  description = "Provides a final answer to the given problem."
7
+ inputs = {'answer': {'type': 'any', 'description': 'The final answer to the problem'}}
8
+ output_type = "any"
9
 
10
+ def forward(self, answer: Any) -> Any:
11
  return answer
12
 
13
  def __init__(self, *args, **kwargs):