@@ -82,8 +82,8 @@ private async Task HandleWebSocket(IServiceProvider services, string agentId, st
             var (eventType, data) = MapEvents(conn, receivedText);
             if (eventType == "start")
             {
-                var states = InitStates(data);
-                await ConnectToModel(hub, webSocket, states);
+                var request = InitRequest(data);
+                await ConnectToModel(hub, webSocket, request?.States);
             }
             else if (eventType == "media")
             {
@@ -157,16 +157,15 @@ private void InitEvents(RealtimeHubConnection conn)
             });
     }
 
-    private List<MessageState> InitStates(string data)
+    private ChatStreamRequest? InitRequest(string data)
     {
         try
         {
-            var states = JsonSerializer.Deserialize<List<MessageState>>(data, BotSharpOptions.defaultJsonOptions);
-            return states ?? [];
+            return JsonSerializer.Deserialize<ChatStreamRequest>(data, BotSharpOptions.defaultJsonOptions);
         }
         catch
         {
-            return [];
+            return null;
         }
     }
 }
@@ -0,0 +1,10 @@
+using BotSharp.Abstraction.Models;
+using System.Text.Json.Serialization;
+
+namespace BotSharp.Plugin.ChatHub.Models.Stream;
+
+public class ChatStreamRequest
+{
+    [JsonPropertyName("states")]
+    public List<MessageState> States { get; set; } = [];
+}
Original file line number	Diff line number	Diff line change
`@@ -82,8 +82,8 @@ private async Task HandleWebSocket(IServiceProvider services, string agentId, st`
`82`	`82`	`var (eventType, data) = MapEvents(conn, receivedText);`
`83`	`83`	`if (eventType == "start")`
`84`	`84`	`{`
`85`		`- var states = InitStates(data);`
`86`		`- await ConnectToModel(hub, webSocket, states);`
	`85`	`+ var request = InitRequest(data);`
	`86`	`+ await ConnectToModel(hub, webSocket, request?.States);`
`87`	`87`	`}`
`88`	`88`	`else if (eventType == "media")`
`89`	`89`	`{`
`@@ -157,16 +157,15 @@ private void InitEvents(RealtimeHubConnection conn)`
`157`	`157`	`});`
`158`	`158`	`}`
`159`	`159`
`160`		`- private List<MessageState> InitStates(string data)`
	`160`	`+ private ChatStreamRequest? InitRequest(string data)`
`161`	`161`	`{`
`162`	`162`	`try`
`163`	`163`	`{`
`164`		`- var states = JsonSerializer.Deserialize<List<MessageState>>(data, BotSharpOptions.defaultJsonOptions);`
`165`		`- return states ?? [];`
	`164`	`+ return JsonSerializer.Deserialize<ChatStreamRequest>(data, BotSharpOptions.defaultJsonOptions);`
`166`	`165`	`}`
`167`	`166`	`catch`
`168`	`167`	`{`
`169`		`- return [];`
	`168`	`+ return null;`
`170`	`169`	`}`
`171`	`170`	`}`
`172`	`171`	`}`