chore: improve inline diff widget render

opensumi · Oct 16, 2024 · f26b3ac · f26b3ac
1 parent 7d79b7e
commit f26b3ac
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/packages/ai-native/src/browser/widget/inline-stream-diff/live-preview.decoration.tsx b/packages/ai-native/src/browser/widget/inline-stream-diff/live-preview.decoration.tsx
@@ -1,6 +1,14 @@
 import { Autowired, INJECTOR_TOKEN, Injectable, Injector } from '@opensumi/di';
 import { StackingLevel } from '@opensumi/ide-core-browser';
-import { ActionSourceEnum, ActionTypeEnum, Disposable, Emitter, Event, IAIReporter } from '@opensumi/ide-core-common';
+import {
+  ActionSourceEnum,
+  ActionTypeEnum,
+  Disposable,
+  Emitter,
+  Event,
+  IAIReporter,
+  runWhenIdle,
+} from '@opensumi/ide-core-common';
 import { ISingleEditOperation } from '@opensumi/ide-editor';
 import { ICodeEditor, IEditorDecorationsCollection, ITextModel, Position, Range } from '@opensumi/ide-monaco';
 import { StandaloneServices } from '@opensumi/ide-monaco/lib/browser/monaco-api/services';
@@ -668,10 +676,11 @@ export class LivePreviewDiffDecorationModel extends Disposable {
   }
 
   public touchRemovedWidget(states: IRemovedWidgetState[]) {
-    this.clearRemovedWidgets();
-
-    states.forEach(({ textLines, position }) => {
-      this.showRemovedWidgetByLineNumber(position.lineNumber, textLines, {});
+    runWhenIdle(() => {
+      this.clearRemovedWidgets();
+      states.forEach(({ textLines, position }) => {
+        this.showRemovedWidgetByLineNumber(position.lineNumber, textLines, {});
+      });
     });
   }