Merge pull request #744 from Flash-LHR/json-chunk

Author: mattsp1290

sdks/community/go/pkg/core/events/decoder_test.go

@@ -93,6 +93,24 @@ func TestEventDecoder(t *testing.T) {
 		assert.Equal(t, "Hello World", msgEvent.Delta)
 	})
 
+	t.Run("DecodeEvent_TextMessageChunk", func(t *testing.T) {
+		decoder := NewEventDecoder(nil)
+		data := []byte(`{"messageId": "msg-123", "role": "assistant", "delta": "Hi"}`)
+
+		event, err := decoder.DecodeEvent("TEXT_MESSAGE_CHUNK", data)
+		require.NoError(t, err)
+		require.NotNil(t, event)
+
+		msgEvent, ok := event.(*TextMessageChunkEvent)
+		require.True(t, ok)
+		require.NotNil(t, msgEvent.MessageID)
+		require.NotNil(t, msgEvent.Role)
+		require.NotNil(t, msgEvent.Delta)
+		assert.Equal(t, "msg-123", *msgEvent.MessageID)
+		assert.Equal(t, "assistant", *msgEvent.Role)
+		assert.Equal(t, "Hi", *msgEvent.Delta)
+	})
+
 	t.Run("DecodeEvent_TextMessageEnd", func(t *testing.T) {
 		decoder := NewEventDecoder(nil)
 		data := []byte(`{"messageId": "msg-123"}`)

sdks/community/go/pkg/core/events/events.go

@@ -365,6 +365,8 @@ func EventFromJSON(data []byte) (Event, error) {
 		event = &TextMessageStartEvent{}
 	case EventTypeTextMessageContent:
 		event = &TextMessageContentEvent{}
+	case EventTypeTextMessageChunk:
+		event = &TextMessageChunkEvent{}
 	case EventTypeTextMessageEnd:
 		event = &TextMessageEndEvent{}
 	case EventTypeToolCallStart:

sdks/community/go/pkg/core/events/events_test.go

@@ -477,6 +477,7 @@ func TestJSONSerialization(t *testing.T) {
 			NewRunStartedEvent("thread-1", "run-1"),
 			NewTextMessageStartEvent("msg-1", WithRole("user")),
 			NewTextMessageContentEvent("msg-1", "Hello"),
+			NewTextMessageChunkEvent(strPtr("msg-1"), strPtr("assistant"), strPtr("Chunk")),
 			NewToolCallStartEvent("tool-1", "get_weather", WithParentMessageID("msg-1")),
 			NewStateSnapshotEvent(map[string]any{"counter": 42}),
 			NewCustomEvent("test-event", WithValue("test-value")),

sdks/community/go/pkg/encoding/json/json_decoder.go

@@ -239,6 +239,13 @@ func (d *JSONDecoder) createEvent(eventType events.EventType, data []byte) (even
 			event = &e
 		}
 
+	case events.EventTypeTextMessageChunk:
+		var e events.TextMessageChunkEvent
+		err = decoder.Decode(&e)
+		if err == nil {
+			event = &e
+		}
+
 	case events.EventTypeTextMessageContent:
 		var e events.TextMessageContentEvent
 		err = decoder.Decode(&e)