Artsem Radzevich commited on
Commit
9073c71
·
1 Parent(s): 03dd32d

Fixed by inherit Tool class.

Browse files
tools/excel_analysis_tool.py CHANGED
@@ -1,7 +1,8 @@
 
1
  import pandas as pd
2
  from huggingface_hub import hf_hub_download
3
 
4
- class ExcelAnalysisTool:
5
  def __init__(self):
6
  self.name = "ExcelAnalysisTool"
7
  self.description = (
 
1
+ from smolagents.tools import Tool
2
  import pandas as pd
3
  from huggingface_hub import hf_hub_download
4
 
5
+ class ExcelAnalysisTool(Tool):
6
  def __init__(self):
7
  self.name = "ExcelAnalysisTool"
8
  self.description = (
tools/wiki_tool.py CHANGED
@@ -1,6 +1,7 @@
 
1
  import wikipedia
2
 
3
- class WikiTool:
4
  def __init__(self):
5
  self.name = "WikiTool"
6
  self.description = (
 
1
+ from smolagents.tools import Tool
2
  import wikipedia
3
 
4
+ class WikiTool(Tool):
5
  def __init__(self):
6
  self.name = "WikiTool"
7
  self.description = (