update

backend/models.py

@@ -0,0 +1,62 @@
+from pydantic import BaseModel, Field
+from typing import Literal
+
+FormatType = Literal[
+    "OnePager",
+    "MiniGuide",
+    "Guide",
+]
+
+ProviderType = Literal["claude", "minimax"]
+
+
+class GuideCreateRequest(BaseModel):
+    topic: str = Field(min_length=1, max_length=100)
+    format: FormatType
+    instructions: str = Field(default="", max_length=2000)
+    provider: ProviderType = "claude"
+
+
+class ProjectResponse(BaseModel):
+    name: str
+
+
+class ProviderInfo(BaseModel):
+    id: str
+    available: bool
+
+
+class GuideResponse(BaseModel):
+    id: str
+    topic: str
+    format: str
+    status: str
+    progress: str | None = None
+    error_msg: str | None = None
+    created_at: str
+    updated_at: str
+
+
+class ChatMessage(BaseModel):
+    role: Literal["user", "assistant"]
+    content: str = Field(min_length=1, max_length=8000)
+
+
+class GuideChatRequest(BaseModel):
+    section: str = Field(default="", max_length=20000)
+    outline: str = Field(default="", max_length=8000)
+    messages: list[ChatMessage] = Field(min_length=1)
+    provider: ProviderType = "claude"
+
+
+class GuideChatResponse(BaseModel):
+    reply: str
+
+
+class ProgressUpdate(BaseModel):
+    chapter: str = Field(min_length=1, max_length=100)
+    done: bool
+
+
+class ProgressResponse(BaseModel):
+    chapters: list[str]