cyberosa commited on
Commit
3c9e5cf
·
1 Parent(s): 6988f95

Fixing the tool

Browse files
Files changed (1) hide show
  1. tools/final_answer.py +7 -1
tools/final_answer.py CHANGED
@@ -1,10 +1,16 @@
1
  from typing import Any, Optional
2
  from smolagents.tools import Tool
3
 
 
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:
 
1
  from typing import Any, Optional
2
  from smolagents.tools import Tool
3
 
4
+
5
  class FinalAnswerTool(Tool):
6
  name = "final_answer"
7
  description = "Provides a final answer to the given problem."
8
+ inputs = {
9
+ "final_answer": {
10
+ "type": "any",
11
+ "description": "The final answer to the problem",
12
+ }
13
+ }
14
  output_type = "any"
15
 
16
  def forward(self, answer: Any) -> Any: