fix: inline chat controller timing

Author: Ricbet

packages/ai-native/src/browser/contrib/problem-fix/problem-fix.handler.ts

@@ -107,11 +107,12 @@ export class ProblemFixHandler extends IAIMonacoContribHandler {
     const model = monacoEditor.getModel();
 
     // 以 marker 的 range 为中心，向上取 2 行，向下取 3 行
+    const endLineNumber = Math.min(part.range.endLineNumber + 3, model!.getLineCount());
     const editRange = new Range(
       Math.max(part.range.startLineNumber - 2, 0),
       1,
-      Math.min(part.range.endLineNumber + 3, model!.getLineCount() ?? 0),
-      model!.getLineMaxColumn(part.range.endLineNumber + 3) ?? 0,
+      endLineNumber,
+      model!.getLineMaxColumn(endLineNumber),
     );
 
     const context = {

packages/ai-native/src/browser/widget/inline-diff/inline-diff.handler.ts

@@ -177,7 +177,6 @@ export class InlineDiffHandler extends IAIMonacoContribHandler {
       previewer.onReady(() => {
         if (InlineChatController.is(chatResponse)) {
           const controller = chatResponse as InlineChatController;
-          controller.listen();
 
           disposable.addDispose([
             controller.onData((data) => {
@@ -198,6 +197,8 @@ export class InlineDiffHandler extends IAIMonacoContribHandler {
               onFinish();
             }),
           ]);
+
+          controller.listen();
         } else {
           previewer.setValue((chatResponse as ReplyResponse).message);
           onFinish();

packages/ai-native/src/browser/widget/inline-input/inline-input.handler.ts

@@ -86,8 +86,6 @@ export class InlineInputHandler extends Disposable {
       if (InlineChatController.is(previewResponse)) {
         const controller = previewResponse as InlineChatController;
 
-        controller.listen();
-
         let latestContent: string | undefined;
         const schedulerEdit: RunOnceScheduler = this.registerDispose(
           new RunOnceScheduler(() => {
@@ -122,6 +120,8 @@ export class InlineInputHandler extends Disposable {
             widget.launchChatStatus(EInlineChatStatus.DONE);
           }),
         ]);
+
+        controller.listen();
       }
     }
   }