jjjulllesss commited on
Commit
480fc0c
·
verified ·
1 Parent(s): 7a0516a

tool renaming

Browse files
Files changed (1) hide show
  1. agent.py +3 -3
agent.py CHANGED
@@ -5,8 +5,8 @@ import requests
5
  import pandas as pd
6
  from huggingface_hub import InferenceClient
7
 
8
- class getFile(Tool):
9
- name = "getFileTool"
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=[getFile(), LoadXlsxFileTool(), LoadTextFileTool(), ImageAnalysisTool()],
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"