38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
"""等待用户补充的问题"""
|
||
|
||
from typing import Self
|
||
|
||
from pydantic import Field, model_validator
|
||
|
||
from app.schemas.base import ExtensibleModel, NonBlankString
|
||
|
||
|
||
class ClarificationOption(ExtensibleModel):
|
||
id: NonBlankString = Field(
|
||
description="本题内唯一的选项 ID；选项对应某个组件时用该组件的 ID"
|
||
)
|
||
label: NonBlankString = Field(description="简短选项名称")
|
||
description: NonBlankString = Field(description="一句话说明选择的影响或推荐依据")
|
||
recommended: bool = False
|
||
|
||
|
||
class Clarification(ExtensibleModel):
|
||
question: NonBlankString = Field(
|
||
description="一个简短、具体、便于用户回答的澄清问题，一次只问一件事"
|
||
)
|
||
options: list[ClarificationOption] = Field(
|
||
default_factory=list,
|
||
description="适合选择时提供 2–3 个互斥选项，否则留空，自由输入由客户端提供",
|
||
)
|
||
|
||
@model_validator(mode="after")
|
||
def validate_options(self) -> Self:
|
||
if len(self.options) not in (0, 2, 3):
|
||
raise ValueError("选项应为空或包含 2–3 项")
|
||
if len({option.id for option in self.options}) != len(self.options):
|
||
raise ValueError("选项 ID 不能重复")
|
||
if sum(option.recommended for option in self.options) > 1:
|
||
raise ValueError("最多一个推荐选项")
|
||
self.options.sort(key=lambda option: not option.recommended)
|
||
return self
|