77 lines
1.6 KiB
Python
77 lines
1.6 KiB
Python
from pydantic import BaseModel, Field
|
|
from typing import Literal
|
|
|
|
FormatType = Literal[
|
|
"OnePager",
|
|
"MiniGuide",
|
|
"Guide",
|
|
"FullGuide",
|
|
]
|
|
|
|
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 BausteineCreateRequest(BaseModel):
|
|
topic: str = Field(min_length=1, max_length=100)
|
|
instructions: str = Field(default="", max_length=2000)
|
|
provider: ProviderType = "claude"
|
|
|
|
|
|
class BausteineStatusResponse(BaseModel):
|
|
ready: bool
|
|
generating: bool
|
|
progress: str | None = None
|
|
error: str | None = None
|
|
|
|
|
|
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]
|