File size: 1,960 Bytes
a709497 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | from smolagents import tool
import requests
import json
import os
@tool
def fetch_questions(api_url: str) -> dict | None:
""" A tool that fetches the questions from the agents-course API and returns the JSON as a dict.
Args:
api_url: The base URL of the agents-course API.
Returns:
A dictionary containing the questions if successful, or None if there was an error.
"""
questions_url = f"{api_url}/questions"
try:
response = requests.get(questions_url)
response.raise_for_status()
return response.json()
except requests.RequestException as e:
print(f"Error fetching questions: {e}")
return None
@tool
def fetch_file(api_url:str, task_id: str, file_name: str) -> None:
""" A tool that fetches a file for a given task ID from the agents-course API.
Args:
api_url: The base URL of the agents-course API.
task_id: The ID of the task for which the file is to be fetched.
file_name: The name of the file to be downloaded.
Returns:
None if the file is downloaded successfully, or an error message if there was an issue.
"""
file_url = f"{api_url}/files/{task_id}"
downsloads_dir = "files"
os.makedirs(downsloads_dir, exist_ok=True)
local_path = os.path.join(downsloads_dir, task_id)
os.makedirs(local_path, exist_ok=True)
file_path = os.path.join(local_path, file_name)
if not os.path.exists(file_path):
print(f"Downloading file {file_name} for task {task_id} ...")
file_response = requests.get(file_url, timeout=30)
if file_response.status_code == 200:
with open(file_path, 'wb') as f:
f.write(file_response.content)
print(f"File {file_name} downloaded successfully.")
else:
print(f"Failed to download file {file_name} for task {task_id}. Status code: {file_response.status_code}")
return None
|