Version: 0.9.47.dev.260427
后端： 1. execute 节点继续拆职责——超大 execute.go 下沉为 node/execute 子包，按决策流、动作路由、上下文锚点、工具执行、状态快照、工具展示与参数解析拆分；顶层 execute.go 收敛为桥接导出，降低单文件编排/业务/模型/工具逻辑混写 2. 节点公共能力继续沉到 shared——抽出 LLM 纠错回灌、完整上下文调试日志、thinking 开关、统一上下文压缩、可见 assistant 文本持久化等 node_* 公共件，减少 execute 独占实现并为其他节点复用铺路 3. speak 文本整理能力独立收口——新增 speak_text 辅助文件，补齐正文归一化的独立承载，继续收缩 execute 主文件体积 前端： 4. NewAgent 时间线接入 business_card 业务卡片协议——schedule_agent.ts 新增 task_query / task_record 卡片载荷类型与 business_card kind；AssistantPanel 增加业务卡片事件存储、时间线恢复、块渲染分支与 BusinessCardRenderer 接入，同时保留 interrupt / status / tool / reasoning 多块并存 5. 新增任务查询卡片与任务记录卡片组件，并补充 DesignDemo 设计预览页与路由，前端可先行验证 business_card 的视觉与交互落点 文档： 6. 新增 newagent business card 前后端对接说明，明确 timeline kind、payload 结构、卡片分类、前后端发射/渲染约束
This commit is contained in:
LoveLosita
File diff suppressed because it is too large
Load Diff
511
backend/newAgent/node/execute/action_router.go
Normal file
511
backend/newAgent/node/execute/action_router.go
Normal file
@@ -0,0 +1,511 @@
|
||||
package newagentexecute
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
newagentshared "github.com/LoveLosita/smartflow/backend/newAgent/shared"
|
||||
"io"
|
||||
"log"
|
||||
"strings"
|
||||
|
||||
infrallm "github.com/LoveLosita/smartflow/backend/infra/llm"
|
||||
newagentmodel "github.com/LoveLosita/smartflow/backend/newAgent/model"
|
||||
newagentrouter "github.com/LoveLosita/smartflow/backend/newAgent/router"
|
||||
newagentstream "github.com/LoveLosita/smartflow/backend/newAgent/stream"
|
||||
newagenttools "github.com/LoveLosita/smartflow/backend/newAgent/tools"
|
||||
"github.com/cloudwego/eino/schema"
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
type executeDecisionStreamOutput struct {
|
||||
decision *newagentmodel.ExecuteDecision
|
||||
rawText string
|
||||
parsedBeforeText string
|
||||
parsedAfterText string
|
||||
streamedSpeak string
|
||||
speakStreamed bool
|
||||
firstChunk bool
|
||||
}
|
||||
|
||||
func collectExecuteDecisionFromLLM(
|
||||
ctx context.Context,
|
||||
input ExecuteNodeInput,
|
||||
flowState *newagentmodel.CommonState,
|
||||
conversationContext *newagentmodel.ConversationContext,
|
||||
emitter *newagentstream.ChunkEmitter,
|
||||
messages []*schema.Message,
|
||||
) (*executeDecisionStreamOutput, error) {
|
||||
reader, err := input.Client.Stream(
|
||||
ctx,
|
||||
messages,
|
||||
infrallm.GenerateOptions{
|
||||
Temperature: 1.0,
|
||||
MaxTokens: 131072,
|
||||
Thinking: newagentshared.ResolveThinkingMode(input.ThinkingEnabled),
|
||||
Metadata: map[string]any{
|
||||
"stage": executeStageName,
|
||||
"step_index": flowState.CurrentStep,
|
||||
"round_used": flowState.RoundUsed,
|
||||
},
|
||||
},
|
||||
)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("执行阶段 Stream 请求失败: %w", err)
|
||||
}
|
||||
|
||||
parser := newagentrouter.NewStreamDecisionParser()
|
||||
output := &executeDecisionStreamOutput{firstChunk: true}
|
||||
var fullText strings.Builder
|
||||
|
||||
for {
|
||||
chunk, recvErr := reader.Recv()
|
||||
if recvErr == io.EOF {
|
||||
break
|
||||
}
|
||||
if recvErr != nil {
|
||||
log.Printf("[WARN] execute stream recv error chat=%s err=%v", flowState.ConversationID, recvErr)
|
||||
break
|
||||
}
|
||||
|
||||
if chunk != nil && strings.TrimSpace(chunk.ReasoningContent) != "" {
|
||||
if emitErr := emitter.EmitReasoningText(
|
||||
executeSpeakBlockID,
|
||||
executeStageName,
|
||||
chunk.ReasoningContent,
|
||||
output.firstChunk,
|
||||
); emitErr != nil {
|
||||
return nil, fmt.Errorf("执行 thinking 推送失败: %w", emitErr)
|
||||
}
|
||||
output.firstChunk = false
|
||||
}
|
||||
|
||||
content := ""
|
||||
if chunk != nil {
|
||||
content = chunk.Content
|
||||
}
|
||||
|
||||
visible, ready, _ := parser.Feed(content)
|
||||
if !ready {
|
||||
continue
|
||||
}
|
||||
|
||||
result := parser.Result()
|
||||
output.rawText = result.RawBuffer
|
||||
output.parsedBeforeText = result.BeforeText
|
||||
output.parsedAfterText = result.AfterText
|
||||
|
||||
if result.Fallback || result.ParseFailed {
|
||||
log.Printf(
|
||||
"[DEBUG] execute LLM 决策解析失败 chat=%s round=%d raw=%s",
|
||||
flowState.ConversationID,
|
||||
flowState.RoundUsed,
|
||||
output.rawText,
|
||||
)
|
||||
flowState.ConsecutiveCorrections++
|
||||
if flowState.ConsecutiveCorrections >= maxConsecutiveCorrections {
|
||||
return nil, fmt.Errorf(
|
||||
"连续 %d 次解析决策 JSON 失败，终止执行。原始输出=%s",
|
||||
flowState.ConsecutiveCorrections,
|
||||
output.rawText,
|
||||
)
|
||||
}
|
||||
|
||||
errorDesc := "未识别到合法的 SMARTFLOW_DECISION 标签，无法继续解析。"
|
||||
optionHint := "请输出一个 <SMARTFLOW_DECISION>{JSON}</SMARTFLOW_DECISION>，然后再在标签外补充可见文本。"
|
||||
if strings.Contains(output.rawText, `"tool_call": [`) || strings.Contains(output.rawText, `"tool_call":[`) {
|
||||
errorDesc = "检测到 tool_call 字段被错误写成数组；每次只允许调用一个工具，不支持数组形式。"
|
||||
optionHint = "请把多次工具调用拆开，每次只保留一个 tool_call，然后再继续下一轮。"
|
||||
}
|
||||
newagentshared.AppendLLMCorrectionWithHint(conversationContext, output.rawText, errorDesc, optionHint)
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
decision, parseErr := infrallm.ParseJSONObject[newagentmodel.ExecuteDecision](result.DecisionJSON)
|
||||
if parseErr != nil {
|
||||
log.Printf(
|
||||
"[DEBUG] execute LLM JSON 解析失败 chat=%s round=%d json=%s raw=%s",
|
||||
flowState.ConversationID,
|
||||
flowState.RoundUsed,
|
||||
result.DecisionJSON,
|
||||
output.rawText,
|
||||
)
|
||||
flowState.ConsecutiveCorrections++
|
||||
if flowState.ConsecutiveCorrections >= maxConsecutiveCorrections {
|
||||
return nil, fmt.Errorf(
|
||||
"连续 %d 次解析决策 JSON 失败，终止执行。原始输出=%s",
|
||||
flowState.ConsecutiveCorrections,
|
||||
output.rawText,
|
||||
)
|
||||
}
|
||||
newagentshared.AppendLLMCorrectionWithHint(
|
||||
conversationContext,
|
||||
"",
|
||||
"决策标签内的 JSON 格式不合法。",
|
||||
"请确保 <SMARTFLOW_DECISION> 标签内是合法 JSON；当 action=next_plan/done 时，goal_check 必须是字符串（不要输出对象）。",
|
||||
)
|
||||
return nil, nil
|
||||
}
|
||||
output.decision = decision
|
||||
|
||||
if visible != "" {
|
||||
if emitErr := emitter.EmitAssistantText(
|
||||
executeSpeakBlockID,
|
||||
executeStageName,
|
||||
visible,
|
||||
output.firstChunk,
|
||||
); emitErr != nil {
|
||||
return nil, fmt.Errorf("执行回答推送失败: %w", emitErr)
|
||||
}
|
||||
output.speakStreamed = true
|
||||
fullText.WriteString(visible)
|
||||
output.firstChunk = false
|
||||
}
|
||||
|
||||
for {
|
||||
chunk2, recvErr2 := reader.Recv()
|
||||
if recvErr2 == io.EOF {
|
||||
break
|
||||
}
|
||||
if recvErr2 != nil {
|
||||
log.Printf("[WARN] execute speak stream error chat=%s err=%v", flowState.ConversationID, recvErr2)
|
||||
break
|
||||
}
|
||||
if chunk2 == nil {
|
||||
continue
|
||||
}
|
||||
if strings.TrimSpace(chunk2.ReasoningContent) != "" {
|
||||
_ = emitter.EmitReasoningText(executeSpeakBlockID, executeStageName, chunk2.ReasoningContent, false)
|
||||
}
|
||||
if chunk2.Content != "" {
|
||||
if emitErr := emitter.EmitAssistantText(
|
||||
executeSpeakBlockID,
|
||||
executeStageName,
|
||||
chunk2.Content,
|
||||
output.firstChunk,
|
||||
); emitErr != nil {
|
||||
return nil, fmt.Errorf("执行回答推送失败: %w", emitErr)
|
||||
}
|
||||
output.speakStreamed = true
|
||||
fullText.WriteString(chunk2.Content)
|
||||
output.firstChunk = false
|
||||
}
|
||||
}
|
||||
break
|
||||
}
|
||||
|
||||
if output.decision == nil {
|
||||
if strings.TrimSpace(output.rawText) == "" {
|
||||
log.Printf(
|
||||
"[WARN] execute LLM 返回空文本 chat=%s round=%d consecutive=%d/%d",
|
||||
flowState.ConversationID,
|
||||
flowState.RoundUsed,
|
||||
flowState.ConsecutiveCorrections+1,
|
||||
maxConsecutiveCorrections,
|
||||
)
|
||||
flowState.ConsecutiveCorrections++
|
||||
if flowState.ConsecutiveCorrections >= maxConsecutiveCorrections {
|
||||
return nil, fmt.Errorf("连续 %d 次模型返回空文本，终止执行", flowState.ConsecutiveCorrections)
|
||||
}
|
||||
newagentshared.AppendLLMCorrectionWithHint(
|
||||
conversationContext,
|
||||
"",
|
||||
"模型没有返回任何内容。",
|
||||
"请至少返回一个 <SMARTFLOW_DECISION>{JSON}</SMARTFLOW_DECISION> 形式的执行决策。",
|
||||
)
|
||||
return nil, nil
|
||||
}
|
||||
return nil, fmt.Errorf("执行阶段模型输出中未提取到决策标签")
|
||||
}
|
||||
|
||||
output.streamedSpeak = fullText.String()
|
||||
output.decision.Speak = pickExecuteVisibleSpeak(
|
||||
output.streamedSpeak,
|
||||
output.parsedAfterText,
|
||||
output.parsedBeforeText,
|
||||
output.decision,
|
||||
)
|
||||
log.Printf(
|
||||
"[DEBUG] execute LLM 响应 chat=%s round=%d action=%s speak_len=%d raw_len=%d raw_preview=%.200s",
|
||||
flowState.ConversationID,
|
||||
flowState.RoundUsed,
|
||||
output.decision.Action,
|
||||
len(output.decision.Speak),
|
||||
len(output.rawText),
|
||||
output.rawText,
|
||||
)
|
||||
return output, nil
|
||||
}
|
||||
|
||||
func handleExecuteDecision(
|
||||
ctx context.Context,
|
||||
input ExecuteNodeInput,
|
||||
runtimeState *newagentmodel.AgentRuntimeState,
|
||||
flowState *newagentmodel.CommonState,
|
||||
conversationContext *newagentmodel.ConversationContext,
|
||||
emitter *newagentstream.ChunkEmitter,
|
||||
output *executeDecisionStreamOutput,
|
||||
) error {
|
||||
if output == nil || output.decision == nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
decision := output.decision
|
||||
if decision.Action == newagentmodel.ExecuteActionDone &&
|
||||
decision.ToolCall != nil &&
|
||||
strings.EqualFold(strings.TrimSpace(decision.ToolCall.Name), newagenttools.ToolNameContextToolsRemove) {
|
||||
decision.ToolCall = nil
|
||||
}
|
||||
|
||||
if err := decision.Validate(); err != nil {
|
||||
flowState.ConsecutiveCorrections++
|
||||
log.Printf(
|
||||
"[WARN] execute 决策不合法 chat=%s round=%d consecutive=%d/%d err=%s",
|
||||
flowState.ConversationID,
|
||||
flowState.RoundUsed,
|
||||
flowState.ConsecutiveCorrections,
|
||||
maxConsecutiveCorrections,
|
||||
err.Error(),
|
||||
)
|
||||
if flowState.ConsecutiveCorrections >= maxConsecutiveCorrections {
|
||||
return fmt.Errorf(
|
||||
"连续 %d 次决策不合法，终止执行。%s (原始输出: %s)",
|
||||
flowState.ConsecutiveCorrections,
|
||||
err.Error(),
|
||||
output.rawText,
|
||||
)
|
||||
}
|
||||
_ = emitter.EmitStatus(
|
||||
executeStatusBlockID,
|
||||
executeStageName,
|
||||
"executing",
|
||||
fmt.Sprintf("执行校验：决策不合法：%s，已请求模型重试。", err.Error()),
|
||||
false,
|
||||
)
|
||||
newagentshared.AppendLLMCorrectionWithHint(
|
||||
conversationContext,
|
||||
"",
|
||||
fmt.Sprintf("本次执行决策不合法：%s", err.Error()),
|
||||
"合法的 action 包括：continue（继续当前步骤）、ask_user（追问用户）、confirm（写操作确认）、next_plan（推进到下一步）、done（任务完成）、abort（正式终止本轮流程）。",
|
||||
)
|
||||
return nil
|
||||
}
|
||||
|
||||
flowState.ConsecutiveCorrections = 0
|
||||
decision.Speak = pickExecuteVisibleSpeak(
|
||||
decision.Speak,
|
||||
output.parsedAfterText,
|
||||
output.parsedBeforeText,
|
||||
decision,
|
||||
)
|
||||
decision.Speak = normalizeSpeak(decision.Speak)
|
||||
|
||||
if decision.Action == newagentmodel.ExecuteActionConfirm &&
|
||||
decision.ToolCall != nil &&
|
||||
input.ToolRegistry != nil &&
|
||||
!input.ToolRegistry.IsWriteTool(decision.ToolCall.Name) {
|
||||
decision.Action = newagentmodel.ExecuteActionContinue
|
||||
}
|
||||
|
||||
if decision.Action == newagentmodel.ExecuteActionContinue &&
|
||||
decision.ToolCall != nil &&
|
||||
newagenttools.IsContextManagementTool(decision.ToolCall.Name) {
|
||||
decision.Speak = ""
|
||||
}
|
||||
|
||||
if !output.speakStreamed && strings.TrimSpace(decision.Speak) != "" {
|
||||
if emitErr := emitter.EmitAssistantText(
|
||||
executeSpeakBlockID,
|
||||
executeStageName,
|
||||
decision.Speak,
|
||||
output.firstChunk,
|
||||
); emitErr != nil {
|
||||
return fmt.Errorf("执行回答补发失败: %w", emitErr)
|
||||
}
|
||||
output.speakStreamed = true
|
||||
output.firstChunk = false
|
||||
}
|
||||
|
||||
if output.speakStreamed {
|
||||
if tail := buildExecuteNormalizedSpeakTail(output.streamedSpeak, decision.Speak); tail != "" {
|
||||
if emitErr := emitter.EmitAssistantText(
|
||||
executeSpeakBlockID,
|
||||
executeStageName,
|
||||
tail,
|
||||
output.firstChunk,
|
||||
); emitErr != nil {
|
||||
return fmt.Errorf("执行回答尾段补发失败: %w", emitErr)
|
||||
}
|
||||
output.firstChunk = false
|
||||
}
|
||||
}
|
||||
|
||||
if flowState.HasPlan() &&
|
||||
(decision.Action == newagentmodel.ExecuteActionNextPlan ||
|
||||
decision.Action == newagentmodel.ExecuteActionDone) {
|
||||
if strings.TrimSpace(decision.GoalCheck) == "" {
|
||||
flowState.ConsecutiveCorrections++
|
||||
if flowState.ConsecutiveCorrections >= maxConsecutiveCorrections {
|
||||
return fmt.Errorf("连续 %d 次 goal_check 为空，终止执行", flowState.ConsecutiveCorrections)
|
||||
}
|
||||
_ = emitter.EmitStatus(
|
||||
executeStatusBlockID,
|
||||
executeStageName,
|
||||
"executing",
|
||||
fmt.Sprintf("执行校验：action=%s 缺少 goal_check，已请求模型重试。", decision.Action),
|
||||
false,
|
||||
)
|
||||
newagentshared.AppendLLMCorrectionWithHint(
|
||||
conversationContext,
|
||||
"",
|
||||
fmt.Sprintf("你输出了 action=%s，但 goal_check 为空。", decision.Action),
|
||||
fmt.Sprintf("输出 %s 时，必须在 goal_check 中对照 done_when 逐条说明完成依据。", decision.Action),
|
||||
)
|
||||
return nil
|
||||
}
|
||||
}
|
||||
|
||||
askUserHistoryAppended := false
|
||||
if strings.TrimSpace(decision.Speak) != "" {
|
||||
isConfirmWithCard := decision.Action == newagentmodel.ExecuteActionConfirm && !input.AlwaysExecute
|
||||
isAskUser := decision.Action == newagentmodel.ExecuteActionAskUser
|
||||
isAbort := decision.Action == newagentmodel.ExecuteActionAbort
|
||||
|
||||
if !isConfirmWithCard && !isAskUser && !isAbort {
|
||||
msg := schema.AssistantMessage(decision.Speak, nil)
|
||||
newagentshared.PersistVisibleAssistantMessage(ctx, input.PersistVisibleMessage, flowState, msg)
|
||||
}
|
||||
if !isAbort {
|
||||
conversationContext.AppendHistory(&schema.Message{
|
||||
Role: schema.Assistant,
|
||||
Content: decision.Speak,
|
||||
})
|
||||
if isAskUser {
|
||||
askUserHistoryAppended = true
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
switch decision.Action {
|
||||
case newagentmodel.ExecuteActionContinue:
|
||||
if decision.ToolCall != nil {
|
||||
if input.ToolRegistry != nil && input.ToolRegistry.IsWriteTool(decision.ToolCall.Name) {
|
||||
flowState.ConsecutiveCorrections++
|
||||
log.Printf(
|
||||
"[WARN] execute 决策协议违背 chat=%s round=%d action=continue tool=%s consecutive=%d/%d",
|
||||
flowState.ConversationID,
|
||||
flowState.RoundUsed,
|
||||
strings.TrimSpace(decision.ToolCall.Name),
|
||||
flowState.ConsecutiveCorrections,
|
||||
maxConsecutiveCorrections,
|
||||
)
|
||||
if flowState.ConsecutiveCorrections >= maxConsecutiveCorrections {
|
||||
return fmt.Errorf("连续 %d 次输出 continue+写工具，终止执行", flowState.ConsecutiveCorrections)
|
||||
}
|
||||
_ = emitter.EmitStatus(
|
||||
executeStatusBlockID,
|
||||
executeStageName,
|
||||
"executing",
|
||||
fmt.Sprintf(
|
||||
"执行校验：写工具 %q 未执行。原因：模型输出了 action=continue；所有写工具都必须使用 action=confirm。",
|
||||
strings.TrimSpace(decision.ToolCall.Name),
|
||||
),
|
||||
false,
|
||||
)
|
||||
llmOutput := decision.Speak
|
||||
if strings.TrimSpace(llmOutput) == "" {
|
||||
llmOutput = decision.Reason
|
||||
}
|
||||
newagentshared.AppendLLMCorrectionWithHint(
|
||||
conversationContext,
|
||||
llmOutput,
|
||||
fmt.Sprintf("你输出了 action=continue，但同时提供了 %q 这个写工具。", decision.ToolCall.Name),
|
||||
"所有写工具都必须使用 action=confirm，并放在同一个 tool_call 中；continue 仅用于读工具。如果写操作尚未执行，请直接回发 confirm。",
|
||||
)
|
||||
return nil
|
||||
}
|
||||
if shouldForceFeasibilityNegotiation(flowState, input.ToolRegistry, decision.ToolCall.Name) {
|
||||
runtimeState.OpenAskUserInteraction(
|
||||
uuid.NewString(),
|
||||
buildInfeasibleNegotiationQuestion(flowState),
|
||||
strings.TrimSpace(input.ResumeNode),
|
||||
)
|
||||
return nil
|
||||
}
|
||||
return executeToolCall(
|
||||
ctx,
|
||||
flowState,
|
||||
conversationContext,
|
||||
decision.ToolCall,
|
||||
emitter,
|
||||
input.ToolRegistry,
|
||||
input.ScheduleState,
|
||||
input.WriteSchedulePreview,
|
||||
)
|
||||
}
|
||||
if strings.TrimSpace(decision.Speak) == "" && strings.TrimSpace(decision.Reason) != "" {
|
||||
conversationContext.AppendHistory(&schema.Message{
|
||||
Role: schema.Assistant,
|
||||
Content: decision.Reason,
|
||||
})
|
||||
}
|
||||
return nil
|
||||
|
||||
case newagentmodel.ExecuteActionAskUser:
|
||||
question := resolveExecuteAskUserText(decision)
|
||||
runtimeState.OpenAskUserInteraction(uuid.NewString(), question, strings.TrimSpace(input.ResumeNode))
|
||||
runtimeState.SetPendingInteractionMetadata(newagentmodel.PendingMetaAskUserSpeakStreamed, output.speakStreamed)
|
||||
runtimeState.SetPendingInteractionMetadata(newagentmodel.PendingMetaAskUserHistoryAppended, askUserHistoryAppended)
|
||||
return nil
|
||||
|
||||
case newagentmodel.ExecuteActionConfirm:
|
||||
if decision.ToolCall != nil && shouldForceFeasibilityNegotiation(flowState, input.ToolRegistry, decision.ToolCall.Name) {
|
||||
runtimeState.OpenAskUserInteraction(
|
||||
uuid.NewString(),
|
||||
buildInfeasibleNegotiationQuestion(flowState),
|
||||
strings.TrimSpace(input.ResumeNode),
|
||||
)
|
||||
return nil
|
||||
}
|
||||
if input.AlwaysExecute && decision.ToolCall != nil {
|
||||
return executeToolCall(
|
||||
ctx,
|
||||
flowState,
|
||||
conversationContext,
|
||||
decision.ToolCall,
|
||||
emitter,
|
||||
input.ToolRegistry,
|
||||
input.ScheduleState,
|
||||
input.WriteSchedulePreview,
|
||||
)
|
||||
}
|
||||
return handleExecuteActionConfirm(decision, runtimeState, flowState)
|
||||
|
||||
case newagentmodel.ExecuteActionNextPlan:
|
||||
if !flowState.AdvanceStep() {
|
||||
flowState.Done()
|
||||
}
|
||||
appendExecuteStepAdvancedMarker(conversationContext)
|
||||
syncExecutePinnedContext(conversationContext, flowState)
|
||||
return nil
|
||||
|
||||
case newagentmodel.ExecuteActionDone:
|
||||
flowState.Done()
|
||||
return nil
|
||||
|
||||
case newagentmodel.ExecuteActionAbort:
|
||||
return handleExecuteActionAbort(decision, flowState)
|
||||
|
||||
default:
|
||||
llmOutput := decision.Speak
|
||||
if strings.TrimSpace(llmOutput) == "" {
|
||||
llmOutput = decision.Reason
|
||||
}
|
||||
newagentshared.AppendLLMCorrectionWithHint(
|
||||
conversationContext,
|
||||
llmOutput,
|
||||
fmt.Sprintf("你输出的 action %q 不是合法的执行动作。", decision.Action),
|
||||
"合法的 action 包括：continue（继续当前步骤）、ask_user（追问用户）、confirm（写操作确认）、next_plan（推进到下一步）、done（任务完成）、abort（正式终止本轮流程）。",
|
||||
)
|
||||
return nil
|
||||
}
|
||||
}
|
||||
119
backend/newAgent/node/execute/action_text.go
Normal file
119
backend/newAgent/node/execute/action_text.go
Normal file
@@ -0,0 +1,119 @@
|
||||
package newagentexecute
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
newagentmodel "github.com/LoveLosita/smartflow/backend/newAgent/model"
|
||||
)
|
||||
|
||||
func resolveExecuteAskUserText(decision *newagentmodel.ExecuteDecision) string {
|
||||
if decision == nil {
|
||||
return "执行过程中遇到不确定的情况，需要向你确认。"
|
||||
}
|
||||
if strings.TrimSpace(decision.Speak) != "" {
|
||||
return strings.TrimSpace(decision.Speak)
|
||||
}
|
||||
if strings.TrimSpace(decision.Reason) != "" {
|
||||
return strings.TrimSpace(decision.Reason)
|
||||
}
|
||||
return "执行过程中遇到不确定的情况，需要向你确认。"
|
||||
}
|
||||
|
||||
func pickExecuteVisibleSpeak(
|
||||
streamed string,
|
||||
afterText string,
|
||||
beforeText string,
|
||||
decision *newagentmodel.ExecuteDecision,
|
||||
) string {
|
||||
if text := strings.TrimSpace(streamed); text != "" {
|
||||
return text
|
||||
}
|
||||
if text := strings.TrimSpace(afterText); text != "" {
|
||||
return text
|
||||
}
|
||||
if text := strings.TrimSpace(beforeText); text != "" {
|
||||
return text
|
||||
}
|
||||
return buildExecuteSpeakWithFallback(decision)
|
||||
}
|
||||
|
||||
func buildExecuteSpeakWithFallback(decision *newagentmodel.ExecuteDecision) string {
|
||||
if decision == nil {
|
||||
return ""
|
||||
}
|
||||
|
||||
speak := strings.TrimSpace(decision.Speak)
|
||||
if speak != "" {
|
||||
return speak
|
||||
}
|
||||
|
||||
switch decision.Action {
|
||||
case newagentmodel.ExecuteActionContinue,
|
||||
newagentmodel.ExecuteActionAskUser,
|
||||
newagentmodel.ExecuteActionConfirm:
|
||||
if reason := strings.TrimSpace(decision.Reason); reason != "" {
|
||||
return reason
|
||||
}
|
||||
switch decision.Action {
|
||||
case newagentmodel.ExecuteActionAskUser:
|
||||
return "我还缺少一条关键信息，想先向你确认。"
|
||||
case newagentmodel.ExecuteActionConfirm:
|
||||
return "我先整理好这一步操作，等待你的确认。"
|
||||
default:
|
||||
return "我先继续这一步处理，马上给你结果。"
|
||||
}
|
||||
default:
|
||||
return speak
|
||||
}
|
||||
}
|
||||
|
||||
func handleExecuteActionConfirm(
|
||||
decision *newagentmodel.ExecuteDecision,
|
||||
runtimeState *newagentmodel.AgentRuntimeState,
|
||||
flowState *newagentmodel.CommonState,
|
||||
) error {
|
||||
toolCall := decision.ToolCall
|
||||
|
||||
argsJSON := ""
|
||||
if toolCall.Arguments != nil {
|
||||
if raw, err := json.Marshal(toolCall.Arguments); err == nil {
|
||||
argsJSON = string(raw)
|
||||
}
|
||||
}
|
||||
|
||||
runtimeState.PendingConfirmTool = &newagentmodel.PendingToolCallSnapshot{
|
||||
ToolName: toolCall.Name,
|
||||
ArgsJSON: argsJSON,
|
||||
Summary: strings.TrimSpace(decision.Speak),
|
||||
}
|
||||
|
||||
flowState.Phase = newagentmodel.PhaseWaitingConfirm
|
||||
return nil
|
||||
}
|
||||
|
||||
func handleExecuteActionAbort(
|
||||
decision *newagentmodel.ExecuteDecision,
|
||||
flowState *newagentmodel.CommonState,
|
||||
) error {
|
||||
if decision == nil || decision.Abort == nil {
|
||||
return fmt.Errorf("abort 动作缺少终止信息")
|
||||
}
|
||||
if flowState == nil {
|
||||
return fmt.Errorf("abort 动作缺少流程状态")
|
||||
}
|
||||
|
||||
internalReason := strings.TrimSpace(decision.Abort.InternalReason)
|
||||
if internalReason == "" {
|
||||
internalReason = strings.TrimSpace(decision.Reason)
|
||||
}
|
||||
|
||||
flowState.Abort(
|
||||
executeStageName,
|
||||
decision.Abort.Code,
|
||||
decision.Abort.UserMessage,
|
||||
internalReason,
|
||||
)
|
||||
return nil
|
||||
}
|
||||
162
backend/newAgent/node/execute/args.go
Normal file
162
backend/newAgent/node/execute/args.go
Normal file
@@ -0,0 +1,162 @@
|
||||
package newagentexecute
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
func intSliceToSet(values []int) map[int]struct{} {
|
||||
result := make(map[int]struct{}, len(values))
|
||||
for _, value := range values {
|
||||
result[value] = struct{}{}
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
func readIntAnyFromMap(args map[string]any, keys ...string) (int, bool) {
|
||||
for _, key := range keys {
|
||||
if args == nil {
|
||||
continue
|
||||
}
|
||||
raw, exists := args[key]
|
||||
if !exists {
|
||||
continue
|
||||
}
|
||||
if value, ok := parseAnyToInt(raw); ok {
|
||||
return value, true
|
||||
}
|
||||
}
|
||||
return 0, false
|
||||
}
|
||||
|
||||
func readIntSliceAnyFromMap(args map[string]any, keys ...string) []int {
|
||||
for _, key := range keys {
|
||||
if args == nil {
|
||||
continue
|
||||
}
|
||||
raw, exists := args[key]
|
||||
if !exists {
|
||||
continue
|
||||
}
|
||||
values := parseAnyToIntSlice(raw)
|
||||
if len(values) > 0 {
|
||||
return values
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func readStringAnyFromMap(args map[string]any, keys ...string) string {
|
||||
for _, key := range keys {
|
||||
if args == nil {
|
||||
continue
|
||||
}
|
||||
raw, exists := args[key]
|
||||
if !exists {
|
||||
continue
|
||||
}
|
||||
if text, ok := raw.(string); ok {
|
||||
return text
|
||||
}
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func parseAnyToInt(value any) (int, bool) {
|
||||
switch v := value.(type) {
|
||||
case int:
|
||||
return v, true
|
||||
case int8:
|
||||
return int(v), true
|
||||
case int16:
|
||||
return int(v), true
|
||||
case int32:
|
||||
return int(v), true
|
||||
case int64:
|
||||
return int(v), true
|
||||
case float32:
|
||||
return int(v), true
|
||||
case float64:
|
||||
return int(v), true
|
||||
case json.Number:
|
||||
if iv, err := v.Int64(); err == nil {
|
||||
return int(iv), true
|
||||
}
|
||||
if fv, err := v.Float64(); err == nil {
|
||||
return int(fv), true
|
||||
}
|
||||
case string:
|
||||
text := strings.TrimSpace(v)
|
||||
if text == "" {
|
||||
return 0, false
|
||||
}
|
||||
iv, err := strconv.Atoi(text)
|
||||
if err == nil {
|
||||
return iv, true
|
||||
}
|
||||
}
|
||||
return 0, false
|
||||
}
|
||||
|
||||
func parseAnyToIntSlice(value any) []int {
|
||||
switch values := value.(type) {
|
||||
case []int:
|
||||
result := make([]int, 0, len(values))
|
||||
for _, value := range values {
|
||||
result = append(result, value)
|
||||
}
|
||||
return result
|
||||
case []any:
|
||||
result := make([]int, 0, len(values))
|
||||
for _, item := range values {
|
||||
iv, ok := parseAnyToInt(item)
|
||||
if !ok {
|
||||
continue
|
||||
}
|
||||
result = append(result, iv)
|
||||
}
|
||||
return result
|
||||
default:
|
||||
return nil
|
||||
}
|
||||
}
|
||||
|
||||
func parseAnyToStringSlice(value any) []string {
|
||||
switch values := value.(type) {
|
||||
case []string:
|
||||
result := make([]string, 0, len(values))
|
||||
for _, item := range values {
|
||||
text := strings.TrimSpace(item)
|
||||
if text == "" {
|
||||
continue
|
||||
}
|
||||
result = append(result, text)
|
||||
}
|
||||
return result
|
||||
case []any:
|
||||
result := make([]string, 0, len(values))
|
||||
for _, item := range values {
|
||||
text := strings.TrimSpace(fmt.Sprintf("%v", item))
|
||||
if text == "" || text == "<nil>" {
|
||||
continue
|
||||
}
|
||||
result = append(result, text)
|
||||
}
|
||||
return result
|
||||
default:
|
||||
return nil
|
||||
}
|
||||
}
|
||||
|
||||
func truncateText(text string, maxLen int) string {
|
||||
text = strings.TrimSpace(text)
|
||||
if len(text) <= maxLen {
|
||||
return text
|
||||
}
|
||||
if maxLen <= 3 {
|
||||
return text[:maxLen]
|
||||
}
|
||||
return text[:maxLen-3] + "..."
|
||||
}
|
||||
157
backend/newAgent/node/execute/context.go
Normal file
157
backend/newAgent/node/execute/context.go
Normal file
@@ -0,0 +1,157 @@
|
||||
package newagentexecute
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
newagentmodel "github.com/LoveLosita/smartflow/backend/newAgent/model"
|
||||
newagentstream "github.com/LoveLosita/smartflow/backend/newAgent/stream"
|
||||
"github.com/cloudwego/eino/schema"
|
||||
)
|
||||
|
||||
const (
|
||||
planCurrentStepKey = "current_step"
|
||||
planCurrentStepTitle = "当前步骤"
|
||||
)
|
||||
|
||||
func prepareExecuteNodeInput(input ExecuteNodeInput) (*newagentmodel.AgentRuntimeState, *newagentmodel.ConversationContext, *newagentstream.ChunkEmitter, error) {
|
||||
if input.RuntimeState == nil {
|
||||
return nil, nil, nil, fmt.Errorf("execute node: runtime state 不能为空")
|
||||
}
|
||||
if input.Client == nil {
|
||||
return nil, nil, nil, fmt.Errorf("execute node: execute client 未注入")
|
||||
}
|
||||
|
||||
input.RuntimeState.EnsureCommonState()
|
||||
if input.ConversationContext == nil {
|
||||
input.ConversationContext = newagentmodel.NewConversationContext("")
|
||||
}
|
||||
if input.ChunkEmitter == nil {
|
||||
input.ChunkEmitter = newagentstream.NewChunkEmitter(newagentstream.NoopPayloadEmitter(), "", "", time.Now().Unix())
|
||||
}
|
||||
return input.RuntimeState, input.ConversationContext, input.ChunkEmitter, nil
|
||||
}
|
||||
|
||||
func syncExecutePinnedContext(
|
||||
conversationContext *newagentmodel.ConversationContext,
|
||||
flowState *newagentmodel.CommonState,
|
||||
) {
|
||||
if conversationContext == nil || flowState == nil {
|
||||
return
|
||||
}
|
||||
|
||||
execContent := buildExecuteContextPinnedMarkdown(flowState)
|
||||
if strings.TrimSpace(execContent) != "" {
|
||||
conversationContext.UpsertPinnedBlock(newagentmodel.ContextBlock{
|
||||
Key: executePinnedKey,
|
||||
Title: "执行上下文",
|
||||
Content: execContent,
|
||||
})
|
||||
}
|
||||
|
||||
if !flowState.HasPlan() {
|
||||
conversationContext.RemovePinnedBlock(planCurrentStepKey)
|
||||
return
|
||||
}
|
||||
|
||||
step, ok := flowState.CurrentPlanStep()
|
||||
if !ok {
|
||||
conversationContext.RemovePinnedBlock(planCurrentStepKey)
|
||||
return
|
||||
}
|
||||
|
||||
current, total := flowState.PlanProgress()
|
||||
title := strings.TrimSpace(planCurrentStepTitle)
|
||||
if title == "" {
|
||||
title = "当前步骤"
|
||||
}
|
||||
conversationContext.UpsertPinnedBlock(newagentmodel.ContextBlock{
|
||||
Key: planCurrentStepKey,
|
||||
Title: title,
|
||||
Content: buildCurrentPlanStepPinnedMarkdown(step, current, total),
|
||||
})
|
||||
}
|
||||
|
||||
func appendExecuteStepAdvancedMarker(conversationContext *newagentmodel.ConversationContext) {
|
||||
if conversationContext == nil {
|
||||
return
|
||||
}
|
||||
|
||||
history := conversationContext.HistorySnapshot()
|
||||
if len(history) > 0 {
|
||||
last := history[len(history)-1]
|
||||
if last != nil && last.Extra != nil {
|
||||
if kind, ok := last.Extra[executeHistoryKindKey].(string); ok && strings.TrimSpace(kind) == executeHistoryKindStepAdvanced {
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
conversationContext.AppendHistory(&schema.Message{
|
||||
Role: schema.Assistant,
|
||||
Content: "",
|
||||
Extra: map[string]any{
|
||||
executeHistoryKindKey: executeHistoryKindStepAdvanced,
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
func buildExecuteContextPinnedMarkdown(flowState *newagentmodel.CommonState) string {
|
||||
if flowState == nil {
|
||||
return ""
|
||||
}
|
||||
|
||||
lines := make([]string, 0, 8)
|
||||
if flowState.HasPlan() {
|
||||
lines = append(lines, "执行模式：计划执行（按步骤推进）")
|
||||
current, total := flowState.PlanProgress()
|
||||
lines = append(lines, fmt.Sprintf("计划进度：第 %d/%d 步", current, total))
|
||||
|
||||
if step, ok := flowState.CurrentPlanStep(); ok {
|
||||
lines = append(lines, "当前步骤："+compactExecutePinnedText(step.Content))
|
||||
doneWhen := compactExecutePinnedText(step.DoneWhen)
|
||||
if doneWhen != "" {
|
||||
lines = append(lines, "完成判定(done_when)："+doneWhen)
|
||||
}
|
||||
lines = append(lines, "动作纪律：未满足 done_when 禁止 next_plan；满足后优先 next_plan。")
|
||||
} else {
|
||||
lines = append(lines, "当前步骤：不可读（可能已执行完成）")
|
||||
}
|
||||
} else {
|
||||
lines = append(lines, "执行模式：自由执行（无预定义步骤）")
|
||||
}
|
||||
|
||||
if flowState.MaxRounds > 0 {
|
||||
lines = append(lines, fmt.Sprintf("轮次预算：%d/%d", flowState.RoundUsed, flowState.MaxRounds))
|
||||
}
|
||||
return strings.TrimSpace(strings.Join(lines, "\n"))
|
||||
}
|
||||
|
||||
func buildCurrentPlanStepPinnedMarkdown(step newagentmodel.PlanStep, current, total int) string {
|
||||
lines := make([]string, 0, 4)
|
||||
lines = append(lines, fmt.Sprintf("步骤进度：第 %d/%d 步", current, total))
|
||||
|
||||
content := compactExecutePinnedText(step.Content)
|
||||
if content == "" {
|
||||
content = "（空）"
|
||||
}
|
||||
lines = append(lines, "步骤内容："+content)
|
||||
|
||||
doneWhen := compactExecutePinnedText(step.DoneWhen)
|
||||
if doneWhen != "" {
|
||||
lines = append(lines, "完成判定："+doneWhen)
|
||||
}
|
||||
|
||||
return strings.TrimSpace(strings.Join(lines, "\n"))
|
||||
}
|
||||
|
||||
func compactExecutePinnedText(text string) string {
|
||||
text = strings.TrimSpace(text)
|
||||
if text == "" {
|
||||
return ""
|
||||
}
|
||||
text = strings.ReplaceAll(text, "\r\n", "\n")
|
||||
text = strings.ReplaceAll(text, "\n", "；")
|
||||
return strings.TrimSpace(text)
|
||||
}
|
||||
150
backend/newAgent/node/execute/run.go
Normal file
150
backend/newAgent/node/execute/run.go
Normal file
@@ -0,0 +1,150 @@
|
||||
package newagentexecute
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
newagentshared "github.com/LoveLosita/smartflow/backend/newAgent/shared"
|
||||
|
||||
infrallm "github.com/LoveLosita/smartflow/backend/infra/llm"
|
||||
newagentmodel "github.com/LoveLosita/smartflow/backend/newAgent/model"
|
||||
newagentprompt "github.com/LoveLosita/smartflow/backend/newAgent/prompt"
|
||||
newagentstream "github.com/LoveLosita/smartflow/backend/newAgent/stream"
|
||||
newagenttools "github.com/LoveLosita/smartflow/backend/newAgent/tools"
|
||||
"github.com/LoveLosita/smartflow/backend/newAgent/tools/schedule"
|
||||
)
|
||||
|
||||
const (
|
||||
executeStageName = "execute"
|
||||
executeStatusBlockID = "execute.status"
|
||||
executeSpeakBlockID = "execute.speak"
|
||||
executePinnedKey = "execution_context"
|
||||
toolAnalyzeHealth = "analyze_health"
|
||||
executeHistoryKindKey = "newagent_history_kind"
|
||||
executeHistoryKindStepAdvanced = "execute_step_advanced"
|
||||
|
||||
maxConsecutiveCorrections = 3
|
||||
)
|
||||
|
||||
type ExecuteNodeInput struct {
|
||||
RuntimeState *newagentmodel.AgentRuntimeState
|
||||
ConversationContext *newagentmodel.ConversationContext
|
||||
UserInput string
|
||||
Client *infrallm.Client
|
||||
ChunkEmitter *newagentstream.ChunkEmitter
|
||||
ResumeNode string
|
||||
ToolRegistry *newagenttools.ToolRegistry
|
||||
ScheduleState *schedule.ScheduleState
|
||||
CompactionStore newagentmodel.CompactionStore
|
||||
WriteSchedulePreview newagentmodel.WriteSchedulePreviewFunc
|
||||
OriginalScheduleState *schedule.ScheduleState
|
||||
AlwaysExecute bool
|
||||
ThinkingEnabled bool
|
||||
PersistVisibleMessage newagentmodel.PersistVisibleMessageFunc
|
||||
}
|
||||
|
||||
type ExecuteRoundObservation struct {
|
||||
Round int `json:"round"`
|
||||
StepIndex int `json:"step_index"`
|
||||
GoalCheck string `json:"goal_check,omitempty"`
|
||||
Decision string `json:"decision,omitempty"`
|
||||
ToolName string `json:"tool_name,omitempty"`
|
||||
ToolParams string `json:"tool_params,omitempty"`
|
||||
ToolSuccess bool `json:"tool_success"`
|
||||
ToolResult string `json:"tool_result,omitempty"`
|
||||
}
|
||||
|
||||
func RunExecuteNode(ctx context.Context, input ExecuteNodeInput) error {
|
||||
runtimeState, conversationContext, emitter, err := prepareExecuteNodeInput(input)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
flowState := runtimeState.EnsureCommonState()
|
||||
applyPendingContextHook(flowState)
|
||||
|
||||
if runtimeState.PendingConfirmTool != nil {
|
||||
return executePendingTool(
|
||||
ctx,
|
||||
runtimeState,
|
||||
conversationContext,
|
||||
input.ToolRegistry,
|
||||
input.ScheduleState,
|
||||
input.OriginalScheduleState,
|
||||
input.WriteSchedulePreview,
|
||||
emitter,
|
||||
)
|
||||
}
|
||||
|
||||
if input.ScheduleState != nil && flowState.RoundUsed == 0 {
|
||||
schedule.ResetTaskProcessingQueue(input.ScheduleState)
|
||||
}
|
||||
|
||||
syncExecutePinnedContext(conversationContext, flowState)
|
||||
|
||||
if flowState.HasCurrentPlanStep() {
|
||||
current, total := flowState.PlanProgress()
|
||||
currentStep, _ := flowState.CurrentPlanStep()
|
||||
if err := emitter.EmitStatus(
|
||||
executeStatusBlockID,
|
||||
executeStageName,
|
||||
"executing",
|
||||
fmt.Sprintf("正在执行第 %d/%d 步：%s", current, total, truncateText(currentStep.Content, 60)),
|
||||
false,
|
||||
); err != nil {
|
||||
return fmt.Errorf("执行阶段状态推送失败: %w", err)
|
||||
}
|
||||
} else {
|
||||
if err := emitter.EmitStatus(
|
||||
executeStatusBlockID,
|
||||
executeStageName,
|
||||
"executing",
|
||||
"正在处理你的请求...",
|
||||
false,
|
||||
); err != nil {
|
||||
return fmt.Errorf("执行阶段状态推送失败: %w", err)
|
||||
}
|
||||
}
|
||||
|
||||
if !flowState.NextRound() {
|
||||
flowState.Exhaust(
|
||||
executeStageName,
|
||||
"本轮执行已达到安全轮次上限，当前先停止继续操作。如需继续，我可以在你确认后接着处理剩余步骤。",
|
||||
"execute rounds exhausted before task completion",
|
||||
)
|
||||
return nil
|
||||
}
|
||||
|
||||
messages := newagentprompt.BuildExecuteMessages(flowState, conversationContext)
|
||||
messages = newagentshared.CompactUnifiedMessagesIfNeeded(ctx, messages, newagentshared.UnifiedCompactInput{
|
||||
Client: input.Client,
|
||||
CompactionStore: input.CompactionStore,
|
||||
FlowState: flowState,
|
||||
Emitter: emitter,
|
||||
StageName: executeStageName,
|
||||
StatusBlockID: executeStatusBlockID,
|
||||
})
|
||||
|
||||
newagentshared.LogNodeLLMContext(executeStageName, "decision", flowState, messages)
|
||||
|
||||
decisionOutput, err := collectExecuteDecisionFromLLM(
|
||||
ctx,
|
||||
input,
|
||||
flowState,
|
||||
conversationContext,
|
||||
emitter,
|
||||
messages,
|
||||
)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return handleExecuteDecision(
|
||||
ctx,
|
||||
input,
|
||||
runtimeState,
|
||||
flowState,
|
||||
conversationContext,
|
||||
emitter,
|
||||
decisionOutput,
|
||||
)
|
||||
}
|
||||
332
backend/newAgent/node/execute/state_snapshot.go
Normal file
332
backend/newAgent/node/execute/state_snapshot.go
Normal file
@@ -0,0 +1,332 @@
|
||||
package newagentexecute
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
newagentmodel "github.com/LoveLosita/smartflow/backend/newAgent/model"
|
||||
newagenttools "github.com/LoveLosita/smartflow/backend/newAgent/tools"
|
||||
)
|
||||
|
||||
func shouldForceFeasibilityNegotiation(
|
||||
flowState *newagentmodel.CommonState,
|
||||
registry *newagenttools.ToolRegistry,
|
||||
toolName string,
|
||||
) bool {
|
||||
if flowState == nil || registry == nil {
|
||||
return false
|
||||
}
|
||||
if !flowState.HealthCheckDone || flowState.HealthIsFeasible {
|
||||
return false
|
||||
}
|
||||
if !registry.IsWriteTool(toolName) || !registry.RequiresScheduleState(toolName) {
|
||||
return false
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func buildInfeasibleNegotiationQuestion(flowState *newagentmodel.CommonState) string {
|
||||
capacityGap := 0
|
||||
reasonCode := "capacity_insufficient"
|
||||
if flowState != nil {
|
||||
capacityGap = flowState.HealthCapacityGap
|
||||
if strings.TrimSpace(flowState.HealthReasonCode) != "" {
|
||||
reasonCode = strings.TrimSpace(flowState.HealthReasonCode)
|
||||
}
|
||||
}
|
||||
return fmt.Sprintf(
|
||||
"当前计划不可行：analyze_health 判断当前约束不可行（capacity_gap=%d，reason=%s）。在继续写操作前，请先与用户协商：扩展时间窗、放宽约束、缩减范围或预算，或接受风险收口。",
|
||||
capacityGap,
|
||||
reasonCode,
|
||||
)
|
||||
}
|
||||
|
||||
func buildInfeasibleBlockedResult(flowState *newagentmodel.CommonState) string {
|
||||
capacityGap := 0
|
||||
reasonCode := "capacity_insufficient"
|
||||
if flowState != nil {
|
||||
capacityGap = flowState.HealthCapacityGap
|
||||
if strings.TrimSpace(flowState.HealthReasonCode) != "" {
|
||||
reasonCode = strings.TrimSpace(flowState.HealthReasonCode)
|
||||
}
|
||||
}
|
||||
return fmt.Sprintf(
|
||||
"已阻断本次写操作：analyze_health 判定当前约束不可行（capacity_gap=%d，reason=%s）。请先与用户协商：扩展时间窗 / 放宽约束 / 缩减范围或预算 / 接受风险收口。",
|
||||
capacityGap,
|
||||
reasonCode,
|
||||
)
|
||||
}
|
||||
|
||||
type contextToolsResultEnvelope struct {
|
||||
Tool string `json:"tool"`
|
||||
Success bool `json:"success"`
|
||||
Domain string `json:"domain,omitempty"`
|
||||
Packs []string `json:"packs,omitempty"`
|
||||
Mode string `json:"mode,omitempty"`
|
||||
All bool `json:"all,omitempty"`
|
||||
}
|
||||
|
||||
type analyzeHealthResultEnvelope struct {
|
||||
Tool string `json:"tool"`
|
||||
Success bool `json:"success"`
|
||||
Feasibility *analyzeHealthFeasibilityBrief `json:"feasibility,omitempty"`
|
||||
Decision *analyzeHealthDecisionBrief `json:"decision,omitempty"`
|
||||
}
|
||||
|
||||
type analyzeHealthFeasibilityBrief struct {
|
||||
IsFeasible bool `json:"is_feasible"`
|
||||
CapacityGap int `json:"capacity_gap"`
|
||||
ReasonCode string `json:"reason_code"`
|
||||
}
|
||||
|
||||
type analyzeHealthDecisionBrief struct {
|
||||
ShouldContinueOptimize bool `json:"should_continue_optimize"`
|
||||
PrimaryProblem string `json:"primary_problem,omitempty"`
|
||||
RecommendedOperation string `json:"recommended_operation,omitempty"`
|
||||
IsForcedImperfection bool `json:"is_forced_imperfection"`
|
||||
ImprovementSignal string `json:"improvement_signal,omitempty"`
|
||||
}
|
||||
|
||||
type upsertTaskClassResultEnvelope struct {
|
||||
Tool string `json:"tool"`
|
||||
Success bool `json:"success"`
|
||||
Validation *upsertTaskClassValidationPart `json:"validation,omitempty"`
|
||||
Error string `json:"error,omitempty"`
|
||||
ErrorCode string `json:"error_code,omitempty"`
|
||||
}
|
||||
|
||||
type upsertTaskClassValidationPart struct {
|
||||
OK bool `json:"ok"`
|
||||
Issues []string `json:"issues"`
|
||||
}
|
||||
|
||||
func updateActiveToolDomainSnapshot(flowState *newagentmodel.CommonState, toolName string, result string) {
|
||||
if flowState == nil || !newagenttools.IsContextManagementTool(toolName) {
|
||||
return
|
||||
}
|
||||
|
||||
var envelope contextToolsResultEnvelope
|
||||
if err := json.Unmarshal([]byte(result), &envelope); err != nil {
|
||||
return
|
||||
}
|
||||
if !envelope.Success {
|
||||
return
|
||||
}
|
||||
|
||||
switch strings.TrimSpace(toolName) {
|
||||
case newagenttools.ToolNameContextToolsAdd:
|
||||
domain := newagenttools.NormalizeToolDomain(envelope.Domain)
|
||||
if domain == "" {
|
||||
return
|
||||
}
|
||||
nextPacks := newagenttools.ResolveEffectiveToolPacks(domain, envelope.Packs)
|
||||
mode := strings.ToLower(strings.TrimSpace(envelope.Mode))
|
||||
if mode == "merge" && newagenttools.NormalizeToolDomain(flowState.ActiveToolDomain) == domain {
|
||||
merged := make([]string, 0, len(flowState.ActiveToolPacks)+len(nextPacks))
|
||||
seen := make(map[string]struct{}, len(flowState.ActiveToolPacks)+len(nextPacks))
|
||||
current := newagenttools.ResolveEffectiveToolPacks(domain, flowState.ActiveToolPacks)
|
||||
for _, pack := range current {
|
||||
if _, exists := seen[pack]; exists {
|
||||
continue
|
||||
}
|
||||
seen[pack] = struct{}{}
|
||||
merged = append(merged, pack)
|
||||
}
|
||||
for _, pack := range nextPacks {
|
||||
if _, exists := seen[pack]; exists {
|
||||
continue
|
||||
}
|
||||
seen[pack] = struct{}{}
|
||||
merged = append(merged, pack)
|
||||
}
|
||||
nextPacks = merged
|
||||
}
|
||||
flowState.ActiveToolDomain = domain
|
||||
flowState.ActiveToolPacks = nextPacks
|
||||
case newagenttools.ToolNameContextToolsRemove:
|
||||
if envelope.All {
|
||||
flowState.ActiveToolDomain = ""
|
||||
flowState.ActiveToolPacks = nil
|
||||
return
|
||||
}
|
||||
domain := newagenttools.NormalizeToolDomain(envelope.Domain)
|
||||
if domain == "" {
|
||||
return
|
||||
}
|
||||
currentDomain := newagenttools.NormalizeToolDomain(flowState.ActiveToolDomain)
|
||||
if currentDomain != domain {
|
||||
return
|
||||
}
|
||||
|
||||
removedPacks := newagenttools.NormalizeToolPacks(domain, envelope.Packs)
|
||||
if len(removedPacks) == 0 {
|
||||
flowState.ActiveToolDomain = ""
|
||||
flowState.ActiveToolPacks = nil
|
||||
return
|
||||
}
|
||||
|
||||
currentEffective := newagenttools.ResolveEffectiveToolPacks(domain, flowState.ActiveToolPacks)
|
||||
if len(currentEffective) == 0 {
|
||||
flowState.ActiveToolDomain = ""
|
||||
flowState.ActiveToolPacks = nil
|
||||
return
|
||||
}
|
||||
|
||||
removedSet := make(map[string]struct{}, len(removedPacks))
|
||||
for _, pack := range removedPacks {
|
||||
removedSet[pack] = struct{}{}
|
||||
}
|
||||
remaining := make([]string, 0, len(currentEffective))
|
||||
for _, pack := range currentEffective {
|
||||
if _, shouldRemove := removedSet[pack]; shouldRemove {
|
||||
continue
|
||||
}
|
||||
remaining = append(remaining, pack)
|
||||
}
|
||||
if len(remaining) == 0 {
|
||||
flowState.ActiveToolDomain = ""
|
||||
flowState.ActiveToolPacks = nil
|
||||
return
|
||||
}
|
||||
flowState.ActiveToolPacks = remaining
|
||||
}
|
||||
}
|
||||
|
||||
func updateHealthFeasibilitySnapshot(flowState *newagentmodel.CommonState, toolName string, result string) {
|
||||
if flowState == nil || !strings.EqualFold(strings.TrimSpace(toolName), toolAnalyzeHealth) {
|
||||
return
|
||||
}
|
||||
|
||||
flowState.HealthCheckDone = false
|
||||
flowState.HealthIsFeasible = true
|
||||
flowState.HealthCapacityGap = 0
|
||||
flowState.HealthReasonCode = ""
|
||||
|
||||
var envelope analyzeHealthResultEnvelope
|
||||
if err := json.Unmarshal([]byte(result), &envelope); err != nil {
|
||||
return
|
||||
}
|
||||
if !envelope.Success || envelope.Feasibility == nil {
|
||||
return
|
||||
}
|
||||
|
||||
flowState.HealthCheckDone = true
|
||||
flowState.HealthIsFeasible = envelope.Feasibility.IsFeasible
|
||||
flowState.HealthCapacityGap = envelope.Feasibility.CapacityGap
|
||||
flowState.HealthReasonCode = strings.TrimSpace(envelope.Feasibility.ReasonCode)
|
||||
}
|
||||
|
||||
func updateTaskClassUpsertSnapshot(flowState *newagentmodel.CommonState, toolName string, result string) {
|
||||
if flowState == nil || !strings.EqualFold(strings.TrimSpace(toolName), "upsert_task_class") {
|
||||
return
|
||||
}
|
||||
|
||||
flowState.TaskClassUpsertLastTried = true
|
||||
flowState.TaskClassUpsertLastSuccess = false
|
||||
flowState.TaskClassUpsertLastIssues = nil
|
||||
|
||||
var envelope upsertTaskClassResultEnvelope
|
||||
if err := json.Unmarshal([]byte(result), &envelope); err != nil {
|
||||
flowState.TaskClassUpsertConsecutiveFailures++
|
||||
return
|
||||
}
|
||||
|
||||
success := envelope.Success
|
||||
issues := make([]string, 0)
|
||||
if envelope.Validation != nil {
|
||||
issues = append(issues, parseAnyToStringSlice(any(envelope.Validation.Issues))...)
|
||||
if !envelope.Validation.OK {
|
||||
success = false
|
||||
}
|
||||
}
|
||||
if !success && strings.TrimSpace(envelope.Error) != "" && len(issues) == 0 {
|
||||
issues = append(issues, strings.TrimSpace(envelope.Error))
|
||||
}
|
||||
issues = uniqueNonEmptyStrings(issues)
|
||||
|
||||
flowState.TaskClassUpsertLastSuccess = success
|
||||
flowState.TaskClassUpsertLastIssues = issues
|
||||
if success {
|
||||
flowState.TaskClassUpsertConsecutiveFailures = 0
|
||||
return
|
||||
}
|
||||
flowState.TaskClassUpsertConsecutiveFailures++
|
||||
}
|
||||
|
||||
func uniqueNonEmptyStrings(values []string) []string {
|
||||
if len(values) == 0 {
|
||||
return nil
|
||||
}
|
||||
seen := make(map[string]struct{}, len(values))
|
||||
result := make([]string, 0, len(values))
|
||||
for _, value := range values {
|
||||
text := strings.TrimSpace(value)
|
||||
if text == "" {
|
||||
continue
|
||||
}
|
||||
if _, exists := seen[text]; exists {
|
||||
continue
|
||||
}
|
||||
seen[text] = struct{}{}
|
||||
result = append(result, text)
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
func updateHealthSnapshotV2(flowState *newagentmodel.CommonState, toolName string, result string) {
|
||||
if flowState == nil || !strings.EqualFold(strings.TrimSpace(toolName), toolAnalyzeHealth) {
|
||||
return
|
||||
}
|
||||
|
||||
prevSignal := strings.TrimSpace(flowState.HealthImprovementSignal)
|
||||
flowState.HealthCheckDone = false
|
||||
flowState.HealthIsFeasible = true
|
||||
flowState.HealthCapacityGap = 0
|
||||
flowState.HealthReasonCode = ""
|
||||
flowState.HealthShouldContinueOptimize = false
|
||||
flowState.HealthTightnessLevel = ""
|
||||
flowState.HealthPrimaryProblem = ""
|
||||
flowState.HealthRecommendedOperation = ""
|
||||
flowState.HealthIsForcedImperfection = false
|
||||
flowState.HealthImprovementSignal = ""
|
||||
|
||||
var envelope struct {
|
||||
Success bool `json:"success"`
|
||||
Feasibility *analyzeHealthFeasibilityBrief `json:"feasibility,omitempty"`
|
||||
Metrics struct {
|
||||
Tightness *struct {
|
||||
TightnessLevel string `json:"tightness_level"`
|
||||
} `json:"tightness,omitempty"`
|
||||
} `json:"metrics"`
|
||||
Decision *analyzeHealthDecisionBrief `json:"decision,omitempty"`
|
||||
}
|
||||
if err := json.Unmarshal([]byte(result), &envelope); err != nil {
|
||||
flowState.HealthStagnationCount = 0
|
||||
return
|
||||
}
|
||||
if !envelope.Success || envelope.Feasibility == nil {
|
||||
flowState.HealthStagnationCount = 0
|
||||
return
|
||||
}
|
||||
|
||||
flowState.HealthCheckDone = true
|
||||
flowState.HealthIsFeasible = envelope.Feasibility.IsFeasible
|
||||
flowState.HealthCapacityGap = envelope.Feasibility.CapacityGap
|
||||
flowState.HealthReasonCode = strings.TrimSpace(envelope.Feasibility.ReasonCode)
|
||||
if envelope.Metrics.Tightness != nil {
|
||||
flowState.HealthTightnessLevel = strings.TrimSpace(envelope.Metrics.Tightness.TightnessLevel)
|
||||
}
|
||||
if envelope.Decision != nil {
|
||||
flowState.HealthShouldContinueOptimize = envelope.Decision.ShouldContinueOptimize
|
||||
flowState.HealthPrimaryProblem = strings.TrimSpace(envelope.Decision.PrimaryProblem)
|
||||
flowState.HealthRecommendedOperation = strings.TrimSpace(envelope.Decision.RecommendedOperation)
|
||||
flowState.HealthIsForcedImperfection = envelope.Decision.IsForcedImperfection
|
||||
flowState.HealthImprovementSignal = strings.TrimSpace(envelope.Decision.ImprovementSignal)
|
||||
}
|
||||
if signal := strings.TrimSpace(flowState.HealthImprovementSignal); signal != "" && prevSignal != "" && signal == prevSignal {
|
||||
flowState.HealthStagnationCount++
|
||||
return
|
||||
}
|
||||
flowState.HealthStagnationCount = 0
|
||||
}
|
||||
436
backend/newAgent/node/execute/tool_runtime.go
Normal file
436
backend/newAgent/node/execute/tool_runtime.go
Normal file
@@ -0,0 +1,436 @@
|
||||
package newagentexecute
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
newagentshared "github.com/LoveLosita/smartflow/backend/newAgent/shared"
|
||||
"log"
|
||||
"regexp"
|
||||
"strings"
|
||||
|
||||
newagentmodel "github.com/LoveLosita/smartflow/backend/newAgent/model"
|
||||
newagentstream "github.com/LoveLosita/smartflow/backend/newAgent/stream"
|
||||
newagenttools "github.com/LoveLosita/smartflow/backend/newAgent/tools"
|
||||
"github.com/LoveLosita/smartflow/backend/newAgent/tools/schedule"
|
||||
"github.com/cloudwego/eino/schema"
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
func appendToolCallResultHistory(
|
||||
conversationContext *newagentmodel.ConversationContext,
|
||||
toolName string,
|
||||
args map[string]any,
|
||||
result string,
|
||||
) {
|
||||
if conversationContext == nil {
|
||||
return
|
||||
}
|
||||
|
||||
argsJSON := "{}"
|
||||
if args != nil {
|
||||
if raw, err := json.Marshal(args); err == nil {
|
||||
argsJSON = string(raw)
|
||||
}
|
||||
}
|
||||
toolCallID := uuid.NewString()
|
||||
conversationContext.AppendHistory(&schema.Message{
|
||||
Role: schema.Assistant,
|
||||
Content: "",
|
||||
ToolCalls: []schema.ToolCall{
|
||||
{
|
||||
ID: toolCallID,
|
||||
Type: "function",
|
||||
Function: schema.FunctionCall{
|
||||
Name: toolName,
|
||||
Arguments: argsJSON,
|
||||
},
|
||||
},
|
||||
},
|
||||
})
|
||||
conversationContext.AppendHistory(&schema.Message{
|
||||
Role: schema.Tool,
|
||||
Content: result,
|
||||
ToolCallID: toolCallID,
|
||||
ToolName: toolName,
|
||||
})
|
||||
}
|
||||
|
||||
func executeToolCall(
|
||||
ctx context.Context,
|
||||
flowState *newagentmodel.CommonState,
|
||||
conversationContext *newagentmodel.ConversationContext,
|
||||
toolCall *newagentmodel.ToolCallIntent,
|
||||
emitter *newagentstream.ChunkEmitter,
|
||||
registry *newagenttools.ToolRegistry,
|
||||
scheduleState *schedule.ScheduleState,
|
||||
writePreview newagentmodel.WriteSchedulePreviewFunc,
|
||||
) error {
|
||||
if toolCall == nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
toolName := strings.TrimSpace(toolCall.Name)
|
||||
if toolName == "" {
|
||||
return fmt.Errorf("工具调用缺少工具名称")
|
||||
}
|
||||
|
||||
if err := emitter.EmitToolCallStart(
|
||||
executeStatusBlockID,
|
||||
executeStageName,
|
||||
toolName,
|
||||
buildToolCallStartSummary(toolName, toolCall.Arguments),
|
||||
buildToolArgumentsPreviewCN(toolCall.Arguments),
|
||||
false,
|
||||
); err != nil {
|
||||
return fmt.Errorf("工具调用开始事件发送失败: %w", err)
|
||||
}
|
||||
|
||||
if registry == nil {
|
||||
return fmt.Errorf("工具注册表未注入")
|
||||
}
|
||||
if scheduleState == nil && registry.RequiresScheduleState(toolName) {
|
||||
return fmt.Errorf("日程状态未加载，无法执行工具 %q", toolName)
|
||||
}
|
||||
if registry.IsToolTemporarilyDisabled(toolName) {
|
||||
flowState.ConsecutiveCorrections++
|
||||
if flowState.ConsecutiveCorrections >= maxConsecutiveCorrections {
|
||||
return fmt.Errorf("连续 %d 次调用临时禁用工具，终止执行: %s",
|
||||
flowState.ConsecutiveCorrections, toolName)
|
||||
}
|
||||
blockedResult := buildTemporarilyDisabledToolResult(toolName)
|
||||
_ = emitter.EmitToolCallResult(
|
||||
executeStatusBlockID,
|
||||
executeStageName,
|
||||
toolName,
|
||||
"blocked",
|
||||
blockedResult,
|
||||
buildToolArgumentsPreviewCN(toolCall.Arguments),
|
||||
false,
|
||||
)
|
||||
appendToolCallResultHistory(conversationContext, toolName, toolCall.Arguments, blockedResult)
|
||||
newagentshared.AppendLLMCorrectionWithHint(
|
||||
conversationContext,
|
||||
"",
|
||||
fmt.Sprintf("工具 %q 当前暂时禁用。", toolName),
|
||||
"请改用 move/swap/batch_move/unplace 等排程微调工具继续推进。",
|
||||
)
|
||||
return nil
|
||||
}
|
||||
if !registry.HasTool(toolName) {
|
||||
flowState.ConsecutiveCorrections++
|
||||
if flowState.ConsecutiveCorrections >= maxConsecutiveCorrections {
|
||||
return fmt.Errorf("连续 %d 次调用未知工具，终止执行: %s；可用工具：%s。",
|
||||
flowState.ConsecutiveCorrections, toolName, strings.Join(registry.ToolNames(), "、"))
|
||||
}
|
||||
log.Printf("[WARN] execute 工具名不合法 chat=%s round=%d tool=%s consecutive=%d/%d available=%v",
|
||||
flowState.ConversationID, flowState.RoundUsed, toolName,
|
||||
flowState.ConsecutiveCorrections, maxConsecutiveCorrections, registry.ToolNames())
|
||||
newagentshared.AppendLLMCorrectionWithHint(
|
||||
conversationContext,
|
||||
"",
|
||||
fmt.Sprintf("你调用的工具 %q 不存在。", toolName),
|
||||
fmt.Sprintf("可用工具：%s。请检查拼写后重试。", strings.Join(registry.ToolNames(), "、")),
|
||||
)
|
||||
return nil
|
||||
}
|
||||
if !isToolVisibleForCurrentExecuteMode(flowState, registry, toolName) {
|
||||
flowState.ConsecutiveCorrections++
|
||||
if flowState.ConsecutiveCorrections >= maxConsecutiveCorrections {
|
||||
return fmt.Errorf("连续 %d 次调用未激活工具，终止执行: %s（active_domain=%q active_packs=%v）",
|
||||
flowState.ConsecutiveCorrections,
|
||||
toolName,
|
||||
flowState.ActiveToolDomain,
|
||||
newagenttools.ResolveEffectiveToolPacks(flowState.ActiveToolDomain, flowState.ActiveToolPacks))
|
||||
}
|
||||
|
||||
addHint := `请先调用 context_tools_add 激活目标工具域后再继续。`
|
||||
if flowState != nil && flowState.ActiveOptimizeOnly {
|
||||
addHint = `当前处于“粗排后主动优化专用模式”，只允许使用 analyze_health、move、swap；不要再尝试 query_target_tasks / query_available_slots 等全窗搜索工具。`
|
||||
} else if domain, pack, ok := newagenttools.ResolveToolDomainPack(toolName); ok {
|
||||
if newagenttools.IsFixedToolPack(domain, pack) {
|
||||
addHint = fmt.Sprintf(`请先调用 context_tools_add，参数 domain="%s"。`, domain)
|
||||
} else {
|
||||
addHint = fmt.Sprintf(`请先调用 context_tools_add，参数 domain="%s", packs=["%s"]。`, domain, pack)
|
||||
}
|
||||
}
|
||||
|
||||
newagentshared.AppendLLMCorrectionWithHint(
|
||||
conversationContext,
|
||||
"",
|
||||
fmt.Sprintf("你调用的工具 %q 当前不在已激活工具域内。", toolName),
|
||||
addHint,
|
||||
)
|
||||
return nil
|
||||
}
|
||||
|
||||
if shouldForceFeasibilityNegotiation(flowState, registry, toolName) {
|
||||
blockedResult := buildInfeasibleBlockedResult(flowState)
|
||||
_ = emitter.EmitToolCallResult(
|
||||
executeStatusBlockID,
|
||||
executeStageName,
|
||||
toolName,
|
||||
"blocked",
|
||||
blockedResult,
|
||||
buildToolArgumentsPreviewCN(toolCall.Arguments),
|
||||
false,
|
||||
)
|
||||
appendToolCallResultHistory(conversationContext, toolName, toolCall.Arguments, blockedResult)
|
||||
return nil
|
||||
}
|
||||
|
||||
beforeDigest := summarizeScheduleStateForDebug(scheduleState)
|
||||
if !registry.RequiresScheduleState(toolName) {
|
||||
if toolCall.Arguments == nil {
|
||||
toolCall.Arguments = make(map[string]any)
|
||||
}
|
||||
toolCall.Arguments["_user_id"] = flowState.UserID
|
||||
}
|
||||
result := registry.Execute(scheduleState, toolName, toolCall.Arguments)
|
||||
updateHealthSnapshotV2(flowState, toolName, result)
|
||||
updateTaskClassUpsertSnapshot(flowState, toolName, result)
|
||||
updateActiveToolDomainSnapshot(flowState, toolName, result)
|
||||
afterDigest := summarizeScheduleStateForDebug(scheduleState)
|
||||
log.Printf(
|
||||
"[DEBUG] execute tool chat=%s round=%d tool=%s args=%s before=%s after=%s result_preview=%.200s",
|
||||
flowState.ConversationID,
|
||||
flowState.RoundUsed,
|
||||
toolName,
|
||||
marshalArgsForDebug(toolCall.Arguments),
|
||||
beforeDigest,
|
||||
afterDigest,
|
||||
flattenForLog(result),
|
||||
)
|
||||
_ = emitter.EmitToolCallResult(
|
||||
executeStatusBlockID,
|
||||
executeStageName,
|
||||
toolName,
|
||||
resolveToolEventResultStatus(result),
|
||||
buildToolEventResultSummary(result),
|
||||
buildToolArgumentsPreviewCN(toolCall.Arguments),
|
||||
false,
|
||||
)
|
||||
|
||||
appendToolCallResultHistory(conversationContext, toolName, toolCall.Arguments, result)
|
||||
|
||||
if registry.IsScheduleMutationTool(toolName) {
|
||||
flowState.HasScheduleWriteOps = true
|
||||
flowState.HasScheduleChanges = true
|
||||
}
|
||||
|
||||
tryWritePreviewAfterWriteTool(ctx, flowState, scheduleState, registry, toolName, writePreview)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func applyPendingContextHook(flowState *newagentmodel.CommonState) {
|
||||
if flowState == nil || flowState.PendingContextHook == nil {
|
||||
return
|
||||
}
|
||||
hook := flowState.PendingContextHook
|
||||
domain := newagenttools.NormalizeToolDomain(hook.Domain)
|
||||
if domain == "" {
|
||||
flowState.PendingContextHook = nil
|
||||
return
|
||||
}
|
||||
flowState.ActiveToolDomain = domain
|
||||
flowState.ActiveToolPacks = newagenttools.ResolveEffectiveToolPacks(domain, hook.Packs)
|
||||
flowState.PendingContextHook = nil
|
||||
}
|
||||
|
||||
func isToolVisibleForCurrentExecuteMode(
|
||||
flowState *newagentmodel.CommonState,
|
||||
registry *newagenttools.ToolRegistry,
|
||||
toolName string,
|
||||
) bool {
|
||||
if registry == nil {
|
||||
return false
|
||||
}
|
||||
activeDomain := ""
|
||||
var activePacks []string
|
||||
if flowState != nil {
|
||||
activeDomain = flowState.ActiveToolDomain
|
||||
activePacks = flowState.ActiveToolPacks
|
||||
}
|
||||
if !registry.IsToolVisibleInDomain(activeDomain, activePacks, toolName) {
|
||||
return false
|
||||
}
|
||||
if flowState != nil && flowState.ActiveOptimizeOnly && !newagenttools.IsToolAllowedInActiveOptimize(toolName) {
|
||||
return false
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func buildTemporarilyDisabledToolResult(toolName string) string {
|
||||
return fmt.Sprintf("工具 %q 当前暂时禁用。请改用 move/swap/batch_move/unplace 等排程微调工具。", strings.TrimSpace(toolName))
|
||||
}
|
||||
|
||||
func executePendingTool(
|
||||
ctx context.Context,
|
||||
runtimeState *newagentmodel.AgentRuntimeState,
|
||||
conversationContext *newagentmodel.ConversationContext,
|
||||
registry *newagenttools.ToolRegistry,
|
||||
scheduleState *schedule.ScheduleState,
|
||||
originalState *schedule.ScheduleState,
|
||||
writePreview newagentmodel.WriteSchedulePreviewFunc,
|
||||
emitter *newagentstream.ChunkEmitter,
|
||||
) error {
|
||||
pending := runtimeState.PendingConfirmTool
|
||||
if pending == nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
var args map[string]any
|
||||
if err := json.Unmarshal([]byte(pending.ArgsJSON), &args); err != nil {
|
||||
return fmt.Errorf("解析待确认工具参数失败: %w", err)
|
||||
}
|
||||
|
||||
if err := emitter.EmitToolCallStart(
|
||||
executeStatusBlockID,
|
||||
executeStageName,
|
||||
pending.ToolName,
|
||||
buildToolCallStartSummary(pending.ToolName, args),
|
||||
buildToolArgumentsPreviewCN(args),
|
||||
false,
|
||||
); err != nil {
|
||||
return fmt.Errorf("工具调用开始事件发送失败: %w", err)
|
||||
}
|
||||
|
||||
if scheduleState == nil {
|
||||
return fmt.Errorf("日程状态未加载，无法执行已确认的写工具 %s", pending.ToolName)
|
||||
}
|
||||
flowState := runtimeState.EnsureCommonState()
|
||||
if registry.IsToolTemporarilyDisabled(pending.ToolName) {
|
||||
blockedResult := buildTemporarilyDisabledToolResult(pending.ToolName)
|
||||
_ = emitter.EmitToolCallResult(
|
||||
executeStatusBlockID,
|
||||
executeStageName,
|
||||
pending.ToolName,
|
||||
"blocked",
|
||||
blockedResult,
|
||||
buildToolArgumentsPreviewCN(args),
|
||||
false,
|
||||
)
|
||||
appendToolCallResultHistory(conversationContext, pending.ToolName, args, blockedResult)
|
||||
runtimeState.PendingConfirmTool = nil
|
||||
return nil
|
||||
}
|
||||
|
||||
if shouldForceFeasibilityNegotiation(flowState, registry, pending.ToolName) {
|
||||
blockedResult := buildInfeasibleBlockedResult(flowState)
|
||||
_ = emitter.EmitToolCallResult(
|
||||
executeStatusBlockID,
|
||||
executeStageName,
|
||||
pending.ToolName,
|
||||
"blocked",
|
||||
blockedResult,
|
||||
buildToolArgumentsPreviewCN(args),
|
||||
false,
|
||||
)
|
||||
appendToolCallResultHistory(conversationContext, pending.ToolName, args, blockedResult)
|
||||
runtimeState.PendingConfirmTool = nil
|
||||
return nil
|
||||
}
|
||||
|
||||
beforeDigest := summarizeScheduleStateForDebug(scheduleState)
|
||||
if !registry.RequiresScheduleState(pending.ToolName) {
|
||||
if args == nil {
|
||||
args = make(map[string]any)
|
||||
}
|
||||
args["_user_id"] = flowState.UserID
|
||||
}
|
||||
result := registry.Execute(scheduleState, pending.ToolName, args)
|
||||
updateHealthSnapshotV2(flowState, pending.ToolName, result)
|
||||
updateTaskClassUpsertSnapshot(flowState, pending.ToolName, result)
|
||||
updateActiveToolDomainSnapshot(flowState, pending.ToolName, result)
|
||||
afterDigest := summarizeScheduleStateForDebug(scheduleState)
|
||||
log.Printf(
|
||||
"[DEBUG] execute pending tool chat=%s round=%d tool=%s args=%s before=%s after=%s result_preview=%.200s",
|
||||
flowState.ConversationID,
|
||||
flowState.RoundUsed,
|
||||
pending.ToolName,
|
||||
marshalArgsForDebug(args),
|
||||
beforeDigest,
|
||||
afterDigest,
|
||||
flattenForLog(result),
|
||||
)
|
||||
_ = emitter.EmitToolCallResult(
|
||||
executeStatusBlockID,
|
||||
executeStageName,
|
||||
pending.ToolName,
|
||||
resolveToolEventResultStatus(result),
|
||||
buildToolEventResultSummary(result),
|
||||
buildToolArgumentsPreviewCN(args),
|
||||
false,
|
||||
)
|
||||
|
||||
appendToolCallResultHistory(conversationContext, pending.ToolName, args, result)
|
||||
|
||||
if registry.IsScheduleMutationTool(pending.ToolName) {
|
||||
flowState.HasScheduleWriteOps = true
|
||||
flowState.HasScheduleChanges = true
|
||||
}
|
||||
|
||||
tryWritePreviewAfterWriteTool(ctx, flowState, scheduleState, registry, pending.ToolName, writePreview)
|
||||
|
||||
runtimeState.PendingConfirmTool = nil
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func tryWritePreviewAfterWriteTool(
|
||||
ctx context.Context,
|
||||
flowState *newagentmodel.CommonState,
|
||||
scheduleState *schedule.ScheduleState,
|
||||
registry *newagenttools.ToolRegistry,
|
||||
toolName string,
|
||||
writePreview newagentmodel.WriteSchedulePreviewFunc,
|
||||
) {
|
||||
if flowState == nil || scheduleState == nil || registry == nil || writePreview == nil {
|
||||
return
|
||||
}
|
||||
if !registry.IsScheduleMutationTool(toolName) {
|
||||
return
|
||||
}
|
||||
|
||||
if err := writePreview(ctx, scheduleState, flowState.UserID, flowState.ConversationID, flowState.TaskClassIDs); err != nil {
|
||||
log.Printf(
|
||||
"[WARN] execute realtime preview write failed chat=%s tool=%s err=%v",
|
||||
flowState.ConversationID,
|
||||
toolName,
|
||||
err,
|
||||
)
|
||||
return
|
||||
}
|
||||
|
||||
log.Printf(
|
||||
"[DEBUG] execute realtime preview write success chat=%s tool=%s",
|
||||
flowState.ConversationID,
|
||||
toolName,
|
||||
)
|
||||
}
|
||||
|
||||
var listItemRe = regexp.MustCompile(`([^\n])([2-9][\.、]\s)`)
|
||||
|
||||
func normalizeSpeak(speak string) string {
|
||||
speak = strings.TrimSpace(speak)
|
||||
if speak == "" {
|
||||
return speak
|
||||
}
|
||||
if !strings.Contains(speak, "\n") {
|
||||
speak = listItemRe.ReplaceAllString(speak, "$1\n$2")
|
||||
}
|
||||
return speak + "\n"
|
||||
}
|
||||
|
||||
func buildExecuteNormalizedSpeakTail(streamed, normalized string) string {
|
||||
streamed = strings.ReplaceAll(streamed, "\r\n", "\n")
|
||||
normalized = strings.ReplaceAll(normalized, "\r\n", "\n")
|
||||
if streamed == "" || normalized == "" {
|
||||
return ""
|
||||
}
|
||||
if !strings.HasPrefix(normalized, streamed) {
|
||||
return ""
|
||||
}
|
||||
return normalized[len(streamed):]
|
||||
}
|
||||
420
backend/newAgent/node/execute/tool_view.go
Normal file
420
backend/newAgent/node/execute/tool_view.go
Normal file
@@ -0,0 +1,420 @@
|
||||
package newagentexecute
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"github.com/LoveLosita/smartflow/backend/newAgent/tools/schedule"
|
||||
)
|
||||
|
||||
func summarizeScheduleStateForDebug(state *schedule.ScheduleState) string {
|
||||
if state == nil {
|
||||
return "state=nil"
|
||||
}
|
||||
|
||||
total := len(state.Tasks)
|
||||
pendingNoSlot := 0
|
||||
suggestedTotal := 0
|
||||
existingTotal := 0
|
||||
taskItemWithSlot := 0
|
||||
eventWithSlot := 0
|
||||
|
||||
for i := range state.Tasks {
|
||||
t := &state.Tasks[i]
|
||||
hasSlot := len(t.Slots) > 0
|
||||
|
||||
switch {
|
||||
case schedule.IsPendingTask(*t):
|
||||
pendingNoSlot++
|
||||
case schedule.IsSuggestedTask(*t):
|
||||
suggestedTotal++
|
||||
case schedule.IsExistingTask(*t):
|
||||
existingTotal++
|
||||
}
|
||||
|
||||
if hasSlot {
|
||||
if t.Source == "task_item" {
|
||||
taskItemWithSlot++
|
||||
}
|
||||
if t.Source == "event" {
|
||||
eventWithSlot++
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return fmt.Sprintf(
|
||||
"tasks=%d pending=%d suggested=%d existing=%d task_item_with_slot=%d event_with_slot=%d",
|
||||
total,
|
||||
pendingNoSlot,
|
||||
suggestedTotal,
|
||||
existingTotal,
|
||||
taskItemWithSlot,
|
||||
eventWithSlot,
|
||||
)
|
||||
}
|
||||
|
||||
func marshalArgsForDebug(args map[string]any) string {
|
||||
if len(args) == 0 {
|
||||
return "{}"
|
||||
}
|
||||
raw, err := json.Marshal(args)
|
||||
if err != nil {
|
||||
return "<marshal_error>"
|
||||
}
|
||||
return string(raw)
|
||||
}
|
||||
|
||||
func flattenForLog(text string) string {
|
||||
text = strings.ReplaceAll(text, "\n", " ")
|
||||
text = strings.ReplaceAll(text, "\r", " ")
|
||||
return strings.TrimSpace(text)
|
||||
}
|
||||
|
||||
func resolveToolEventResultStatus(result string) string {
|
||||
normalized := strings.TrimSpace(result)
|
||||
if normalized == "" {
|
||||
return "done"
|
||||
}
|
||||
if strings.Contains(normalized, "失败") {
|
||||
return "failed"
|
||||
}
|
||||
lower := strings.ToLower(normalized)
|
||||
if strings.Contains(lower, "error") || strings.Contains(lower, "failed") {
|
||||
return "failed"
|
||||
}
|
||||
return "done"
|
||||
}
|
||||
|
||||
func buildToolEventResultSummary(result string) string {
|
||||
flat := flattenForLog(result)
|
||||
if flat == "" {
|
||||
return "工具已执行完成。"
|
||||
}
|
||||
|
||||
if summary, ok := tryExtractToolResultSummaryCN(flat); ok {
|
||||
return summary
|
||||
}
|
||||
|
||||
runes := []rune(flat)
|
||||
if len(runes) <= 48 {
|
||||
return flat
|
||||
}
|
||||
return string(runes[:48]) + "..."
|
||||
}
|
||||
|
||||
func tryExtractToolResultSummaryCN(raw string) (string, bool) {
|
||||
trimmed := strings.TrimSpace(raw)
|
||||
if trimmed == "" {
|
||||
return "", false
|
||||
}
|
||||
|
||||
var payload map[string]any
|
||||
if err := json.Unmarshal([]byte(trimmed), &payload); err != nil {
|
||||
return "", false
|
||||
}
|
||||
|
||||
toolRaw := strings.TrimSpace(readStringAnyFromMap(payload, "tool"))
|
||||
toolName := resolveToolDisplayNameCN(toolRaw)
|
||||
|
||||
if strings.EqualFold(toolRaw, "upsert_task_class") {
|
||||
if summary, ok := buildUpsertTaskClassSummaryCN(payload); ok {
|
||||
return truncateToolSummaryCN(summary), true
|
||||
}
|
||||
}
|
||||
|
||||
if errText := strings.TrimSpace(readStringAnyFromMap(payload, "error", "err")); errText != "" {
|
||||
return truncateToolSummaryCN(fmt.Sprintf("%s失败：%s", toolName, errText)), true
|
||||
}
|
||||
|
||||
if success, exists := payload["success"]; exists {
|
||||
if ok, isBool := success.(bool); isBool && !ok {
|
||||
reason := strings.TrimSpace(readStringAnyFromMap(payload, "reason", "message"))
|
||||
if reason != "" {
|
||||
return truncateToolSummaryCN(fmt.Sprintf("%s失败：%s", toolName, reason)), true
|
||||
}
|
||||
return truncateToolSummaryCN(fmt.Sprintf("%s执行失败。", toolName)), true
|
||||
}
|
||||
}
|
||||
|
||||
if message := strings.TrimSpace(readStringAnyFromMap(payload, "result", "message", "reason")); message != "" {
|
||||
return truncateToolSummaryCN(message), true
|
||||
}
|
||||
|
||||
pending, hasPending := readIntAnyFromMap(payload, "pending_count")
|
||||
completed, hasCompleted := readIntAnyFromMap(payload, "completed_count")
|
||||
if hasPending || hasCompleted {
|
||||
skipped, _ := readIntAnyFromMap(payload, "skipped_count")
|
||||
return fmt.Sprintf("队列状态：待处理 %d，已完成 %d，已跳过 %d。", pending, completed, skipped), true
|
||||
}
|
||||
|
||||
if hasHead, exists := payload["has_head"]; exists {
|
||||
if b, isBool := hasHead.(bool); isBool {
|
||||
if b {
|
||||
return "已获取当前队首任务。", true
|
||||
}
|
||||
return "当前队列没有可处理任务。", true
|
||||
}
|
||||
}
|
||||
|
||||
if _, ok := payload["slot_candidates"]; ok {
|
||||
if total, exists := readIntAnyFromMap(payload, "total"); exists {
|
||||
return fmt.Sprintf("共找到 %d 个可用时段。", total), true
|
||||
}
|
||||
}
|
||||
|
||||
if toolRaw != "" {
|
||||
return fmt.Sprintf("已完成“%s”操作。", toolName), true
|
||||
}
|
||||
|
||||
return "", false
|
||||
}
|
||||
|
||||
func buildUpsertTaskClassSummaryCN(payload map[string]any) (string, bool) {
|
||||
validationRaw, hasValidation := payload["validation"]
|
||||
if !hasValidation {
|
||||
return "", false
|
||||
}
|
||||
validation, ok := validationRaw.(map[string]any)
|
||||
if !ok {
|
||||
return "", false
|
||||
}
|
||||
|
||||
validationOK, hasValidationOK := validation["ok"].(bool)
|
||||
issues := parseAnyToStringSlice(validation["issues"])
|
||||
|
||||
if hasValidationOK && !validationOK {
|
||||
if len(issues) > 0 {
|
||||
return fmt.Sprintf("任务类写入未通过校验：%s。", strings.Join(issues, "；")), true
|
||||
}
|
||||
return "任务类写入未通过校验，请先补齐缺失字段。", true
|
||||
}
|
||||
|
||||
success, hasSuccess := payload["success"].(bool)
|
||||
if hasSuccess && success {
|
||||
if taskClassID, ok := readIntAnyFromMap(payload, "task_class_id"); ok && taskClassID > 0 {
|
||||
return fmt.Sprintf("任务类写入成功，task_class_id=%d。", taskClassID), true
|
||||
}
|
||||
return "任务类写入成功。", true
|
||||
}
|
||||
|
||||
return "", false
|
||||
}
|
||||
|
||||
func truncateToolSummaryCN(text string) string {
|
||||
runes := []rune(strings.TrimSpace(text))
|
||||
if len(runes) <= 48 {
|
||||
return string(runes)
|
||||
}
|
||||
return string(runes[:48]) + "..."
|
||||
}
|
||||
|
||||
func buildToolCallStartSummary(toolName string, args map[string]any) string {
|
||||
displayName := resolveToolDisplayNameCN(toolName)
|
||||
argSummary := buildToolArgumentsPreviewCN(args)
|
||||
if argSummary == "" {
|
||||
return fmt.Sprintf("已调用工具：%s。", displayName)
|
||||
}
|
||||
return fmt.Sprintf("已调用工具：%s（%s）。", displayName, argSummary)
|
||||
}
|
||||
|
||||
func buildToolArgumentsPreviewCN(args map[string]any) string {
|
||||
if len(args) <= 0 {
|
||||
return ""
|
||||
}
|
||||
|
||||
type argPair struct {
|
||||
Key string
|
||||
Label string
|
||||
}
|
||||
|
||||
orderedPairs := []argPair{
|
||||
{Key: "title", Label: "任务标题"},
|
||||
{Key: "task_name", Label: "任务名称"},
|
||||
{Key: "deadline_at", Label: "截止时间"},
|
||||
{Key: "new_day", Label: "目标日期"},
|
||||
{Key: "new_slot_start", Label: "目标开始时段"},
|
||||
{Key: "day", Label: "日期"},
|
||||
{Key: "day_start", Label: "开始日"},
|
||||
{Key: "day_end", Label: "结束日"},
|
||||
{Key: "day_scope", Label: "日期范围"},
|
||||
{Key: "day_of_week", Label: "星期"},
|
||||
{Key: "week", Label: "周"},
|
||||
{Key: "week_from", Label: "起始周"},
|
||||
{Key: "week_to", Label: "结束周"},
|
||||
{Key: "week_filter", Label: "周筛选"},
|
||||
{Key: "slot_start", Label: "开始时段"},
|
||||
{Key: "slot_end", Label: "结束时段"},
|
||||
{Key: "slot_type", Label: "时段类型"},
|
||||
{Key: "slot_types", Label: "时段类型"},
|
||||
{Key: "task_id", Label: "任务 ID"},
|
||||
{Key: "task_ids", Label: "任务 ID 列表"},
|
||||
{Key: "task_item_id", Label: "任务项 ID"},
|
||||
{Key: "task_item_ids", Label: "任务项 ID 列表"},
|
||||
{Key: "query", Label: "查询词"},
|
||||
{Key: "keyword", Label: "关键词"},
|
||||
{Key: "domain", Label: "工具域"},
|
||||
{Key: "mode", Label: "激活模式"},
|
||||
{Key: "all", Label: "移除全部"},
|
||||
{Key: "top_k", Label: "返回数量"},
|
||||
{Key: "url", Label: "链接"},
|
||||
{Key: "reason", Label: "原因"},
|
||||
{Key: "limit", Label: "数量"},
|
||||
}
|
||||
|
||||
items := make([]string, 0, 2)
|
||||
for _, pair := range orderedPairs {
|
||||
rawValue, exists := args[pair.Key]
|
||||
if !exists {
|
||||
continue
|
||||
}
|
||||
valueText := formatToolArgValueByKeyCN(pair.Key, rawValue)
|
||||
if valueText == "" {
|
||||
continue
|
||||
}
|
||||
items = append(items, fmt.Sprintf("%s：%s", pair.Label, valueText))
|
||||
if len(items) >= 2 {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
return strings.Join(items, "，")
|
||||
}
|
||||
|
||||
func resolveToolDisplayNameCN(toolName string) string {
|
||||
name := strings.TrimSpace(toolName)
|
||||
if name == "" {
|
||||
return "未知工具"
|
||||
}
|
||||
|
||||
displayNameMap := map[string]string{
|
||||
"get_overview": "查看总览",
|
||||
"query_range": "查询时间范围",
|
||||
"queue_status": "查看任务队列",
|
||||
"queue_pop_head": "获取队首任务",
|
||||
"queue_apply_head_move": "应用队首任务时段",
|
||||
"queue_skip_head": "跳过队首任务",
|
||||
"query_target_tasks": "查询目标任务",
|
||||
"query_available_slots": "查询可用时段",
|
||||
"get_task_info": "查看任务信息",
|
||||
"analyze_health": "综合体检",
|
||||
"analyze_rhythm": "分析学习节律",
|
||||
"web_search": "网页搜索",
|
||||
"web_fetch": "网页抓取",
|
||||
"move": "移动任务",
|
||||
"place": "放置任务",
|
||||
"swap": "交换任务",
|
||||
"batch_move": "批量移动任务",
|
||||
"unplace": "移出任务安排",
|
||||
"upsert_task_class": "写入任务类",
|
||||
"context_tools_add": "激活工具域",
|
||||
"context_tools_remove": "移除工具域",
|
||||
}
|
||||
|
||||
if label, ok := displayNameMap[name]; ok {
|
||||
return label
|
||||
}
|
||||
return name
|
||||
}
|
||||
|
||||
func formatToolArgValueByKeyCN(key string, value any) string {
|
||||
switch key {
|
||||
case "day_scope":
|
||||
scope := strings.ToLower(strings.TrimSpace(formatToolArgValueCN(value)))
|
||||
switch scope {
|
||||
case "workday":
|
||||
return "工作日"
|
||||
case "weekend":
|
||||
return "周末"
|
||||
case "all":
|
||||
return "全部日期"
|
||||
default:
|
||||
return scope
|
||||
}
|
||||
case "day_of_week":
|
||||
weekdays := parseAnyToIntSlice(value)
|
||||
if len(weekdays) <= 0 {
|
||||
return formatToolArgValueCN(value)
|
||||
}
|
||||
labels := make([]string, 0, len(weekdays))
|
||||
for _, day := range weekdays {
|
||||
labels = append(labels, fmt.Sprintf("周%d", day))
|
||||
if len(labels) >= 4 {
|
||||
break
|
||||
}
|
||||
}
|
||||
return strings.Join(labels, "、")
|
||||
case "task_ids", "task_item_ids", "week_filter":
|
||||
values := parseAnyToIntSlice(value)
|
||||
if len(values) <= 0 {
|
||||
return formatToolArgValueCN(value)
|
||||
}
|
||||
items := make([]string, 0, len(values))
|
||||
for _, current := range values {
|
||||
items = append(items, strconv.Itoa(current))
|
||||
if len(items) >= 4 {
|
||||
break
|
||||
}
|
||||
}
|
||||
return strings.Join(items, "、")
|
||||
case "url":
|
||||
return truncateToolSummaryCN(formatToolArgValueCN(value))
|
||||
case "reason", "title", "task_name", "query", "keyword":
|
||||
return truncateToolSummaryCN(formatToolArgValueCN(value))
|
||||
default:
|
||||
return formatToolArgValueCN(value)
|
||||
}
|
||||
}
|
||||
|
||||
func formatToolArgValueCN(value any) string {
|
||||
switch v := value.(type) {
|
||||
case string:
|
||||
text := strings.TrimSpace(v)
|
||||
if text == "" {
|
||||
return ""
|
||||
}
|
||||
return text
|
||||
case int:
|
||||
return strconv.Itoa(v)
|
||||
case int8:
|
||||
return strconv.Itoa(int(v))
|
||||
case int16:
|
||||
return strconv.Itoa(int(v))
|
||||
case int32:
|
||||
return strconv.Itoa(int(v))
|
||||
case int64:
|
||||
return strconv.Itoa(int(v))
|
||||
case float32:
|
||||
return strings.TrimSpace(strconv.FormatFloat(float64(v), 'f', -1, 32))
|
||||
case float64:
|
||||
return strings.TrimSpace(strconv.FormatFloat(v, 'f', -1, 64))
|
||||
case bool:
|
||||
if v {
|
||||
return "是"
|
||||
}
|
||||
return "否"
|
||||
case []any:
|
||||
values := make([]string, 0, len(v))
|
||||
for _, item := range v {
|
||||
text := formatToolArgValueCN(item)
|
||||
if text == "" {
|
||||
continue
|
||||
}
|
||||
values = append(values, text)
|
||||
if len(values) >= 3 {
|
||||
break
|
||||
}
|
||||
}
|
||||
return strings.Join(values, "、")
|
||||
default:
|
||||
if value == nil {
|
||||
return ""
|
||||
}
|
||||
text := strings.TrimSpace(fmt.Sprintf("%v", value))
|
||||
if text == "" || text == "<nil>" || text == "map[]" {
|
||||
return ""
|
||||
}
|
||||
return text
|
||||
}
|
||||
}
|
||||
21
backend/newAgent/node/speak_text.go
Normal file
21
backend/newAgent/node/speak_text.go
Normal file
@@ -0,0 +1,21 @@
|
||||
package newagentnode
|
||||
|
||||
import (
|
||||
"regexp"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// listItemRe 匹配被粘连在一起的列表序号，用于正文归一化时自动补换行。
|
||||
var listItemRe = regexp.MustCompile(`([^\n])([2-9][\.、]\s)`)
|
||||
|
||||
// normalizeSpeak 统一整理要展示给用户的正文。
|
||||
func normalizeSpeak(speak string) string {
|
||||
speak = strings.TrimSpace(speak)
|
||||
if speak == "" {
|
||||
return speak
|
||||
}
|
||||
if !strings.Contains(speak, "\n") {
|
||||
speak = listItemRe.ReplaceAllString(speak, "$1\n$2")
|
||||
}
|
||||
return speak + "\n"
|
||||
}
|
||||
117
backend/newAgent/shared/node_correction.go
Normal file
117
backend/newAgent/shared/node_correction.go
Normal file
@@ -0,0 +1,117 @@
|
||||
package newagentshared
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
newagentmodel "github.com/LoveLosita/smartflow/backend/newAgent/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 *newagentmodel.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 *newagentmodel.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 *newagentmodel.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,
|
||||
})
|
||||
}
|
||||
121
backend/newAgent/shared/node_llm_debug.go
Normal file
121
backend/newAgent/shared/node_llm_debug.go
Normal file
@@ -0,0 +1,121 @@
|
||||
package newagentshared
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"strings"
|
||||
|
||||
newagentmodel "github.com/LoveLosita/smartflow/backend/newAgent/model"
|
||||
"github.com/cloudwego/eino/schema"
|
||||
)
|
||||
|
||||
// LogNodeLLMContext 将某个节点即将送入 LLM 的完整消息上下文按统一格式打印到日志。
|
||||
//
|
||||
// 步骤化说明：
|
||||
// 1. 统一输出 stage / phase / chat / round，方便按一次请求内的多次 LLM 调用串联排查；
|
||||
// 2. 完整展开 messages，不做截断，保证问题复现时能直接对照 prompt 组装结果；
|
||||
// 3. 该函数只负责调试日志，不参与任何业务判断，也不修改上下文内容。
|
||||
func LogNodeLLMContext(
|
||||
stage string,
|
||||
phase string,
|
||||
flowState *newagentmodel.CommonState,
|
||||
messages []*schema.Message,
|
||||
) {
|
||||
chatID := ""
|
||||
roundUsed := 0
|
||||
if flowState != nil {
|
||||
chatID = flowState.ConversationID
|
||||
roundUsed = flowState.RoundUsed
|
||||
}
|
||||
|
||||
log.Printf(
|
||||
"[DEBUG] %s LLM context begin phase=%s chat=%s round=%d message_count=%d\n%s\n[DEBUG] %s LLM context end phase=%s chat=%s round=%d",
|
||||
stage,
|
||||
strings.TrimSpace(phase),
|
||||
chatID,
|
||||
roundUsed,
|
||||
len(messages),
|
||||
formatLLMMessagesForDebug(messages),
|
||||
stage,
|
||||
strings.TrimSpace(phase),
|
||||
chatID,
|
||||
roundUsed,
|
||||
)
|
||||
}
|
||||
|
||||
// formatLLMMessagesForDebug 将本轮送入 LLM 的完整消息上下文展开成可读多行日志。
|
||||
//
|
||||
// 说明：
|
||||
// 1. 按消息索引逐条输出，便于和上游上下文构造步骤逐项对齐；
|
||||
// 2. 完整输出 content / reasoning_content / tool_calls / extra，不做截断；
|
||||
// 3. 仅用于调试打点，不参与业务决策。
|
||||
func formatLLMMessagesForDebug(messages []*schema.Message) string {
|
||||
if len(messages) == 0 {
|
||||
return "(empty messages)"
|
||||
}
|
||||
|
||||
var sb strings.Builder
|
||||
for i, msg := range messages {
|
||||
sb.WriteString(fmt.Sprintf("----- message[%d] -----\n", i))
|
||||
if msg == nil {
|
||||
sb.WriteString("role: <nil>\n\n")
|
||||
continue
|
||||
}
|
||||
|
||||
sb.WriteString(fmt.Sprintf("role: %s\n", msg.Role))
|
||||
|
||||
if strings.TrimSpace(msg.ToolCallID) != "" {
|
||||
sb.WriteString(fmt.Sprintf("tool_call_id: %s\n", msg.ToolCallID))
|
||||
}
|
||||
if strings.TrimSpace(msg.ToolName) != "" {
|
||||
sb.WriteString(fmt.Sprintf("tool_name: %s\n", msg.ToolName))
|
||||
}
|
||||
|
||||
if len(msg.ToolCalls) > 0 {
|
||||
sb.WriteString("tool_calls:\n")
|
||||
for j, call := range msg.ToolCalls {
|
||||
sb.WriteString(fmt.Sprintf(" - [%d] id=%s type=%s function=%s\n", j, call.ID, call.Type, call.Function.Name))
|
||||
sb.WriteString(" arguments:\n")
|
||||
sb.WriteString(indentMultilineForDebug(call.Function.Arguments, " "))
|
||||
sb.WriteString("\n")
|
||||
}
|
||||
}
|
||||
|
||||
if strings.TrimSpace(msg.ReasoningContent) != "" {
|
||||
sb.WriteString("reasoning_content:\n")
|
||||
sb.WriteString(indentMultilineForDebug(msg.ReasoningContent, " "))
|
||||
sb.WriteString("\n")
|
||||
}
|
||||
|
||||
sb.WriteString("content:\n")
|
||||
sb.WriteString(indentMultilineForDebug(msg.Content, " "))
|
||||
sb.WriteString("\n")
|
||||
|
||||
if len(msg.Extra) > 0 {
|
||||
sb.WriteString("extra:\n")
|
||||
raw, err := json.MarshalIndent(msg.Extra, "", " ")
|
||||
if err != nil {
|
||||
sb.WriteString(indentMultilineForDebug("<marshal_error>", " "))
|
||||
} else {
|
||||
sb.WriteString(indentMultilineForDebug(string(raw), " "))
|
||||
}
|
||||
sb.WriteString("\n")
|
||||
}
|
||||
|
||||
sb.WriteString("\n")
|
||||
}
|
||||
return sb.String()
|
||||
}
|
||||
|
||||
// indentMultilineForDebug 为多行文本统一添加前缀缩进，避免日志折行后难以阅读。
|
||||
func indentMultilineForDebug(text, prefix string) string {
|
||||
if text == "" {
|
||||
return prefix + "<empty>"
|
||||
}
|
||||
lines := strings.Split(text, "\n")
|
||||
for i := range lines {
|
||||
lines[i] = prefix + lines[i]
|
||||
}
|
||||
return strings.Join(lines, "\n")
|
||||
}
|
||||
10
backend/newAgent/shared/node_thinking.go
Normal file
10
backend/newAgent/shared/node_thinking.go
Normal file
@@ -0,0 +1,10 @@
|
||||
package newagentshared
|
||||
|
||||
import infrallm "github.com/LoveLosita/smartflow/backend/infra/llm"
|
||||
|
||||
func ResolveThinkingMode(enabled bool) infrallm.ThinkingMode {
|
||||
if enabled {
|
||||
return infrallm.ThinkingModeEnabled
|
||||
}
|
||||
return infrallm.ThinkingModeDisabled
|
||||
}
|
||||
290
backend/newAgent/shared/node_unified_compact.go
Normal file
290
backend/newAgent/shared/node_unified_compact.go
Normal file
@@ -0,0 +1,290 @@
|
||||
package newagentshared
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
|
||||
infrallm "github.com/LoveLosita/smartflow/backend/infra/llm"
|
||||
newagentmodel "github.com/LoveLosita/smartflow/backend/newAgent/model"
|
||||
newagentprompt "github.com/LoveLosita/smartflow/backend/newAgent/prompt"
|
||||
newagentstream "github.com/LoveLosita/smartflow/backend/newAgent/stream"
|
||||
"github.com/LoveLosita/smartflow/backend/pkg"
|
||||
"github.com/cloudwego/eino/schema"
|
||||
)
|
||||
|
||||
// UnifiedCompactInput 是统一压缩入口的参数。
|
||||
//
|
||||
// 设计说明：
|
||||
// 1. 从各节点输入中提取压缩所需的公共字段，消除对具体节点实现的直接依赖；
|
||||
// 2. 各节点（Plan/Chat/Deliver/Execute）构造此参数时，只需填充自己已有的运行时能力；
|
||||
// 3. StageName 和 StatusBlockID 用于区分日志来源与 SSE 状态推送目标。
|
||||
type UnifiedCompactInput struct {
|
||||
// Client 用于调用 LLM 压缩 msg1/msg2。
|
||||
Client *infrallm.Client
|
||||
// CompactionStore 用于持久化压缩摘要和 token 统计，为 nil 时跳过持久化。
|
||||
CompactionStore newagentmodel.CompactionStore
|
||||
// FlowState 提供 userID / conversationID / roundUsed 等定位信息。
|
||||
FlowState *newagentmodel.CommonState
|
||||
// Emitter 用于推送压缩进度 SSE 事件。
|
||||
Emitter *newagentstream.ChunkEmitter
|
||||
// StageName 标识当前阶段，如 execute / plan / chat / deliver。
|
||||
StageName string
|
||||
// StatusBlockID 是 SSE 状态推送的 block ID，各节点使用自己的 block ID。
|
||||
StatusBlockID string
|
||||
}
|
||||
|
||||
// CompactUnifiedMessagesIfNeeded 检查统一消息结构的 token 预算，
|
||||
// 超限时对 msg1（历史对话）和 msg2（阶段工作区）执行 LLM 压缩。
|
||||
//
|
||||
// 消息布局约定（由统一消息构造器返回）：
|
||||
// [0] system - msg0: 系统规则 + 工具简表
|
||||
// [1] assistant - msg1: 历史对话上下文
|
||||
// [2] assistant - msg2: 阶段工作区（Execute=ReAct Loop，其余通常为“暂无”）
|
||||
// [3] system - msg3: 阶段状态 + 记忆 + 指令
|
||||
//
|
||||
// 压缩策略：
|
||||
// 1. msg1 超过可用预算一半时触发 LLM 压缩（合并已有摘要 + 新内容）；
|
||||
// 2. msg1 压缩后仍超限，则对 msg2 也做 LLM 压缩；
|
||||
// 3. 压缩结果持久化到 CompactionStore，下一轮可复用摘要避免重复计算。
|
||||
func CompactUnifiedMessagesIfNeeded(
|
||||
ctx context.Context,
|
||||
messages []*schema.Message,
|
||||
input UnifiedCompactInput,
|
||||
) []*schema.Message {
|
||||
if input.FlowState == nil {
|
||||
log.Printf("[COMPACT:%s] FlowState is nil, skip token stats refresh", input.StageName)
|
||||
return messages
|
||||
}
|
||||
|
||||
// 1. 非严格 4 段式时，退化成按角色汇总的统计，确保 context_token_stats 仍能刷新。
|
||||
if len(messages) != 4 {
|
||||
breakdown := estimateFallbackStageTokenBreakdown(messages)
|
||||
log.Printf(
|
||||
"[COMPACT:%s] fallback token stats refresh: total=%d budget=%d count=%d (msg0=%d msg1=%d msg2=%d msg3=%d)",
|
||||
input.StageName, breakdown.Total, breakdown.Budget, len(messages),
|
||||
breakdown.Msg0, breakdown.Msg1, breakdown.Msg2, breakdown.Msg3,
|
||||
)
|
||||
saveUnifiedTokenStats(ctx, input, breakdown)
|
||||
return messages
|
||||
}
|
||||
|
||||
// 2. 提取四条消息的文本内容，供预算检查与后续压缩使用。
|
||||
msg0 := messages[0].Content
|
||||
msg1 := messages[1].Content
|
||||
msg2 := messages[2].Content
|
||||
msg3 := messages[3].Content
|
||||
|
||||
// 3. 执行 token 预算检查，判断是否需要压缩历史对话或阶段工作区。
|
||||
breakdown, overBudget, needCompactMsg1, needCompactMsg2 := pkg.CheckStageTokenBudget(msg0, msg1, msg2, msg3)
|
||||
|
||||
log.Printf(
|
||||
"[COMPACT:%s] token budget check: total=%d budget=%d over=%v compactMsg1=%v compactMsg2=%v (msg0=%d msg1=%d msg2=%d msg3=%d)",
|
||||
input.StageName, breakdown.Total, breakdown.Budget, overBudget, needCompactMsg1, needCompactMsg2,
|
||||
breakdown.Msg0, breakdown.Msg1, breakdown.Msg2, breakdown.Msg3,
|
||||
)
|
||||
|
||||
if !overBudget {
|
||||
// 4. 未超限时仅记录 token 分布，不做压缩。
|
||||
saveUnifiedTokenStats(ctx, input, breakdown)
|
||||
return messages
|
||||
}
|
||||
|
||||
// 5. 先压缩 msg1（历史对话），它通常是最主要的 token 消耗来源。
|
||||
if needCompactMsg1 {
|
||||
msg1 = compactUnifiedMsg1(ctx, input, msg1)
|
||||
messages[1].Content = msg1
|
||||
breakdown = pkg.EstimateStageMessagesTokens(msg0, msg1, msg2, msg3)
|
||||
}
|
||||
|
||||
// 6. 若 msg1 压缩后仍超限，再压缩 msg2（阶段工作区 / ReAct 记录）。
|
||||
if needCompactMsg2 || breakdown.Total > pkg.StageTokenBudget {
|
||||
msg2 = compactUnifiedMsg2(ctx, input, msg2)
|
||||
messages[2].Content = msg2
|
||||
breakdown = pkg.EstimateStageMessagesTokens(msg0, msg1, msg2, msg3)
|
||||
}
|
||||
|
||||
// 7. 记录最终 token 分布，供后续调试与监控使用。
|
||||
saveUnifiedTokenStats(ctx, input, breakdown)
|
||||
|
||||
log.Printf(
|
||||
"[COMPACT:%s] after compaction: total=%d budget=%d (msg0=%d msg1=%d msg2=%d msg3=%d)",
|
||||
input.StageName, breakdown.Total, breakdown.Budget,
|
||||
breakdown.Msg0, breakdown.Msg1, breakdown.Msg2, breakdown.Msg3,
|
||||
)
|
||||
return messages
|
||||
}
|
||||
|
||||
// estimateFallbackStageTokenBreakdown 在非统一 4 段式场景下按消息角色做近似统计。
|
||||
//
|
||||
// 步骤说明：
|
||||
// 1. 先按消息类型汇总 token，保证总量准确；
|
||||
// 2. 再把最后一个 user 消息尽量视作 msg3，保留阶段指令语义；
|
||||
// 3. 其他历史内容归入 msg1 / msg2，确保上下文统计不会因为结构不标准而断更。
|
||||
func estimateFallbackStageTokenBreakdown(messages []*schema.Message) pkg.StageTokenBreakdown {
|
||||
breakdown := pkg.StageTokenBreakdown{Budget: pkg.StageTokenBudget}
|
||||
if len(messages) == 0 {
|
||||
return breakdown
|
||||
}
|
||||
|
||||
lastUserIndex := -1
|
||||
for i := len(messages) - 1; i >= 0; i-- {
|
||||
msg := messages[i]
|
||||
if msg == nil {
|
||||
continue
|
||||
}
|
||||
if msg.Role == schema.User {
|
||||
lastUserIndex = i
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
for i, msg := range messages {
|
||||
if msg == nil {
|
||||
continue
|
||||
}
|
||||
tokens := pkg.EstimateMessageTokens(msg)
|
||||
breakdown.Total += tokens
|
||||
|
||||
switch msg.Role {
|
||||
case schema.System:
|
||||
breakdown.Msg0 += tokens
|
||||
case schema.User:
|
||||
if i == lastUserIndex {
|
||||
breakdown.Msg3 += tokens
|
||||
} else {
|
||||
breakdown.Msg1 += tokens
|
||||
}
|
||||
case schema.Tool:
|
||||
breakdown.Msg2 += tokens
|
||||
case schema.Assistant:
|
||||
if len(msg.ToolCalls) > 0 {
|
||||
breakdown.Msg2 += tokens
|
||||
} else {
|
||||
breakdown.Msg1 += tokens
|
||||
}
|
||||
default:
|
||||
breakdown.Msg1 += tokens
|
||||
}
|
||||
}
|
||||
|
||||
return breakdown
|
||||
}
|
||||
|
||||
// compactUnifiedMsg1 对 msg1（历史对话）执行 LLM 压缩。
|
||||
//
|
||||
// 步骤化说明：
|
||||
// 1. CompactionStore 为 nil 时跳过（测试环境 / 骨架期）；
|
||||
// 2. 先加载该阶段已有的压缩摘要，与当前 msg1 合并后调 LLM 压缩；
|
||||
// 3. 压缩失败时降级为原始文本，不中断主流程；
|
||||
// 4. 压缩成功后持久化新摘要，供下一轮复用。
|
||||
func compactUnifiedMsg1(
|
||||
ctx context.Context,
|
||||
input UnifiedCompactInput,
|
||||
msg1 string,
|
||||
) string {
|
||||
if input.CompactionStore == nil {
|
||||
log.Printf("[COMPACT:%s] CompactionStore is nil, skip msg1 compaction", input.StageName)
|
||||
return msg1
|
||||
}
|
||||
|
||||
existingSummary, _, err := input.CompactionStore.LoadStageCompaction(ctx, input.FlowState.UserID, input.FlowState.ConversationID, input.StageName)
|
||||
if err != nil {
|
||||
log.Printf("[COMPACT:%s] load existing compaction failed: %v, proceed without cache", input.StageName, err)
|
||||
}
|
||||
|
||||
tokenBefore := pkg.EstimateTextTokens(msg1)
|
||||
_ = input.Emitter.EmitStatus(
|
||||
input.StatusBlockID, input.StageName, "context_compact_start",
|
||||
fmt.Sprintf("正在压缩对话历史（%d tokens）...", tokenBefore),
|
||||
false,
|
||||
)
|
||||
|
||||
newSummary, err := newagentprompt.CompactMsg1(ctx, input.Client, msg1, existingSummary)
|
||||
if err != nil {
|
||||
log.Printf("[COMPACT:%s] compact msg1 failed: %v", input.StageName, err)
|
||||
_ = input.Emitter.EmitStatus(
|
||||
input.StatusBlockID, input.StageName, "context_compact_done",
|
||||
"对话历史压缩失败，使用原始文本",
|
||||
false,
|
||||
)
|
||||
return msg1
|
||||
}
|
||||
|
||||
tokenAfter := pkg.EstimateTextTokens(newSummary)
|
||||
_ = input.Emitter.EmitStatus(
|
||||
input.StatusBlockID, input.StageName, "context_compact_done",
|
||||
fmt.Sprintf("对话历史已压缩：%d → %d tokens", tokenBefore, tokenAfter),
|
||||
false,
|
||||
)
|
||||
|
||||
if err := input.CompactionStore.SaveStageCompaction(ctx, input.FlowState.UserID, input.FlowState.ConversationID, input.StageName, newSummary, input.FlowState.RoundUsed); err != nil {
|
||||
log.Printf("[COMPACT:%s] save compaction failed: %v", input.StageName, err)
|
||||
}
|
||||
|
||||
return newSummary
|
||||
}
|
||||
|
||||
// compactUnifiedMsg2 对 msg2（阶段工作区）执行 LLM 压缩。
|
||||
//
|
||||
// 步骤化说明：
|
||||
// 1. 非 Execute 阶段的 msg2 通常内容较少，压缩即使收益有限也不应出错；
|
||||
// 2. Execute 阶段的 msg2 包含 ReAct loop 记录，压缩可显著节省 token；
|
||||
// 3. 压缩失败时降级为原始文本，不中断主流程。
|
||||
func compactUnifiedMsg2(
|
||||
ctx context.Context,
|
||||
input UnifiedCompactInput,
|
||||
msg2 string,
|
||||
) string {
|
||||
tokenBefore := pkg.EstimateTextTokens(msg2)
|
||||
_ = input.Emitter.EmitStatus(
|
||||
input.StatusBlockID, input.StageName, "context_compact_start",
|
||||
fmt.Sprintf("正在压缩执行记录（%d tokens）...", tokenBefore),
|
||||
false,
|
||||
)
|
||||
|
||||
compressed, err := newagentprompt.CompactMsg2(ctx, input.Client, msg2)
|
||||
if err != nil {
|
||||
log.Printf("[COMPACT:%s] compact msg2 failed: %v", input.StageName, err)
|
||||
_ = input.Emitter.EmitStatus(
|
||||
input.StatusBlockID, input.StageName, "context_compact_done",
|
||||
"执行记录压缩失败，使用原始文本",
|
||||
false,
|
||||
)
|
||||
return msg2
|
||||
}
|
||||
|
||||
tokenAfter := pkg.EstimateTextTokens(compressed)
|
||||
_ = input.Emitter.EmitStatus(
|
||||
input.StatusBlockID, input.StageName, "context_compact_done",
|
||||
fmt.Sprintf("执行记录已压缩：%d → %d tokens", tokenBefore, tokenAfter),
|
||||
false,
|
||||
)
|
||||
|
||||
return compressed
|
||||
}
|
||||
|
||||
// saveUnifiedTokenStats 持久化当前 token 分布到存储层。
|
||||
//
|
||||
// 步骤化说明：
|
||||
// 1. CompactionStore 为 nil 时跳过（测试环境 / 骨架期）；
|
||||
// 2. 序列化失败只记日志，不中断主流程；
|
||||
// 3. 写入失败只记日志，不中断主流程。
|
||||
func saveUnifiedTokenStats(
|
||||
ctx context.Context,
|
||||
input UnifiedCompactInput,
|
||||
breakdown pkg.StageTokenBreakdown,
|
||||
) {
|
||||
if input.CompactionStore == nil || input.FlowState == nil {
|
||||
return
|
||||
}
|
||||
statsJSON, err := json.Marshal(breakdown)
|
||||
if err != nil {
|
||||
log.Printf("[COMPACT:%s] marshal token stats failed: %v", input.StageName, err)
|
||||
return
|
||||
}
|
||||
if err := input.CompactionStore.SaveContextTokenStats(ctx, input.FlowState.UserID, input.FlowState.ConversationID, string(statsJSON)); err != nil {
|
||||
log.Printf("[COMPACT:%s] save token stats failed: %v", input.StageName, err)
|
||||
}
|
||||
}
|
||||
37
backend/newAgent/shared/node_visible_message.go
Normal file
37
backend/newAgent/shared/node_visible_message.go
Normal file
@@ -0,0 +1,37 @@
|
||||
package newagentshared
|
||||
|
||||
import (
|
||||
"context"
|
||||
"log"
|
||||
"strings"
|
||||
|
||||
newagentmodel "github.com/LoveLosita/smartflow/backend/newAgent/model"
|
||||
"github.com/cloudwego/eino/schema"
|
||||
)
|
||||
|
||||
// PersistVisibleAssistantMessage 负责把“真正要展示给用户”的 assistant 文本交给 service 层持久化。
|
||||
//
|
||||
// 职责边界：
|
||||
// 1. 只处理可见的 assistant 消息，不处理内部纠错提示、工具调用结果和纯状态文案；
|
||||
// 2. 持久化失败只记日志，不反向中断节点主流程，避免“已经对外输出但后端补写失败”时把用户请求打断；
|
||||
// 3. 具体的 Redis / MySQL / 乐观缓存写入由 service 回调统一完成。
|
||||
func PersistVisibleAssistantMessage(
|
||||
ctx context.Context,
|
||||
persist newagentmodel.PersistVisibleMessageFunc,
|
||||
state *newagentmodel.CommonState,
|
||||
msg *schema.Message,
|
||||
) {
|
||||
if persist == nil || state == nil || msg == nil {
|
||||
return
|
||||
}
|
||||
|
||||
role := strings.TrimSpace(string(msg.Role))
|
||||
content := strings.TrimSpace(msg.Content)
|
||||
if role != string(schema.Assistant) || content == "" {
|
||||
return
|
||||
}
|
||||
|
||||
if err := persist(ctx, state, msg); err != nil {
|
||||
log.Printf("[WARN] persist visible assistant message failed chat=%s phase=%s err=%v", state.ConversationID, state.Phase, err)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user