diff --git a/example/src/main/java/example/chat/StreamChatExample.java b/example/src/main/java/example/chat/StreamChatExample.java index ca44fa1..ececce2 100644 --- a/example/src/main/java/example/chat/StreamChatExample.java +++ b/example/src/main/java/example/chat/StreamChatExample.java @@ -58,7 +58,8 @@ public static void main(String[] args) { System.out.print(event.getMessage().getContent()); } if (ChatEventType.CONVERSATION_CHAT_COMPLETED.equals(event.getEvent())) { - if (MessageType.FOLLOW_UP.equals(event.getMessage().getType())) { + if (event.getMessage() != null + && MessageType.FOLLOW_UP.equals(event.getMessage().getType())) { System.out.println(event.getMessage().getContent()); } else { System.out.println("Token usage:" + event.getChat().getUsage().getTokenCount());