Implement agent busy state

Author: Yahiewi

src/agent.ts

@@ -435,6 +435,20 @@ export class AgentManager {
     return this._tokenUsageChanged;
   }
 
+  /**
+   * Signal emitted when the agent's busy state changes (generating response or executing tools)
+   */
+  get busyChanged(): ISignal<this, boolean> {
+    return this._busyChanged;
+  }
+
+  /**
+   * Whether the agent is currently generating a response or executing tools
+   */
+  get busy(): boolean {
+    return this._busy;
+  }
+
   /**
    * Refresh the skills snapshot and rebuild the agent if resources are ready.
    */
@@ -592,6 +606,9 @@ export class AgentManager {
    * @param message The user message to respond to (may include processed attachment content)
    */
   async generateResponse(message: string): Promise<void> {
+    // Abort any ongoing streaming immediately
+    this.stopStreaming();
+    this._setBusy(true);
     this._controller = new AbortController();
 
     try {
@@ -659,6 +676,7 @@ export class AgentManager {
       }
     } finally {
       this._controller = null;
+      this._setBusy(false);
     }
   }
 
@@ -1191,6 +1209,16 @@ WEB RETRIEVAL POLICY:
     return `Supported MIME types in this session: ${safeMimeTypes.join(', ')}`;
   }
 
+  /**
+   * Set busy state and emit signal
+   */
+  private _setBusy(busy: boolean): void {
+    if (this._busy !== busy) {
+      this._busy = busy;
+      this._busyChanged.emit(busy);
+    }
+  }
+
   // Private attributes
   private _settingsModel: AISettingsModel;
   private _toolRegistry?: IToolRegistry;
@@ -1201,6 +1229,8 @@ WEB RETRIEVAL POLICY:
   private _agent: ToolLoopAgent<never, ToolMap> | null;
   private _history: ModelMessage[];
   private _mcpTools: ToolMap;
+  private _busy: boolean = false;
+  private _busyChanged = new Signal<this, boolean>(this);
   private _controller: AbortController | null;
   private _agentEvent: Signal<this, IAgentEvent>;
   private _tokenUsage: ITokenUsage;

src/chat-model.ts

@@ -109,11 +109,24 @@ export class AIChatModel extends AbstractChatModel {
     this._agentManager = options.agentManager;
     this._trans = options.trans;
 
-    // Listen for agent events
+    // Listen for agent events and busy state
     this._agentManager.agentEvent.connect(this._onAgentEvent, this);
+    this._agentManager.busyChanged.connect((_, busy) => {
+      this._busyChanged.emit(busy);
+    });
 
     // Listen for settings changes to update chat behavior
     this._settingsModel.stateChanged.connect(this._onSettingsChanged, this);
+
+    // Prevent clearing input field when agent is busy.
+    const originalSend = this.input.send;
+    this.input.send = (content: string) => {
+      if (this._agentManager.busy) {
+        return;
+      }
+      originalSend(content);
+    };
+
     this.setReady();
   }
 
@@ -156,6 +169,20 @@ export class AIChatModel extends AbstractChatModel {
     return this._agentManager;
   }
 
+  /**
+   * Whether the agent is currently busy generating a response or executing tools.
+   */
+  get busy(): boolean {
+    return this._agentManager.busy;
+  }
+
+  /**
+   * A signal emitted when the busy state changes.
+   */
+  get busyChanged(): ISignal<this, boolean> {
+    return this._busyChanged;
+  }
+
   /**
    * Creates a chat context for the current conversation.
    */
@@ -214,6 +241,11 @@ export class AIChatModel extends AbstractChatModel {
       return;
     }
 
+    // Prevent sending multiple messages concurrently
+    if (this._agentManager.busy) {
+      return;
+    }
+
     // Add user message to chat
     const userMessage: IMessageContent = {
       body: message.body,
@@ -916,6 +948,7 @@ export class AIChatModel extends AbstractChatModel {
   }
 
   // Private fields
+  private _busyChanged = new Signal<this, boolean>(this);
   private _settingsModel: AISettingsModel;
   private _user: IUser;
   private _toolContexts: Map<string, IToolExecutionContext> = new Map();

src/widgets/main-area-chat.ts

@@ -64,6 +64,12 @@ export class MainAreaChat extends MainAreaWidget<ChatWidget> {
     });
 
     this.model.writersChanged.connect(this._writersChanged);
+    // Toggle busy class based on agent state
+    this.model.busyChanged.connect((_, busy) => {
+      this.content.toggleClass('jp-ai-busy', busy);
+    });
+    // Set initial state
+    this.content.toggleClass('jp-ai-busy', this.model.busy);
   }
 
   dispose(): void {

style/base.css

@@ -344,3 +344,17 @@
 .jp-ai-completion-status .jp-ai-completion-disabled path {
   fill: var(--jp-layout-color2);
 }
+
+/**
+ * Busy state styling for locking the chat input.
+ */
+.jp-ai-busy .jp-chat-input-container {
+  pointer-events: none;
+  opacity: 0.6;
+  filter: grayscale(0.5);
+  cursor: not-allowed;
+}
+
+.jp-ai-busy .jp-chat-input-textfield {
+  cursor: not-allowed;
+}