Spaces:
Running on Zero
Running on Zero
File size: 2,829 Bytes
97ac1ad | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 | """
Unified MCP tool error response format.
Error code enumeration:
- INVALID_PARAMS: Parameter validation failed (missing required fields, type errors, value out of range)
- MODEL_NOT_FOUND: The specified model name does not exist
- ARCHITECTURE_NOT_FOUND: The specified architecture name does not exist
- CHAIN_TYPE_NOT_FOUND: The specified chain/injector type is invalid
- FEATURE_NOT_SUPPORTED: The current model does not support the requested feature
- TASK_NOT_FOUND: The async task ID does not exist
- MODEL_OOM: GPU out of memory
- INTERNAL_ERROR: Internal server error
"""
def make_error(code: str, message: str, details: dict = None) -> dict:
"""
Construct a unified MCP tool error response.
Args:
code: Error code (UPPER_SNAKE_CASE format)
message: Human-readable error description
details: Optional details dictionary
Returns:
Standardized error response dictionary
"""
error = {
"error": {
"code": code,
"message": message,
}
}
if details:
error["error"]["details"] = details
return error
def make_validation_error(
message: str = "Request validation failed.",
missing_fields: list = None,
invalid_fields: dict = None,
) -> dict:
"""
Construct a parameter validation failure error response.
Args:
message: Error description
missing_fields: List of missing required field names
invalid_fields: Key-value pairs of invalid fields, key=field name, value=reason description
Returns:
Standardized INVALID_PARAMS error response
"""
details = {}
if missing_fields:
details["missing_fields"] = missing_fields
if invalid_fields:
details["invalid_fields"] = invalid_fields
return make_error("INVALID_PARAMS", message, details if details else None)
def make_not_found_error(resource_type: str, resource_id: str) -> dict:
"""
Construct a resource-not-found error response.
Args:
resource_type: Resource type (e.g., "model", "architecture", "chain_type", "task")
resource_id: Resource identifier
Returns:
Standardized *_NOT_FOUND error response
"""
code_map = {
"model": "MODEL_NOT_FOUND",
"architecture": "ARCHITECTURE_NOT_FOUND",
"chain_type": "CHAIN_TYPE_NOT_FOUND",
"task": "TASK_NOT_FOUND",
}
code = code_map.get(resource_type, f"{resource_type.upper()}_NOT_FOUND")
return make_error(
code,
f"The specified {resource_type} '{resource_id}' was not found.",
{"resource_type": resource_type, "resource_id": resource_id},
)
|