From cb552d317f19bc31d84f84d0b1b829947f7db736 Mon Sep 17 00:00:00 2001
From: end-4 <97237370+end-4@users.noreply.github.com>
Date: Fri, 9 May 2025 20:02:49 +0200
Subject: [PATCH] ai: gemini: allow searching

---
 .config/ags/services/gemini.js | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/.config/ags/services/gemini.js b/.config/ags/services/gemini.js
index 6775173c7..38ae4e744 100644
--- a/.config/ags/services/gemini.js
+++ b/.config/ags/services/gemini.js
@@ -299,14 +299,17 @@ class GeminiService extends Service {
             "generationConfig": {
                 "temperature": this._temperature,
             },
-            // "key": this._key,
-            // "apiKey": this._key,
+            "tools": [
+                {
+                    "google_search": {}
+                }
+            ]
         };
         const proxyResolver = new Gio.SimpleProxyResolver({ 'default-proxy': userOptions.ai.proxyUrl });
         const session = new Soup.Session({ 'proxy-resolver': proxyResolver });
         const message = new Soup.Message({
             method: 'POST',
-            uri: GLib.Uri.parse(replaceapidom(`https://generativelanguage.googleapis.com/v1/models/${this.modelName}:streamGenerateContent?key=${this._key}`), GLib.UriFlags.NONE),
+            uri: GLib.Uri.parse(replaceapidom(`https://generativelanguage.googleapis.com/v1beta/models/${this.modelName}:streamGenerateContent?key=${this._key}`), GLib.UriFlags.NONE),
         });
         message.request_headers.append('Content-Type', `application/json`);
         message.set_request_body_from_bytes('application/json', new GLib.Bytes(JSON.stringify(body)));