fix missing user message issue

sdcb · Dec 23, 2024 · fc0a40d · fc0a40d
1 parent 2815586
commit fc0a40d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/BE/Controllers/Chats/Chats/ChatController.cs b/src/BE/Controllers/Chats/Chats/ChatController.cs
@@ -149,7 +149,7 @@ public async Task<IActionResult> StartConversationStreamed(
                 ParentId = req.MessageId,
             };
             db.Messages.Add(dbUserMessage);
-            existingMessages.Add(dbUserMessage.Id, MessageLiteDto.FromDB(dbUserMessage));
+            existingMessages.Add(-1, MessageLiteDto.FromDB(dbUserMessage));
         }
 
         Response.Headers.ContentType = "text/event-stream";
@@ -249,7 +249,7 @@ private static async Task<ChatSpanResponse> ProcessChatSpan(
         OpenAIChatMessage[] messageToSend = await ((MessageLiteDto[])
         [
             ..systemMessages.Where(x => x.Role == DBChatRole.System && x.SpanId == span.Id || x.SpanId == null),
-            ..GetMessageTree(existingMessages, req.MessageId),
+            ..GetMessageTree(filteredMessages, req.MessageId ?? -1),
         ])
         .ToAsyncEnumerable()
         .SelectAwait(async x => await x.ToOpenAI(fup, cancellationToken))