Skip to content

Commit d9fd652

Browse files
committed
Fix SSE response format.
1 parent e75fd5d commit d9fd652

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

src/Infrastructure/BotSharp.OpenAPI/Controllers/ConversationController.cs

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)