@@ -251,6 +251,8 @@ public async Task SendMessageSse([FromRoute] string agentId,
251251 await conv . TruncateConversation ( conversationId , input . TruncateMessageId , inputMsg . MessageId ) ;
252252 }
253253
254+ var state = _services . GetRequiredService < IConversationStateService > ( ) ;
255+
254256 var routing = _services . GetRequiredService < IRoutingService > ( ) ;
255257 routing . Context . SetMessageId ( conversationId , inputMsg . MessageId ) ;
256258
@@ -278,6 +280,7 @@ await conv.SendMessage(agentId, inputMsg,
278280 response . RichContent = msg . SecondaryRichContent ?? msg . RichContent ;
279281 response . Instruction = msg . Instruction ;
280282 response . Data = msg . Data ;
283+ response . States = state . GetStates ( ) ;
281284
282285 await OnChunkReceived ( Response , response ) ;
283286 } ,
@@ -290,6 +293,7 @@ await conv.SendMessage(agentId, inputMsg,
290293 MessageId = msg . MessageId ,
291294 Text = msg . Indication ,
292295 Function = "indicating" ,
296+ States = new Dictionary < string , string > ( )
293297 } ;
294298 await OnChunkReceived ( Response , indicator ) ;
295299 } ,
@@ -299,7 +303,6 @@ await conv.SendMessage(agentId, inputMsg,
299303
300304 } ) ;
301305
302- var state = _services . GetRequiredService < IConversationStateService > ( ) ;
303306 response . States = state . GetStates ( ) ;
304307 response . MessageId = inputMsg . MessageId ;
305308 response . ConversationId = conversationId ;
@@ -309,7 +312,10 @@ await conv.SendMessage(agentId, inputMsg,
309312
310313 private async Task OnChunkReceived ( HttpResponse response , ChatResponseModel message )
311314 {
312- var json = JsonSerializer . Serialize ( message ) ;
315+ var json = JsonSerializer . Serialize ( message , new JsonSerializerOptions
316+ {
317+ PropertyNamingPolicy = JsonNamingPolicy . CamelCase ,
318+ } ) ;
313319
314320 var buffer = Encoding . UTF8 . GetBytes ( $ "data:{ json } \n ") ;
315321 await response . Body . WriteAsync ( buffer , 0 , buffer . Length ) ;
0 commit comments