@@ -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" )})
0 commit comments