import json
import sys
import urllib.request

endpoint, model = sys.argv[1:3]
try:
    with urllib.request.urlopen("{}/api/tags".format(endpoint.rstrip("/")), timeout=5) as response:
        tags = json.loads(response.read().decode("utf-8"))
    models = {item.get("name", "") for item in tags.get("models", [])}
    sys.exit(0 if model in models or "{}:latest".format(model) in models else 1)
except Exception:
    sys.exit(1)