ag-common 0.0.779 → 0.0.781

package/dist/api/helpers/gemini/index.js

@@ -136,15 +136,38 @@ const geminiPromptDirect = (_a) => __awaiter(void 0, [_a], void 0, function* ({
     }
 });
 exports.geminiPromptDirect = geminiPromptDirect;
-// if url contains "vertexaisearch.cloud.google.com", follow redirect
 const resolveGroundedUrl = (url) => __awaiter(void 0, void 0, void 0, function* () {
-    if (url.includes('vertexaisearch.cloud.google.com')) {
-        const response = yield fetch(url);
-        const redirectUrl = response.headers.get('Location');
-        if (redirectUrl) {
-            return redirectUrl;
+    if (!url.includes('vertexaisearch.cloud.google.com')) {
+        (0, log_1.debug)('not a grounded url', url);
+        return url;
+    }
+    try {
+        (0, log_1.debug)('resolving grounded url', url);
+        // Fetch the grounded URL with redirect following disabled
+        const response = yield fetch(url, {
+            method: 'HEAD',
+            redirect: 'manual',
+        });
+        // Check for redirect responses (301, 302, 307, 308)
+        if (response.status >= 300 && response.status < 400) {
+            const location = response.headers.get('location') || response.headers.get('Location');
+            if (location) {
+                (0, log_1.debug)('resolved grounded url to', location);
+                return location;
+            }
         }
+        // If no redirect, try a GET request to follow any JavaScript redirects
+        const fullResponse = yield fetch(url, {
+            redirect: 'follow',
+        });
+        const finalUrl = fullResponse.url;
+        (0, log_1.debug)('resolved grounded url via GET to', finalUrl);
+        return finalUrl;
+    }
+    catch (error) {
+        (0, log_1.debug)('error resolving grounded url', error);
+        // Return original URL if resolution fails
+        return url;
     }
-    return url;
 });
 exports.resolveGroundedUrl = resolveGroundedUrl;

package/package.json

@@ -1,5 +1,5 @@
 {
-  "version": "0.0.779",
+  "version": "0.0.781",
   "name": "ag-common",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",