feat(chat): implement Google Search integration and update workflows

- Add Google Search toggle to chat interface
- Implement chat/google-search command in ChatController
- Add isGoogleSearchEnabled parameter to completion and chat handlers
- Update googleChatClient with conditional Google search tools
- Add Google search toggle methods to ChatBuilder
- Add isGoogleSearchEnabled prop to HumanMessageEditor
- Update version to 1.71.0+1
- Remove experimental and nightly GitHub workflow files

Author: PriNova

.github/workflows/release-vscode-experimental.yml

@@ -83,7 +83,8 @@ export async function googleChatClient({
             })
         }
     }
-    const hasSearch = (model as Model).clientSideConfig?.options?.googleSearch
+    const hasSearch =
+        (model as Model).clientSideConfig?.options?.googleSearch && params.isGoogleSearchEnabled
     const tools = hasSearch ? [{ google_search: {} }] : []
     const configs = isGeminiThinkModel ? { thinkingConfig: { includeThoughts: true } } : {}
 

.github/workflows/release-vscode-nightly.yml

@@ -56,6 +56,7 @@ export interface CompletionParameters {
     model?: string
     stream?: boolean
     images?: ImageData[]
+    isGoogleSearchEnabled?: boolean
     // Configuration for a Predicted Output, which can greatly improve response
     // times when large parts of the model response are known ahead of time.
     // https://platform.openai.com/docs/guides/latency-optimization#use-predicted-outputs

lib/shared/src/llm-providers/google/chat-client.ts

@@ -3,7 +3,7 @@
   "name": "cody-ai",
   "private": true,
   "displayName": "Cody: AI Code Assistant",
-  "version": "1.71.0+0",
+  "version": "1.71.0+1",
   "publisher": "sourcegraph",
   "license": "Apache-2.0",
   "icon": "resources/sourcegraph.png",

lib/shared/src/sourcegraph-api/completions/types.ts

@@ -96,7 +96,8 @@ export class ChatBuilder {
         public readonly sessionID: string = new Date(Date.now()).toUTCString(),
         private messages: ChatMessage[] = [],
         private customChatTitle?: string,
-        private images: ImageData[] = []
+        private images: ImageData[] = [],
+        private isGoogleSearchEnabled = false
     ) {}
 
     /** An observable that emits whenever the {@link ChatBuilder}'s chat changes. */
@@ -388,6 +389,24 @@ export class ChatBuilder {
         // Default to jpeg if unknown
         return 'image/jpeg'
     }
+
+    /**
+     * Sets the Google search toggle to enabled.
+     */
+    public async setGoogleSearchToggle(): Promise<void> {
+        this.isGoogleSearchEnabled = true
+    }
+
+    /**
+     * Retrieves the current state of the Google search toggle and resets it to disabled.
+     *
+     * @returns The previous state of the Google search toggle, indicating whether it was enabled or disabled.
+     */
+    public getAndResetGoogleSearchToggle(): boolean {
+        const isGoogleSearchEnabled = this.isGoogleSearchEnabled
+        this.isGoogleSearchEnabled = false
+        return isGoogleSearchEnabled
+    }
 }
 
 function messageToSerializedChatInteraction(

vscode/package.json

@@ -561,6 +561,10 @@ export class ChatController implements vscode.Disposable, vscode.WebviewViewProv
                 await this.chatBuilder.addImage(message.image)
                 break
             }
+            case 'chat/google-search': {
+                await this.chatBuilder.setGoogleSearchToggle()
+                break
+            }
         }
     }
 

vscode/src/chat/chat-view/ChatBuilder.ts

@@ -155,6 +155,7 @@ export class ChatHandler implements AgentHandler {
                 model,
                 maxTokensToSample: contextWindow.output,
                 images: chatBuilder.getAndResetImages(),
+                isGoogleSearchEnabled: chatBuilder.getAndResetGoogleSearchToggle(),
             } as CompletionParameters
 
             // Set stream param only when the model is disabled for streaming.

vscode/src/chat/chat-view/ChatController.ts

@@ -177,6 +177,9 @@ export type WebviewMessage =
           command: 'openRelativeFile'
           uri: Uri
       }
+    | {
+          command: 'chat/google-search'
+      }
 
 export interface SmartApplyResult {
     taskId: FixupTaskID

vscode/src/chat/chat-view/handlers/ChatHandler.ts

@@ -223,14 +223,14 @@ export const HumanMessageEditor: FunctionComponent<{
             }
             setImageFile(undefined)
             processImage()
-            /* const processGoogleSearch = async () => {
+            const processGoogleSearch = async () => {
                 if (isGoogleSearchEnabled) {
                     getVSCodeAPI().postMessage({
                         command: 'chat/google-search',
                     })
                 }
             }
-            processGoogleSearch() */
+            processGoogleSearch()
 
             parentOnSubmit(intent)
 
@@ -257,6 +257,7 @@ export const HumanMessageEditor: FunctionComponent<{
             isSent,
             imageFile,
             setImageFile,
+            isGoogleSearchEnabled,
         ]
     )