From 7750ae9f7a4629dfc13375f5969a01c7dc8d25f3 Mon Sep 17 00:00:00 2001
From: Nikita Hrytsai <cookies.in.chocolate@gmail.com>
Date: Thu, 13 Feb 2025 18:24:22 +0200
Subject: [PATCH] some fixes

---
 node/chat/chat.ts | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/node/chat/chat.ts b/node/chat/chat.ts
index 0cb1fd6..ecc5f4e 100644
--- a/node/chat/chat.ts
+++ b/node/chat/chat.ts
@@ -210,7 +210,7 @@ export function init({ nvim, lsp }: { nvim: Nvim; lsp: Lsp }) {
               lastUserMessage,
               counter,
             );
-            break;
+            return [model];
           }
           case "regenerate": {
             const searchEndIndex = messageIndex + 1;
@@ -224,7 +224,7 @@ export function init({ nvim, lsp }: { nvim: Nvim; lsp: Lsp }) {
 
             if (lastUserMessageIndex === -1) {
               nvim.logger?.error(
-                "Cannot regenarte message, no previous user message found",
+                "Cannot regenerate message, no previous user message found",
               );
               return [model];
             }
@@ -236,14 +236,13 @@ export function init({ nvim, lsp }: { nvim: Nvim; lsp: Lsp }) {
 
           case "edit": {
             // TODO: not implemented yet
-            break;
+            return [model];
           }
 
           default:
             assertUnreachable(action);
         }
-
-        return [model];
+        break;
       }
 
       case "conversation-state": {