Add search term prompting and improve navigation loops in agentic search

- agentic search should use [Fragment] instead of asking the user
- should display the instructions for the user as a proper first step in the flow to confirm/edit, only then show available agents to select [Commit]
- agentic search should copy the instructions, not use reference (as changing input box value incorrectly propagates into the provider)
- esc key should move back if possible, not close
- don't prevent outside clicks in the flow

└── apps
    └── editor
        └── src
            ├── i18n
            │   └── translations
            │       └── views
            │           └── prompt
            │               └── handlers
            │                   └── handle-agentic-search.ts
            └── views
                └── prompt
                    └── backend
                        ├── message-handlers
                        │   └── handle-agentic-search
                        │       ├── agentic-search.ts
                        │       ├── agents
                        │       │   ├── antigravity.ts
                        │       │   ├── claude.ts
                        │       │   ├── codex.ts
                        │       │   ├── cursor.ts
                        │       │   ├── grok.ts
                        │       │   ├── muse.ts
                        │       │   └── opencode.ts
                        │       ├── handle-agentic-search.ts
                        │       ├── index.ts
                        │       ├── types.ts
                        │       └── utils
                        │           ├── build-agent-prompt.ts
                        │           ├── check-command-exists.ts
                        │           ├── format-duration.ts
                        │           ├── get-progress-dots.ts
                        │           ├── index.ts
                        │           ├── prompt-for-search-term.ts
                        │           └── show-search-results-quick-pick.ts
                        └── prompt-view-provider.ts

Author: robertpiosik

apps/editor/src/views/prompt/backend/message-handlers/handle-agentic-search/agentic-search.ts

@@ -12,6 +12,7 @@ import { extract_paths_from_bullet_list } from '@/utils/extract-paths-from-bulle
 import { get_all_workspace_files } from '@/context/helpers/get-all-workspace-files'
 import { Logger } from '@shared/utils/logger'
 import { show_search_results_quick_pick } from './utils/show-search-results-quick-pick'
+import { prompt_for_search_term } from './utils/prompt-for-search-term'
 import { antigravity_agent } from './agents/antigravity'
 import { claude_agent } from './agents/claude'
 import { codex_agent } from './agents/codex'
@@ -57,10 +58,29 @@ export const agentic_search = async (params: {
   }
 
   const output_channel = get_output_channel()
-  const query = params.query
+  let query = params.query
 
   while (true) {
-    const available_agents = AGENTS.filter((a) => a.is_installed())
+    const query_result = await prompt_for_search_term(query)
+
+    if (query_result.back) {
+      return 'back'
+    }
+
+    if (query_result.value === undefined) {
+      return undefined
+    }
+
+    query = query_result.value
+
+    if (query.trim() == '') {
+      continue
+    }
+
+    let go_back_to_query = false
+
+    while (true) {
+      const available_agents = AGENTS.filter((a) => a.is_installed())
 
         if (available_agents.length == 0) {
           vscode.window.showInformationMessage(
@@ -160,7 +180,7 @@ export const agentic_search = async (params: {
         agent_quick_pick.placeholder = t(
           'views.prompt.handlers.handle-agentic-search.agent.select-agent-placeholder'
         )
-        agent_quick_pick.buttons = [close_button]
+        agent_quick_pick.buttons = [vscode.QuickInputButtons.Back, close_button]
         agent_quick_pick.ignoreFocusOut = true
 
         const config_listener = vscode.workspace.onDidChangeConfiguration(
@@ -183,7 +203,11 @@ export const agentic_search = async (params: {
             let is_resolved = false
 
             agent_quick_pick.onDidTriggerButton((button) => {
-              if (button === close_button) {
+              if (button === vscode.QuickInputButtons.Back) {
+                is_resolved = true
+                resolve('back')
+                agent_quick_pick.hide()
+              } else if (button === close_button) {
                 is_resolved = true
                 resolve(undefined)
                 agent_quick_pick.hide()
@@ -272,7 +296,7 @@ export const agentic_search = async (params: {
 
             agent_quick_pick.onDidHide(() => {
               if (!is_resolved) {
-                resolve(undefined)
+                resolve('back')
               }
               config_listener.dispose()
               agent_quick_pick.dispose()
@@ -282,7 +306,12 @@ export const agentic_search = async (params: {
           }
         )
 
-        if (!agent_selection_result || agent_selection_result === 'back') {
+        if (agent_selection_result === 'back') {
+          go_back_to_query = true
+          break
+        }
+
+        if (!agent_selection_result) {
           return undefined
         }
 
@@ -557,11 +586,13 @@ export const agentic_search = async (params: {
             vscode.window.showErrorMessage(
               t('views.prompt.handlers.handle-agentic-search.error.failed', { error: String(err) })
             )
-            return undefined
+            go_back_to_query = true
+            break
           }
 
           if (is_cancelled) {
-            return undefined
+            go_back_to_query = true
+            break
           }
 
           const duration = format_duration(Date.now() - start_time)
@@ -573,7 +604,8 @@ export const agentic_search = async (params: {
             vscode.window.showInformationMessage(
               t('views.prompt.handlers.handle-agentic-search.info.no-files')
             )
-            return undefined
+            go_back_to_query = true
+            break
           }
 
           const all_workspace_files = await get_all_workspace_files({
@@ -605,7 +637,8 @@ export const agentic_search = async (params: {
             vscode.window.showInformationMessage(
               t('views.prompt.handlers.handle-agentic-search.info.no-files')
             )
-            return undefined
+            go_back_to_query = true
+            break
           }
 
           let should_go_back_to_workspace = false
@@ -658,5 +691,12 @@ export const agentic_search = async (params: {
         break
       }
 
+      if (go_back_to_query) {
+        continue
+      }
+
+      break
+    }
+
     return undefined
 }

apps/editor/src/views/prompt/backend/message-handlers/handle-agentic-search/handle-agentic-search.ts

@@ -17,10 +17,6 @@ export const handle_agentic_search = async (
   }
 
   const query = prompt_view_provider.current_instructions
-  if (query.trim() === '') {
-    vscode.window.showErrorMessage(t('feature.search-files.validation-empty'))
-    return
-  }
 
   agentic_search_in_progress = true
   try {

apps/editor/src/views/prompt/backend/message-handlers/handle-agentic-search/utils/index.ts

@@ -2,4 +2,5 @@ export * from './build-agent-prompt'
 export * from './check-command-exists'
 export * from './format-duration'
 export * from './get-progress-dots'
+export * from './prompt-for-search-term'
 export * from './show-search-results-quick-pick'

apps/editor/src/views/prompt/backend/message-handlers/handle-agentic-search/utils/prompt-for-search-term.ts

@@ -0,0 +1,64 @@
+import * as vscode from 'vscode'
+import { t } from '@/i18n'
+
+export const prompt_for_search_term = async (
+  initial_search_term: string,
+  on_change?: (value: string) => void
+): Promise<{ value: string | undefined; back?: boolean }> => {
+  const close_button = {
+    iconPath: new vscode.ThemeIcon('close'),
+    tooltip: t('common.close')
+  }
+
+  const input_box = vscode.window.createInputBox()
+  input_box.title = t('views.prompt.handlers.handle-agentic-search.title')
+  input_box.prompt = t('feature.search-files.prompt.intelligent')
+  input_box.placeholder = t('feature.search-files.placeholder')
+  input_box.value = initial_search_term
+  input_box.buttons = [close_button]
+
+  return new Promise<{ value: string | undefined; back?: boolean }>(
+    (resolve) => {
+      let is_resolved = false
+      const disposables: vscode.Disposable[] = []
+
+      disposables.push(
+        input_box.onDidTriggerButton((button) => {
+          if (button === close_button) {
+            is_resolved = true
+            resolve({ value: undefined })
+            input_box.hide()
+          }
+        }),
+        input_box.onDidAccept(() => {
+          const value = input_box.value.trim()
+          if (value.length == 0) {
+            input_box.validationMessage = t(
+              'feature.search-files.validation-empty'
+            )
+            return
+          }
+          is_resolved = true
+          resolve({ value })
+          input_box.hide()
+        }),
+        input_box.onDidChangeValue(() => {
+          if (input_box.value.trim().length > 0) {
+            input_box.validationMessage = undefined
+          }
+          if (on_change) {
+            on_change(input_box.value)
+          }
+        }),
+        input_box.onDidHide(() => {
+          if (!is_resolved) {
+            resolve({ value: undefined })
+          }
+          disposables.forEach((d) => d.dispose())
+          input_box.dispose()
+        })
+      )
+      input_box.show()
+    }
+  )
+}

apps/editor/src/views/prompt/backend/message-handlers/handle-agentic-search/utils/show-search-results-quick-pick.ts

@@ -121,7 +121,7 @@ export const show_search_results_quick_pick = async (params: {
     quick_pick.onDidHide(() => {
       if (is_showing_folder_quick_pick) return
       if (!is_resolved) {
-        resolve('back')
+        resolve(params.show_back_button ? 'back' : undefined)
       }
       quick_pick.dispose()
     })