Artsem Radzevich
commited on
Commit
·
9073c71
1
Parent(s):
03dd32d
Fixed by inherit Tool class.
Browse files- tools/excel_analysis_tool.py +2 -1
- tools/wiki_tool.py +2 -1
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 = (
|