Initial commit

backend/models.py

@@ -0,0 +1,47 @@
+from __future__ import annotations
+
+from typing import Optional
+from uuid import UUID
+
+from pydantic import BaseModel, Field
+
+
+class LevelInfo(BaseModel):
+    id: int
+    title: str
+    description: str
+
+
+class QueryRequest(BaseModel):
+    session_id: UUID
+    text: str = Field(min_length=1)
+    hard_mode: bool = False
+
+
+class QueryResponse(BaseModel):
+    session_id: UUID
+    response_text: str
+    success: bool
+    session_rotated: bool
+    level_id: int
+
+
+class FilterDecision(BaseModel):
+    triggered: bool
+    reason: str = ""
+
+
+class SessionState(BaseModel):
+    session_id: UUID
+    password: str
+    request_count: int = 0
+
+
+class AgentResponse(BaseModel):
+    session_id: UUID
+    response_text: str
+    success: bool
+    session_rotated: bool
+    level_id: int
+    filter_request: Optional[FilterDecision] = None
+    filter_response: Optional[FilterDecision] = None