Update app.py
Browse files
app.py
CHANGED
@@ -4,6 +4,7 @@ import requests
|
|
4 |
import inspect
|
5 |
import pandas as pd
|
6 |
from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, VisitWebpageTool, Tool, HfApiModel, ToolCallingAgent
|
|
|
7 |
# (Keep Constants as is)
|
8 |
# --- Constants ---
|
9 |
DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
@@ -17,7 +18,7 @@ class AttachmentDownloadTool(Tool):
|
|
17 |
"description": "task_id that needs to download attachment files."
|
18 |
}
|
19 |
}
|
20 |
-
output_type =
|
21 |
|
22 |
def forward(self, task_id):
|
23 |
download_url = f"{api_url}/files/"
|
|
|
4 |
import inspect
|
5 |
import pandas as pd
|
6 |
from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, VisitWebpageTool, Tool, HfApiModel, ToolCallingAgent
|
7 |
+
import io
|
8 |
# (Keep Constants as is)
|
9 |
# --- Constants ---
|
10 |
DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
|
|
18 |
"description": "task_id that needs to download attachment files."
|
19 |
}
|
20 |
}
|
21 |
+
output_type = io.BytesIO
|
22 |
|
23 |
def forward(self, task_id):
|
24 |
download_url = f"{api_url}/files/"
|