107 lines
2.4 KiB
Python
107 lines
2.4 KiB
Python
"""故事相关 Pydantic 模型。"""
|
|
|
|
from datetime import datetime
|
|
from typing import Literal
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
MAX_DATA_LENGTH = 2000
|
|
MAX_EDU_THEME_LENGTH = 200
|
|
MAX_TTS_LENGTH = 4000
|
|
|
|
|
|
# ==================== 故事模型 ====================
|
|
|
|
|
|
class GenerateRequest(BaseModel):
|
|
"""Story generation request."""
|
|
|
|
type: Literal["keywords", "full_story"]
|
|
data: str = Field(..., min_length=1, max_length=MAX_DATA_LENGTH)
|
|
education_theme: str | None = Field(default=None, max_length=MAX_EDU_THEME_LENGTH)
|
|
child_profile_id: str | None = None
|
|
universe_id: str | None = None
|
|
|
|
|
|
class StoryResponse(BaseModel):
|
|
"""Story response."""
|
|
|
|
id: int
|
|
title: str
|
|
story_text: str
|
|
cover_prompt: str | None
|
|
image_url: str | None
|
|
mode: str
|
|
child_profile_id: str | None = None
|
|
universe_id: str | None = None
|
|
|
|
|
|
class StoryListItem(BaseModel):
|
|
"""Story list item."""
|
|
|
|
id: int
|
|
title: str
|
|
image_url: str | None
|
|
created_at: datetime
|
|
mode: str
|
|
|
|
|
|
class FullStoryResponse(BaseModel):
|
|
"""完整故事响应（含图片和音频状态）。"""
|
|
|
|
id: int
|
|
title: str
|
|
story_text: str
|
|
cover_prompt: str | None
|
|
image_url: str | None
|
|
audio_ready: bool
|
|
mode: str
|
|
errors: dict[str, str | None] = Field(default_factory=dict)
|
|
child_profile_id: str | None = None
|
|
universe_id: str | None = None
|
|
|
|
|
|
# ==================== 绘本模型 ====================
|
|
|
|
|
|
class StorybookRequest(BaseModel):
|
|
"""Storybook 生成请求。"""
|
|
|
|
keywords: str = Field(..., min_length=1, max_length=200)
|
|
page_count: int = Field(default=6, ge=4, le=12)
|
|
education_theme: str | None = Field(default=None, max_length=MAX_EDU_THEME_LENGTH)
|
|
generate_images: bool = Field(default=False, description="是否同时生成插图")
|
|
child_profile_id: str | None = None
|
|
universe_id: str | None = None
|
|
|
|
|
|
class StorybookPageResponse(BaseModel):
|
|
"""故事书单页响应。"""
|
|
|
|
page_number: int
|
|
text: str
|
|
image_prompt: str
|
|
image_url: str | None = None
|
|
|
|
|
|
class StorybookResponse(BaseModel):
|
|
"""故事书响应。"""
|
|
|
|
id: int | None = None
|
|
title: str
|
|
main_character: str
|
|
art_style: str
|
|
pages: list[StorybookPageResponse]
|
|
cover_prompt: str
|
|
cover_url: str | None = None
|
|
|
|
|
|
# ==================== 成就模型 ====================
|
|
|
|
|
|
class AchievementItem(BaseModel):
|
|
type: str
|
|
description: str
|
|
obtained_at: str | None = None
|