76 lines
2.2 KiB
Go
76 lines
2.2 KiB
Go
package app
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"infinite-canvas/backend/internal/model"
|
|
)
|
|
|
|
// cloudAgentContinuationReply 生成「上一轮接着聊」用的两段内容：
|
|
//
|
|
// reply —— 上一轮 assistant 实际说过的话（作为 assistant 的历史消息）
|
|
// context —— 上一轮收束摘要（独立上下文，不能并进 reply）
|
|
//
|
|
// 只保留状态、失败原因和已提交生成任务，不把上一轮工具流水当成本轮目标。
|
|
func cloudAgentContinuationReply(task *model.Task, run *CloudAgentRun) (string, string, error) {
|
|
text := ""
|
|
if task.Status == model.TaskStatusSucceeded {
|
|
text = taskResultText(task.ResultJSON)
|
|
}
|
|
submitted := make([]string, 0)
|
|
seen := map[string]bool{}
|
|
addSubmitted := func(id string) {
|
|
id = strings.TrimSpace(id)
|
|
if id == "" || seen[id] {
|
|
return
|
|
}
|
|
seen[id] = true
|
|
submitted = append(submitted, id)
|
|
}
|
|
for _, event := range run.Events {
|
|
if event.Type == "assistant_message" {
|
|
text = stringValue(event.Payload["text"])
|
|
}
|
|
if event.Type == "generation_task_created" {
|
|
addSubmitted(stringValue(event.Payload["taskId"]))
|
|
}
|
|
if result, ok := event.Payload["result"].(map[string]any); ok {
|
|
if submittedFlag, _ := result["taskSubmitted"].(bool); submittedFlag {
|
|
addSubmitted(stringValue(result["taskId"]))
|
|
}
|
|
}
|
|
}
|
|
context := cloudAgentContinuationContext(run, submitted)
|
|
return text, context, nil
|
|
}
|
|
|
|
func cloudAgentContinuationContext(run *CloudAgentRun, submitted []string) string {
|
|
if run == nil {
|
|
return ""
|
|
}
|
|
failed := run.Status == "failed" || strings.TrimSpace(run.FailureMessage) != ""
|
|
if run.Status == "completed" && !failed && len(submitted) == 0 {
|
|
return ""
|
|
}
|
|
var b strings.Builder
|
|
b.WriteString("上一轮已结束（")
|
|
b.WriteString(firstNonEmpty(run.Status, "unknown"))
|
|
b.WriteString("）。")
|
|
if failed {
|
|
reason := strings.TrimSpace(run.FailureMessage)
|
|
if reason == "" {
|
|
reason = run.Status
|
|
}
|
|
b.WriteString(" 上一轮失败：")
|
|
b.WriteString(truncateRunes(reason, 240))
|
|
}
|
|
if len(submitted) > 0 {
|
|
if len(submitted) > 8 {
|
|
submitted = submitted[:8]
|
|
}
|
|
b.WriteString(" 已提交生成任务：")
|
|
b.WriteString(strings.Join(submitted, ", "))
|
|
}
|
|
return b.String()
|
|
}
|