request_models.py

from pydantic import BaseModel, Field, field_validator
from typing import Optional, List, Dict, Any
from datetime import datetime


# Request Models
class ChatRequest(BaseModel):
    message: str = Field(..., min_length=1, max_length=50000, description="Chat message")
    session: str = Field(..., description="Session ID")
    attachments: Optional[List[str]] = Field(default=[], description="Attachment IDs")
    use_web: Optional[bool] = Field(default=False, description="Enable web search")
    use_research: Optional[bool] = Field(default=False, description="Enable deep research")
    time_filter: Optional[str] = Field(default=None, description="Time filter for search")
    preset_id: Optional[str] = Field(default=None, description="Preset identifier")
    
    @field_validator('message')
    @classmethod
    def clean_message(cls, v):
        return v.strip()
    
    @field_validator('time_filter')
    @classmethod
    def validate_time_filter(cls, v):
        if v is not None and v not in ['day', 'week', 'month', 'year']:
            return None  # Just set to None if invalid rather than raising error
        return v


class SessionCreateRequest(BaseModel):
    name: Optional[str] = Field(default="", max_length=200, description="Session name")
    endpoint_url: str = Field(..., description="LLM endpoint URL")
    model: Optional[str] = Field(default="", description="Model ID")
    rag: Optional[bool] = Field(default=False, description="Enable RAG")


class MemoryAddRequest(BaseModel):
    text: str = Field(..., min_length=1, max_length=5000, description="Memory text")
    category: str = Field(default="fact", description="Memory category")
    source: str = Field(default="user", description="Memory source")
    session_id: Optional[str] = Field(default=None, description="Associated session ID")

    @field_validator('category')
    @classmethod
    def validate_category(cls, v):
        if v not in ['fact', 'contact', 'task', 'preference', 'identity', 'project', 'goal']:
            return 'fact'  # Default to 'fact' if invalid
        return v


class MemoryUpdateRequest(BaseModel):
    text: str = Field(..., min_length=1, max_length=5000, description="Updated memory text")
    category: Optional[str] = Field(default=None, pattern="^(fact|contact|task|preference|identity|project|goal)$", description="Memory category")


class PresetUpdateRequest(BaseModel):
    """Request model for updating custom preset configuration."""
    name: str = Field(
        "",
        max_length=50,
        description="Character display name (shown next to model name)"
    )
    enabled: bool = Field(
        True,
        description="Whether this character is active"
    )
    temperature: float = Field(
        1.0,
        ge=0.0,
        le=2.0,
        description="Temperature parameter for text generation (0.0-2.0)"
    )
    max_tokens: int = Field(
        0,
        ge=0,
        le=8192,
        description="Maximum number of tokens to generate (0 = no limit)"
    )
    system_prompt: str = Field(
        "",
        max_length=10000,
        description="System prompt to guide assistant behavior (empty = default)"
    )
    inject_prefix: str = Field(
        "",
        max_length=5000,
        description="Text to prepend to each outgoing user message"
    )
    inject_suffix: str = Field(
        "",
        max_length=5000,
        description="Text to append to each outgoing user message"
    )


class DirectoryRequest(BaseModel):
    """Request model for directory operations."""
    directory: str = Field(
        ...,
        min_length=1,
        max_length=500,
        description="Path to the directory"
    )


# Response Models
class ErrorResponse(BaseModel):
    error: str = Field(..., description="Error code")
    message: str = Field(..., description="Error message")
    details: Optional[Dict[str, Any]] = Field(default=None, description="Additional error details")


class UploadResponse(BaseModel):
    id: str = Field(..., description="File ID")
    name: str = Field(..., description="Sanitized filename")
    mime: str = Field(..., description="MIME type")
    size: int = Field(..., description="File size in bytes")
    hash: str = Field(..., description="SHA-256 hash")
    uploaded_at: datetime = Field(..., description="Upload timestamp")
    is_duplicate: bool = Field(default=False, description="Whether file is a duplicate")


class SessionResponse(BaseModel):
    id: str = Field(..., description="Session ID")
    name: str = Field(..., description="Session name")
    model: str = Field(..., description="Model being used")
    rag: bool = Field(default=False, description="RAG enabled")
    archived: bool = Field(default=False, description="Whether session is archived")


class MemoryResponse(BaseModel):
    id: str = Field(..., description="Memory ID")
    text: str = Field(..., description="Memory text")
    category: str = Field(..., description="Memory category")
    source: str = Field(..., description="Memory source")
    timestamp: int = Field(..., description="Unix timestamp")
    session_id: Optional[str] = Field(default=None, description="Associated session")