Add callbacks for direct callers

Author: AdityaHegde

web-common/src/features/chat/core/conversation.ts

@@ -121,7 +121,10 @@ export class Conversation {
    * @param context - Chat context to be sent with the message
    * @param options - Callback functions for different stages of message sending
    */
-  public async sendMessage(context: RuntimeServiceCompleteBody): Promise<void> {
+  public async sendMessage(
+    context: RuntimeServiceCompleteBody,
+    options?: { onStreamStart?: () => void },
+  ): Promise<void> {
     // Prevent concurrent message sending
     if (get(this.isStreaming)) {
       this.streamError.set("Please wait for the current response to complete");
@@ -140,7 +143,8 @@ export class Conversation {
     const userMessage = this.addOptimisticUserMessage(prompt);
 
     try {
-      this.events.emit("stream-start");
+      options?.onStreamStart?.(); // Callback for direct callers
+      this.events.emit("stream-start"); // Event for external listeners
       // Start streaming - this establishes the connection
       const streamPromise = this.startStreaming(prompt, context);
 

web-common/src/features/chat/core/input/ChatInput.svelte

@@ -27,14 +27,6 @@
   $: draftMessageStore = currentConversation.draftMessage;
   $: isStreamingStore = currentConversation.isStreaming;
 
-  let streamStartUnsub: (() => void) | undefined = undefined;
-  $: {
-    streamStartUnsub?.();
-    streamStartUnsub = currentConversation.on("stream-start", () => {
-      editor.commands.setContent("");
-    });
-  }
-
   $: value = $draftMessageStore;
   $: disabled = $getConversationQuery?.isLoading || $isStreamingStore;
   $: canSend = !disabled && value.trim();
@@ -48,7 +40,9 @@
 
     // Message handling with input focus
     try {
-      await currentConversation.sendMessage($additionalContextStore);
+      await currentConversation.sendMessage($additionalContextStore, {
+        onStreamStart: () => editor.commands.setContent(""),
+      });
       onSend?.();
     } catch (error) {
       console.error("Failed to send message:", error);