Chris Coutinho
51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
"""Base Pydantic models for common response patterns."""
|
|
|
|
from datetime import datetime
|
|
from typing import Any, Dict, Optional, Union
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class BaseResponse(BaseModel):
|
|
"""Base response model for all MCP tool responses."""
|
|
|
|
model_config = {"json_encoders": {datetime: lambda v: v.isoformat()}}
|
|
|
|
success: bool = Field(
|
|
default=True, description="Whether the operation was successful"
|
|
)
|
|
timestamp: datetime = Field(
|
|
default_factory=datetime.now, description="Response timestamp"
|
|
)
|
|
|
|
|
|
class ErrorResponse(BaseResponse):
|
|
"""Response model for error cases."""
|
|
|
|
success: bool = Field(default=False, description="Always False for error responses")
|
|
error: str = Field(description="Error message")
|
|
error_code: Optional[str] = Field(None, description="Optional error code")
|
|
details: Optional[Dict[str, Any]] = Field(
|
|
None, description="Additional error details"
|
|
)
|
|
|
|
|
|
class SuccessResponse(BaseResponse):
|
|
"""Generic success response."""
|
|
|
|
message: Optional[str] = Field(None, description="Optional success message")
|
|
data: Optional[Dict[str, Any]] = Field(None, description="Optional response data")
|
|
|
|
|
|
class IdResponse(BaseResponse):
|
|
"""Response model for operations that return a new ID."""
|
|
|
|
id: Union[int, str] = Field(description="ID of the created or affected resource")
|
|
|
|
|
|
class StatusResponse(BaseResponse):
|
|
"""Response model for operations that return just a status."""
|
|
|
|
status_code: Optional[int] = Field(None, description="HTTP status code")
|
|
message: Optional[str] = Field(None, description="Status message")
|