Fixes

sourcegraph · Apr 4, 2024 · a912047 · a912047
1 parent 6a5d236
commit a912047
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/main/kotlin/com/sourcegraph/cody/history/HistoryService.kt b/src/main/kotlin/com/sourcegraph/cody/history/HistoryService.kt
@@ -37,9 +37,17 @@ class HistoryService(private val project: Project) :
     val found = getOrCreateChat(internalId)
     if (found.messages.size < chatMessages.size) {
       found.setUpdatedTimeAt(LocalDateTime.now())
-      found.messages = chatMessages.map(::convertToMessageState).toMutableList()
-      synchronized(listeners) { listeners.forEach { it(found) } }
     }
+
+    chatMessages.map(::convertToMessageState).forEachIndexed { index, messageState ->
+      val messageToUpdate = found.messages.getOrNull(index)
+      if (messageToUpdate != null) {
+        found.messages[index] = messageState
+      } else {
+        found.messages.add(messageState)
+      }
+    }
+    synchronized(listeners) { listeners.forEach { it(found) } }
   }
 
   @Synchronized