d7184b776b
后端： 1.收口阶段 6 agent 结构迁移，将 newAgent 内核与 agentsvc 编排层迁入 services/agent - 切换 Agent 启动装配与 HTTP handler 直连 agent sv，移除旧 service agent bridge - 补齐 Agent 对 memory、task、task-class、schedule 的 RPC 适配与契约字段 - 扩展 schedule、task、task-class RPC/contract 支撑 Agent 查询、写入与 provider 切流 - 更新迁移文档、README 与相关注释，明确 agent 当前切流点和剩余 memory 迁移面
118 lines
3.5 KiB
Go
118 lines
3.5 KiB
Go
package agentshared
|
||
|
||
import (
|
||
"fmt"
|
||
"strings"
|
||
|
||
agentmodel "github.com/LoveLosita/smartflow/backend/services/agent/model"
|
||
"github.com/cloudwego/eino/schema"
|
||
)
|
||
|
||
const (
|
||
correctionHistoryKindKey = "newagent_history_kind"
|
||
correctionHistoryKindCorrectionUser = "llm_correction_prompt"
|
||
)
|
||
|
||
// AppendLLMCorrection 追加 LLM 修正提示到对话历史。
|
||
//
|
||
// 设计目的：
|
||
// 1. 当 LLM 输出不符合预期（如不支持的 action、格式错误等），不应直接报错终止；
|
||
// 2. 应该给 LLM 一个自我修正的机会，把错误反馈写回历史，让它重新生成；
|
||
// 3. 该函数封装了“追加 assistant 消息 + 追加纠正提示”的通用流程。
|
||
//
|
||
// 参数说明：
|
||
// - conversationContext: 对话上下文，用于追加历史消息；
|
||
// - llmOutput: LLM 的原始输出内容，会作为 assistant 消息追加；
|
||
// - validOptionsDesc: 合法选项的描述，用于构造纠正提示。
|
||
func AppendLLMCorrection(
|
||
conversationContext *agentmodel.ConversationContext,
|
||
llmOutput string,
|
||
validOptionsDesc string,
|
||
) {
|
||
if conversationContext == nil {
|
||
return
|
||
}
|
||
|
||
assistantContent := strings.TrimSpace(llmOutput)
|
||
appendCorrectionAssistantIfNeeded(conversationContext, assistantContent)
|
||
|
||
correctionContent := fmt.Sprintf(
|
||
"你的输出不符合预期。%s 请重新分析当前状态，输出正确的内容。",
|
||
validOptionsDesc,
|
||
)
|
||
conversationContext.AppendHistory(&schema.Message{
|
||
Role: schema.User,
|
||
Content: correctionContent,
|
||
Extra: map[string]any{
|
||
correctionHistoryKindKey: correctionHistoryKindCorrectionUser,
|
||
},
|
||
})
|
||
}
|
||
|
||
// AppendLLMCorrectionWithHint 追加 LLM 修正提示（带自定义错误描述）。
|
||
//
|
||
// 相比 AppendLLMCorrection，该函数允许调用方提供更详细的错误描述，
|
||
// 适用于需要明确告知 LLM 具体哪里出错的场景。
|
||
func AppendLLMCorrectionWithHint(
|
||
conversationContext *agentmodel.ConversationContext,
|
||
llmOutput string,
|
||
errorDesc string,
|
||
validOptionsDesc string,
|
||
) {
|
||
if conversationContext == nil {
|
||
return
|
||
}
|
||
|
||
assistantContent := strings.TrimSpace(llmOutput)
|
||
appendCorrectionAssistantIfNeeded(conversationContext, assistantContent)
|
||
|
||
correctionContent := fmt.Sprintf(
|
||
"%s %s 请重新分析当前状态，输出正确的内容。",
|
||
errorDesc,
|
||
validOptionsDesc,
|
||
)
|
||
conversationContext.AppendHistory(&schema.Message{
|
||
Role: schema.User,
|
||
Content: correctionContent,
|
||
Extra: map[string]any{
|
||
correctionHistoryKindKey: correctionHistoryKindCorrectionUser,
|
||
},
|
||
})
|
||
}
|
||
|
||
// appendCorrectionAssistantIfNeeded 在纠错回灌前做最小降噪。
|
||
//
|
||
// 1. 空文本直接跳过，避免写入“占位噪音”；
|
||
// 2. 若与“最近一条 assistant 文本”完全一致则跳过，避免同句反复回灌；
|
||
// 3. 仅负责“是否回灌”判定，不负责生成纠错 user 提示。
|
||
func appendCorrectionAssistantIfNeeded(
|
||
conversationContext *agentmodel.ConversationContext,
|
||
assistantContent string,
|
||
) {
|
||
if conversationContext == nil {
|
||
return
|
||
}
|
||
assistantContent = strings.TrimSpace(assistantContent)
|
||
if assistantContent == "" {
|
||
return
|
||
}
|
||
|
||
history := conversationContext.HistorySnapshot()
|
||
for i := len(history) - 1; i >= 0; i-- {
|
||
msg := history[i]
|
||
if msg == nil || msg.Role != schema.Assistant {
|
||
continue
|
||
}
|
||
if strings.TrimSpace(msg.Content) == assistantContent {
|
||
return
|
||
}
|
||
// 只看最近一条 assistant，避免误去重很久以前的正常重复表达。
|
||
break
|
||
}
|
||
|
||
conversationContext.AppendHistory(&schema.Message{
|
||
Role: schema.Assistant,
|
||
Content: assistantContent,
|
||
})
|
||
}
|