feat(config): 新增反截断功能防止模型输出截断

在任务配置中增加 anti_truncation 开关,支持在生成大段文本时防止被意外截断。启用后会在 prompt 中添加指示模型输出完成信令的指令,并在检测到未正常结束时触发自动重试机制。(我相信这次的一定不会炸飞你们了)
This commit is contained in:
minecraft1024a
2025-08-21 20:29:37 +08:00
committed by Windpicker-owo
parent fcb7a85e69
commit be769e29c3
3 changed files with 38 additions and 6 deletions

View File

@@ -83,6 +83,7 @@ class TaskConfig(ValidatedConfigBase):
max_tokens: int = Field(default=800, description="任务最大输出token数")
temperature: float = Field(default=0.7, description="模型温度")
concurrency_count: int = Field(default=1, description="并发请求数量")
anti_truncation: bool = Field(default=False, description="是否启用反截断功能,防止模型输出被截断")
@field_validator('model_list')
@classmethod