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"
|