Spaces:
Runtime error
Runtime error
File size: 1,820 Bytes
ed4d993 |
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 52 53 54 |
"""Util that calls several NASA APIs."""
import json
import requests
from langchain_core.pydantic_v1 import BaseModel
IMAGE_AND_VIDEO_LIBRARY_URL = "https://images-api.nasa.gov"
class NasaAPIWrapper(BaseModel):
"""Wrapper for NASA API."""
def get_media(self, query: str) -> str:
params = json.loads(query)
if params.get("q"):
queryText = params["q"]
params.pop("q")
else:
queryText = ""
response = requests.get(
IMAGE_AND_VIDEO_LIBRARY_URL + "/search?q=" + queryText, params=params
)
data = response.json()
return data
def get_media_metadata_manifest(self, query: str) -> str:
response = requests.get(IMAGE_AND_VIDEO_LIBRARY_URL + "/asset/" + query)
return response.json()
def get_media_metadata_location(self, query: str) -> str:
response = requests.get(IMAGE_AND_VIDEO_LIBRARY_URL + "/metadata/" + query)
return response.json()
def get_video_captions_location(self, query: str) -> str:
response = requests.get(IMAGE_AND_VIDEO_LIBRARY_URL + "/captions/" + query)
return response.json()
def run(self, mode: str, query: str) -> str:
if mode == "search_media":
output = self.get_media(query)
elif mode == "get_media_metadata_manifest":
output = self.get_media_metadata_manifest(query)
elif mode == "get_media_metadata_location":
output = self.get_media_metadata_location(query)
elif mode == "get_video_captions_location":
output = self.get_video_captions_location(query)
else:
output = f"ModeError: Got unexpected mode {mode}."
try:
return json.dumps(output)
except Exception:
return str(output)
|