update:1.新增章节内容批量生成功能

This commit is contained in:
xiamuceer
2025-11-06 12:36:32 +08:00
parent d76c79e733
commit e8396f0cbd
11 changed files with 1381 additions and 28 deletions
+41 -1
View File
@@ -65,4 +65,44 @@ class ChapterGenerateRequest(BaseModel):
description="目标字数,默认3000字",
ge=500, # 最小500字
le=10000 # 最大10000字
)
)
class BatchGenerateRequest(BaseModel):
"""批量生成章节的请求模型"""
start_chapter_number: int = Field(..., description="起始章节序号")
count: int = Field(..., description="生成章节数量", ge=1, le=20)
style_id: Optional[int] = Field(None, description="写作风格ID")
target_word_count: Optional[int] = Field(
3000,
description="目标字数,默认3000字",
ge=500,
le=10000
)
enable_analysis: bool = Field(False, description="是否启用同步分析")
max_retries: int = Field(3, description="每个章节的最大重试次数", ge=0, le=5)
class BatchGenerateResponse(BaseModel):
"""批量生成响应模型"""
batch_id: str = Field(..., description="批次ID")
message: str = Field(..., description="响应消息")
chapters_to_generate: list[dict] = Field(..., description="待生成章节列表")
estimated_time_minutes: int = Field(..., description="预估耗时(分钟)")
class BatchGenerateStatusResponse(BaseModel):
"""批量生成状态响应模型"""
batch_id: str
status: str
total: int
completed: int
current_chapter_id: Optional[str] = None
current_chapter_number: Optional[int] = None
current_retry_count: Optional[int] = None
max_retries: Optional[int] = None
failed_chapters: list[dict] = []
created_at: Optional[str] = None
started_at: Optional[str] = None
completed_at: Optional[str] = None
error_message: Optional[str] = None