Limits & Errors

{
  "code": "INVALID_ARGUMENT",
  "message": "Missing 'type' or 'id' parameter."
}

{
  "code": "INVALID_ARGUMENT",
  "message": "Parameter \"sub1\" exceeds the maximum length of 600 characters (got 847)."
}

import json

def call_tool(agent_response):
    for block in agent_response.content:
        if not hasattr(block, "text"):
            continue

        # Attempt to detect an error envelope: {"code": "...", "message": "..."}
        try:
            payload = json.loads(block.text)
        except json.JSONDecodeError:
            # Not JSON — treat as a successful plain-text result
            return block.text

        # Valid JSON but not an error envelope — successful structured result
        if "code" not in payload:
            return block.text

        code = payload.get("code")
        message = payload.get("message", "Unknown error")

        if code == "INVALID_ARGUMENT":
            raise ValueError(f"Bad request: {message}")
        elif code == "UNAUTHENTICATED":
            raise PermissionError(f"Auth failed: {message}")
        elif code == "PERMISSION_DENIED":
            raise PermissionError(f"Insufficient permissions: {message}")
        elif code == "NOT_FOUND":
            raise LookupError(f"Not found: {message}")
        elif code == "RESOURCE_EXHAUSTED":
            raise RuntimeError(f"Rate limited: {message}")
        elif code == "INTERNAL":
            raise RuntimeError(f"Server error: {message}")