13 lines
428 B
Python
13 lines
428 B
Python
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)
|