@lobehub/lobehub 2.0.0-next.101 → 2.0.0-next.102

package/src/store/chat/slices/aiChat/actions/__tests__/conversationLifecycle.test.ts

@@ -2,6 +2,7 @@ import { act, renderHook } from '@testing-library/react';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
 import { aiChatService } from '@/services/aiChat';
+import { getSessionStoreState } from '@/store/session';
 
 import { useChatStore } from '../../../../store';
 import { TEST_CONTENT, TEST_IDS, createMockMessage } from './fixtures';
@@ -30,6 +31,8 @@ beforeEach(() => {
   resetTestEnvironment();
   setupMockSelectors();
   spyOnMessageService();
+  const sessionStore = getSessionStoreState();
+  vi.spyOn(sessionStore, 'triggerSessionUpdate').mockResolvedValue(undefined);
 
   act(() => {
     useChatStore.setState({

package/src/store/chat/slices/aiChat/actions/streamingExecutor.ts

@@ -530,6 +530,17 @@ export const streamingExecutor: StateCreator<
               get().completeOperation(reasoningOperationId);
               reasoningOperationId = undefined;
             }
+            break;
+          }
+
+          case 'stop': {
+            // Complete reasoning operation when receiving stop signal
+            if (!duration && reasoningOperationId) {
+              duration = Date.now() - thinkingStartAt;
+              get().completeOperation(reasoningOperationId);
+              reasoningOperationId = undefined;
+            }
+            break;
           }
         }
       },