Create fetch_file_url.py

tools/fetch_file_url.py

@@ -0,0 +1,18 @@
+from typing import Any, Optional
+from smolagents.tools import Tool
+from gradio_client import Client
+
+class FetchFileURLTool(Tool):
+  name = "fetch_file_url"
+  description = "Get the correct URL at which a task's file is located for download."
+  inputs = {'lookup_link': {'type': 'any', 'description': 'The link at which to get the URL of the file. It has the pattern https://agents-course-unit4-scoring.hf.space/files/{{ task_id }}'}}
+  output_type = "any"
+
+  def forward(self, lookup_link: Any) -> Any:
+    print('*$%: Performing fetching of file URL at '+lookup_link)
+    requests.get(lookup_link)
+
+    return result['markdown_content']
+
+  def __init__(self, *args, **kwargs):
+    self.is_initialized = False