Method Connect

internal/multi_test.go:189–265

(ctx context.Context, conversationID string, execID string, start *proto.AgentStart, e agent.Executor, handler agent.OutputHandler)

func (m *mockPlanner) Close() error { return nil }
func (m *mockPlanner) Connect(ctx context.Context, conversationID string, execID string, start *proto.AgentStart, e agent.Executor, handler agent.OutputHandler) error {
	var lastText string
	for _, m := range start.Messages {
		if textMsg := m.GetContent().GetText(); textMsg != nil {
			lastText = textMsg.Text
		}
	}

	// Step 1: User -> Local
	if strings.HasPrefix(lastText, "Send the word") {
		inputs := append(start.Messages, &proto.Message{
			Role: "assistant",
			Content: &proto.Content{
				Type: &proto.Content_Text{
					Text: &proto.TextContent{Text: "oRanGe"},
				},
			},
		})
		_, err := e.Exec(ctx, conversationID, "local-echo", &proto.AgentStart{
			AgentId:  "local-echo-agent",
			Messages: inputs,
		}, handler)
		return err
	}

	// Step 2: Local -> Remote
	if lastText == "orange" {
		inputs := append(start.Messages, &proto.Message{
			Role: "assistant",
			Content: &proto.Content{
				Type: &proto.Content_Text{
					Text: &proto.TextContent{Text: lastText},
				},
			},
		})
		_, err := e.Exec(ctx, conversationID, "remote-text", &proto.AgentStart{
			AgentId:  "remote-text-processor",
			Messages: inputs,
		}, handler)
		return err
	}

	// Step 3: Remote -> Uppercase
	if strings.HasPrefix(lastText, "Remote Prefix:") {
		inputs := append(start.Messages, &proto.Message{
			Role: "assistant",
			Content: &proto.Content{
				Type: &proto.Content_Text{
					Text: &proto.TextContent{Text: lastText},
				},
			},
		})
		_, err := e.Exec(ctx, conversationID, "uppercase-task", &proto.AgentStart{
			AgentId:  "uppercase",
			Messages: inputs,
		}, handler)
		return err
	}