update:1.新增章节内容批量生成功能
This commit is contained in:
@@ -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
|
||||
Reference in New Issue
Block a user