Add files via upload

internal/multiagent/eino_turn_loop_runtime_test.go

@@ -0,0 +1,145 @@
+package multiagent
+
+import (
+	"context"
+	"strings"
+	"sync"
+	"testing"
+	"time"
+
+	"github.com/cloudwego/eino/adk"
+	"github.com/cloudwego/eino/components/model"
+	"github.com/cloudwego/eino/schema"
+)
+
+type turnLoopBlockingModel struct {
+	started chan struct{}
+	release chan struct{}
+
+	mu     sync.Mutex
+	inputs [][]*schema.Message
+}
+
+func newTurnLoopBlockingModel() *turnLoopBlockingModel {
+	return &turnLoopBlockingModel{
+		started: make(chan struct{}, 8),
+		release: make(chan struct{}),
+	}
+}
+
+func (m *turnLoopBlockingModel) Generate(ctx context.Context, input []*schema.Message, _ ...model.Option) (*schema.Message, error) {
+	m.mu.Lock()
+	m.inputs = append(m.inputs, cloneSchemaMessages(input))
+	callNo := len(m.inputs)
+	m.mu.Unlock()
+
+	select {
+	case m.started <- struct{}{}:
+	default:
+	}
+	if callNo == 1 {
+		select {
+		case <-ctx.Done():
+			return nil, ctx.Err()
+		case <-m.release:
+		}
+	}
+	return schema.AssistantMessage("done", nil), nil
+}
+
+func (m *turnLoopBlockingModel) Stream(ctx context.Context, input []*schema.Message, opts ...model.Option) (*schema.StreamReader[*schema.Message], error) {
+	msg, err := m.Generate(ctx, input, opts...)
+	if err != nil {
+		return nil, err
+	}
+	return schema.StreamReaderFromArray([]*schema.Message{msg}), nil
+}
+
+func (m *turnLoopBlockingModel) snapshotInputs() [][]*schema.Message {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+	out := make([][]*schema.Message, len(m.inputs))
+	for i := range m.inputs {
+		out[i] = cloneSchemaMessages(m.inputs[i])
+	}
+	return out
+}
+
+func TestEinoTurnLoopRuntimePushInterruptStartsNextTurn(t *testing.T) {
+	ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
+	defer cancel()
+
+	mockModel := newTurnLoopBlockingModel()
+	agent, err := adk.NewChatModelAgent(ctx, &adk.ChatModelAgentConfig{
+		Name:  "turn-loop-agent",
+		Model: mockModel,
+	})
+	if err != nil {
+		t.Fatalf("NewChatModelAgent: %v", err)
+	}
+
+	runtime := NewEinoTurnLoopRuntime(EinoTurnLoopRuntimeConfig{
+		Agent:            agent,
+		InitialMessages:  []*schema.Message{schema.UserMessage("initial task")},
+		InterruptTimeout: 20 * time.Millisecond,
+	})
+	runtime.Run(ctx)
+
+	select {
+	case <-mockModel.started:
+	case <-ctx.Done():
+		t.Fatal("first model call did not start")
+	}
+	if !runtime.PushInterruptContinue("focus on ssh") {
+		t.Fatal("interrupt continue push was rejected")
+	}
+	select {
+	case <-mockModel.started:
+	case <-ctx.Done():
+		t.Fatal("second model call did not start after interrupt push")
+	}
+
+	runtime.StopWhenIdle()
+	state := runtime.Wait()
+	if state == nil {
+		t.Fatal("expected turn loop exit state")
+	}
+	if state.ExitReason != nil {
+		t.Fatalf("exit reason = %v", state.ExitReason)
+	}
+
+	inputs := mockModel.snapshotInputs()
+	if len(inputs) < 2 {
+		t.Fatalf("model calls = %d, want at least 2", len(inputs))
+	}
+	if got := inputs[0][0].Content; got != "initial task" {
+		t.Fatalf("first input = %q, want initial task", got)
+	}
+	lastInput := inputs[len(inputs)-1]
+	if len(lastInput) == 0 || !strings.Contains(lastInput[len(lastInput)-1].Content, "focus on ssh") {
+		t.Fatalf("last input = %#v, want interrupt note", lastInput)
+	}
+}
+
+func TestMergeEinoTurnLoopMessagesClonesInput(t *testing.T) {
+	original := schema.UserMessage("hello")
+	msgs := mergeEinoTurnLoopMessages([]EinoTurnLoopItem{{Messages: []*schema.Message{original}}})
+	if len(msgs) != 1 || msgs[0].Content != "hello" {
+		t.Fatalf("merged = %#v", msgs)
+	}
+	msgs[0].Content = "changed"
+	if original.Content != "hello" {
+		t.Fatalf("original message was mutated: %#v", original)
+	}
+}
+
+func TestFormatInterruptContinuePrompt(t *testing.T) {
+	got := formatInterruptContinuePrompt("focus ports")
+	if !strings.Contains(got, "focus ports") || !strings.Contains(got, "不要重复") {
+		t.Fatalf("prompt = %q", got)
+	}
+	empty := formatInterruptContinuePrompt(" ")
+	if !strings.Contains(empty, "不要重复") {
+		t.Fatalf("empty prompt = %q", empty)
+	}
+}