diff --git a/scripts/ollama-proxy.py b/scripts/ollama-proxy.py index 7b587c5..e7be8e0 100644 --- a/scripts/ollama-proxy.py +++ b/scripts/ollama-proxy.py @@ -1282,7 +1282,7 @@ class Handler(http.server.BaseHTTPRequestHandler): "is_active": (fname[:-4] == CURRENT_SESSION), }) self._send(200, {"sessions": sorted(sessions, key=lambda s: s["size_bytes"], reverse=True)}) - elif path == "/v1/models": + elif path in ("/v1/models", "/models"): self._send(200, { "object": "list", "data": [ @@ -1295,6 +1295,10 @@ class Handler(http.server.BaseHTTPRequestHandler): "max_context_length": CTX_SIZE, "max_tokens": MAX_OUTPUT, "max_output_tokens": MAX_OUTPUT, + "input_modalities": ["text", "image"], + "modalities": ["text", "image"], + "supports_vision": True, + "supports_image": True, } for tag in MODEL_TAGS ],