Fix bugs

Author: rugvedS07

src/subcommands/chat/react/inputReducer.test.ts

@@ -1501,6 +1501,23 @@ describe("chatInputStateReducers", () => {
       expect(result.cursorOnSegmentIndex).toBe(0);
       expect(result.cursorInSegmentOffset).toBe((firstSegmentText + secondSegmentText).length);
     });
+
+    it("jumps to end of largePaste in one move when cursor is mid-word before largePaste", () => {
+      const initialState = createChatUserInputState(
+        [
+          { type: "text", content: "ddf" },
+          { type: "largePaste", content: "x".repeat(1495) },
+          { type: "text", content: "Helh" },
+        ],
+        0,
+        "dd".length,
+      );
+
+      const result = moveCursorWordRight(initialState);
+
+      expect(result.cursorOnSegmentIndex).toBe(1);
+      expect(result.cursorInSegmentOffset).toBe(0);
+    });
   });
 
   describe("moveCursorToLineStart", () => {

src/subcommands/chat/react/inputReducer.ts

@@ -833,7 +833,11 @@ export function moveCursorWordRight(state: ChatUserInputState): ChatUserInputSta
         // Find next word boundary within current segment and move cursor there
         const newCursorOffset = findNextWordBoundaryInSegment(segmentContent, cursorOffset);
         draft.cursorInSegmentOffset = newCursorOffset;
-        return;
+        // If we reached the end of the segment, continue to jump to next segment
+        if (newCursorOffset < segmentLength) {
+          return;
+        }
+        // Fall through to jump to next segment
       }
     }