Skip to content

Commit 7a3fd1f

Browse files
committed
refactor(skill): use GetMessage method to handle both streaming and non-streaming messages
1 parent d7bd738 commit 7a3fd1f

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

adk/middlewares/skill/skill.go

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -391,14 +391,15 @@ func (s *skillTool) runAgentMode(ctx context.Context, skill Skill, forkHistory b
391391
if !ok {
392392
break
393393
}
394-
if event == nil {
394+
if event == nil || event.Output == nil || event.Output.MessageOutput == nil {
395395
continue
396396
}
397-
if event.Output != nil && event.Output.MessageOutput != nil {
398-
msgOutput := event.Output.MessageOutput
399-
if !msgOutput.IsStreaming && msgOutput.Message.Content != "" {
400-
results = append(results, msgOutput.Message.Content)
401-
}
397+
msg, err := event.Output.MessageOutput.GetMessage()
398+
if err != nil {
399+
return "", fmt.Errorf("failed to get message from event: %w", err)
400+
}
401+
if msg != nil && msg.Content != "" {
402+
results = append(results, msg.Content)
402403
}
403404
}
404405

0 commit comments

Comments
 (0)