Skip to content

Commit 3ce8587

Browse files
fix(adk): preserve agentic role when copying events (#1067)
1 parent 72796c1 commit 3ce8587

2 files changed

Lines changed: 6 additions & 3 deletions

File tree

adk/agentic_test.go

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1206,7 +1206,8 @@ func TestCoverage_CopyAgenticEvent(t *testing.T) {
12061206
RunPath: []RunStep{{agentName: "root"}, {agentName: "agent1"}},
12071207
Output: &TypedAgentOutput[*schema.AgenticMessage]{
12081208
MessageOutput: &TypedMessageVariant[*schema.AgenticMessage]{
1209-
Message: agenticMsg("hello"),
1209+
Message: agenticMsg("hello"),
1210+
AgenticRole: schema.AgenticRoleTypeAssistant,
12101211
},
12111212
},
12121213
Action: &AgentAction{
@@ -1218,6 +1219,7 @@ func TestCoverage_CopyAgenticEvent(t *testing.T) {
12181219
assert.Equal(t, original.AgentName, copied.AgentName)
12191220
assert.Equal(t, len(original.RunPath), len(copied.RunPath))
12201221
assert.Equal(t, original.Action, copied.Action)
1222+
assert.Equal(t, schema.AgenticRoleTypeAssistant, copied.Output.MessageOutput.AgenticRole)
12211223

12221224
copied.RunPath[0].agentName = "mutated"
12231225
assert.NotEqual(t, original.RunPath[0].agentName, copied.RunPath[0].agentName)
@@ -1506,7 +1508,7 @@ func TestAgenticRetryWithShouldRetry_Stream(t *testing.T) {
15061508
require.NoError(t, err)
15071509

15081510
runner := NewTypedRunner(TypedRunnerConfig[*schema.AgenticMessage]{
1509-
Agent: agent,
1511+
Agent: agent,
15101512
EnableStreaming: true,
15111513
})
15121514
iter := runner.Run(ctx, []*schema.AgenticMessage{schema.UserAgenticMessage("hello")})
@@ -1639,7 +1641,7 @@ func TestAgenticFailoverStream_MidStreamError(t *testing.T) {
16391641
require.NoError(t, err)
16401642

16411643
runner := NewTypedRunner(TypedRunnerConfig[*schema.AgenticMessage]{
1642-
Agent: agent,
1644+
Agent: agent,
16431645
EnableStreaming: true,
16441646
})
16451647
iter := runner.Run(ctx, []*schema.AgenticMessage{schema.UserAgenticMessage("hello")})

adk/utils.go

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -261,6 +261,7 @@ func copyTypedAgentEvent[M MessageType](ae *TypedAgentEvent[M]) *TypedAgentEvent
261261
copied.Output.MessageOutput = &TypedMessageVariant[M]{
262262
IsStreaming: mv.IsStreaming,
263263
Role: mv.Role,
264+
AgenticRole: mv.AgenticRole,
264265
ToolName: mv.ToolName,
265266
}
266267
if mv.IsStreaming {

0 commit comments

Comments
 (0)