Remove unnecessary task_id argument from tools

custom_tools.py

@@ -138,22 +138,21 @@ def wiki_search(query: str) -> str:
         return f"An unexpected error occurred: {e}"
 
 @tool
-def sum_excel_cols(task_id: str, file_name: str, column_names: List[str]) -> float:
+def sum_excel_cols(file_name: str, column_names: List[str]) -> float:
     """
     Sum the values of specified columns in a pandas DataFrame read from an Excel file.
 
     Args:
-        task_id (str): The ID of the task.
         file_name (str): The path to the Excel file.
-        column_names (List[str]): A list of column names to sum.
+        column_names (List[str]): A list of column names to sum. The column names need to exist and should be extracted from the input of the read_file_content tool
 
     Returns:
         float: The sum of the specified columns.
 
     Example:
-        sum_excel_cols("task123", "data.xlsx", ["Column1", "Column2"]) -> 100.0
+        sum_excel_cols("data.xlsx", ["Column1", "Column2"]) -> 100.0
     """
-    file_status = download_file(task_id, file_name)
+    file_status = download_file(file_name)
 
     if not os.path.exists(file_name):
         return f"File {file_name} does not exist."
@@ -211,18 +210,17 @@ def youtube_transcript(url: str) -> str:
 
 
 @tool 
-def read_file_content(task_id: str, file_name: str) -> str:
+def read_file_content(file_name: str) -> str:
     """
-    Read the text from a file and return its content as a string.
+    Read the text from an input file and return its content as a string.
 
     Args:
-        task_id (str): The unique identifier for the task.
         file_name (str): The name of the file.
 
     Returns:
         str: The content of the file, or a detailed error message.
     """
-    download_state = download_file(task_id, file_name)
+    download_state = download_file(file_name)
     if download_state == f"Success downloading {file_name}":
         file_content = read_file(file_name)
     
@@ -268,11 +266,10 @@ def analyse_youtube_video(url: str, video_question: str):
 
 
 @tool
-def analyze_image(task_id: str, file_name: str, question: str) -> str:
+def analyze_image(file_name: str, question: str) -> str:
     """    
     Download and analyze an image based on a given question.
     Args:
-        task_id (str): The ID of the task.
         file_name (str): The name of the image file.
         question (str): The question to be answered about the image.
     Returns:
@@ -280,7 +277,7 @@ def analyze_image(task_id: str, file_name: str, question: str) -> str:
     """
     try:        
         if not os.path.exists(file_name):
-            file_status = download_file(task_id, file_name)
+            file_status = download_file(file_name)
         
         if not os.path.exists(file_name):
             return f"File {file_name} does not exist : {file_status}"
@@ -301,17 +298,16 @@ def analyze_image(task_id: str, file_name: str, question: str) -> str:
 
 # Build a tool to transcript a sound .mp3 file with a LLM, based on the filename as a parameter
 @tool
-def transcript_audio(task_id: str, file_name: str) -> str:
+def transcript_audio(file_name: str) -> str:
     """    Generate a transcript for an audio file using a language model.
     Args:
-        task_id (str): The ID of the task.
         file_name (str): The name of the image file.
     Returns:
         str: A transcript of the audio.
     """
     # Download the image file if not already present
     if not os.path.exists(file_name):
-        file_status = download_file(task_id, file_name)
+        file_status = download_file(file_name)
     
     # Check if the file exists
     if not os.path.exists(file_name):

utils.py

@@ -12,17 +12,17 @@ from typing import List
 
 
 
-def download_file(task_id: str, file_name: str) -> str:
+def download_file(file_name: str) -> str:
     """
     Download a file from a given task ID and save it to the local filesystem.
 
     Args:
-        task_id (str): The ID of the task whose file is to be downloaded.
         file_name (str): The name of the file to be saved.
 
     Returns:
         str: A message indicating the result of the download operation.
     """
+    task_id = file_name.split(".")[0].split("/")[-1]
     url = f"https://agents-course-unit4-scoring.hf.space/files/{task_id}"
 
     try: