unification

handler.py

@@ -181,6 +181,30 @@ class EndpointHandler:
         if width and height:
             out.update({"width": width, "height": height})
         out.update({"task": task})
+
+        # Print prompt, dimensions, and raw output
+        try:
+            print(f"[moondream-endpoint] Prompt: {text_piece}")
+        except Exception:
+            pass
+        if width and height:
+            try:
+                print(f"[moondream-endpoint] Received image size: {width}x{height}")
+            except Exception:
+                pass
+        try:
+            print(f"[moondream-endpoint] Raw output: {json.dumps(out)}")
+        except Exception:
+            pass
+
+        # Ensure strict shape for point task: include points[] and raw
+        if task == "point":
+            # Ensure points array exists
+            if not isinstance(out.get("points"), list) or not out["points"]:
+                return {"error": "No points returned"}
+            # Attach raw for strict client and drop width/height from payload
+            return {"points": out["points"], "raw": out}
+
         return out
 
     @staticmethod