Allow adding extra agent invocations during agentic CLI execution

- while agentic search run, we show notification. it should have a button 'Add invocation' and should behave like having invocation count + 1 [Accuracy boost]
- on the notification 'add invocation', use text: 'More agentic runs might increase search accuracy.'

└── apps
    └── editor
        ├── package.json
        └── src
            ├── commands
            │   └── agentic-search-command
            │       ├── agentic-search-command.ts
            │       └── index.ts
            ├── features
            │   └── agentic-search
            │       ├── agentic-search-state.ts
            │       ├── agentic-search.ts
            │       ├── index.ts
            │       └── utils
            │           ├── build-agent-prompt.ts
            │           ├── index.ts
            │           └── prompt-for-search-term.ts
            ├── i18n
            │   ├── index.ts
            │   └── translations
            │       ├── features
            │       │   └── agentic-search.ts
            │       └── utils
            │           ├── agentic-cli-invocation.ts
            │           └── index.ts
            ├── utils
            │   └── agentic-cli-invocation
            │       ├── index.ts
            │       └── invoke-agentic-cli.ts
            └── views
                └── prompt
                    └── backend
                        └── message-handlers
                            └── handle-agentic-search.ts

Author: robertpiosik

apps/editor/src/features/agentic-search/agentic-search.ts

@@ -74,6 +74,7 @@ export const agentic_search = async (params: {
       default_agent_key: 'isDefaultForAgenticSearch',
       show_back_button: false,
       invocation_count,
+      allow_adding_invocations: true,
       target_folder: params.folder_path,
       build_prompt: async () => {
         while (true) {

apps/editor/src/i18n/translations/utils/agentic-cli-invocation.ts

@@ -197,6 +197,42 @@ export const translations = {
     hu: 'várakozás az ügynökre...',
     bg: 'изчакване на агент...'
   },
+  'utils.agentic-cli-invocation.agent.add-invocation-message': {
+    en: 'More agentic runs might increase search accuracy.',
+    pl: 'Więcej przebiegów agenta może zwiększyć dokładność wyszukiwania.',
+    'zh-cn': '更多智能体运行可能会提高搜索准确性。',
+    ja: 'エージェントの実行回数を増やすと、検索の精度が向上する可能性があります。',
+    'zh-tw': '更多智能體運行可能會提高搜索準確性。',
+    de: 'Mehr Agentenläufe könnten die Suchgenauigkeit erhöhen.',
+    es: 'Más ejecuciones de agentes podrían aumentar la precisión de búsqueda.',
+    fr: "Plus d'exécutions d'agents pourraient augmenter la précision de la recherche.",
+    'pt-br': 'Mais execuções de agentes podem aumentar a precisão da pesquisa.',
+    ru: 'Дополнительные запуски агента могут повысить точность поиска.',
+    ko: '더 많은 에이전트 실행이 검색 정확도를 높일 수 있습니다.',
+    it: 'Più esecuzioni di agenti potrebbero aumentare la precisione della ricerca.',
+    tr: 'Daha fazla aracı çalıştırması arama doğruluğunu artırabilir.',
+    cs: 'Více běhů agentů může zvýšit přesnost vyhledávání.',
+    hu: 'Több ügynöki futtatás növelheti a keresés pontosságát.',
+    bg: 'Повече изпълнения на агенти може да увеличат точността на търсенето.'
+  },
+  'utils.agentic-cli-invocation.agent.add-invocation': {
+    en: 'Add invocation',
+    pl: 'Dodaj wywołanie',
+    'zh-cn': '添加调用',
+    ja: '呼び出しを追加',
+    'zh-tw': '新增呼叫',
+    de: 'Aufruf hinzufügen',
+    es: 'Añadir invocación',
+    fr: 'Ajouter une invocation',
+    'pt-br': 'Adicionar invocação',
+    ru: 'Добавить вызов',
+    ko: '호출 추가',
+    it: 'Aggiungi invocazione',
+    tr: 'Çağrı ekle',
+    cs: 'Přidat vyvolání',
+    hu: 'Hívás hozzáadása',
+    bg: 'Добавяне на извикване'
+  },
   'utils.agentic-cli-invocation.error.not-installed': {
     en: 'The {agent} CLI is not installed on this computer.',
     pl: 'CLI {agent} nie jest zainstalowane na tym komputerze.',

apps/editor/src/utils/agentic-cli-invocation/invoke-agentic-cli.ts

@@ -42,6 +42,7 @@ export const invoke_agentic_cli = async (params: {
   use_quick_pick?: boolean
   default_agent_key?: keyof ConfigAgentConfigurationFormat
   invocation_count?: number
+  allow_adding_invocations?: boolean
   target_folder?: string
 }): Promise<
   | {
@@ -627,16 +628,17 @@ export const invoke_agentic_cli = async (params: {
         }
       }
 
-      const invocation_count = params.invocation_count || 1
+      const initial_invocation_count = params.invocation_count || 1
+      let total_invocations = 0
+      let is_finished = false
       let cancelled_count = 0
       const agent_outputs: string[] = []
 
       output_channel.clear()
       const start_time = Date.now()
 
       try {
-        await Promise.all(
-          Array.from({ length: invocation_count }).map(async (_, index) => {
+        const run_invocation = async (index: number) => {
             return vscode.window.withProgress(
               {
                 location: vscode.ProgressLocation.Notification,
@@ -756,7 +758,7 @@ export const invoke_agentic_cli = async (params: {
                     if (
                       params.generate_diff_for_temp_dir &&
                       temp_dir_path &&
-                      invocation_count === 1
+                      total_invocations === 1
                     ) {
                       try {
                         const diff_markdown = await generate_diff_markdown(
@@ -801,8 +803,43 @@ export const invoke_agentic_cli = async (params: {
                 })
               }
             )
-          })
-        )
+        }
+
+        const tasks: Promise<void>[] = []
+        const start_invocation = () => {
+          const index = tasks.length
+          total_invocations++
+          tasks.push(run_invocation(index))
+        }
+
+        const show_add_invocation_button = () => {
+          const label = t('utils.agentic-cli-invocation.agent.add-invocation')
+          vscode.window
+            .showInformationMessage(
+              t('utils.agentic-cli-invocation.agent.add-invocation-message'),
+              label
+            )
+            .then((selection) => {
+              if (selection !== label || is_finished) return
+              start_invocation()
+              show_add_invocation_button()
+            })
+        }
+
+        for (let i = 0; i < initial_invocation_count; i++) {
+          start_invocation()
+        }
+
+        if (params.allow_adding_invocations) {
+          show_add_invocation_button()
+        }
+
+        let awaited_count = 0
+        while (awaited_count < tasks.length) {
+          const batch = tasks.slice(awaited_count)
+          awaited_count = tasks.length
+          await Promise.all(batch)
+        }
       } catch (err) {
         vscode.window.showErrorMessage(
           t('utils.agentic-cli-invocation.error.failed', {
@@ -812,12 +849,13 @@ export const invoke_agentic_cli = async (params: {
         go_back_to_caller = true
         break
       } finally {
+        is_finished = true
         await fs.promises
           .rm(temp_prompt_dir, { recursive: true, force: true })
           .catch(() => {})
       }
 
-      if (cancelled_count === invocation_count) {
+      if (cancelled_count === total_invocations) {
         go_back_to_caller = true
         break
       }