feat: adapt action behavior

toeverything · Mar 28, 2024 · 957ed72 · 957ed72
1 parent 6089c7e
commit 957ed72
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/packages/backend/server/src/plugins/copilot/session.ts b/packages/backend/server/src/plugins/copilot/session.ts
@@ -58,6 +58,10 @@ export class ChatSession implements AsyncDisposable {
   }
 
   private takeMessages(): ChatMessage[] {
+    if (this.state.action) {
+      const messages = this.state.messages;
+      return messages.slice(messages.length - 1);
+    }
     const ret = [];
     const messages = [...this.state.messages];
     messages.reverse();
@@ -76,7 +80,7 @@ export class ChatSession implements AsyncDisposable {
 
   finish(): ChatMessage[] {
     const messages = this.takeMessages();
-    return [...(this.state.prompt || []), ...messages];
+    return [...this.state.prompt, ...messages];
   }
 
   async save() {