Writing changes fixed locally

tools/nasa_neo_data_fetcher.py

@@ -1,11 +1,12 @@
 import os
 import requests
-import json
 from calendar import monthrange
 from smolagents.tools import Tool
 from typing import Generator
 from dotenv import load_dotenv
 
+load_dotenv()
+
 
 class NasaNeoDataFetcher(Tool):
     name = "nasa_neo_data_fetcher"
@@ -23,17 +24,16 @@ class NasaNeoDataFetcher(Tool):
     output_type = "string"
 
     def __init__(self):
-        load_dotenv()
         self.api_key = os.getenv("NASA_API_KEY")
         self.root_url = "https://api.nasa.gov/neo/rest/v1/feed?"
         self.is_initialized = False
 
-    def forward(self, start_date: str, end_date: str) -> list:
+    def forward(self, start_date: str, end_date: str) -> list[tuple[str, dict]]:
         """A function to fetch Near Earth Object data from NASA API for a given date range.
         Args:
             start_date: A string representing the start date of the data to be fetched.
             end_date: A string representing the end date of the data to be fetched.
-        Returns: The data fetched from the API as a list of tuples containing JSON-like dictionaries.
+        Returns: The data fetched from the API as a list of tuples containing the date and the JSON-like dictionaries.
         """
         return self._fetch_neo_data_in_chunks(start_date, end_date)
 
@@ -93,13 +93,15 @@ class NasaNeoDataFetcher(Tool):
             data = self._get_nasa_neo_data(start_date=start_date, end_date=end_date)
             yield data
 
-    def _fetch_neo_data_in_chunks(self, start_date: str, end_date: str) -> list:
+    def _fetch_neo_data_in_chunks(
+        self, start_date: str, end_date: str
+    ) -> list[tuple[str, dict]]:
         """A function that fetches Near Earth Object data from NASA API in chunks of 7 days. NB: The API
         returns an error when you try to pull data covering longer than 7 days.
         Args:
             start_date: A string representing the start date of the data to be fetched.
             end_date: A string representing the end date of the data to be fetched.
-        Returns: A list of JSON-like dictionaries of Near Earth Object data fetched from the API.
+        Returns: A list of tuples containing the date and the JSON-like dictionaries of Near Earth Object data fetched from the API.
         """
         neo_data = []
         start_year, start_month, start_day = self._split_date_into_ymd(start_date)
@@ -127,4 +129,4 @@ class NasaNeoDataFetcher(Tool):
                     for item in data:
                         if "near_earth_objects" in item:
                             neo_data.extend(item["near_earth_objects"].items())
-        return neo_data
+        return neo_data