Function Example

internal/controller/executor/executor_test.go:38–66

()

}

func Example() {
	ctx := context.Background()

	registry := map[string]agent.Agent{
		"planner": AgentFunc(func(inputs []*proto.Message, tm agent.Executor, o agent.OutputHandler) {
			if _, err := tm.Exec(ctx, "test-conv", "deep-research-task", &proto.AgentStart{
				AgentId:  "deep-research",
				Messages: inputs,
			}, o); err != nil {
				return
			}

			if _, err := tm.Exec(ctx, "test-conv", "pub-med-lookup-task", &proto.AgentStart{
				AgentId:  "pub-med-index",
				Messages: inputs,
			}, o); err != nil {
				return
			}
		}),
	}

	tm := DefaultExecutor(memoryEventLog(), registry)
	if _, err := tm.Exec(ctx, "test-conv", "test", &proto.AgentStart{
		AgentId:  "planner",
		Messages: []*proto.Message{text("user", "Hello, I'd like to research cancer treatment options.")},
	}, nil); err != nil {
		log.Fatal(err)
	}
}

func TestTaskManager(t *testing.T) {
	ctx := context.Background()