Merge pull request #188 from sr5434/iss

Add an ISS position tracker to the Astro plugin

src/autogpt_plugins/astro/README.md

@@ -1,5 +1,5 @@
 # Auto-GPT Space Plugin
-This plugin enables AutoGPT to see how many people are in space. This can help enable AutoGPT to better achieve its goals
+This plugin enables AutoGPT to see how many people are in space and see the position of the ISS. This can help enable AutoGPT to better achieve its goals.
 
 ## Use cases
  - Researching how many people are in space

src/autogpt_plugins/astro/__init__.py

@@ -1,4 +1,4 @@
-"""This is the email plugin for Auto-GPT."""
+"""This is the space plugin for Auto-GPT."""
 import os
 from typing import Any, Dict, List, Optional, Tuple, TypedDict, TypeVar
 
@@ -34,6 +34,12 @@ def post_prompt(self, prompt: PromptGenerator) -> PromptGenerator:
                 {},
                 get_num_astronauts,
             )
+            prompt.add_command(
+                "Get the coordinates of the ISS",
+                "get_coords_iss",
+                {},
+                get_coords_iss,
+            )
 
         return prompt
 
@@ -246,4 +252,4 @@ def can_handle_report(self) -> bool:
         return False
 
     def report(self, message: str) -> None:
-        pass
+        pass

src/autogpt_plugins/astro/astronauts.py

@@ -16,3 +16,18 @@ def get_num_astronauts():
     data = response.json()
     #Extract the number and return it
     return data["number"]
+
+def get_coords_iss():
+    """Get the coordinates of the ISS
+    Args:
+        None
+    Returns:
+        int: The latitude of the ISS.
+        int: The longitude of the ISS.
+    """
+    #Get the data
+    response = requests.get("http://api.open-notify.org/iss-now.json")
+    #Convert it to JSON
+    data = response.json()
+    #Extract the number and return it
+    return float(data["iss_position"]["latitude"]), float(data["iss_position"]["longitude"])

src/autogpt_plugins/astro/test_astro_plugin.py

@@ -1,5 +1,10 @@
 from .astronauts import get_num_astronauts
-
+from .astronauts import get_coords_iss
 
 def test_astro():
-    assert type(get_num_astronauts())==int
+    assert type(get_num_astronauts())==int
+
+def test_iss():
+    latitude, longitude = get_coords_iss()
+    assert type(latitude)==float
+    assert type(longitude)==float