Manage history message cache

devchat-ai · Dec 6, 2023 · 2dc6a78 · 2dc6a78
1 parent 4af88a2
commit 2dc6a78
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/main/kotlin/ai/devchat/devchat/handler/SendMessageRequestHandler.kt b/src/main/kotlin/ai/devchat/devchat/handler/SendMessageRequestHandler.kt
@@ -4,6 +4,7 @@ import ai.devchat.cli.DevChatResponse
 import ai.devchat.common.Log
 import ai.devchat.devchat.BaseActionHandler
 import ai.devchat.devchat.DevChatActions
+import ai.devchat.idea.storage.ActiveConversation
 import com.alibaba.fastjson.JSONObject
 import java.io.File
 import java.io.FileWriter
@@ -53,6 +54,14 @@ class SendMessageRequestHandler(metadata: JSONObject?, payload: JSONObject?) : B
             response.update(line)
             promptCallback(response)
         }
+        val currentTopic = ActiveConversation.topic ?: response.promptHash!!
+        val newMessage = wrapper.logTopic(currentTopic, 1).getJSONObject(0)
+
+        if (currentTopic == ActiveConversation.topic) {
+            ActiveConversation.addMessage(newMessage)
+        } else {
+            ActiveConversation.reset(currentTopic, listOf(newMessage))
+        }
     }
 
     override fun except(exception: Exception) {