tool renaming
Browse files
agent.py
CHANGED
@@ -5,8 +5,8 @@ import requests
|
|
5 |
import pandas as pd
|
6 |
from huggingface_hub import InferenceClient
|
7 |
|
8 |
-
class
|
9 |
-
name = "
|
10 |
description = "This tool allows to download the file attached to the question"
|
11 |
output_type = "string"
|
12 |
inputs = {
|
@@ -187,7 +187,7 @@ manager_agent = CodeAgent(
|
|
187 |
model_id="gemini/gemini-2.5-flash-preview-04-17",
|
188 |
api_key=os.getenv("GOOGLE_API_KEY"),
|
189 |
),
|
190 |
-
tools=[
|
191 |
managed_agents=[web_agent, audio_agent],
|
192 |
additional_authorized_imports=[
|
193 |
"pandas"
|
|
|
5 |
import pandas as pd
|
6 |
from huggingface_hub import InferenceClient
|
7 |
|
8 |
+
class GetFileTool(Tool):
|
9 |
+
name = "get_file_tool"
|
10 |
description = "This tool allows to download the file attached to the question"
|
11 |
output_type = "string"
|
12 |
inputs = {
|
|
|
187 |
model_id="gemini/gemini-2.5-flash-preview-04-17",
|
188 |
api_key=os.getenv("GOOGLE_API_KEY"),
|
189 |
),
|
190 |
+
tools=[GetFileTool(), LoadXlsxFileTool(), LoadTextFileTool(), ImageAnalysisTool()],
|
191 |
managed_agents=[web_agent, audio_agent],
|
192 |
additional_authorized_imports=[
|
193 |
"pandas"
|