Add files via upload

internal/multiagent/eino_turn_loop_event_bridge.go

@@ -0,0 +1,105 @@
+package multiagent
+
+import (
+	"context"
+	"strings"
+	"sync/atomic"
+
+	"github.com/cloudwego/eino/adk"
+	"github.com/cloudwego/eino/schema"
+)
+
+type einoTurnLoopEventBridge struct {
+	conversationID string
+	orchestration  string
+	progress       func(eventType, message string, data interface{})
+	gen            *adk.AsyncGenerator[*adk.AgentEvent]
+	forwardedErr   atomic.Bool
+}
+
+func newEinoTurnLoopEventBridge(
+	conversationID string,
+	orchestration string,
+	progress func(eventType, message string, data interface{}),
+	gen *adk.AsyncGenerator[*adk.AgentEvent],
+) *einoTurnLoopEventBridge {
+	return &einoTurnLoopEventBridge{
+		conversationID: conversationID,
+		orchestration:  orchestration,
+		progress:       progress,
+		gen:            gen,
+	}
+}
+
+func (b *einoTurnLoopEventBridge) OnAgentEvents(
+	_ context.Context,
+	tc *adk.TurnContext[EinoTurnLoopItem, *schema.Message],
+	events *adk.AsyncIterator[*adk.AgentEvent],
+) error {
+	for {
+		ev, ok := events.Next()
+		if !ok {
+			return nil
+		}
+		if ev == nil {
+			continue
+		}
+		if ev.Err != nil && isEinoVoluntaryCancelErr(ev.Err) {
+			// TurnLoop owns cancel routing. Returning CancelError /
+			// ErrStreamCanceled from OnAgentEvents aborts the whole loop and
+			// races with the Preempted signal becoming observable. A preempt
+			// must return nil so the queued user supplement can start the next
+			// turn; a terminal stop is surfaced through TurnLoopExitState.
+			if isEinoTurnLoopPreemptCancel(tc, ev.Err) {
+				b.emitPreempted()
+			}
+			return nil
+		}
+		if b.gen != nil {
+			b.gen.Send(ev)
+		}
+		if ev.Err != nil {
+			b.forwardedErr.Store(true)
+			return ev.Err
+		}
+	}
+}
+
+func (b *einoTurnLoopEventBridge) ForwardedError() bool {
+	if b == nil {
+		return false
+	}
+	return b.forwardedErr.Load()
+}
+
+func (b *einoTurnLoopEventBridge) emitPreempted() {
+	if b == nil || b.progress == nil {
+		return
+	}
+	b.progress("progress", "Eino TurnLoop 已在安全点切换到用户补充后的下一轮。", map[string]interface{}{
+		"conversationId": b.conversationID,
+		"source":         "eino",
+		"orchestration":  b.orchestration,
+		"kind":           "turn_loop_preempted",
+	})
+}
+
+func isEinoTurnLoopPreemptCancel(tc *adk.TurnContext[EinoTurnLoopItem, *schema.Message], err error) bool {
+	if tc == nil || !isEinoVoluntaryCancelErr(err) {
+		return false
+	}
+	select {
+	case <-tc.Preempted:
+		return true
+	default:
+		return false
+	}
+}
+
+func einoTurnLoopInterruptTimelineSummary(note string) string {
+	note = strings.TrimSpace(note)
+	if note == "" {
+		return "用户选择「中断并继续」，未填写说明；已推入 Eino TurnLoop 并等待安全点续跑。"
+	}
+	return "用户中断说明（Eino TurnLoop 原生续跑）：\n\n" + note
+}