ag-common 0.0.908 → 0.0.910

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

@@ -1,239 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.resolveGroundedUrl = exports.geminiPromptDirect = exports.geminiPromptImage = exports.isTextGenerationModel = void 0;
-const genai_1 = require("@google/genai");
-const array_1 = require("../../../common/helpers/array");
-const async_1 = require("../../../common/helpers/async");
-const fetch_1 = require("../../../common/helpers/fetch");
-const log_1 = require("../../../common/helpers/log");
-const node_cache_1 = require("../../../common/helpers/node-cache");
-const retryOnError_1 = require("../retryOnError");
-const apikey_1 = require("./apikey");
-let genAIs = [];
-const geminiModelsCache = new node_cache_1.TypedNodeCache({ stdTTL: 86400 });
-const geminiModelsCacheKey = "gemini-models-v1";
-const FALLBACK_GEMINI_MODELS = [
-    "gemini-3-flash-preview",
-    "gemini-3-pro-preview",
-    "gemini-2.5-pro",
-    "gemini-2.5-flash",
-    "gemini-2.5-flash-lite",
-];
-// Helper to sort models based on preference
-const sortModelsByPreference = (models, prefer) => {
-    const modelArray = [...models];
-    if (prefer === "quality") {
-        // Sort pro models first
-        return modelArray.sort((a, b) => {
-            const aIsPro = a.includes("pro");
-            const bIsPro = b.includes("pro");
-            if (aIsPro && !bIsPro)
-                return -1;
-            if (!aIsPro && bIsPro)
-                return 1;
-            return 0;
-        });
-    }
-    else if (prefer === "fast") {
-        // Sort flash models first
-        return modelArray.sort((a, b) => {
-            const aIsFlash = a.includes("flash");
-            const bIsFlash = b.includes("flash");
-            if (aIsFlash && !bIsFlash)
-                return -1;
-            if (!aIsFlash && bIsFlash)
-                return 1;
-            return 0;
-        });
-    }
-    // Default behavior - no sorting
-    return modelArray;
-};
-const normalizeModelName = (name) => name.replace(/^models\//, "");
-const NON_TEXT_MODEL_CAPABILITIES = [
-    "audio",
-    "computer-use",
-    "embedding",
-    "image",
-    "live",
-    "robotics",
-    "tts",
-];
-const isTextGenerationModel = (name) => {
-    const normalizedName = name.toLowerCase();
-    return (normalizedName.startsWith("gemini") &&
-        !NON_TEXT_MODEL_CAPABILITIES.some((capability) => normalizedName.includes(capability)));
-};
-exports.isTextGenerationModel = isTextGenerationModel;
-const getAvailableGeminiModels = async () => {
-    const cachedModels = geminiModelsCache.get(geminiModelsCacheKey);
-    if (cachedModels && cachedModels.length > 0) {
-        return cachedModels;
-    }
-    const keyServiceCombinations = (0, apikey_1.getAvailableCombinations)("gemini");
-    const key = keyServiceCombinations[0]?.key;
-    if (!key) {
-        (0, log_1.warn)("No GOOGLE_API_KEY available. Falling back to default Gemini models.");
-        return [...FALLBACK_GEMINI_MODELS];
-    }
-    try {
-        const response = await fetch(`https://generativelanguage.googleapis.com/v1beta/models?key=${encodeURIComponent(key)}`);
-        if (!response.ok) {
-            throw new Error(`model list fetch failed with status ${response.status}`);
-        }
-        const payload = (await response.json());
-        const discoveredModels = [
-            ...new Set((payload.models ?? [])
-                .filter((m) => (m.supportedGenerationMethods ?? []).includes("generateContent"))
-                .map((m) => normalizeModelName(m.name ?? ""))
-                .filter(exports.isTextGenerationModel)),
-        ];
-        if (discoveredModels.length === 0) {
-            throw new Error("no gemini generateContent models discovered");
-        }
-        geminiModelsCache.set(geminiModelsCacheKey, discoveredModels);
-        (0, log_1.info)(`loaded ${discoveredModels.length} Gemini models from API`);
-        return discoveredModels;
-    }
-    catch (e) {
-        (0, log_1.warn)(`Failed to load Gemini models from API. Falling back to defaults. ${String(e)}`);
-        return [...FALLBACK_GEMINI_MODELS];
-    }
-};
-// Helper to get available key+model combinations
-const getAvailableGeminiCombinations = async (prefer) => {
-    if (genAIs.length === 0) {
-        const keyServiceCombinations = (0, apikey_1.getAvailableCombinations)("gemini");
-        const keys = keyServiceCombinations.map((combo) => combo.key);
-        genAIs = keys.map((k) => [k, new genai_1.GoogleGenAI({ apiKey: k })]);
-    }
-    const availableModels = await getAvailableGeminiModels();
-    const sortedModels = sortModelsByPreference(availableModels, prefer);
-    const combinations = [];
-    for (const [key, ai] of genAIs) {
-        for (const model of sortedModels) {
-            const keyServiceCombinations = (0, apikey_1.getAvailableCombinations)(`gemini-${model}`, false);
-            const isAvailable = keyServiceCombinations.some((k) => k.key === key);
-            if (isAvailable) {
-                combinations.push([key, ai, model]);
-            }
-        }
-    }
-    return combinations;
-};
-const geminiPromptImage = async ({ prompt, urls, ident, prefer, onGenerated, }) => {
-    let images = [];
-    if (urls && urls.length > 0) {
-        images = await (0, async_1.asyncMap)(urls, (i) => (0, fetch_1.fetchToMemory)(i));
-    }
-    if (images.some((image) => image === undefined)) {
-        throw new Error("image not downloaded correctly");
-    }
-    const r = await (0, exports.geminiPromptDirect)({
-        prompt,
-        images: images.filter(array_1.notEmpty),
-        ident,
-        prefer,
-        onGenerated,
-    });
-    return r;
-};
-exports.geminiPromptImage = geminiPromptImage;
-const geminiPromptDirect = async ({ prompt, images = [], ident, prefer, groundedSearch = false, onGenerated, }) => {
-    const parts = images.map((i) => ({
-        inlineData: {
-            data: Buffer.from(i.arraybuffer).toString("base64"),
-            mimeType: i.type,
-        },
-    }));
-    return (0, retryOnError_1.retryOnError)(`geminiPromptDirect:${ident ?? "unknown"}`, async () => {
-        const combinations = await getAvailableGeminiCombinations(prefer);
-        if (combinations.length === 0) {
-            throw new Error("No available API key and model combinations");
-        }
-        let lastFailure;
-        for (const [key, ai, selectedModel] of combinations) {
-            // Another concurrent prompt may have exhausted this combination already.
-            if (!(0, apikey_1.getAvailableCombinations)(`gemini-${selectedModel}`, false).some((entry) => entry.key === key))
-                continue;
-            // Prepare the request configuration
-            const requestConfig = {
-                model: selectedModel,
-                contents: [{ parts: [{ text: prompt }, ...parts] }],
-                config: {},
-            };
-            // Add grounded search configuration if enabled
-            if (groundedSearch) {
-                requestConfig.config.tools = [{ googleSearch: {} }];
-            }
-            (0, log_1.info)("gem query on:" + selectedModel, requestConfig);
-            try {
-                // oxlint-disable-next-line no-await-in-loop -- fallback depends on the previous model failing
-                const response = await ai.models.generateContent(requestConfig);
-                const rawtext = (response.text ?? "")
-                    .replace(/```(json)?/gi, "")
-                    .replace(/:[ ]+undefined/gim, ": null");
-                (0, log_1.info)("gem response");
-                (0, log_1.debug)("gem prompt:" + prompt, ident);
-                (0, log_1.debug)("gem response:" + rawtext);
-                (0, log_1.debug)("gem query usage:" + JSON.stringify(response.usageMetadata));
-                onGenerated?.({ model: response.modelVersion || selectedModel, generatedAt: Date.now() });
-                return rawtext;
-            }
-            catch (e) {
-                const mod = `gemini-${selectedModel}`;
-                const status = e.status;
-                const message = e instanceof Error ? e.message : String(e);
-                const unavailableModel = status === 404 && message.includes(selectedModel);
-                if (status === 429 || (0, retryOnError_1.isOverloadedApiKeyError)(e) || unavailableModel) {
-                    (0, log_1.warn)("Gemini model attempt failed; trying next available combination", {
-                        model: selectedModel,
-                        status,
-                        reason: unavailableModel ? "model unavailable" : "quota or capacity exhausted",
-                    });
-                    (0, apikey_1.blockKeyService)(key, mod);
-                    lastFailure = e;
-                    continue;
-                }
-                throw e;
-            }
-        }
-        throw lastFailure ?? new Error("No available API key and model combinations");
-    }, 1, 5000);
-};
-exports.geminiPromptDirect = geminiPromptDirect;
-const resolveGroundedUrl = async (url) => {
-    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 = await 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 = await 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;
-    }
-};
-exports.resolveGroundedUrl = resolveGroundedUrl;