Update agent.py

agent.py

@@ -18,12 +18,20 @@ def download_and_get_path_for_provided_file(path: str):
         bytes: The binary content of the downloaded file
 
     """
-    file_path = hf_hub_download(
-        repo_id="gaia-benchmark/GAIA",
-        filename="2023/test/063800f6-8832-4856-972b-17b877612533.png",
-        repo_type="dataset",
-        token=os.environ['HF_TOKEN']
-    )    
+    for dataset in ["test","validation"]:
+        try:
+            file_path = hf_hub_download(
+            repo_id="gaia-benchmark/GAIA",
+            filename=f"2023/{dataset}/{file_path}",
+            repo_type="dataset",
+            token=os.environ['HF_TOKEN'])
+            if file_path:
+                return file_path
+        except Exception as e:
+            continue
+
+    file_name = "99c9cc74-fdc8-46c6-8f8d-3ce2d3bfeea3.mp3"
+    file_path = get_path(file_name)
     return file_path