correctly stop streams when creating a new chat

Author: sawka

frontend/app/aipanel/aipanel.tsx

@@ -205,7 +205,7 @@ const AIPanelComponentInner = memo(({ className, onClose }: AIPanelProps) => {
     const telemetryEnabled = jotai.useAtomValue(getSettingsKeyAtom("telemetry:enabled")) ?? false;
     const isPanelVisible = jotai.useAtomValue(WorkspaceLayoutModel.getInstance().panelVisibleAtom);
 
-    const { messages, sendMessage, status, setMessages, error } = useChat({
+    const { messages, sendMessage, status, setMessages, error, stop } = useChat({
         transport: new DefaultChatTransport({
             api: `${getWebServerEndpoint()}/api/post-chat-message`,
             prepareSendMessagesRequest: (opts) => {
@@ -232,7 +232,7 @@ const AIPanelComponentInner = memo(({ className, onClose }: AIPanelProps) => {
         },
     });
 
-    model.registerUseChatData(sendMessage, setMessages, status);
+    model.registerUseChatData(sendMessage, setMessages, status, stop);
 
     // console.log("AICHAT messages", messages);
 

frontend/app/aipanel/waveai-model.tsx

@@ -30,6 +30,7 @@ export class WaveAIModel {
     private useChatSendMessage: UseChatSendMessageType | null = null;
     private useChatSetMessages: UseChatSetMessagesType | null = null;
     private useChatStatus: ChatStatus = "ready";
+    private useChatStop: (() => void) | null = null;
     // Used for injecting Wave-specific message data into DefaultChatTransport's prepareSendMessagesRequest
     realMessage: AIMessage | null = null;
 
@@ -135,6 +136,7 @@ export class WaveAIModel {
     }
 
     clearChat() {
+        this.useChatStop?.();
         this.clearFiles();
         this.isChatEmpty = true;
         const newChatId = crypto.randomUUID();
@@ -165,10 +167,16 @@ export class WaveAIModel {
         this.scrollToBottomCallback = callback;
     }
 
-    registerUseChatData(sendMessage: UseChatSendMessageType, setMessages: UseChatSetMessagesType, status: ChatStatus) {
+    registerUseChatData(
+        sendMessage: UseChatSendMessageType,
+        setMessages: UseChatSetMessagesType,
+        status: ChatStatus,
+        stop: () => void
+    ) {
         this.useChatSendMessage = sendMessage;
         this.useChatSetMessages = setMessages;
         this.useChatStatus = status;
+        this.useChatStop = stop;
     }
 
     scrollToBottom() {