first commit

app/schemas/tryout.py

@@ -0,0 +1,97 @@
+"""
+Pydantic schemas for Tryout API endpoints.
+"""
+
+from datetime import datetime
+from typing import List, Literal, Optional
+
+from pydantic import BaseModel, Field
+
+
+class TryoutConfigResponse(BaseModel):
+    """Response schema for tryout configuration."""
+
+    id: int
+    website_id: int
+    tryout_id: str
+    name: str
+    description: Optional[str]
+
+    # Scoring configuration
+    scoring_mode: Literal["ctt", "irt", "hybrid"]
+    selection_mode: Literal["fixed", "adaptive", "hybrid"]
+    normalization_mode: Literal["static", "dynamic", "hybrid"]
+
+    # Normalization settings
+    min_sample_for_dynamic: int
+    static_rataan: float
+    static_sb: float
+
+    # AI generation
+    ai_generation_enabled: bool
+
+    # Hybrid mode settings
+    hybrid_transition_slot: Optional[int]
+
+    # IRT settings
+    min_calibration_sample: int
+    theta_estimation_method: Literal["mle", "map", "eap"]
+    fallback_to_ctt_on_error: bool
+
+    # Current stats
+    current_stats: Optional["TryoutStatsResponse"]
+
+    # Timestamps
+    created_at: datetime
+    updated_at: datetime
+
+    model_config = {"from_attributes": True}
+
+
+class TryoutStatsResponse(BaseModel):
+    """Response schema for tryout statistics."""
+
+    participant_count: int
+    rataan: Optional[float]
+    sb: Optional[float]
+    min_nm: Optional[int]
+    max_nm: Optional[int]
+    last_calculated: Optional[datetime]
+
+    model_config = {"from_attributes": True}
+
+
+class TryoutConfigBrief(BaseModel):
+    """Brief tryout config for list responses."""
+
+    tryout_id: str
+    name: str
+    scoring_mode: str
+    selection_mode: str
+    normalization_mode: str
+    participant_count: Optional[int] = None
+
+    model_config = {"from_attributes": True}
+
+
+class NormalizationUpdateRequest(BaseModel):
+    """Request schema for updating normalization settings."""
+
+    normalization_mode: Optional[Literal["static", "dynamic", "hybrid"]] = None
+    static_rataan: Optional[float] = Field(None, ge=0)
+    static_sb: Optional[float] = Field(None, gt=0)
+
+
+class NormalizationUpdateResponse(BaseModel):
+    """Response schema for normalization update."""
+
+    tryout_id: str
+    normalization_mode: str
+    static_rataan: float
+    static_sb: float
+    will_switch_to_dynamic_at: int
+    current_participant_count: int
+
+
+# Update forward reference
+TryoutConfigResponse.model_rebuild()