fix disable AI-Q new session during shallow activity

Block the logo new-session action while shallow chat streaming or HITL is active, but keep it available during deep research jobs.

Made-with: Cursor

Author: exactlyallan

frontends/ui/src/features/layout/components/AppBar.spec.tsx

@@ -89,6 +89,14 @@ describe('AppBar', () => {
     expect(onNewSession).toHaveBeenCalledOnce()
   })
 
+  test('disables new session button when shallow navigation is blocked', () => {
+    render(<AppBar isAuthenticated={true} isNewSessionDisabled={true} />)
+
+    expect(screen.getByRole('button', { name: /create new session/i })).toBeDisabled()
+    // Other action buttons remain enabled.
+    expect(screen.getByRole('button', { name: /toggle sessions sidebar/i })).not.toBeDisabled()
+  })
+
   test('toggles sessions panel when menu button clicked', async () => {
     const user = userEvent.setup()
 

frontends/ui/src/features/layout/components/AppBar.tsx

@@ -35,6 +35,8 @@ interface AppBarProps {
   }
   /** Callback when a new session is requested */
   onNewSession?: () => void
+  /** Disable creating a new session while shallow research/HITL is active */
+  isNewSessionDisabled?: boolean
   /** Callback when sign in is clicked */
   onSignIn?: () => void
   /** Callback when sign out is clicked */
@@ -51,6 +53,7 @@ export const AppBar: FC<AppBarProps> = ({
   authRequired = false,
   user,
   onNewSession,
+  isNewSessionDisabled = false,
   onSignIn,
   onSignOut,
 }) => {
@@ -85,9 +88,9 @@ export const AppBar: FC<AppBarProps> = ({
   }, [])
 
   const handleNewSessionClick = useCallback(() => {
-    if (!isAuthenticated) return
+    if (!isAuthenticated || isNewSessionDisabled) return
     onNewSession?.()
-  }, [isAuthenticated, onNewSession])
+  }, [isAuthenticated, isNewSessionDisabled, onNewSession])
 
   const handleSignOut = useCallback(() => {
     setIsUserMenuOpen(false)
@@ -103,9 +106,13 @@ export const AppBar: FC<AppBarProps> = ({
             kind="tertiary"
             size="small"
             onClick={handleNewSessionClick}
-            disabled={!isAuthenticated}
+            disabled={!isAuthenticated || isNewSessionDisabled}
             aria-label="Create new session"
-            title="Create new session"
+            title={
+              isNewSessionDisabled
+                ? 'Cannot create new session while shallow research is active'
+                : 'Create new session'
+            }
           >
             <Flex align="center" gap="density-lg">
               <Logo kind="logo-only" size="small" />

frontends/ui/src/features/layout/components/MainLayout.spec.tsx

@@ -33,6 +33,8 @@ vi.mock('@/features/chat', () => ({
     deleteConversation: mockDeleteConversation,
     deleteAllConversations: mockDeleteAllConversations,
     updateConversationTitle: mockUpdateConversationTitle,
+    isStreaming: false,
+    pendingInteraction: null,
     isDeepResearchStreaming: false,
     deepResearchOwnerConversationId: null,
   })),
@@ -58,10 +60,18 @@ vi.mock('../store', () => ({
 
 // Mock child components
 vi.mock('./AppBar', () => ({
-  AppBar: ({ sessionTitle, onNewSession }: { sessionTitle: string; onNewSession?: () => void }) => (
+  AppBar: ({
+    sessionTitle,
+    onNewSession,
+    isNewSessionDisabled,
+  }: {
+    sessionTitle: string
+    onNewSession?: () => void
+    isNewSessionDisabled?: boolean
+  }) => (
     <>
       <div data-testid="app-bar">{sessionTitle}</div>
-      <button type="button" onClick={onNewSession}>
+      <button type="button" onClick={onNewSession} disabled={isNewSessionDisabled}>
         Header New Session
       </button>
     </>
@@ -127,6 +137,8 @@ describe('MainLayout', () => {
       deleteConversation: vi.fn(),
       deleteAllConversations: vi.fn(),
       updateConversationTitle: vi.fn(),
+      isStreaming: false,
+      pendingInteraction: null,
       isDeepResearchStreaming: false,
       deepResearchOwnerConversationId: null,
     } as unknown as ReturnType<typeof useChatStore>)
@@ -163,6 +175,26 @@ describe('MainLayout', () => {
     expect(mockCloseRightPanel).toHaveBeenCalledOnce()
   })
 
+  test('disables new session action while shallow streaming is active', () => {
+    vi.mocked(useChatStore).mockReturnValue({
+      currentConversation: { id: 'session-1', title: 'Test Session' },
+      getUserConversations: vi.fn(() => []),
+      selectConversation: vi.fn(),
+      startNewSessionDraft: vi.fn(),
+      deleteConversation: vi.fn(),
+      deleteAllConversations: vi.fn(),
+      updateConversationTitle: vi.fn(),
+      isStreaming: true,
+      pendingInteraction: null,
+      isDeepResearchStreaming: false,
+      deepResearchOwnerConversationId: null,
+    } as unknown as ReturnType<typeof useChatStore>)
+
+    render(<MainLayout />)
+
+    expect(screen.getByRole('button', { name: /header new session/i })).toBeDisabled()
+  })
+
   test('adjusts chat width when details panel is open', () => {
     vi.mocked(useLayoutStore).mockReturnValue({
       rightPanel: 'research',

frontends/ui/src/features/layout/components/MainLayout.tsx

@@ -68,6 +68,8 @@ export const MainLayout: FC<MainLayoutProps> = ({
     deleteConversation,
     deleteAllConversations,
     updateConversationTitle,
+    isStreaming,
+    pendingInteraction,
     isDeepResearchStreaming,
     deepResearchOwnerConversationId,
   } = useChatStore()
@@ -117,6 +119,7 @@ export const MainLayout: FC<MainLayoutProps> = ({
 
   // Check if research panel is open (pushes content instead of overlaying)
   const isResearchPanelOpen = rightPanel === 'research'
+  const isNavigationBlocked = isStreaming || pendingInteraction !== null
 
   // Get only conversations for the current authenticated user
   const userConversations = getUserConversations()
@@ -142,6 +145,7 @@ export const MainLayout: FC<MainLayoutProps> = ({
         authRequired={authRequired}
         user={user}
         onNewSession={handleNewSession}
+        isNewSessionDisabled={isNavigationBlocked}
         onSignIn={onSignIn}
         onSignOut={onSignOut}
       />