Function TestTransferTaskAllowsSubAgent

pkg/runtime/runtime_test.go:2298–2333

(t *testing.T)

}

func TestTransferTaskAllowsSubAgent(t *testing.T) {
	t.Parallel()

	// Verify that transfer_task to a valid sub-agent is NOT rejected by the validation.
	// We can't fully run the child session without a real model, so we just confirm
	// it gets past validation (it will fail later due to mock stream being empty,
	// which is fine — we only care that it's not blocked by the sub-agent check).
	prov := &mockProvider{id: "test/mock-model", stream: newStreamBuilder().AddContent("done").AddStopWithUsage(10, 5).Build()}

	librarian := agent.New("librarian", "Library agent", agent.WithModel(prov))
	root := agent.New("root", "Root agent", agent.WithModel(prov))

	agent.WithSubAgents(librarian)(root)

	tm := team.New(team.WithAgents(root, librarian))

	rt, err := NewLocalRuntime(t.Context(), tm, WithSessionCompaction(false), WithModelStore(mockModelStore{}))
	require.NoError(t, err)

	sess := session.New(session.WithUserMessage("Test"), session.WithToolsApproved(true))
	evts := make(chan Event, 128)

	toolCall := tools.ToolCall{
		ID:   "call_1",
		Type: "function",
		Function: tools.FunctionCall{
			Name:      "transfer_task",
			Arguments: `{"agent":"librarian","task":"find a book","expected_output":"book title"}`,
		},
	}

	result, err := rt.handleTaskTransfer(t.Context(), sess, toolCall, NewChannelSink(evts))
	require.NoError(t, err)
	require.NotNil(t, result)
	assert.False(t, result.IsError, "transfer to valid sub-agent should succeed")
}

// TestTransferTaskPersistsSubSessionOnError covers the case where a sub-agent's
// run loop emits an ErrorEvent — for example because the model stream failed,