address bugbot comments

Author: icecrasher321

apps/sim/app/workspace/[workspaceId]/w/[workflowId]/components/panel/components/editor/components/sub-block/components/code/code.tsx

@@ -301,15 +301,14 @@ export const Code = memo(function Code({
   const updatePromptValue = wandHook?.updatePromptValue || (() => {})
   const cancelGeneration = wandHook?.cancelGeneration || (() => {})
 
-  const { recordChange, clearHistory, recordReplace, flushPending, startSession, undo, redo } =
-    useCodeUndoRedo({
-      blockId,
-      subBlockId,
-      value: code,
-      enabled: isFunctionCode,
-      isReadOnly: readOnly || disabled || isPreview,
-      isStreaming: isAiStreaming,
-    })
+  const { recordChange, recordReplace, flushPending, startSession, undo, redo } = useCodeUndoRedo({
+    blockId,
+    subBlockId,
+    value: code,
+    enabled: isFunctionCode,
+    isReadOnly: readOnly || disabled || isPreview,
+    isStreaming: isAiStreaming,
+  })
 
   const [storeValue, setStoreValue] = useSubBlockValue(blockId, subBlockId, false, {
     isStreaming: isAiStreaming,

apps/sim/hooks/use-code-undo-redo.ts

@@ -154,7 +154,7 @@ export function useCodeUndoRedo({
   )
 
   const flushPending = useCallback(() => {
-    if (!pendingBeforeRef.current) return
+    if (pendingBeforeRef.current === null) return
     clearTimer()
     commitPending()
   }, [clearTimer, commitPending])
@@ -201,10 +201,21 @@ export function useCodeUndoRedo({
   }, [activeWorkflowId, applyValue, blockId, isEnabled, subBlockId])
 
   useEffect(() => {
-    if (!pendingBeforeRef.current && !isApplyingRef.current) {
-      lastCommittedValueRef.current = value ?? ''
+    if (isApplyingRef.current) return
+
+    const nextValue = value ?? ''
+
+    if (pendingBeforeRef.current !== null) {
+      if (pendingAfterRef.current !== nextValue) {
+        clearTimer()
+        resetPending()
+        lastCommittedValueRef.current = nextValue
+      }
+      return
     }
-  }, [value])
+
+    lastCommittedValueRef.current = nextValue
+  }, [clearTimer, resetPending, value])
 
   useEffect(() => {
     return () => {