92 lines
3.6 KiB
Python
92 lines
3.6 KiB
Python
"""需求理解提示词，字段细节由 Requirement 模型描述"""
|
||
|
||
import json
|
||
|
||
from app.prompt.domain import EXISTING_TARGET_RULE, INPUT_AS_DATA
|
||
from app.schemas.capabilities import DATA_BINDING_GUIDE
|
||
from app.schemas.input_context import InputContext
|
||
from app.schemas.operations import OPERATION_GUIDE
|
||
from app.schemas.requirement import Intent
|
||
|
||
REQUIREMENT_SYSTEM_PROMPT = f"""你负责结合上下文，将用户本次的大屏需求整理为 Requirement
|
||
|
||
## 需求理解
|
||
1. 沿用已确认的 intent，不重新分类
|
||
2. 结合历史会话和本次澄清记录，理解“这个”“它”“刚才那个”等指代
|
||
3. 将需求改写为独立完整的 goal，保留目标、操作和约束，不扩大修改范围
|
||
4. 用户后续明确补充或纠正的内容优先，保留仍适用的约束，不重复执行历史中已经完成的操作
|
||
5. 提供“用户已同意的提议”时，用户本次的回答就是同意执行这条提议，以它作为本次需求
|
||
|
||
## 变更提取
|
||
1. 将本次变更写入 changes，保留相对变化与单位
|
||
operation 取值：{OPERATION_GUIDE}
|
||
2. {EXISTING_TARGET_RULE}；无法唯一确定目标时标记 unresolved，
|
||
并在 candidate_ids 中列出当前大屏里所有可能是该目标的组件 ID，没有可能的组件时留空
|
||
按用户的原话理解操作，不猜测目标，也不因目标不存在而改写成新增等其他操作
|
||
3. 默认先执行：能用合理默认值补齐的信息，写入 assumptions 后继续，不要求澄清
|
||
4. 新增组件时，位置、尺寸、标题、样式、数据来源均不阻塞，由后续规划采用默认值，用户之后可再次提出调整
|
||
{DATA_BINDING_GUIDE}
|
||
5. 存在 unresolved 目标，或无法提取任何变更时，needs_clarification 为 true 并给出问题；
|
||
最终执行、追问还是直接告知用户，由系统根据 changes 和 candidate_ids 决定
|
||
|
||
## 输出边界
|
||
按照 Requirement 结构输出，不生成执行动作
|
||
修改范围和是否需要规划由系统根据 changes 推导，target_scope 和 operation 要准确
|
||
{INPUT_AS_DATA}
|
||
不声称读取了未解析的文件
|
||
"""
|
||
|
||
REQUIREMENT_INPUT_TEMPLATE = """历史会话：
|
||
{history_text}
|
||
|
||
本次需求的澄清记录：
|
||
{clarification_text}
|
||
|
||
当前用户问题：
|
||
{query}
|
||
|
||
已确认的意图：
|
||
{intent}
|
||
|
||
当前大屏：
|
||
{screen_json}
|
||
|
||
编辑上下文：
|
||
{context_json}
|
||
|
||
附件信息：
|
||
{attachments_json}
|
||
"""
|
||
|
||
|
||
def build_requirement_input(
|
||
context: InputContext,
|
||
intent: Intent,
|
||
*,
|
||
history_text: str | None = None,
|
||
accepted_offer: str | None = None,
|
||
) -> str:
|
||
"""整理本次模型输入，历史会话默认取输入上下文中由服务端写入的记录"""
|
||
data = context.model_dump(mode="json", exclude_unset=True)
|
||
# 会话历史
|
||
history = data.pop("history", [])
|
||
if history_text is None:
|
||
history_text = json.dumps(history, ensure_ascii=False) if history else "未提供"
|
||
content = REQUIREMENT_INPUT_TEMPLATE.format(
|
||
history_text=history_text,
|
||
clarification_text=json.dumps(
|
||
data.pop("clarification_history", []), ensure_ascii=False
|
||
),
|
||
query=data.pop("prompt"),
|
||
intent=intent,
|
||
screen_json=json.dumps(data.pop("screen"), ensure_ascii=False),
|
||
context_json=json.dumps(data.pop("context", {}), ensure_ascii=False),
|
||
attachments_json=json.dumps(data.pop("attachments", []), ensure_ascii=False),
|
||
)
|
||
if accepted_offer:
|
||
content += f"\n用户已同意的提议：\n{accepted_offer}\n"
|
||
# 保留扩展字段，避免拆分模板时丢失额外输入
|
||
if data:
|
||
content += "\n补充上下文：\n" + json.dumps(data, ensure_ascii=False)
|
||
return content
|