improvement(home): make Assistant a peer of Build and Search in the mode switcher

Author: waleedlatif1

apps/sim/app/workspace/[workspaceId]/home/components/suggested-actions/suggested-actions.test.tsx

@@ -128,4 +128,14 @@ describe('SuggestedActions', () => {
     expect(document.querySelector('[data-testid="search-sources"]')).not.toBeNull()
     expect(rows()).toHaveLength(0)
   })
+
+  it('shows the sources in Assistant mode, which answers from them', () => {
+    mount()
+
+    act(() => useMothershipModeStore.getState().setMode('assistant'))
+
+    expect(heading()).toBe('Sources')
+    expect(document.querySelector('[data-testid="search-sources"]')).not.toBeNull()
+    expect(rows()).toHaveLength(0)
+  })
 })

apps/sim/app/workspace/[workspaceId]/home/components/suggested-actions/suggested-actions.tsx

@@ -235,6 +235,7 @@ const INITIAL_ACTIONS: Action[] = [
 const HEADINGS: Record<MothershipMode, string> = {
   build: 'Suggested actions',
   search: 'Sources',
+  assistant: 'Sources',
 }
 
 interface SuggestedActionsProps {
@@ -372,7 +373,7 @@ export function SuggestedActions({ onSelectPrompt }: SuggestedActionsProps) {
               `collapsible-up`/`-down` interpolate height alone, so a margin here
               would hold its full value through the close and then vanish on unmount,
               snapping the content below up. */}
-          {mode === 'search' && workspaceId ? (
+          {mode !== 'build' && workspaceId ? (
             <div className='pt-1.5'>
               <SearchSources workspaceId={workspaceId} />
             </div>

apps/sim/app/workspace/[workspaceId]/home/components/user-input/components/index.ts

@@ -36,4 +36,3 @@ export { PromptEditor, usePromptEditor } from './prompt-editor'
 export { SendButton } from './send-button'
 export type { SkillsMenuHandle } from './skills-menu-dropdown/skills-menu-dropdown'
 export { SkillsMenuDropdown } from './skills-menu-dropdown/skills-menu-dropdown'
-export { SourcesModeToggle } from './sources-mode-toggle'

apps/sim/app/workspace/[workspaceId]/home/components/user-input/components/mode-switcher/mode-switcher.test.tsx

@@ -78,14 +78,15 @@ describe('ModeSwitcher', () => {
     expect(button.querySelector('svg')).toBeNull()
   })
 
-  it('lists both modes and checks the active one', () => {
+  it('lists every mode and checks the active one', () => {
     mount()
     openMenu()
 
     const rows = items()
-    expect(rows.map((row) => row.textContent)).toEqual(['Build', 'Sources'])
+    expect(rows.map((row) => row.textContent)).toEqual(['Build', 'Search', 'Assistant'])
     expect(rows[0].querySelector('svg')).not.toBeNull()
     expect(rows[1].querySelector('svg')).toBeNull()
+    expect(rows[2].querySelector('svg')).toBeNull()
   })
 
   it('switches the shared mode and reports the change', () => {
@@ -97,15 +98,15 @@ describe('ModeSwitcher', () => {
     })
 
     expect(useMothershipModeStore.getState().mode).toBe('search')
-    expect(trigger().textContent).toBe('Sources')
+    expect(trigger().textContent).toBe('Search')
     expect(mockCaptureEvent).toHaveBeenCalledWith(null, 'chat_mode_changed', {
       workspace_id: 'workspace-1',
       mode: 'search',
     })
     expect(mockSetSearchQuery).not.toHaveBeenCalled()
   })
 
-  it('drops the search query from the URL when leaving Sources', () => {
+  it('drops the search query from the URL when leaving Search', () => {
     useMothershipModeStore.getState().setMode('search')
     mount()
     openMenu()

apps/sim/app/workspace/[workspaceId]/home/components/user-input/components/mode-switcher/mode-switcher.tsx

@@ -28,11 +28,12 @@ import {
 
 const MODE_LABELS: Record<MothershipMode, string> = {
   build: 'Build',
-  search: 'Sources',
+  search: 'Search',
+  assistant: 'Assistant',
 }
 
 /**
- * The composer's Build / Sources switcher: a label-only `Chip` in its `round`
+ * The composer's Build / Search / Assistant switcher: a label-only `Chip` in its `round`
  * shape — chip chrome throughout (`--text-body` label, `--surface-hover` on
  * hover, no text-color shift), fully round to sit in the toolbar's row of
  * round controls — opening a menu that checks the active mode, as
@@ -47,7 +48,7 @@ export const ModeSwitcher = memo(function ModeSwitcher() {
   const [, setSearchQueryParam] = useQueryState(searchQueryParam.key, searchQueryParam.parser)
   const [, setSearchFilters] = useQueryStates(searchFilterParsers, resourceUrlKeys)
 
-  /** Leaving Sources drops the query from the URL, so a clean URL always means no search is showing. */
+  /** Leaving Search drops the query from the URL, so a clean URL always means no search is showing. */
   const handleSelect = (next: MothershipMode) => {
     if (next === mode) return
     setMode(next)

apps/sim/app/workspace/[workspaceId]/home/components/user-input/components/sources-mode-toggle/index.ts

@@ -28,7 +28,6 @@ import {
   ModeSwitcher,
   PromptEditor,
   SendButton,
-  SourcesModeToggle,
   usePromptEditor,
 } from '@/app/workspace/[workspaceId]/home/components/user-input/components'
 import { handleMothershipAddContextEvent } from '@/app/workspace/[workspaceId]/home/components/user-input/mothership-context-event'
@@ -720,7 +719,6 @@ const UserInputImpl = forwardRef<UserInputHandle, UserInputProps>(function UserI
           </Tooltip.Root>
         </div>
         <div className='flex items-center gap-1.5'>
-          {canSearch && <SourcesModeToggle />}
           {canSearch && <ModeSwitcher />}
           {isSttSupported && (
             <MicButton

apps/sim/app/workspace/[workspaceId]/home/components/user-input/components/sources-mode-toggle/sources-mode-toggle.test.tsx

@@ -195,7 +195,6 @@ export function Home({ chatId, userName, userId }: HomeProps) {
     if (searchQuery) useMothershipModeStore.getState().setMode('search')
   }, [searchQuery])
   const composerMode = useMothershipModeStore((state) => state.mode)
-  const assistantMode = useMothershipModeStore((state) => state.assistant)
   /** The bases an Ask turn is grounded in; read through a ref so a list refresh never rebuilds the submit handler. */
   const { data: knowledgeBases = EMPTY_KNOWLEDGE_BASES } = useKnowledgeBasesQuery(workspaceId)
   const knowledgeBasesRef = useRef(knowledgeBases)
@@ -488,13 +487,13 @@ export function Home({ chatId, userName, userId }: HomeProps) {
       })
 
       /**
-       * Sources mode's Search lists documents, not a turn of the agent, and only
-       * a query can be searched: attachments alone have nothing to search for.
-       * Its Assistant makes the query a turn of the agent grounded in the sources.
+       * Search lists documents, not a turn of the agent, and only a query can
+       * be searched: attachments alone have nothing to search for. Assistant
+       * makes the query a turn of the agent grounded in the sources.
        */
-      const { mode, assistant } = useMothershipModeStore.getState()
-      const answering = mode === 'search' && assistant
-      if (mode === 'search' && !assistant) {
+      const mode = useMothershipModeStore.getState().mode
+      const answering = mode === 'assistant'
+      if (mode === 'search') {
         if (trimmed) setSearchQuery(trimmed)
         return
       }
@@ -525,21 +524,18 @@ export function Home({ chatId, userName, userId }: HomeProps) {
 
   /** Summarize or Answer on a result: switch to Assistant and hand the question to it. */
   const handleSummarize = (prompt: string) => {
-    const store = useMothershipModeStore.getState()
-    store.setMode('search')
-    store.setAssistant(true)
+    useMothershipModeStore.getState().setMode('assistant')
     setSearchQuery('')
     handleSubmit(prompt)
   }
   /**
-   * A chat that already exists never opens in Sources' document-listing
-   * Search: its transcript is a conversation, and search results never join
-   * it. Build and the Assistant both carry over, so a follow-up stays grounded
-   * in the sources.
+   * A chat that already exists never opens in Search: its transcript is a
+   * conversation, and search results never join it. Build and Assistant both
+   * carry over, so a follow-up stays grounded in the sources.
    */
   useEffect(() => {
     const store = useMothershipModeStore.getState()
-    if (chatId && store.mode === 'search' && !store.assistant) store.setMode('build')
+    if (chatId && store.mode === 'search') store.setMode('build')
   }, [chatId])
   const showSearchResults = composerMode === 'search' && searchQuery.trim().length > 0
   const searchResults = showSearchResults ? (
@@ -772,7 +768,7 @@ export function Home({ chatId, userName, userId }: HomeProps) {
                     draftScopeKey={draftScopeKey}
                     onSubmit={handleSubmit}
                     canSearch
-                    clearOnSubmit={composerMode !== 'search' || assistantMode}
+                    clearOnSubmit={composerMode !== 'search'}
                     onCleared={clearSearch}
                     isSending={isSending}
                     onStopGeneration={handleStopGeneration}
@@ -801,7 +797,7 @@ export function Home({ chatId, userName, userId }: HomeProps) {
             isLoading={showChatSkeleton}
             onSubmit={handleSubmit}
             canSearch
-            clearOnSubmit={composerMode !== 'search' || assistantMode}
+            clearOnSubmit={composerMode !== 'search'}
             onCleared={clearSearch}
             onStopGeneration={handleStopGeneration}
             messageQueue={messageQueue}