from pydantic import BaseModel
from typing import Optional

# Request body model
class CodeExecuteRequest(BaseModel):
    code: str
    session_id: Optional[str] = None
    message_id: Optional[int] = None
    
class CodeEditRequest(BaseModel):
    original_code: str
    user_prompt: str
    
class CodeFixRequest(BaseModel):
    code: str
    error: str
    
class CodeCleanRequest(BaseModel):
    code: str
    
class GetLatestCodeRequest(BaseModel):
    message_id: int