Add files via upload

internal/multiagent/eino_main_assistant_complete_handler.go

@@ -0,0 +1,49 @@
+package multiagent
+
+import "strings"
+
+type einoMainAssistantCompleteHandler struct {
+	agentName        string
+	emitter          *einoMainResponseStreamEmitter
+	stdoutSuppressor *einoExecuteStdoutSuppressor
+	assistantOutput  *einoAssistantOutputAccumulator
+}
+
+type einoMainAssistantCompleteHandlerConfig struct {
+	AgentName        string
+	Emitter          *einoMainResponseStreamEmitter
+	StdoutSuppressor *einoExecuteStdoutSuppressor
+	AssistantOutput  *einoAssistantOutputAccumulator
+}
+
+func newEinoMainAssistantCompleteHandler(cfg einoMainAssistantCompleteHandlerConfig) *einoMainAssistantCompleteHandler {
+	return &einoMainAssistantCompleteHandler{
+		agentName:        cfg.AgentName,
+		emitter:          cfg.Emitter,
+		stdoutSuppressor: cfg.StdoutSuppressor,
+		assistantOutput:  cfg.AssistantOutput,
+	}
+}
+
+func (h *einoMainAssistantCompleteHandler) EmitComplete(content string) bool {
+	if h == nil {
+		return false
+	}
+	body := strings.TrimSpace(content)
+	if body == "" {
+		return false
+	}
+	if h.stdoutSuppressor != nil {
+		if dup := h.stdoutSuppressor.Consume(); dup != "" && body == dup {
+			if h.assistantOutput != nil {
+				h.assistantOutput.RecordMainAssistant(h.agentName, body)
+			}
+			return false
+		}
+	}
+	emitted := h.emitter.EmitDelta(body, body)
+	if h.assistantOutput != nil {
+		h.assistantOutput.RecordMainAssistant(h.agentName, body)
+	}
+	return emitted
+}