Deep Cody: remove TOOL context item after review (#6079)

This change removes the `TOOLCONTEXT` context item from the `DeepCodyAgent` after the review process is complete. The `TOOLCONTEXT` context item is only used during the review process and should not be included in the final context. ## Test plan  After: the search history (which should only be used during the review process) should not showing in the UI ![image](https://github.com/user-attachments/assets/ecc07576-a680-4d4a-98e4-70e21a2a7c77) Before: the search history and chat memory are showing in the UI ![image](https://github.com/user-attachments/assets/675be551-0e5f-40cb-b4d3-3af692e54b59) ## Changelog
sourcegraph · Nov 7, 2024 · 45b9ff2 · 45b9ff2
1 parent 876a2a3
commit 45b9ff2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/vscode/src/chat/agentic/CodyTool.ts b/vscode/src/chat/agentic/CodyTool.ts
@@ -193,7 +193,7 @@ class SearchTool extends CodyTool {
             content: 'Queries performed: ' + Array.from(this.performedSearch).join(', '),
             uri: URI.file('search-history'),
             source: ContextItemSource.Agentic,
-            title: 'TOOL',
+            title: 'TOOLCONTEXT',
         } satisfies ContextItem
         context.push(searchQueryItem)
         return context.slice(-maxSearchItems)

diff --git a/vscode/src/chat/agentic/DeepCody.ts b/vscode/src/chat/agentic/DeepCody.ts
@@ -64,9 +64,7 @@ export class DeepCodyAgent extends CodyChatAgent {
                 category: 'billable',
             },
         })
-
-        // Remove the TOOL context item that is only used during the review process.
-        return this.context.filter(c => c.title !== 'TOOL')
+        return this.context
     }
 
     private async reviewLoop(
@@ -81,7 +79,9 @@ export class DeepCodyAgent extends CodyChatAgent {
             const newContext = await this.review(span, chatAbortSignal)
             if (!newContext.length) break
 
-            this.context.push(...newContext)
+            // Remove the TOOL context item that is only used during the review process.
+            this.context.push(...newContext.filter(c => c.title !== 'TOOLCONTEXT'))
+
             context += newContext.length
             loop++
         }