@juspay/neurolink 11.13.0 → 11.13.2

package/dist/lib/providers/googleAiStudio/client.js

@@ -1,1661 +0,0 @@
-import { ErrorCategory, ErrorSeverity, GoogleAIModels, } from "../../constants/enums.js";
-import { BaseProvider } from "../../core/baseProvider.js";
-import { IMAGE_GENERATION_MODELS, TOOL_STORAGE_TIMEOUT_MS, } from "../../core/constants.js";
-import { mergeMediaFileAliases, normalizeVisionImageFormats, processUnifiedFilesArray, } from "../../utils/messageBuilder.js";
-import { ATTR, tracers, withClientSpan, withClientStreamSpan, withSpan, } from "../../telemetry/index.js";
-import { AuthenticationError, InvalidModelError, NetworkError, ProviderError, RateLimitError, } from "../../types/index.js";
-import { ERROR_CODES, NeuroLinkError } from "../../utils/errorHandling.js";
-import { logger } from "../../utils/logger.js";
-import { createGeminiLoopAdapter } from "../../core/geminiLoopAdapter.js";
-import { runAgenticLoop } from "../../core/loopEngine.js";
-import { DEFAULT_TOOL_MAX_RETRIES } from "../../core/constants.js";
-import { isToolsSchemaExclusionInForce } from "../../core/modules/structuredOutputPolicy.js";
-import { GEMINI_ELISION_NOTE, planGeminiLoopReclaim, previewGeminiToolResponseText, } from "../../context/geminiLoopGuard.js";
-import { getAvailableInputTokens, getContextWindowSize, } from "../../constants/contextWindows.js";
-import { composeAbortSignals, createTimeoutController, TimeoutError, } from "../../utils/timeout.js";
-import { withTimeout } from "../../utils/async/index.js";
-import { estimateTokens } from "../../utils/tokenEstimation.js";
-import { transformToolExecutions } from "../../utils/transformationUtils.js";
-import { resolveToolExecutionRecords } from "../../core/toolExecutionRecorder.js";
-import { buildGeminiResponseSchema, buildNativeConfig, computeMaxSteps, createContextGuard, buildUserPartsWithMultimodal, extractThoughtSignature, handleMaxStepsTermination, prependConversationMessages, } from "../googleNativeGemini3/index.js";
-import { createStreamChannel } from "../../core/streamChannel.js";
-import { toNativeToolDeclarations } from "../../core/nativeToolFormat.js";
-import { warnGoogleSdkIgnoresProxy } from "../../proxy/proxyFetch.js";
-// Google AI Live API types now imported from ../types/providerSpecific.js
-// Import proper types for multimodal message handling
-// Create Google GenAI client
-async function createGoogleGenAIClient(apiKey, baseURL) {
-    const mod = await import("@google/genai");
-    const ctor = mod.GoogleGenAI;
-    if (!ctor) {
-        throw new NeuroLinkError({
-            code: ERROR_CODES.INVALID_CONFIGURATION,
-            message: "@google/genai does not export GoogleGenAI",
-            category: ErrorCategory.CONFIGURATION,
-            severity: ErrorSeverity.CRITICAL,
-            retriable: false,
-            context: { module: "@google/genai", expectedExport: "GoogleGenAI" },
-        });
-    }
-    const Ctor = ctor;
-    // httpOptions carries the endpoint override and nothing else. It used to
-    // also pass a proxy fetch, which the SDK silently ignored — see
-    // warnGoogleSdkIgnoresProxy for why that is not fixable here.
-    //
-    // baseUrl is only included when resolved — verified against
-    // @google/genai's ApiClient (dist/node/index.cjs) that it falls back to
-    // its own default whenever httpOptions.baseUrl is undefined, so omitting
-    // the key and passing `baseUrl: undefined` behave identically; the key is
-    // still omitted outright for a cleaner outbound config object.
-    warnGoogleSdkIgnoresProxy("GoogleAIStudio");
-    return new Ctor({
-        apiKey,
-        httpOptions: {
-            ...(baseURL ? { baseUrl: baseURL } : {}),
-        },
-    });
-}
-/**
- * Google AI Studio provider implementation using BaseProvider
- * Migrated from original GoogleAIStudio class to new factory pattern
- *
- * @important Structured Output Limitation
- * Google Gemini models cannot combine function calling (tools) with structured
- * output (JSON schema). When using schemas with output.format: "json", you MUST
- * set disableTools: true.
- *
- * Error without disableTools:
- * "Function calling with a response mime type: 'application/json' is unsupported"
- *
- * This is a Google API limitation documented at:
- * https://ai.google.dev/gemini-api/docs/function-calling
- *
- * @example
- * ```typescript
- * // ✅ Correct usage with schemas
- * const provider = new GoogleAIStudioProvider("gemini-2.5-flash");
- * const result = await provider.generate({
- *   input: { text: "Analyze data" },
- *   schema: MySchema,
- *   output: { format: "json" },
- *   disableTools: true  // Required
- * });
- * ```
- *
- * @note Gemini 3 Pro Preview (November 2025) will support combining tools + schemas
- * @note "Too many states for serving" errors can occur with complex schemas + tools.
- *       Solution: Simplify schema or use disableTools: true
- */
-/**
- * Reclaim context from an AI Studio loop history IN PLACE.
- *
- * This loop had NO in-turn guard at all — it appended a model turn plus a tool
- * turn every step with nothing bounding growth, so a long agentic run walked
- * into a provider "context length exceeded" and lost every completed step.
- * Shares its reclaim policy with the other provider loops via loopGuardCore.
- *
- * Returns true when something was reclaimed.
- */
-function reclaimAiStudioContext(contents, modelName, observedPromptTokens) {
-    const plan = planGeminiLoopReclaim({
-        contents,
-        availableInputTokens: getAvailableInputTokens("googleAiStudio", modelName),
-        provider: "googleAiStudio",
-        ...(observedPromptTokens ? { observedPromptTokens } : {}),
-    });
-    if (!plan) {
-        return false;
-    }
-    const dropSet = new Set(plan.drop);
-    const truncateSet = new Set(plan.truncate);
-    const rebuilt = [];
-    for (let i = 0; i < contents.length; i++) {
-        if (dropSet.has(i)) {
-            continue;
-        }
-        const content = contents[i];
-        if (truncateSet.has(i) && Array.isArray(content.parts)) {
-            rebuilt.push({
-                ...content,
-                parts: content.parts.map((part) => {
-                    const record = part;
-                    if (!record.functionResponse) {
-                        return part;
-                    }
-                    const text = JSON.stringify(record.functionResponse.response) ?? "";
-                    if (text.length <= 2048) {
-                        return part;
-                    }
-                    return {
-                        functionResponse: {
-                            name: record.functionResponse.name,
-                            response: { result: previewGeminiToolResponseText(text) },
-                        },
-                    };
-                }),
-            });
-            continue;
-        }
-        rebuilt.push(content);
-    }
-    if (dropSet.size > 0) {
-        let noteIndex = rebuilt.findIndex((c) => Array.isArray(c.parts) &&
-            c.parts.some((part) => !!part.functionCall ||
-                !!part.functionResponse));
-        if (noteIndex < 0) {
-            noteIndex = Math.min(1, rebuilt.length);
-        }
-        rebuilt.splice(noteIndex, 0, {
-            role: "user",
-            parts: [{ text: GEMINI_ELISION_NOTE }],
-        });
-    }
-    contents.length = 0;
-    contents.push(...rebuilt);
-    return true;
-}
-export class GoogleAIStudioProvider extends BaseProvider {
-    credentials;
-    constructor(modelName, sdk, credentials) {
-        super(modelName, "google-ai", sdk);
-        this.credentials = credentials;
-        logger.debug("GoogleAIStudioProvider initialized", {
-            model: this.modelName,
-            provider: this.providerName,
-            sdkProvided: !!sdk,
-        });
-    }
-    // ===================
-    // ABSTRACT METHOD IMPLEMENTATIONS
-    // ===================
-    getProviderName() {
-        return "google-ai";
-    }
-    getDefaultModel() {
-        return process.env.GOOGLE_AI_MODEL || GoogleAIModels.GEMINI_2_5_FLASH;
-    }
-    /**
-     * AI SDK model instance — no longer used.
-     * All models are routed through native @google/genai SDK directly.
-     */
-    getAISDKModel() {
-        throw new NeuroLinkError({
-            code: ERROR_CODES.INVALID_CONFIGURATION,
-            message: "GoogleAIStudioProvider no longer uses @ai-sdk/google. All models use native @google/genai SDK.",
-            category: ErrorCategory.CONFIGURATION,
-            severity: ErrorSeverity.CRITICAL,
-            retriable: false,
-            context: { provider: this.providerName, model: this.modelName },
-        });
-    }
-    formatProviderError(error) {
-        if (error instanceof TimeoutError) {
-            return new NetworkError(error.message, this.providerName);
-        }
-        const errorRecord = error;
-        const message = typeof errorRecord?.message === "string"
-            ? errorRecord.message
-            : "Unknown error";
-        const statusCode = typeof errorRecord?.status === "number"
-            ? errorRecord.status
-            : typeof errorRecord?.statusCode === "number"
-                ? errorRecord.statusCode
-                : undefined;
-        // Authentication errors
-        if (message.includes("API_KEY_INVALID") ||
-            message.includes("Invalid API key") ||
-            statusCode === 401) {
-            return new AuthenticationError("Invalid Google AI API key. Please check your GOOGLE_AI_API_KEY environment variable.", this.providerName);
-        }
-        // Rate limit errors
-        if (message.includes("RATE_LIMIT_EXCEEDED") ||
-            message.includes("rate limit") ||
-            message.includes("429") ||
-            statusCode === 429) {
-            return new RateLimitError("Google AI rate limit exceeded. Please try again later.", this.providerName);
-        }
-        // Model not found errors — gate on a 404 status when available; fall
-        // back to literal phrase matching only when we have no status code at
-        // all. Avoids misclassifying permission/validation errors that happen
-        // to mention model resource paths (e.g. "...models/foo permission...").
-        if (statusCode === 404 ||
-            (statusCode === undefined &&
-                (message.includes("model not found") ||
-                    message.includes("Model not found")))) {
-            return new InvalidModelError(`Model '${this.modelName}' not found. Please check the model name and ensure it is available.`, this.providerName);
-        }
-        // Network connectivity errors
-        if (message.includes("ECONNRESET") ||
-            message.includes("ENOTFOUND") ||
-            message.includes("ETIMEDOUT") ||
-            message.includes("ECONNREFUSED") ||
-            message.includes("network") ||
-            message.includes("connection")) {
-            return new NetworkError(`Connection error: ${message}`, this.providerName);
-        }
-        // Server errors (5xx)
-        if (message.includes("500") ||
-            message.includes("502") ||
-            message.includes("503") ||
-            message.includes("504") ||
-            message.includes("server error") ||
-            message.includes("Internal Server Error") ||
-            (statusCode && statusCode >= 500 && statusCode < 600)) {
-            return new ProviderError(`Google AI server error: ${message}. Please try again later.`, this.providerName);
-        }
-        return new ProviderError(`Google AI error: ${message}`, this.providerName);
-    }
-    /**
-     * Overrides the BaseProvider's image generation method to implement it for Google AI.
-     * This method calls the Google AI API to generate an image from a prompt.
-     * @param options The generation options containing the prompt.
-     * @returns A promise that resolves to the generation result, including the image data.
-     */
-    async executeImageGeneration(options) {
-        // Image-to-image generation takes reference images through the same
-        // `input.images` array, and generate() routes here BEFORE reaching its own
-        // normalization call — so a reference photo in HEIC/BMP/AVIF would arrive
-        // at the API untranscoded. Normalizing at the top of this method covers
-        // every route in, rather than relying on each caller to remember.
-        // Idempotent, so a request that was already normalized pays nothing.
-        await normalizeVisionImageFormats(options.input);
-        const prompt = options.prompt || options.input?.text || "";
-        const imageModelName = options.model || this.modelName;
-        const startTime = Date.now();
-        const apiKey = this.getApiKey();
-        logger.info("🎨 Starting Google AI Studio image generation", {
-            model: imageModelName,
-            prompt: prompt.substring(0, 100),
-            provider: this.providerName,
-        });
-        // Use the @google/genai client for image generation
-        let client;
-        try {
-            client = await createGoogleGenAIClient(apiKey, this.getBaseURL());
-        }
-        catch {
-            throw new AuthenticationError("Missing '@google/genai'. Install with: npm install @google/genai", this.providerName);
-        }
-        try {
-            // Build content array with multimodal support
-            const imageParts = await Promise.all((options.input?.images || []).map(async (image) => {
-                // Handle ImageWithAltText objects
-                if (typeof image === "object" && "url" in image) {
-                    const imageUrl = image.url;
-                    if (imageUrl.startsWith("http")) {
-                        const response = await fetch(imageUrl);
-                        if (!response.ok) {
-                            throw new Error(`Failed to fetch image from ${imageUrl}: ${response.status} ${response.statusText}`);
-                        }
-                        const arrayBuffer = await response.arrayBuffer();
-                        const buffer = Buffer.from(arrayBuffer);
-                        const mimeType = this.detectImageType(buffer);
-                        logger.debug(`Downloaded and detected image MIME type: ${mimeType}`);
-                        return {
-                            inlineData: {
-                                mimeType,
-                                data: buffer.toString("base64"),
-                            },
-                        };
-                    }
-                    // Base64 URL in ImageWithAltText
-                    const buffer = Buffer.from(imageUrl, "base64");
-                    const mimeType = this.detectImageType(buffer);
-                    return {
-                        inlineData: {
-                            mimeType,
-                            data: buffer.toString("base64"),
-                        },
-                    };
-                }
-                // Handle string URLs
-                if (typeof image === "string" && image.startsWith("http")) {
-                    const response = await fetch(image);
-                    if (!response.ok) {
-                        throw new Error(`Failed to fetch image from ${image}: ${response.status} ${response.statusText}`);
-                    }
-                    const arrayBuffer = await response.arrayBuffer();
-                    const buffer = Buffer.from(arrayBuffer);
-                    const mimeType = this.detectImageType(buffer);
-                    logger.debug(`Downloaded and detected image MIME type: ${mimeType}`);
-                    return {
-                        inlineData: {
-                            mimeType,
-                            data: buffer.toString("base64"),
-                        },
-                    };
-                }
-                // Handle Buffer or base64 string
-                const buffer = Buffer.isBuffer(image)
-                    ? image
-                    : typeof image === "string"
-                        ? Buffer.from(image, "base64")
-                        : Buffer.from(""); // Fallback for unexpected types
-                const mimeType = this.detectImageType(buffer);
-                logger.debug(`Detected image MIME type: ${mimeType}`);
-                return {
-                    inlineData: {
-                        mimeType,
-                        data: buffer.toString("base64"),
-                    },
-                };
-            }));
-            const contents = [
-                {
-                    role: "user",
-                    parts: [{ text: prompt }, ...imageParts],
-                },
-            ];
-            // Configure for image generation
-            const generateConfig = {
-                responseModalities: ["IMAGE", "TEXT"], // This is the key setting for image generation
-            };
-            logger.debug("Starting image generation request", {
-                model: imageModelName,
-                contentParts: contents[0].parts.length,
-                responseModalities: generateConfig.responseModalities,
-            });
-            // Try streaming approach first
-            let imageData = null;
-            let textContent = "";
-            try {
-                // Await the Promise to get the AsyncIterable
-                const stream = await client.models.generateContentStream({
-                    model: imageModelName,
-                    contents: contents,
-                    config: generateConfig,
-                });
-                // Process the stream
-                for await (const chunk of stream) {
-                    logger.debug("Received chunk", {
-                        hasCandidate: !!chunk.candidates?.[0],
-                        hasContent: !!chunk.candidates?.[0]?.content,
-                        hasParts: !!chunk.candidates?.[0]?.content?.parts,
-                    });
-                    const candidate = chunk.candidates?.[0];
-                    if (candidate?.content?.parts) {
-                        for (const part of candidate.content.parts) {
-                            // Check for image data
-                            if ("inlineData" in part && part.inlineData?.data) {
-                                const foundImageData = part.inlineData.data;
-                                imageData = foundImageData;
-                                const mimeType = part.inlineData.mimeType || "image/png";
-                                logger.info("Image generation successful", {
-                                    model: imageModelName,
-                                    mimeType,
-                                    dataLength: foundImageData.length,
-                                    responseTime: Date.now() - startTime,
-                                });
-                                const result = {
-                                    content: `Generated image using ${imageModelName} (${mimeType})`,
-                                    imageOutput: {
-                                        base64: foundImageData,
-                                    },
-                                    provider: this.providerName,
-                                    model: imageModelName,
-                                    usage: {
-                                        input: this.estimateTokenCount(prompt),
-                                        output: 0,
-                                        total: this.estimateTokenCount(prompt),
-                                    },
-                                };
-                                return await this.enhanceResult(result, options, startTime);
-                            }
-                            // Check for text content
-                            if ("text" in part && part.text) {
-                                textContent += part.text;
-                                logger.debug("Received text content", {
-                                    text: part.text.substring(0, 100),
-                                });
-                            }
-                        }
-                    }
-                }
-            }
-            catch (streamError) {
-                logger.debug("Streaming failed, trying non-streaming approach", {
-                    error: streamError instanceof Error
-                        ? streamError.message
-                        : String(streamError),
-                });
-            }
-            // If no image was found, try non-streaming approach
-            if (!imageData) {
-                logger.debug("Trying non-streaming approach");
-                const response = await client.models.generateContent({
-                    model: imageModelName,
-                    contents: contents,
-                    config: generateConfig,
-                });
-                const candidate = response.candidates?.[0];
-                if (candidate?.content?.parts) {
-                    for (const part of candidate.content.parts) {
-                        if ("inlineData" in part && part.inlineData?.data) {
-                            const foundImageData = part.inlineData.data;
-                            imageData = foundImageData;
-                            const mimeType = part.inlineData.mimeType || "image/png";
-                            logger.info("Image generation successful (non-streaming)", {
-                                model: imageModelName,
-                                mimeType,
-                                dataLength: foundImageData.length,
-                                responseTime: Date.now() - startTime,
-                            });
-                            const result = {
-                                content: `Generated image using ${imageModelName} (${mimeType})`,
-                                imageOutput: {
-                                    base64: foundImageData,
-                                },
-                                provider: this.providerName,
-                                model: imageModelName,
-                                usage: {
-                                    input: this.estimateTokenCount(prompt),
-                                    output: 0,
-                                    total: this.estimateTokenCount(prompt),
-                                },
-                            };
-                            return await this.enhanceResult(result, options, startTime);
-                        }
-                        if ("text" in part && part.text) {
-                            textContent += part.text;
-                        }
-                    }
-                }
-            }
-            // If we reach here, no image was generated
-            logger.warn("No image data found in response", {
-                model: imageModelName,
-                prompt: prompt.substring(0, 100),
-                hasTextContent: !!textContent,
-                textContent: textContent.substring(0, 200),
-            });
-            throw new ProviderError(textContent ||
-                `Image generation completed but no image data was returned. This may indicate an issue with the model "${imageModelName}" or the prompt: "${prompt}". Please try again or use a different model.`, this.providerName);
-        }
-        catch (error) {
-            logger.error("Image generation failed", {
-                error: error instanceof Error ? error.message : String(error),
-                model: imageModelName,
-                prompt: prompt.substring(0, 100),
-            });
-            throw this.handleProviderError(error);
-        }
-    }
-    /**
-     * Detect image MIME type from buffer
-     */
-    detectImageType(buffer) {
-        // Check PNG signature
-        if (buffer.length >= 8 &&
-            buffer[0] === 0x89 &&
-            buffer[1] === 0x50 &&
-            buffer[2] === 0x4e &&
-            buffer[3] === 0x47) {
-            return "image/png";
-        }
-        // Check JPEG signature
-        if (buffer.length >= 3 &&
-            buffer[0] === 0xff &&
-            buffer[1] === 0xd8 &&
-            buffer[2] === 0xff) {
-            return "image/jpeg";
-        }
-        // Check WebP signature
-        if (buffer.length >= 12 &&
-            buffer[0] === 0x52 &&
-            buffer[1] === 0x49 &&
-            buffer[2] === 0x46 &&
-            buffer[3] === 0x46 &&
-            buffer[8] === 0x57 &&
-            buffer[9] === 0x45 &&
-            buffer[10] === 0x42 &&
-            buffer[11] === 0x50) {
-            return "image/webp";
-        }
-        // Check GIF signature
-        if (buffer.length >= 6 &&
-            buffer[0] === 0x47 &&
-            buffer[1] === 0x49 &&
-            buffer[2] === 0x46) {
-            return "image/gif";
-        }
-        // Default to PNG if unknown
-        return "image/png";
-    }
-    /**
-     * Estimate token count from text using centralized estimation with provider multipliers
-     */
-    estimateTokenCount(text) {
-        return estimateTokens(text, "google-ai");
-    }
-    // executeGenerate removed - BaseProvider handles all generation with tools
-    /**
-     * Run the file preprocessing this provider's native paths depend on.
-     *
-     * AI Studio overrides both `generate()` and `executeStream()` and routes
-     * straight to the native SDK, so neither reaches
-     * `buildMultimodalMessagesArray` — the place that turns `input.files` into
-     * text, images, PDFs and `nativeAudioFiles`. `BaseProvider.stream()` does
-     * build messages, but onto a throwaway clone whose result is discarded, so
-     * the real `options.input` came through untouched.
-     *
-     * The consequence was asymmetric and easy to miss: `generate()` did this
-     * inline and worked, while `stream()` silently dropped every attached file —
-     * not just audio, but the metadata summary too. Vertex hit the identical bug
-     * (#1258) and solved it with exactly this shape, called from both entry
-     * points.
-     */
-    async preprocessNativeFileInput(options) {
-        // The user-facing aliases (`input.audioFiles`, `input.videoFiles`) are
-        // folded into `input.files` here, exactly as the Vertex client does. Only
-        // `files` is processed below, so without this a caller who used the
-        // documented `audioFiles` field had it silently ignored on both of this
-        // provider's paths.
-        if (options.input) {
-            mergeMediaFileAliases(options.input);
-        }
-        if (options.input?.files && options.input.files.length > 0) {
-            try {
-                // Mutates options.input.text / .images / .pdfFiles / .nativeAudioFiles
-                // in place.
-                await processUnifiedFilesArray(options, 100 * 1024 * 1024, this.providerName);
-            }
-            catch (fileError) {
-                logger.warn(`[GoogleAIStudio] processUnifiedFilesArray threw, continuing without file content: ${fileError instanceof Error ? fileError.message : String(fileError)}`);
-            }
-        }
-        // Runs even without input.files: a caller can populate input.images
-        // directly, and this native path never reaches the shared multimodal
-        // builder that would otherwise normalize the formats.
-        await normalizeVisionImageFormats(options.input);
-    }
-    async executeStream(options, analysisSchema) {
-        const modelName = options.model || this.modelName;
-        // Phase 1: if audio input present, bridge to Gemini Live (Studio) using @google/genai
-        if (options.input?.audio) {
-            return await this.executeAudioStreamViaGeminiLive(options);
-        }
-        // #1258, for this provider: stream() must run the same file preprocessing
-        // generate() does, or attached files are dropped on this path alone.
-        await this.preprocessNativeFileInput(options);
-        // Structured output (analysisSchema, JSON format, or schema) is incompatible with tools on Gemini.
-        const wantsStructuredOutput = Boolean(analysisSchema || options.output?.format === "json" || options.schema);
-        // Tool filter (a0269210): trust options.tools — caller (BaseProvider.stream)
-        // already merged MCP/built-in tools with user tools and applied any
-        // enabledToolNames filter. Re-attaching getAllTools() here would clobber
-        // that filter and re-introduce filtered-out tools.
-        const shouldUseTools = !options.disableTools && this.supportsTools() && !wantsStructuredOutput;
-        const optionTools = options.tools || {};
-        // Merge into options for native SDK path
-        let mergedOptions = {
-            ...options,
-            tools: optionTools,
-        };
-        // Tools + JSON conflict (Gemini), via the shared predicate.
-        const wantsJsonOutput = options.output?.format === "json" || options.schema;
-        const exclusionInForce = isToolsSchemaExclusionInForce(this.providerName, modelName, !mergedOptions.disableTools, Object.keys(mergedOptions.tools ?? {}).length);
-        if (wantsJsonOutput && exclusionInForce) {
-            logger.warn("[GoogleAIStudio] Gemini does not support tools and JSON schema output simultaneously. Disabling tools for this request.");
-            mergedOptions = { ...mergedOptions, disableTools: true, tools: {} };
-        }
-        const hasActiveTools = shouldUseTools &&
-            !mergedOptions.disableTools &&
-            mergedOptions.tools &&
-            Object.keys(mergedOptions.tools).length > 0;
-        if (hasActiveTools) {
-            logger.info("[GoogleAIStudio] Routing to native @google/genai SDK for tool calling", {
-                model: modelName,
-                totalToolCount: Object.keys(mergedOptions.tools ?? {}).length,
-            });
-        }
-        // Route ALL models through native @google/genai SDK (no more @ai-sdk/google dependency)
-        return this.executeNativeGemini3Stream(mergedOptions);
-    }
-    /**
-     * Execute stream using native @google/genai SDK
-     * Uses @google/genai directly for all Gemini models (2.0, 2.5, 3.x)
-     */
-    async executeNativeGemini3Stream(options) {
-        const modelName = options.model || this.modelName;
-        return withClientStreamSpan({
-            name: "neurolink.provider.stream",
-            tracer: tracers.provider,
-            attributes: {
-                [ATTR.GEN_AI_SYSTEM]: "google-ai",
-                [ATTR.GEN_AI_MODEL]: modelName,
-                [ATTR.GEN_AI_OPERATION]: "stream",
-                [ATTR.NL_PROVIDER]: this.providerName,
-            },
-        }, async (span) => {
-            const startTime = Date.now();
-            const timeout = this.getTimeout(options);
-            const timeoutController = createTimeoutController(timeout, this.providerName, "stream");
-            try {
-                const apiKey = this.getApiKey();
-                const client = await createGoogleGenAIClient(apiKey, this.getBaseURL());
-                logger.debug("[GoogleAIStudio] Using native @google/genai for Gemini 3", {
-                    model: modelName,
-                    hasTools: !!options.tools && Object.keys(options.tools).length > 0,
-                });
-                // Build contents from input. Prepend prior conversation turns so
-                // multi-turn callers (memory, loop REPL, agent flows) actually
-                // carry context — the previous build started fresh from the
-                // current user input only, which silently dropped history.
-                //
-                // `buildUserPartsWithMultimodal` is the shared helper that also
-                // attaches `input.images` and `input.pdfFiles` as `inlineData`
-                // parts. The previous AI Studio path pushed only `{ text }` and
-                // silently dropped both, which is why the model legitimately
-                // reported "no image attached" on multimodal calls.
-                const currentContents = [];
-                prependConversationMessages(currentContents, options.conversationMessages);
-                const userParts = await buildUserPartsWithMultimodal(options.input, options.input.text, "[GoogleAIStudio:stream]");
-                currentContents.push({
-                    role: "user",
-                    parts: userParts,
-                });
-                // Convert tools
-                let toolsConfig;
-                let declarationsResult;
-                if (options.tools &&
-                    Object.keys(options.tools).length > 0 &&
-                    !options.disableTools) {
-                    const result = toNativeToolDeclarations(options.tools, "functionDeclarations");
-                    declarationsResult = result;
-                    toolsConfig = result.toolsConfig;
-                    logger.debug("[GoogleAIStudio] Converted tools for native SDK", {
-                        toolCount: toolsConfig[0].functionDeclarations.length,
-                        toolNames: toolsConfig[0].functionDeclarations.map((t) => t.name),
-                    });
-                }
-                // Native JSON / schema enforcement: when no tools are being sent
-                // (the AI Studio orchestrator above already force-disables tools
-                // whenever JSON/schema output is requested), enforce the response
-                // shape natively via responseMimeType / responseSchema. Without
-                // this, JSON output was best-effort prompting only.
-                const wantsNativeJson = !toolsConfig &&
-                    (options.output?.format === "json" || !!options.schema);
-                const nativeResponseSchema = wantsNativeJson && options.schema
-                    ? buildGeminiResponseSchema(options.schema)
-                    : undefined;
-                const config = buildNativeConfig({
-                    ...options,
-                    // Effective model (falls back to the instance default) so the
-                    // registry sampling-support check sees the model actually used.
-                    model: modelName,
-                    wantsJsonOutput: wantsNativeJson,
-                    responseSchema: nativeResponseSchema,
-                }, toolsConfig);
-                const maxSteps = computeMaxSteps(options.maxSteps);
-                // Compose abort signal from user signal + timeout
-                const composedSignal = composeAbortSignals(options.abortSignal, timeoutController?.controller.signal);
-                // Create a push-based text channel so the caller receives tokens as
-                // they arrive from the network rather than after full buffering.
-                const channel = createStreamChannel();
-                // Shared mutable state updated by the background agentic loop.
-                const allToolCalls = [];
-                // Mirror the Vertex Gemini stream path: track tool executions so
-                // the storage hook can persist real outputs and StreamResult can
-                // surface toolsUsed/toolExecutions for tool-bearing turns.
-                const toolExecutions = [];
-                // analyticsResolvers lets the background loop settle the analytics
-                // promise once token counts are known (after the loop completes).
-                let analyticsResolve;
-                let analyticsReject;
-                const analyticsPromise = new Promise((res, rej) => {
-                    analyticsResolve = res;
-                    analyticsReject = rej;
-                });
-                // Shared metadata object mutated by the background loop so the
-                // returned object reflects the final values after stream completion.
-                const metadata = {
-                    streamId: `native-${Date.now()}`,
-                    startTime,
-                    responseTime: 0,
-                    totalToolExecutions: 0,
-                };
-                // Run the agentic loop in the background without awaiting it here,
-                // so we can return the StreamResult (with channel.iterable) immediately.
-                const loopPromise = (async () => {
-                    let lastStepText = "";
-                    let totalInputTokens = 0;
-                    let totalOutputTokens = 0;
-                    let totalCacheReadTokens = 0;
-                    let totalReasoningTokens = 0;
-                    let step = 0;
-                    // Cheap trigger for the in-turn reclaim, mirroring the Vertex twin.
-                    // Planning serializes the WHOLE accumulated history to estimate it,
-                    // so running it unconditionally charges that once per step for the
-                    // life of the turn; the guard tracks real prompt counts plus
-                    // measured growth instead, and it supplies the observed count that
-                    // calibrates the planner's char estimate.
-                    const contextGuard = createContextGuard(getContextWindowSize("googleAiStudio", modelName));
-                    try {
-                        // Agentic loop for tool calling
-                        // The turn itself now runs on the shared engine: the step cap,
-                        // tool dispatch, the failure breaker and usage accumulation are
-                        // engine-owned. Everything below is the provider half — building
-                        // one request, and the per-step side effects the old loop
-                        // performed inline.
-                        const baseAdapter = createGeminiLoopAdapter({
-                            providerLabel: "GoogleAIStudio",
-                            maxSteps,
-                            // Same threshold the hand-rolled dispatcher used, which is
-                            // what makes an always-failing tool dispatch exactly twice.
-                            toolFailureBreaker: { maxRetries: DEFAULT_TOOL_MAX_RETRIES },
-                            liveTools: options.tools ?? {},
-                            ...(declarationsResult
-                                ? { declarations: declarationsResult }
-                                : {}),
-                            buildRequest: (contents) => ({
-                                model: modelName,
-                                contents,
-                                config,
-                                ...(composedSignal
-                                    ? { httpOptions: { signal: composedSignal } }
-                                    : {}),
-                            }),
-                            sendStep: async (request) => client.models.generateContentStream(request),
-                            noteUsage: (inputTokens, outputTokens) => {
-                                contextGuard.noteUsage(inputTokens, outputTokens);
-                            },
-                            // Pure: the engine assigns what this returns. The old loop
-                            // reclaimed in place because it owned `currentContents`.
-                            planReclaim: (contents, stepIndex) => {
-                                if (stepIndex !== 0 && !contextGuard.shouldStop()) {
-                                    return undefined;
-                                }
-                                const working = [...contents];
-                                if (!reclaimAiStudioContext(working, modelName, contextGuard.projectedNextPromptTokens)) {
-                                    return undefined;
-                                }
-                                contextGuard.resetAfterReclaim();
-                                return working;
-                            },
-                        });
-                        // Wrapped because these fire once PER STEP in the loop this
-                        // replaces, and buildToolResultMessages is the only hook that
-                        // runs per step with exactly that step's results. Reading them
-                        // off the turn's final result would batch every step into one
-                        // late write and lose the per-step thought signature.
-                        const adapter = {
-                            ...baseAdapter,
-                            buildToolResultMessages: (contents, stepResult, toolResults) => {
-                                step++;
-                                for (const call of stepResult.toolCalls) {
-                                    span.addEvent("gen_ai.tool_call", {
-                                        "tool.name": call.name,
-                                        "tool.step": step,
-                                    });
-                                }
-                                lastStepText = stepResult.text || lastStepText;
-                                for (const call of stepResult.toolCalls) {
-                                    allToolCalls.push({
-                                        toolName: call.name,
-                                        args: call.args,
-                                    });
-                                }
-                                for (const result of toolResults) {
-                                    toolExecutions.push({
-                                        name: result.name,
-                                        input: result.args,
-                                        output: result.output,
-                                    });
-                                }
-                                if (toolResults.length > 0) {
-                                    const stepThoughtSig = extractThoughtSignature(stepResult.raw.rawResponseParts);
-                                    withTimeout(this.handleToolExecutionStorage(stepResult.toolCalls.map((call, i) => ({
-                                        toolName: call.name,
-                                        args: call.args,
-                                        ...(i === 0 && stepThoughtSig
-                                            ? { thoughtSignature: stepThoughtSig }
-                                            : {}),
-                                        stepIndex: step,
-                                    })), toolResults.map((result) => ({
-                                        toolName: result.name,
-                                        output: result.output,
-                                        stepIndex: step,
-                                    })), options, new Date()), TOOL_STORAGE_TIMEOUT_MS, "tool storage write timed out").catch((error) => {
-                                        logger.warn("[GoogleAIStudio] Failed to store native tool executions", {
-                                            error: error instanceof Error
-                                                ? error.message
-                                                : String(error),
-                                        });
-                                    });
-                                }
-                                const next = baseAdapter.buildToolResultMessages(contents, stepResult, toolResults);
-                                // Project this step's growth: the appended tool results ride
-                                // the next prompt, which the provider has not reported on yet.
-                                try {
-                                    const appended = next[next.length - 1];
-                                    contextGuard.noteAppendedChars(JSON.stringify(appended?.parts ?? []).length);
-                                }
-                                catch {
-                                    /* estimation is best-effort — never break the loop */
-                                }
-                                return next;
-                            },
-                        };
-                        const engineTools = {};
-                        for (const [name, tool] of Object.entries(options.tools ?? {})) {
-                            const execute = tool?.execute;
-                            if (!execute) {
-                                continue;
-                            }
-                            engineTools[name] = {
-                                execute: async (args, opts) => execute(args, opts),
-                            };
-                        }
-                        const { stream: engineStream, resultPromise } = runAgenticLoop(adapter, currentContents, {
-                            tools: engineTools,
-                            ...(composedSignal ? { abortSignal: composedSignal } : {}),
-                        });
-                        const pump = (async () => {
-                            for await (const chunk of engineStream) {
-                                channel.push(chunk);
-                            }
-                        })();
-                        let engineResult;
-                        try {
-                            engineResult = await resultPromise;
-                        }
-                        catch (error) {
-                            await pump.catch(() => { });
-                            logger.error("[GoogleAIStudio] Native SDK error", error);
-                            throw this.handleProviderError(error);
-                        }
-                        await pump;
-                        totalInputTokens += engineResult.usage.inputTokens;
-                        totalOutputTokens += engineResult.usage.outputTokens;
-                        totalCacheReadTokens += engineResult.usage.cacheReadTokens ?? 0;
-                        totalReasoningTokens += engineResult.usage.reasoningTokens ?? 0;
-                        // The turn produced a final answer when the model stopped
-                        // calling tools of its own accord, rather than being cut off at
-                        // the cap.
-                        const completedWithFinalAnswer = engineResult.toolCalls.length === 0 ||
-                            engineResult.finishReason !== "tool-calls";
-                        // Handle max-steps termination: if the model was still calling
-                        // tools when we hit the limit, push a synthetic final message.
-                        const hitStepLimitWithoutFinalAnswer = step >= maxSteps && !completedWithFinalAnswer;
-                        if (hitStepLimitWithoutFinalAnswer) {
-                            const fallback = handleMaxStepsTermination("[GoogleAIStudio]", step, maxSteps, "", // finalText is empty — model didn't stop on its own
-                            lastStepText);
-                            if (fallback) {
-                                channel.push({ content: fallback });
-                            }
-                        }
-                        const responseTime = Date.now() - startTime;
-                        // Update shared metadata so the returned object reflects final values.
-                        metadata.responseTime = responseTime;
-                        metadata.totalToolExecutions = allToolCalls.length;
-                        // Set token usage and finish reason on the span
-                        span.setAttribute(ATTR.GEN_AI_INPUT_TOKENS, totalInputTokens);
-                        span.setAttribute(ATTR.GEN_AI_OUTPUT_TOKENS, totalOutputTokens);
-                        span.setAttribute(ATTR.GEN_AI_FINISH_REASON, hitStepLimitWithoutFinalAnswer ? "max_steps" : "stop");
-                        // Gemini promptTokenCount is OVERLAPPING: it already includes
-                        // cachedContentTokenCount. Subtract once here so calculateCost
-                        // bills the cached portion at the cheaper cacheRead rate without
-                        // double-counting. Total billable tokens are conserved.
-                        const adjustedInputTokens = Math.max(0, totalInputTokens - totalCacheReadTokens);
-                        analyticsResolve({
-                            provider: this.providerName,
-                            model: modelName,
-                            tokenUsage: {
-                                input: adjustedInputTokens,
-                                // Thinking tokens are billed at the output rate but Gemini
-                                // does NOT include them in candidatesTokenCount, so they
-                                // are folded into `output` — what calculateCost bills at
-                                // the output rate — with `reasoning` as the subset.
-                                output: totalOutputTokens + totalReasoningTokens,
-                                total: adjustedInputTokens +
-                                    totalCacheReadTokens +
-                                    totalOutputTokens +
-                                    totalReasoningTokens,
-                                ...(totalCacheReadTokens > 0
-                                    ? { cacheReadTokens: totalCacheReadTokens }
-                                    : {}),
-                                ...(totalReasoningTokens > 0
-                                    ? { reasoning: totalReasoningTokens }
-                                    : {}),
-                            },
-                            requestDuration: responseTime,
-                            timestamp: new Date().toISOString(),
-                        });
-                        channel.close();
-                    }
-                    catch (err) {
-                        channel.error(err);
-                        analyticsReject(err);
-                    }
-                    finally {
-                        timeoutController?.cleanup();
-                    }
-                })();
-                // Suppress unhandled-rejection warnings on loopPromise — errors are
-                // forwarded to the channel and will surface when the caller iterates.
-                loopPromise.catch(() => undefined);
-                const result = {
-                    stream: channel.iterable,
-                    provider: this.providerName,
-                    model: modelName,
-                    toolCalls: allToolCalls,
-                    analytics: analyticsPromise,
-                    metadata,
-                };
-                // Surface tools-used + executions via getters so they resolve at
-                // access time, after the background loop has populated the live
-                // arrays. Same lazy pattern used for `structuredOutput` elsewhere.
-                Object.defineProperty(result, "toolsUsed", {
-                    enumerable: true,
-                    configurable: true,
-                    get: () => allToolCalls.map((tc) => tc.toolName),
-                });
-                Object.defineProperty(result, "toolExecutions", {
-                    enumerable: true,
-                    configurable: true,
-                    get: () => transformToolExecutions(toolExecutions),
-                });
-                return result;
-            }
-            finally {
-                // Timeout controller cleanup is managed inside the background loop
-            }
-        }, (r) => r.stream, (r, wrapped) => ({ ...r, stream: wrapped }));
-    }
-    /**
-     * Execute generate using native @google/genai SDK for Gemini 3 models
-     * This bypasses @ai-sdk/google to properly handle thought_signature
-     */
-    async executeNativeGemini3Generate(options) {
-        const modelName = options.model || this.modelName;
-        return withClientSpan({
-            name: "neurolink.provider.generate",
-            tracer: tracers.provider,
-            attributes: {
-                [ATTR.GEN_AI_SYSTEM]: "google-ai",
-                [ATTR.GEN_AI_MODEL]: modelName,
-                [ATTR.GEN_AI_OPERATION]: "generate",
-                [ATTR.NL_PROVIDER]: this.providerName,
-            },
-        }, async (span) => {
-            const startTime = Date.now();
-            const timeout = this.getTimeout(options);
-            const timeoutController = createTimeoutController(timeout, this.providerName, "generate");
-            try {
-                const apiKey = this.getApiKey();
-                const client = await createGoogleGenAIClient(apiKey, this.getBaseURL());
-                logger.debug("[GoogleAIStudio] Using native @google/genai for Gemini 3 generate", {
-                    model: modelName,
-                    hasTools: !!options.tools && Object.keys(options.tools).length > 0,
-                });
-                // Build contents from input
-                // Prefer input.text over prompt — processCSVFilesForNativeSDK enriches
-                // input.text with inlined CSV data, so using prompt first would discard it.
-                const promptText = options.input?.text || options.prompt || "";
-                // Prepend prior conversation turns so multi-turn generate calls
-                // see history; otherwise the native generate path silently drops
-                // every turn before the current prompt.
-                //
-                // `buildUserPartsWithMultimodal` also attaches inline image / PDF
-                // parts. Without it the request body was text-only and the model
-                // legitimately reported "no image / PDF attached".
-                const currentContents = [];
-                prependConversationMessages(currentContents, options.conversationMessages);
-                const userParts = await buildUserPartsWithMultimodal(options.input, promptText, "[GoogleAIStudio:generate]");
-                currentContents.push({
-                    role: "user",
-                    parts: userParts,
-                });
-                // Convert tools (a0269210: trust options.tools — already merged + filtered upstream)
-                let toolsConfig;
-                let declarationsResult;
-                const shouldUseTools = !options.disableTools;
-                // Structured output (JSON format or schema) is incompatible with
-                // tools on Gemini — routed through the shared predicate so this
-                // decision matches stream()'s orchestrator (previously this path
-                // had no proactive check and silently dropped the schema instead).
-                const wantsNativeJsonRequested = Boolean(options.output?.format === "json" || options.schema);
-                const exclusionInForce = isToolsSchemaExclusionInForce(this.providerName, modelName, shouldUseTools, Object.keys(options.tools || {}).length);
-                if (wantsNativeJsonRequested && exclusionInForce) {
-                    logger.warn("[GoogleAIStudio] Gemini does not support tools and JSON schema output simultaneously. Disabling tools for this request (generate()).");
-                }
-                // Both conjuncts, matching the warning directly above and the
-                // stream path's gate. `isToolsSchemaExclusionInForce` answers "does
-                // the tools/schema exclusion APPLY to this provider and model", and
-                // is true for any Gemini request that has tools at all — it is not
-                // "the exclusion is triggered". Testing `!exclusionInForce` alone
-                // therefore made this branch reachable only when there were NO
-                // tools, so every caller-supplied tool was dropped on the generate
-                // path whether or not structured output was ever requested.
-                if (shouldUseTools &&
-                    !(wantsNativeJsonRequested && exclusionInForce)) {
-                    const tools = options.tools || {};
-                    if (Object.keys(tools).length > 0) {
-                        const result = toNativeToolDeclarations(tools, "functionDeclarations");
-                        declarationsResult = result;
-                        toolsConfig = result.toolsConfig;
-                        logger.debug("[GoogleAIStudio] Converted tools for native SDK generate", {
-                            toolCount: toolsConfig[0].functionDeclarations.length,
-                            toolNames: toolsConfig[0].functionDeclarations.map((t) => t.name),
-                        });
-                    }
-                }
-                // Native JSON / schema enforcement (generate path). Mirrors the
-                // stream block above; only set when no tools are being sent
-                // because Gemini cannot combine function calling with JSON mime.
-                const wantsNativeJson = !toolsConfig && wantsNativeJsonRequested;
-                const nativeResponseSchema = wantsNativeJson && options.schema
-                    ? buildGeminiResponseSchema(options.schema)
-                    : undefined;
-                const config = buildNativeConfig({
-                    ...options,
-                    // Effective model (falls back to the instance default) so the
-                    // registry sampling-support check sees the model actually used.
-                    model: modelName,
-                    wantsJsonOutput: wantsNativeJson,
-                    responseSchema: nativeResponseSchema,
-                }, toolsConfig);
-                const composedSignal = composeAbortSignals(options.abortSignal, timeoutController?.controller.signal);
-                const maxSteps = computeMaxSteps(options.maxSteps);
-                let finalText = "";
-                let lastStepText = "";
-                let totalInputTokens = 0;
-                let totalOutputTokens = 0;
-                let totalCacheReadTokens = 0;
-                let totalReasoningTokens = 0;
-                const allToolCalls = [];
-                const toolExecutions = [];
-                let step = 0;
-                // Cheap reclaim trigger — see the stream twin.
-                const contextGuard = createContextGuard(getContextWindowSize("googleAiStudio", modelName));
-                // Agentic loop for tool calling
-                // Same shared engine as the streaming twin. This path has no
-                // consumer channel — generate() returns one result rather than
-                // streaming — so the engine's stream is drained and discarded, and
-                // the turn's text comes from the result.
-                const baseAdapter = createGeminiLoopAdapter({
-                    providerLabel: "GoogleAIStudio",
-                    maxSteps,
-                    toolFailureBreaker: { maxRetries: DEFAULT_TOOL_MAX_RETRIES },
-                    liveTools: options.tools ?? {},
-                    ...(declarationsResult ? { declarations: declarationsResult } : {}),
-                    buildRequest: (contents) => ({
-                        model: modelName,
-                        contents,
-                        config,
-                        ...(composedSignal
-                            ? { httpOptions: { signal: composedSignal } }
-                            : {}),
-                    }),
-                    sendStep: async (request) => client.models.generateContentStream(request),
-                    noteUsage: (inputTokens, outputTokens) => {
-                        contextGuard.noteUsage(inputTokens, outputTokens);
-                    },
-                    planReclaim: (contents, stepIndex) => {
-                        if (stepIndex !== 0 && !contextGuard.shouldStop()) {
-                            return undefined;
-                        }
-                        const working = [...contents];
-                        if (!reclaimAiStudioContext(working, modelName, contextGuard.projectedNextPromptTokens)) {
-                            return undefined;
-                        }
-                        contextGuard.resetAfterReclaim();
-                        return working;
-                    },
-                });
-                const adapter = {
-                    ...baseAdapter,
-                    buildToolResultMessages: (contents, stepResult, toolResults) => {
-                        step++;
-                        for (const call of stepResult.toolCalls) {
-                            span.addEvent("gen_ai.tool_call", {
-                                "tool.name": call.name,
-                                "tool.step": step,
-                            });
-                            allToolCalls.push({ toolName: call.name, args: call.args });
-                        }
-                        lastStepText = stepResult.text || lastStepText;
-                        for (const result of toolResults) {
-                            toolExecutions.push({
-                                name: result.name,
-                                input: result.args,
-                                output: result.output,
-                            });
-                        }
-                        if (toolResults.length > 0) {
-                            const stepThoughtSig = extractThoughtSignature(stepResult.raw.rawResponseParts);
-                            withTimeout(this.handleToolExecutionStorage(stepResult.toolCalls.map((call, i) => ({
-                                toolName: call.name,
-                                args: call.args,
-                                ...(i === 0 && stepThoughtSig
-                                    ? { thoughtSignature: stepThoughtSig }
-                                    : {}),
-                                stepIndex: step,
-                            })), toolResults.map((result) => ({
-                                toolName: result.name,
-                                output: result.output,
-                                stepIndex: step,
-                            })), options, new Date()), TOOL_STORAGE_TIMEOUT_MS, "tool storage write timed out").catch((error) => {
-                                logger.warn("[GoogleAIStudio] Failed to store native tool executions", {
-                                    error: error instanceof Error ? error.message : String(error),
-                                });
-                            });
-                        }
-                        const next = baseAdapter.buildToolResultMessages(contents, stepResult, toolResults);
-                        try {
-                            const appended = next[next.length - 1];
-                            contextGuard.noteAppendedChars(JSON.stringify(appended?.parts ?? []).length);
-                        }
-                        catch {
-                            /* estimation is best-effort — never break the loop */
-                        }
-                        return next;
-                    },
-                };
-                const engineTools = {};
-                for (const [name, tool] of Object.entries(options.tools ?? {})) {
-                    const execute = tool?.execute;
-                    if (!execute) {
-                        continue;
-                    }
-                    engineTools[name] = {
-                        execute: async (args, opts) => execute(args, opts),
-                    };
-                }
-                const { stream: engineStream, resultPromise } = runAgenticLoop(adapter, currentContents, {
-                    tools: engineTools,
-                    ...(composedSignal ? { abortSignal: composedSignal } : {}),
-                });
-                // Drained, not consumed: nothing streams out of generate(), but an
-                // undrained channel would stall the engine mid-turn.
-                const drain = (async () => {
-                    for await (const chunk of engineStream) {
-                        void chunk;
-                    }
-                })();
-                let engineResult;
-                try {
-                    engineResult = await resultPromise;
-                }
-                catch (error) {
-                    await drain.catch(() => { });
-                    logger.error("[GoogleAIStudio] Native SDK generate error", error);
-                    throw this.handleProviderError(error);
-                }
-                await drain;
-                totalInputTokens += engineResult.usage.inputTokens;
-                totalOutputTokens += engineResult.usage.outputTokens;
-                totalCacheReadTokens += engineResult.usage.cacheReadTokens ?? 0;
-                totalReasoningTokens += engineResult.usage.reasoningTokens ?? 0;
-                finalText = engineResult.text;
-                finalText = handleMaxStepsTermination("[GoogleAIStudio]", step, maxSteps, finalText, lastStepText);
-                const responseTime = Date.now() - startTime;
-                // Set token usage and finish reason on the span
-                span.setAttribute(ATTR.GEN_AI_INPUT_TOKENS, totalInputTokens);
-                span.setAttribute(ATTR.GEN_AI_OUTPUT_TOKENS, totalOutputTokens);
-                span.setAttribute(ATTR.GEN_AI_FINISH_REASON, step >= maxSteps ? "max_steps" : "stop");
-                // Build EnhancedGenerateResult and route through enhanceResult so
-                // analytics / evaluation / tracing stay attached. The native AI
-                // Studio generate path bypasses BaseProvider.generate(), so
-                // skipping enhanceResult would silently drop those features.
-                // Gemini promptTokenCount is OVERLAPPING (already includes
-                // cachedContentTokenCount). Subtract once so the cached portion is
-                // billed at the cheaper cacheRead rate without double-counting.
-                const adjustedInputTokens = Math.max(0, totalInputTokens - totalCacheReadTokens);
-                const baseResult = {
-                    content: finalText,
-                    provider: this.providerName,
-                    model: modelName,
-                    usage: {
-                        input: adjustedInputTokens,
-                        // Thinking tokens are billed at the output rate but Gemini
-                        // does NOT include them in candidatesTokenCount, so they are
-                        // folded into `output` — what calculateCost bills at the
-                        // output rate — with `reasoning` as the subset.
-                        output: totalOutputTokens + totalReasoningTokens,
-                        total: adjustedInputTokens +
-                            totalCacheReadTokens +
-                            totalOutputTokens +
-                            totalReasoningTokens,
-                        ...(totalCacheReadTokens > 0
-                            ? { cacheReadTokens: totalCacheReadTokens }
-                            : {}),
-                        ...(totalReasoningTokens > 0
-                            ? { reasoning: totalReasoningTokens }
-                            : {}),
-                    },
-                    ...(totalReasoningTokens > 0 && {
-                        reasoningTokens: totalReasoningTokens,
-                    }),
-                    responseTime,
-                    toolsUsed: allToolCalls.map((tc) => tc.toolName),
-                    toolExecutions: resolveToolExecutionRecords(options, toolExecutions),
-                    enhancedWithTools: allToolCalls.length > 0,
-                };
-                return this.enhanceResult(baseResult, options, startTime);
-            }
-            finally {
-                timeoutController?.cleanup();
-            }
-        });
-    }
-    /**
-     * Override generate to route Gemini 3 models with tools to native SDK
-     */
-    async generate(optionsOrPrompt) {
-        // Normalize options
-        const options = typeof optionsOrPrompt === "string"
-            ? { prompt: optionsOrPrompt }
-            : optionsOrPrompt;
-        const modelName = options.model || this.modelName;
-        // Image-generation models reject function-calling. Route them to
-        // executeImageGeneration without merging tools. This must happen
-        // BEFORE getToolsForStream to avoid leaking registered (MCP / built-in)
-        // tools into the image API request, which trips
-        // "Function calling is not enabled for this model".
-        // startsWith (not includes) so a hypothetical text model whose ID
-        // contains an image-model string as a substring isn't silently routed
-        // to executeImageGeneration and stripped of tool support.
-        const isImageModel = IMAGE_GENERATION_MODELS.some((m) => modelName.toLowerCase().startsWith(m.toLowerCase()));
-        if (isImageModel) {
-            logger.info("[GoogleAIStudio] Routing image generation model to executeImageGeneration", { model: modelName });
-            return this.executeImageGeneration(options);
-        }
-        // TTS direct-synthesis mode: synthesise the input text directly (no LLM
-        // call). BaseProvider.runGenerateInActiveContext does the same dispatch
-        // — replicated here because AI Studio's override bypasses that path.
-        if (options.tts?.enabled && !options.tts?.useAiResponse) {
-            logger.info("[GoogleAIStudio] Routing TTS direct-synthesis to handleDirectTTSSynthesis", { model: modelName });
-            return this.handleDirectTTSSynthesis(options, Date.now());
-        }
-        await this.preprocessNativeFileInput(options);
-        // Merge registered (built-in / MCP) tools with caller-supplied tools.
-        // AI Studio's generate() bypasses BaseProvider.generate(), so the
-        // ToolsManager-driven merge that normally injects sdk.registerTool()
-        // entries never runs here. Without this call, registered tools never
-        // reach the native function-calling path.
-        const baseTools = !options.disableTools
-            ? await this.getToolsForStream(options)
-            : {};
-        let mergedOptions = {
-            ...options,
-            tools: baseTools,
-        };
-        // Check for tools + JSON schema conflict (Gemini limitation)
-        const wantsJsonOutput = options.output?.format === "json" || options.schema;
-        if (wantsJsonOutput &&
-            mergedOptions.tools &&
-            Object.keys(mergedOptions.tools).length > 0 &&
-            !mergedOptions.disableTools) {
-            logger.warn("[GoogleAIStudio] Gemini does not support tools and JSON schema output simultaneously. Disabling tools for this request.");
-            mergedOptions = { ...mergedOptions, disableTools: true, tools: {} };
-        }
-        const hasActiveTools = !mergedOptions.disableTools &&
-            mergedOptions.tools &&
-            Object.keys(mergedOptions.tools).length > 0;
-        if (hasActiveTools) {
-            logger.info("[GoogleAIStudio] Routing generate to native @google/genai SDK for tool calling", {
-                model: modelName,
-                totalToolCount: Object.keys(mergedOptions.tools ?? {}).length,
-            });
-        }
-        // Route ALL models through native @google/genai SDK (no more @ai-sdk/google dependency).
-        // Emit Pipeline B `generation:end` so the observability listener
-        // creates a `model.generation` span — AI Studio's native path bypasses
-        // the AI SDK + experimental_telemetry plumbing the same way Vertex's
-        // does, so the event has to be emitted manually.
-        const generateStartTime = Date.now();
-        const inputPrompt = mergedOptions.input?.text ||
-            mergedOptions.prompt ||
-            "";
-        try {
-            // Wrap in `neurolink.executeGeneration` so the observability span
-            // chain (Test: Generate Span Chain) sees a third inner span on the
-            // native @google/genai path — Pipeline A providers get this from
-            // GenerationHandler.executeGeneration; the native path bypasses
-            // GenerationHandler so we add the span here.
-            let result = await withSpan({
-                name: "neurolink.executeGeneration",
-                tracer: tracers.provider,
-                attributes: {
-                    [ATTR.GEN_AI_SYSTEM]: this.providerName,
-                    [ATTR.GEN_AI_MODEL]: modelName,
-                    "neurolink.path": "native.google-genai",
-                },
-            }, async () => this.executeNativeGemini3Generate(mergedOptions));
-            // Pipe through TTS-of-AI-response when caller asks for it. No-op when
-            // tts is disabled or useAiResponse is false.
-            result = await this.synthesizeAIResponseIfNeeded(result, options);
-            this.emitPipelineBGenerationEvent(modelName, result, generateStartTime, true, undefined, inputPrompt);
-            return result;
-        }
-        catch (error) {
-            this.emitPipelineBGenerationEvent(modelName, null, generateStartTime, false, error, inputPrompt);
-            throw error;
-        }
-    }
-    /**
-     * Emit `generation:end` so the Pipeline B observability listener creates
-     * a `model.generation` span for native Google AI Studio generate calls.
-     * Without this hand-off the native path silently disappears from
-     * Pipeline B exporters (Langfuse, custom OTEL collectors).
-     */
-    emitPipelineBGenerationEvent(modelName, result, startTime, success, error, prompt) {
-        const emitter = this.neurolink?.getEventEmitter();
-        if (!emitter) {
-            return;
-        }
-        const usage = result?.usage && typeof result.usage === "object"
-            ? result.usage
-            : { input: 0, output: 0, total: 0 };
-        // Mark on the result so the SDK-level runStandardGenerateRequest knows
-        // this provider already emitted `generation:end` itself and skips its
-        // own duplicate emission. Without this flag the public event listener
-        // (and the observability test) would see two events per generate call.
-        if (result && typeof result === "object") {
-            result._generationEndEmitted =
-                true;
-        }
-        emitter.emit("generation:end", {
-            provider: this.providerName,
-            responseTime: Date.now() - startTime,
-            timestamp: Date.now(),
-            // The Pipeline B listener reads `data.prompt` to populate the
-            // `input` span attribute. Without this, the Observability Spans
-            // test fails with "input capture not working".
-            prompt: prompt || "",
-            result: {
-                content: result?.content || "",
-                usage,
-                model: modelName,
-                provider: this.providerName,
-                finishReason: success ? "stop" : "error",
-            },
-            success,
-            ...(error
-                ? { error: error instanceof Error ? error.message : String(error) }
-                : {}),
-        });
-    }
-    // ===================
-    // HELPER METHODS
-    // ===================
-    async executeAudioStreamViaGeminiLive(options) {
-        const startTime = Date.now();
-        const apiKey = this.getApiKey();
-        // Dynamic import to avoid hard dependency unless audio streaming is used
-        let client;
-        try {
-            client = await createGoogleGenAIClient(apiKey, this.getBaseURL());
-        }
-        catch {
-            throw new AuthenticationError("Missing '@google/genai'. Install with: pnpm add @google/genai", this.providerName);
-        }
-        const model = this.modelName ||
-            process.env.GOOGLE_VOICE_AI_MODEL ||
-            "gemini-2.5-flash-preview-native-audio-dialog";
-        // Simple async queue for yielding audio events to the outer AsyncIterable
-        const queue = [];
-        let resolveNext = null;
-        let done = false;
-        const push = (item) => {
-            if (done) {
-                return;
-            }
-            if (item.type === "audio") {
-                if (resolveNext) {
-                    const fn = resolveNext;
-                    resolveNext = null;
-                    fn({ value: { type: "audio", audio: item.audio }, done: false });
-                    return;
-                }
-            }
-            queue.push(item);
-        };
-        const session = await client.live.connect({
-            model,
-            callbacks: {
-                onopen: () => {
-                    // no-op
-                },
-                onmessage: async (message) => {
-                    try {
-                        const audio = message?.serverContent?.modelTurn?.parts?.[0]?.inlineData;
-                        if (audio?.data) {
-                            const buf = Buffer.from(String(audio.data), "base64");
-                            const chunk = {
-                                data: buf,
-                                sampleRateHz: 24000,
-                                channels: 1,
-                                encoding: "PCM16LE",
-                            };
-                            push({ type: "audio", audio: chunk });
-                        }
-                        if (message?.serverContent?.interrupted) {
-                            // allow consumer to handle; no special action required here
-                        }
-                    }
-                    catch (e) {
-                        push({ type: "error", error: e });
-                    }
-                },
-                onerror: (e) => {
-                    push({ type: "error", error: e });
-                },
-                onclose: (_e) => {
-                    push({ type: "end" });
-                },
-            },
-            config: {
-                responseModalities: ["AUDIO"],
-                speechConfig: {
-                    voiceConfig: { prebuiltVoiceConfig: { voiceName: "Orus" } },
-                },
-            },
-        });
-        // Feed upstream audio frames concurrently
-        (async () => {
-            try {
-                const spec = options.input?.audio;
-                if (!spec) {
-                    logger.debug("[GeminiLive] No audio spec found on input; skipping upstream send");
-                    return;
-                }
-                for await (const frame of spec.frames) {
-                    // Zero-length frame acts as a 'flush' control signal
-                    if (!frame || frame.byteLength === 0) {
-                        try {
-                            if (session.sendInput) {
-                                await session.sendInput({ event: "flush" });
-                            }
-                            else if (session.sendRealtimeInput) {
-                                await session.sendRealtimeInput({ event: "flush" });
-                            }
-                        }
-                        catch (err) {
-                            logger.debug("[GeminiLive] flush control failed (non-fatal)", {
-                                error: err instanceof Error ? err.message : String(err),
-                            });
-                        }
-                        continue;
-                    }
-                    // Convert PCM16LE buffer to base64 and wrap in genai Blob-like object
-                    const base64 = frame.toString("base64");
-                    const mimeType = `audio/pcm;rate=${spec.sampleRateHz || 16000}`;
-                    await session.sendRealtimeInput?.({
-                        media: { data: base64, mimeType },
-                    });
-                }
-                // Best-effort flush signal if supported
-                try {
-                    if (session.sendInput) {
-                        await session.sendInput({ event: "flush" });
-                    }
-                    else if (session.sendRealtimeInput) {
-                        await session.sendRealtimeInput({ event: "flush" });
-                    }
-                }
-                catch (err) {
-                    logger.debug("[GeminiLive] final flush failed (non-fatal)", {
-                        error: err instanceof Error ? err.message : String(err),
-                    });
-                }
-            }
-            catch (e) {
-                push({ type: "error", error: e });
-            }
-        })().catch(() => {
-            // ignore
-        });
-        // AsyncIterable for stream events
-        const asyncIterable = {
-            [Symbol.asyncIterator]() {
-                return {
-                    async next() {
-                        if (queue.length > 0) {
-                            const item = queue.shift();
-                            if (!item) {
-                                // `done: true` selects the IteratorReturnResult arm, whose
-                                // value type accepts undefined.
-                                return { value: undefined, done: true };
-                            }
-                            if (item.type === "audio") {
-                                return {
-                                    value: { type: "audio", audio: item.audio },
-                                    done: false,
-                                };
-                            }
-                            if (item.type === "end") {
-                                done = true;
-                                return { value: undefined, done: true };
-                            }
-                            if (item.type === "error") {
-                                done = true;
-                                throw item.error instanceof Error
-                                    ? item.error
-                                    : new Error(String(item.error));
-                            }
-                        }
-                        if (done) {
-                            return { value: undefined, done: true };
-                        }
-                        return await new Promise((resolve) => {
-                            resolveNext = resolve;
-                        });
-                    },
-                };
-            },
-        };
-        return {
-            stream: asyncIterable,
-            provider: this.providerName,
-            model: model,
-            metadata: {
-                startTime,
-                streamId: `google-ai-audio-${Date.now()}`,
-            },
-        };
-    }
-    getDefaultEmbeddingModel() {
-        return (process.env.GOOGLE_AI_EMBEDDING_MODEL ||
-            process.env.GOOGLE_EMBEDDING_MODEL ||
-            "gemini-embedding-001");
-    }
-    /**
-     * Generate embeddings for text using Google AI Studio embedding models
-     * @param text - The text to embed
-     * @param modelName - The embedding model to use (default: gemini-embedding-001)
-     * @returns Promise resolving to the embedding vector
-     */
-    async embed(text, modelName) {
-        const embeddingModelName = modelName || this.getDefaultEmbeddingModel() || "gemini-embedding-001";
-        logger.debug("Generating embedding", {
-            provider: this.providerName,
-            model: embeddingModelName,
-            textLength: text.length,
-        });
-        try {
-            const apiKey = this.getApiKey();
-            const client = await createGoogleGenAIClient(apiKey, this.getBaseURL());
-            const result = await client.models.embedContent({
-                model: embeddingModelName,
-                contents: [text],
-            });
-            const embedding = result.embeddings?.[0]?.values;
-            if (!embedding) {
-                throw new ProviderError("No embedding returned from Google AI", this.providerName);
-            }
-            logger.debug("Embedding generated successfully", {
-                provider: this.providerName,
-                model: embeddingModelName,
-                embeddingDimension: embedding.length,
-            });
-            return embedding;
-        }
-        catch (error) {
-            logger.error("Embedding generation failed", {
-                error: error instanceof Error ? error.message : String(error),
-                model: embeddingModelName,
-                textLength: text.length,
-            });
-            throw this.handleProviderError(error);
-        }
-    }
-    /**
-     * Generate embeddings for multiple texts in a single batch
-     * @param texts - The texts to embed
-     * @param modelName - The embedding model to use (default: gemini-embedding-001)
-     * @returns Promise resolving to an array of embedding vectors
-     */
-    async embedMany(texts, modelName) {
-        const embeddingModelName = modelName || this.getDefaultEmbeddingModel() || "gemini-embedding-001";
-        logger.debug("Generating batch embeddings", {
-            provider: this.providerName,
-            model: embeddingModelName,
-            count: texts.length,
-        });
-        try {
-            const apiKey = this.getApiKey();
-            const client = await createGoogleGenAIClient(apiKey, this.getBaseURL());
-            const result = await client.models.embedContent({
-                model: embeddingModelName,
-                contents: texts,
-            });
-            const embeddings = (result.embeddings || []).map((e) => e.values || []);
-            logger.debug("Batch embeddings generated successfully", {
-                provider: this.providerName,
-                model: embeddingModelName,
-                count: embeddings.length,
-                embeddingDimension: embeddings[0]?.length,
-            });
-            return embeddings;
-        }
-        catch (error) {
-            logger.error("Batch embedding generation failed", {
-                error: error instanceof Error ? error.message : String(error),
-                model: embeddingModelName,
-                count: texts.length,
-            });
-            throw this.handleProviderError(error);
-        }
-    }
-    getApiKey() {
-        const apiKey = this.credentials?.apiKey ||
-            process.env.GOOGLE_AI_API_KEY ||
-            process.env.GOOGLE_GENERATIVE_AI_API_KEY;
-        if (!apiKey) {
-            throw new AuthenticationError("GOOGLE_AI_API_KEY or GOOGLE_GENERATIVE_AI_API_KEY environment variable is not set", this.providerName);
-        }
-        return apiKey;
-    }
-    // Mirrors mistral.ts's baseURL precedence (credentials override, then env,
-    // then unset — the SDK's own default applies when unset). Blank/whitespace
-    // values are treated as unset so an empty override can't accidentally
-    // clobber the default.
-    getBaseURL() {
-        const baseURL = this.credentials?.baseURL?.trim() ||
-            process.env.GOOGLE_AI_BASE_URL?.trim();
-        return baseURL && baseURL.length > 0 ? baseURL : undefined;
-    }
-}
-export default GoogleAIStudioProvider;
-//# sourceMappingURL=client.js.map