Function TestTransferTaskRejectsNonSubAgent

pkg/runtime/runtime_test.go:2259–2296

(t *testing.T)

}

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

	// root has librarian as sub-agent but NOT planner.
	// planner exists in the team. transfer_task to planner should be rejected.
	prov := &mockProvider{id: "test/mock-model", stream: &mockStream{}}

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

	agent.WithSubAgents(librarian)(root)

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

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

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

	toolCall := tools.ToolCall{
		ID:   "call_1",
		Type: "function",
		Function: tools.FunctionCall{
			Name:      "transfer_task",
			Arguments: `{"agent":"planner","task":"do something","expected_output":""}`,
		},
	}

	result, err := rt.handleTaskTransfer(t.Context(), sess, toolCall, NewChannelSink(evts))
	require.NoError(t, err)
	require.NotNil(t, result)
	assert.True(t, result.IsError, "transfer to non-sub-agent should return an error result")
	assert.Contains(t, result.Output, "cannot transfer task to planner")
	assert.Contains(t, result.Output, "librarian")
	assert.Equal(t, "root", rt.CurrentAgentName(t.Context()), "current agent should remain root")
}

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