nanobot: 2026-09-13 14:37:42 - Ollama model capabilities: log + skript do ai projektu
This commit is contained in:
31
projects/ai/artifacts/ollama-model-capabilities.py
Normal file
31
projects/ai/artifacts/ollama-model-capabilities.py
Normal file
@@ -0,0 +1,31 @@
|
||||
#!/usr/bin/env python3
|
||||
"""List Ollama models on nvidia.hell:11434 with size + capabilities."""
|
||||
import json
|
||||
import urllib.request
|
||||
|
||||
BASE = "http://nvidia.hell:11434"
|
||||
|
||||
with urllib.request.urlopen(f"{BASE}/api/tags", timeout=10) as r:
|
||||
tags = json.load(r)
|
||||
|
||||
models = []
|
||||
for m in tags.get("models", []):
|
||||
name = m.get("name")
|
||||
size = m.get("size", 0)
|
||||
# fetch capabilities via /api/show
|
||||
req = urllib.request.Request(
|
||||
f"{BASE}/api/show",
|
||||
data=json.dumps({"model": name}).encode(),
|
||||
headers={"Content-Type": "application/json"},
|
||||
)
|
||||
try:
|
||||
with urllib.request.urlopen(req, timeout=10) as r:
|
||||
show = json.load(r)
|
||||
caps = show.get("capabilities", [])
|
||||
except Exception as e:
|
||||
caps = [f"ERR: {e}"]
|
||||
models.append({"name": name, "size_gb": round(size / 1e9, 1), "capabilities": caps})
|
||||
|
||||
models.sort(key=lambda x: x["size_gb"])
|
||||
for m in models:
|
||||
print(f"{m['size_gb']:>6.1f} GB {','.join(m['capabilities']):<40} {m['name']}")
|
||||
Reference in New Issue
Block a user