non-building checkpoint 1

backend/app/models/prompt.py

@@ -0,0 +1,88 @@
+"""
+Pydantic models for prompt-related data.
+"""
+
+from typing import List, Optional, Dict, Any
+from pydantic import BaseModel, Field
+
+
+class PromptResponse(BaseModel):
+    """Response model for a single prompt."""
+    key: str = Field(..., description="Prompt key (e.g., 'prompt00')")
+    text: str = Field(..., description="Prompt text content")
+    position: int = Field(..., description="Position in history (0 = most recent)")
+
+    class Config:
+        """Pydantic configuration."""
+        from_attributes = True
+
+
+class PoolStatsResponse(BaseModel):
+    """Response model for pool statistics."""
+    total_prompts: int = Field(..., description="Total prompts in pool")
+    prompts_per_session: int = Field(..., description="Prompts drawn per session")
+    target_pool_size: int = Field(..., description="Target pool volume")
+    available_sessions: int = Field(..., description="Available sessions in pool")
+    needs_refill: bool = Field(..., description="Whether pool needs refilling")
+
+
+class HistoryStatsResponse(BaseModel):
+    """Response model for history statistics."""
+    total_prompts: int = Field(..., description="Total prompts in history")
+    history_capacity: int = Field(..., description="Maximum history capacity")
+    available_slots: int = Field(..., description="Available slots in history")
+    is_full: bool = Field(..., description="Whether history is full")
+
+
+class FeedbackWord(BaseModel):
+    """Model for a feedback word with weight."""
+    key: str = Field(..., description="Feedback key (e.g., 'feedback00')")
+    word: str = Field(..., description="Feedback word")
+    weight: int = Field(..., ge=0, le=6, description="Weight from 0-6")
+
+
+class FeedbackHistoryItem(BaseModel):
+    """Model for a feedback history item (word only, no weight)."""
+    key: str = Field(..., description="Feedback key (e.g., 'feedback00')")
+    word: str = Field(..., description="Feedback word")
+
+
+class GeneratePromptsRequest(BaseModel):
+    """Request model for generating prompts."""
+    count: Optional[int] = Field(
+        None,
+        ge=1,
+        le=20,
+        description="Number of prompts to generate (defaults to settings)"
+    )
+    use_history: bool = Field(
+        True,
+        description="Whether to use historic prompts as context"
+    )
+    use_feedback: bool = Field(
+        True,
+        description="Whether to use feedback words as context"
+    )
+
+
+class GeneratePromptsResponse(BaseModel):
+    """Response model for generated prompts."""
+    prompts: List[str] = Field(..., description="Generated prompts")
+    count: int = Field(..., description="Number of prompts generated")
+    used_history: bool = Field(..., description="Whether history was used")
+    used_feedback: bool = Field(..., description="Whether feedback was used")
+
+
+class RateFeedbackWordsRequest(BaseModel):
+    """Request model for rating feedback words."""
+    ratings: Dict[str, int] = Field(
+        ...,
+        description="Dictionary of word to rating (0-6)"
+    )
+
+
+class RateFeedbackWordsResponse(BaseModel):
+    """Response model for rated feedback words."""
+    feedback_words: List[FeedbackWord] = Field(..., description="Rated feedback words")
+    added_to_history: bool = Field(..., description="Whether added to history")
+