@browserbasehq/orca 3.0.9-alpha-4 → 3.1.0-headers

package/dist/cjs/lib/v3/llm/GoogleClient.js

@@ -0,0 +1,373 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GoogleClient = void 0;
+const genai_1 = require("@google/genai");
+const utils_js_1 = require("../../utils.js");
+const LLMClient_js_1 = require("./LLMClient.js");
+const sdkErrors_js_1 = require("../types/public/sdkErrors.js");
+// Mapping from generic roles to Gemini roles
+const roleMap = {
+    user: "user",
+    assistant: "model",
+    system: "user", // Gemini API prefers system instructions either via system_instruction or at the start of 'user' content
+};
+// Basic safety settings - adjust as needed
+const safetySettings = [
+    {
+        category: genai_1.HarmCategory.HARM_CATEGORY_HARASSMENT,
+        threshold: genai_1.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
+    },
+    {
+        category: genai_1.HarmCategory.HARM_CATEGORY_HATE_SPEECH,
+        threshold: genai_1.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
+    },
+    {
+        category: genai_1.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
+        threshold: genai_1.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
+    },
+    {
+        category: genai_1.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
+        threshold: genai_1.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
+    },
+];
+class GoogleClient extends LLMClient_js_1.LLMClient {
+    type = "google";
+    client;
+    logger;
+    constructor({ logger, // Added logger based on other clients
+    modelName, clientOptions, }) {
+        super(modelName);
+        if (!clientOptions?.apiKey) {
+            // Try to get the API key from the environment variable GOOGLE_API_KEY
+            clientOptions.apiKey = (0, utils_js_1.loadApiKeyFromEnv)("google_legacy", logger);
+        }
+        this.clientOptions = clientOptions;
+        this.client = new genai_1.GoogleGenAI({ apiKey: clientOptions.apiKey });
+        this.modelName = modelName;
+        this.logger = logger;
+        // Determine vision capability based on model name (adjust as needed)
+        this.hasVision =
+            modelName.includes("vision") || modelName.includes("gemini-1.5"); // Example logic
+    }
+    // Helper to convert project's ChatMessage[] to Gemini's Content[]
+    formatMessages(messages, image) {
+        const contents = [];
+        let systemInstruction = null;
+        messages.forEach((msg, index) => {
+            const role = roleMap[msg.role];
+            if (!role) {
+                this.logger({
+                    category: "google",
+                    message: `WARNING: Unsupported role: ${msg.role}`,
+                    level: 1,
+                });
+                return; // Skip unsupported roles
+            }
+            // Handle system messages - prepend to the first user message or use system_instruction if available
+            if (msg.role === "system") {
+                if (typeof msg.content === "string") {
+                    systemInstruction =
+                        (systemInstruction ? systemInstruction + "\n\n" : "") + msg.content;
+                }
+                return; // Don't add system messages directly to contents yet
+            }
+            const parts = [];
+            if (Array.isArray(msg.content)) {
+                msg.content.forEach((partContent) => {
+                    if (partContent.type === "text") {
+                        parts.push({ text: partContent.text });
+                    }
+                    else if (partContent.type === "image_url") {
+                        if ("image_url" in partContent && partContent.image_url?.url) {
+                            // Assuming base64 data URI format: data:[<mediatype>];base64,<data>
+                            const base64Data = partContent.image_url.url.split(",")[1];
+                            const mimeTypeMatch = partContent.image_url.url.match(/^data:(image\/\w+);base64,/);
+                            if (base64Data && mimeTypeMatch) {
+                                parts.push({
+                                    inlineData: { mimeType: mimeTypeMatch[1], data: base64Data },
+                                });
+                            }
+                            else {
+                                this.logger({
+                                    category: "google",
+                                    message: "WARNING: Could not parse image data URI format",
+                                    level: 1,
+                                });
+                            }
+                        }
+                    }
+                });
+            }
+            else if (typeof msg.content === "string") {
+                parts.push({ text: msg.content });
+            }
+            // Add image from options if this is the last message and it's a user message
+            if (image && index === messages.length - 1 && msg.role === "user") {
+                const imageDesc = image.description || LLMClient_js_1.AnnotatedScreenshotText;
+                parts.push({ text: imageDesc }); // Add description first
+                parts.push({
+                    inlineData: {
+                        mimeType: "image/jpeg", // Assuming JPEG, adjust if needed
+                        data: image.buffer.toString("base64"),
+                    },
+                });
+            }
+            // Apply system instruction to the first non-system message if needed
+            if (systemInstruction && contents.length === 0 && role === "user") {
+                const firstPartText = parts.find((p) => "text" in p);
+                if (firstPartText && "text" in firstPartText) {
+                    firstPartText.text = `${systemInstruction}\n\n${firstPartText.text}`;
+                }
+                else {
+                    parts.unshift({ text: systemInstruction });
+                }
+                systemInstruction = null; // Clear after applying
+            }
+            if (parts.length > 0) {
+                contents.push({ role, parts });
+            }
+        });
+        // If system instruction wasn't applied (e.g., no user messages followed it), add it as a final user message
+        if (systemInstruction) {
+            contents.unshift({ role: "user", parts: [{ text: systemInstruction }] });
+        }
+        return contents;
+    }
+    // Helper to convert LLMTool[] to Gemini's Tool[]
+    formatTools(tools) {
+        if (!tools || tools.length === 0) {
+            return undefined;
+        }
+        return [
+            {
+                functionDeclarations: tools.map((tool) => {
+                    let parameters = undefined;
+                    if (tool.parameters) {
+                        parameters = {
+                            type: genai_1.Type.OBJECT,
+                            properties: tool.parameters.properties,
+                            required: tool.parameters.required,
+                        };
+                    }
+                    return {
+                        name: tool.name,
+                        description: tool.description,
+                        parameters: parameters,
+                    };
+                }),
+            },
+        ];
+    }
+    async createChatCompletion({ 
+    // Ensure LLMResponse is compatible
+    options, logger, retries = 3, }) {
+        const { image, requestId, response_model, tools, temperature, top_p, maxOutputTokens, } = options;
+        const formattedMessages = this.formatMessages(options.messages, image);
+        const formattedTools = this.formatTools(tools);
+        const generationConfig = {
+            maxOutputTokens: maxOutputTokens,
+            temperature: temperature,
+            topP: top_p,
+            responseMimeType: response_model ? "application/json" : undefined,
+            responseSchema: response_model
+                ? (0, utils_js_1.toGeminiSchema)(response_model.schema)
+                : undefined,
+        };
+        logger({
+            category: "google",
+            message: "creating chat completion",
+            level: 2,
+            auxiliary: {
+                modelName: { value: this.modelName, type: "string" },
+                requestId: { value: requestId, type: "string" },
+                requestPayloadSummary: {
+                    value: `Model: ${this.modelName}, Messages: ${formattedMessages.length}, Config Keys: ${Object.keys(generationConfig).join(", ")}, Tools: ${formattedTools ? formattedTools.length : 0}, Safety Categories: ${safetySettings.map((s) => s.category).join(", ")}`,
+                    type: "string",
+                },
+            },
+        });
+        // Construct the full request object
+        const requestPayload = {
+            model: this.modelName,
+            contents: formattedMessages,
+            config: {
+                ...generationConfig,
+                safetySettings: safetySettings,
+                tools: formattedTools,
+            },
+        };
+        // Log the full payload safely
+        try {
+            logger({
+                category: "google",
+                message: "Full request payload",
+                level: 2,
+                auxiliary: {
+                    requestId: { value: requestId, type: "string" },
+                    fullPayload: {
+                        value: JSON.stringify(requestPayload),
+                        type: "object",
+                    },
+                },
+            });
+        }
+        catch (e) {
+            logger({
+                category: "google",
+                message: "Failed to stringify full request payload for logging",
+                level: 0,
+                auxiliary: {
+                    requestId: { value: requestId, type: "string" },
+                    error: { value: e.message, type: "string" },
+                },
+            });
+        }
+        try {
+            const result = await this.client.models.generateContent(requestPayload); // Pass the constructed payload
+            logger({
+                category: "google",
+                message: "received response",
+                level: 2,
+                auxiliary: {
+                    requestId: { value: requestId, type: "string" },
+                    response: {
+                        value: JSON.stringify(result),
+                        type: "object",
+                    },
+                },
+            });
+            const finishReason = result.candidates?.[0]?.finishReason || "unknown";
+            const toolCalls = result.functionCalls?.map((fc, index) => ({
+                id: `tool_call_${requestId}_${index}`,
+                type: "function",
+                function: {
+                    name: fc.name,
+                    arguments: JSON.stringify(fc.args),
+                },
+            }));
+            let content = null;
+            try {
+                content = result.text;
+            }
+            catch (e) {
+                logger({
+                    category: "google",
+                    message: `Could not extract text content: ${e.message}`,
+                    level: 1,
+                    auxiliary: { requestId: { value: requestId, type: "string" } },
+                });
+                content = null;
+            }
+            // Construct LLMResponse shape
+            const llmResponse = {
+                id: result.candidates?.[0]?.index?.toString() || requestId,
+                object: "chat.completion",
+                created: Math.floor(Date.now() / 1000),
+                model: this.modelName,
+                choices: [
+                    {
+                        index: 0,
+                        message: {
+                            role: "assistant",
+                            content: content,
+                            tool_calls: toolCalls,
+                        },
+                        finish_reason: finishReason,
+                    },
+                ],
+                usage: {
+                    prompt_tokens: result.usageMetadata?.promptTokenCount || 0,
+                    completion_tokens: result.usageMetadata?.candidatesTokenCount || 0,
+                    total_tokens: result.usageMetadata?.totalTokenCount || 0,
+                },
+            };
+            // Validate schema if response_model was provided
+            if (response_model) {
+                let parsedData;
+                try {
+                    // Need to handle potential markdown fences if the model didn't follow instructions perfectly
+                    const potentialJson = content?.trim().replace(/^```json\n?|\n?```$/g, "") || "{}";
+                    parsedData = JSON.parse(potentialJson);
+                }
+                catch (e) {
+                    logger({
+                        category: "google",
+                        message: `Failed to parse JSON response: ${e.message}`,
+                        level: 0,
+                        auxiliary: {
+                            content: { value: content || "null", type: "string" },
+                        },
+                    });
+                    if (retries > 0) {
+                        return this.createChatCompletion({
+                            options,
+                            logger,
+                            retries: retries - 1,
+                        });
+                    }
+                    throw new sdkErrors_js_1.CreateChatCompletionResponseError(`Failed to parse JSON response: ${e.message}`);
+                }
+                try {
+                    (0, utils_js_1.validateZodSchema)(response_model.schema, parsedData);
+                }
+                catch (err) {
+                    logger({
+                        category: "google",
+                        message: "Response failed Zod schema validation",
+                        level: 0,
+                    });
+                    if (retries > 0) {
+                        return this.createChatCompletion({
+                            options,
+                            logger,
+                            retries: retries - 1,
+                        });
+                    }
+                    throw err;
+                }
+                // If schema validation passes, structure the response for extraction use case
+                const extractionResult = {
+                    data: parsedData,
+                    usage: llmResponse.usage,
+                };
+                return extractionResult;
+            }
+            return llmResponse;
+        }
+        catch (error) {
+            logger({
+                category: "google",
+                message: `Error during Google AI chat completion: ${error.message}`,
+                level: 0,
+                auxiliary: {
+                    errorDetails: {
+                        value: `Message: ${error.message}${error.stack ? "\nStack: " + error.stack : ""}`,
+                        type: "string",
+                    },
+                    requestId: { value: requestId, type: "string" },
+                },
+            });
+            // Basic retry logic
+            if (retries > 0) {
+                logger({
+                    category: "google",
+                    message: `Retrying... (${retries} attempts left)`,
+                    level: 1,
+                });
+                await new Promise((resolve) => setTimeout(resolve, 1000 * (4 - retries))); // Simple backoff
+                return this.createChatCompletion({
+                    options,
+                    logger,
+                    retries: retries - 1,
+                });
+            }
+            // Re-throw specific Stagehand errors or a generic one
+            if (error instanceof sdkErrors_js_1.StagehandError) {
+                throw error;
+            }
+            throw new sdkErrors_js_1.StagehandError(`Google AI API request failed: ${error.message}`);
+        }
+    }
+}
+exports.GoogleClient = GoogleClient;
+//# sourceMappingURL=GoogleClient.js.map

package/dist/cjs/lib/v3/llm/GoogleClient.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"GoogleClient.js","sourceRoot":"","sources":["../../../../../lib/v3/llm/GoogleClient.ts"],"names":[],"mappings":";;;AAAA,yCAUuB;AAIvB,6CAIwB;AACxB,iDAOwB;AACxB,+DAGsC;AAEtC,6CAA6C;AAC7C,MAAM,OAAO,GAA6C;IACxD,IAAI,EAAE,MAAM;IACZ,SAAS,EAAE,OAAO;IAClB,MAAM,EAAE,MAAM,EAAE,yGAAyG;CAC1H,CAAC;AAEF,2CAA2C;AAC3C,MAAM,cAAc,GAAG;IACrB;QACE,QAAQ,EAAE,oBAAY,CAAC,wBAAwB;QAC/C,SAAS,EAAE,0BAAkB,CAAC,sBAAsB;KACrD;IACD;QACE,QAAQ,EAAE,oBAAY,CAAC,yBAAyB;QAChD,SAAS,EAAE,0BAAkB,CAAC,sBAAsB;KACrD;IACD;QACE,QAAQ,EAAE,oBAAY,CAAC,+BAA+B;QACtD,SAAS,EAAE,0BAAkB,CAAC,sBAAsB;KACrD;IACD;QACE,QAAQ,EAAE,oBAAY,CAAC,+BAA+B;QACtD,SAAS,EAAE,0BAAkB,CAAC,sBAAsB;KACrD;CACF,CAAC;AAEF,MAAa,YAAa,SAAQ,wBAAS;IAClC,IAAI,GAAG,QAAiB,CAAC;IACxB,MAAM,CAAc;IAGpB,MAAM,CAA6B;IAE3C,YAAY,EACV,MAAM,EAAE,sCAAsC;IAC9C,SAAS,EACT,aAAa,GAKd;QACC,KAAK,CAAC,SAAS,CAAC,CAAC;QACjB,IAAI,CAAC,aAAa,EAAE,MAAM,EAAE,CAAC;YAC3B,sEAAsE;YACtE,aAAa,CAAC,MAAM,GAAG,IAAA,4BAAiB,EAAC,eAAe,EAAE,MAAM,CAAC,CAAC;QACpE,CAAC;QACD,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,MAAM,GAAG,IAAI,mBAAW,CAAC,EAAE,MAAM,EAAE,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC;QAChE,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,qEAAqE;QACrE,IAAI,CAAC,SAAS;YACZ,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,gBAAgB;IACtF,CAAC;IAED,kEAAkE;IAC1D,cAAc,CACpB,QAAuB,EACvB,KAAsC;QAEtC,MAAM,QAAQ,GAAc,EAAE,CAAC;QAC/B,IAAI,iBAAiB,GAAkB,IAAI,CAAC;QAE5C,QAAQ,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;YAC9B,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAC/B,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,IAAI,CAAC,MAAM,CAAC;oBACV,QAAQ,EAAE,QAAQ;oBAClB,OAAO,EAAE,8BAA8B,GAAG,CAAC,IAAI,EAAE;oBACjD,KAAK,EAAE,CAAC;iBACT,CAAC,CAAC;gBACH,OAAO,CAAC,yBAAyB;YACnC,CAAC;YAED,oGAAoG;YACpG,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC1B,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;oBACpC,iBAAiB;wBACf,CAAC,iBAAiB,CAAC,CAAC,CAAC,iBAAiB,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,OAAO,CAAC;gBACxE,CAAC;gBACD,OAAO,CAAC,qDAAqD;YAC/D,CAAC;YAED,MAAM,KAAK,GAAW,EAAE,CAAC;YAEzB,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC/B,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE;oBAClC,IAAI,WAAW,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBAChC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC;oBACzC,CAAC;yBAAM,IAAI,WAAW,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;wBAC5C,IAAI,WAAW,IAAI,WAAW,IAAI,WAAW,CAAC,SAAS,EAAE,GAAG,EAAE,CAAC;4BAC7D,oEAAoE;4BACpE,MAAM,UAAU,GAAG,WAAW,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;4BAC3D,MAAM,aAAa,GAAG,WAAW,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CACnD,4BAA4B,CAC7B,CAAC;4BACF,IAAI,UAAU,IAAI,aAAa,EAAE,CAAC;gCAChC,KAAK,CAAC,IAAI,CAAC;oCACT,UAAU,EAAE,EAAE,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE;iCAC7D,CAAC,CAAC;4BACL,CAAC;iCAAM,CAAC;gCACN,IAAI,CAAC,MAAM,CAAC;oCACV,QAAQ,EAAE,QAAQ;oCAClB,OAAO,EAAE,gDAAgD;oCACzD,KAAK,EAAE,CAAC;iCACT,CAAC,CAAC;4BACL,CAAC;wBACH,CAAC;oBACH,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC;iBAAM,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;gBAC3C,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YACpC,CAAC;YAED,6EAA6E;YAC7E,IAAI,KAAK,IAAI,KAAK,KAAK,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAClE,MAAM,SAAS,GAAG,KAAK,CAAC,WAAW,IAAI,sCAAuB,CAAC;gBAC/D,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,wBAAwB;gBACzD,KAAK,CAAC,IAAI,CAAC;oBACT,UAAU,EAAE;wBACV,QAAQ,EAAE,YAAY,EAAE,kCAAkC;wBAC1D,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC;qBACtC;iBACF,CAAC,CAAC;YACL,CAAC;YAED,qEAAqE;YACrE,IAAI,iBAAiB,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;gBAClE,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC;gBACrD,IAAI,aAAa,IAAI,MAAM,IAAI,aAAa,EAAE,CAAC;oBAC7C,aAAa,CAAC,IAAI,GAAG,GAAG,iBAAiB,OAAO,aAAa,CAAC,IAAI,EAAE,CAAC;gBACvE,CAAC;qBAAM,CAAC;oBACN,KAAK,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAC;gBAC7C,CAAC;gBACD,iBAAiB,GAAG,IAAI,CAAC,CAAC,uBAAuB;YACnD,CAAC;YAED,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACrB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YACjC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,4GAA4G;QAC5G,IAAI,iBAAiB,EAAE,CAAC;YACtB,QAAQ,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,EAAE,CAAC,CAAC;QAC3E,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,iDAAiD;IACzC,WAAW,CACjB,KAAsC;QAEtC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO;YACL;gBACE,oBAAoB,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;oBACvC,IAAI,UAAU,GAAuB,SAAS,CAAC;oBAC/C,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;wBACpB,UAAU,GAAG;4BACX,IAAI,EAAE,YAAI,CAAC,MAAM;4BACjB,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,UAE3B;4BACD,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,QAAgC;yBAC3D,CAAC;oBACJ,CAAC;oBACD,OAAO;wBACL,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,WAAW,EAAE,IAAI,CAAC,WAAW;wBAC7B,UAAU,EAAE,UAAU;qBACvB,CAAC;gBACJ,CAAC,CAAC;aACH;SACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAkB;IAC1C,mCAAmC;IACnC,OAAO,EACP,MAAM,EACN,OAAO,GAAG,CAAC,GACiB;QAC5B,MAAM,EACJ,KAAK,EACL,SAAS,EACT,cAAc,EACd,KAAK,EACL,WAAW,EACX,KAAK,EACL,eAAe,GAChB,GAAG,OAAO,CAAC;QAEZ,MAAM,iBAAiB,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACvE,MAAM,cAAc,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAE/C,MAAM,gBAAgB,GAAG;YACvB,eAAe,EAAE,eAAe;YAChC,WAAW,EAAE,WAAW;YACxB,IAAI,EAAE,KAAK;YACX,gBAAgB,EAAE,cAAc,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,SAAS;YACjE,cAAc,EAAE,cAAc;gBAC5B,CAAC,CAAC,IAAA,yBAAc,EAAC,cAAc,CAAC,MAAM,CAAC;gBACvC,CAAC,CAAC,SAAS;SACd,CAAC;QAEF,MAAM,CAAC;YACL,QAAQ,EAAE,QAAQ;YAClB,OAAO,EAAE,0BAA0B;YACnC,KAAK,EAAE,CAAC;YACR,SAAS,EAAE;gBACT,SAAS,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACpD,SAAS,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC/C,qBAAqB,EAAE;oBACrB,KAAK,EAAE,UAAU,IAAI,CAAC,SAAS,eAAe,iBAAiB,CAAC,MAAM,kBAAkB,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,cAAc,CAAC,CAAC,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,wBAAwB,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;oBAChQ,IAAI,EAAE,QAAQ;iBACf;aACF;SACF,CAAC,CAAC;QAEH,oCAAoC;QACpC,MAAM,cAAc,GAAG;YACrB,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,QAAQ,EAAE,iBAAiB;YAC3B,MAAM,EAAE;gBACN,GAAG,gBAAgB;gBACnB,cAAc,EAAE,cAAc;gBAC9B,KAAK,EAAE,cAAc;aACtB;SACF,CAAC;QAEF,8BAA8B;QAC9B,IAAI,CAAC;YACH,MAAM,CAAC;gBACL,QAAQ,EAAE,QAAQ;gBAClB,OAAO,EAAE,sBAAsB;gBAC/B,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE;oBACT,SAAS,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE;oBAC/C,WAAW,EAAE;wBACX,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC;wBACrC,IAAI,EAAE,QAAQ;qBACf;iBACF;aACF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,CAAC;gBACL,QAAQ,EAAE,QAAQ;gBAClB,OAAO,EAAE,sDAAsD;gBAC/D,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE;oBACT,SAAS,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE;oBAC/C,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE;iBAC5C;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC,CAAC,+BAA+B;YAExG,MAAM,CAAC;gBACL,QAAQ,EAAE,QAAQ;gBAClB,OAAO,EAAE,mBAAmB;gBAC5B,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE;oBACT,SAAS,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE;oBAC/C,QAAQ,EAAE;wBACR,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;wBAC7B,IAAI,EAAE,QAAQ;qBACf;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,YAAY,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,IAAI,SAAS,CAAC;YACvE,MAAM,SAAS,GAAG,MAAM,CAAC,aAAa,EAAE,GAAG,CACzC,CAAC,EAAgB,EAAE,KAAa,EAAE,EAAE,CAAC,CAAC;gBACpC,EAAE,EAAE,aAAa,SAAS,IAAI,KAAK,EAAE;gBACrC,IAAI,EAAE,UAAmB;gBACzB,QAAQ,EAAE;oBACR,IAAI,EAAE,EAAE,CAAC,IAAI;oBACb,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,IAAI,CAAC;iBACnC;aACF,CAAC,CACH,CAAC;YAEF,IAAI,OAAO,GAAkB,IAAI,CAAC;YAClC,IAAI,CAAC;gBACH,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC;YACxB,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,MAAM,CAAC;oBACL,QAAQ,EAAE,QAAQ;oBAClB,OAAO,EAAE,mCAAmC,CAAC,CAAC,OAAO,EAAE;oBACvD,KAAK,EAAE,CAAC;oBACR,SAAS,EAAE,EAAE,SAAS,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;iBAC/D,CAAC,CAAC;gBACH,OAAO,GAAG,IAAI,CAAC;YACjB,CAAC;YAED,8BAA8B;YAC9B,MAAM,WAAW,GAAgB;gBAC/B,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,SAAS;gBAC1D,MAAM,EAAE,iBAAiB;gBACzB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;gBACtC,KAAK,EAAE,IAAI,CAAC,SAAS;gBACrB,OAAO,EAAE;oBACP;wBACE,KAAK,EAAE,CAAC;wBACR,OAAO,EAAE;4BACP,IAAI,EAAE,WAAW;4BACjB,OAAO,EAAE,OAAO;4BAChB,UAAU,EAAE,SAAS;yBACtB;wBACD,aAAa,EAAE,YAAY;qBAC5B;iBACF;gBACD,KAAK,EAAE;oBACL,aAAa,EAAE,MAAM,CAAC,aAAa,EAAE,gBAAgB,IAAI,CAAC;oBAC1D,iBAAiB,EAAE,MAAM,CAAC,aAAa,EAAE,oBAAoB,IAAI,CAAC;oBAClE,YAAY,EAAE,MAAM,CAAC,aAAa,EAAE,eAAe,IAAI,CAAC;iBACzD;aACF,CAAC;YAEF,iDAAiD;YACjD,IAAI,cAAc,EAAE,CAAC;gBACnB,IAAI,UAAU,CAAC;gBACf,IAAI,CAAC;oBACH,6FAA6F;oBAC7F,MAAM,aAAa,GACjB,OAAO,EAAE,IAAI,EAAE,CAAC,OAAO,CAAC,sBAAsB,EAAE,EAAE,CAAC,IAAI,IAAI,CAAC;oBAC9D,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;gBACzC,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,MAAM,CAAC;wBACL,QAAQ,EAAE,QAAQ;wBAClB,OAAO,EAAE,kCAAkC,CAAC,CAAC,OAAO,EAAE;wBACtD,KAAK,EAAE,CAAC;wBACR,SAAS,EAAE;4BACT,OAAO,EAAE,EAAE,KAAK,EAAE,OAAO,IAAI,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE;yBACtD;qBACF,CAAC,CAAC;oBACH,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;wBAChB,OAAO,IAAI,CAAC,oBAAoB,CAAC;4BAC/B,OAAO;4BACP,MAAM;4BACN,OAAO,EAAE,OAAO,GAAG,CAAC;yBACrB,CAAC,CAAC;oBACL,CAAC;oBACD,MAAM,IAAI,gDAAiC,CACzC,kCAAkC,CAAC,CAAC,OAAO,EAAE,CAC9C,CAAC;gBACJ,CAAC;gBAED,IAAI,CAAC;oBACH,IAAA,4BAAiB,EAAC,cAAc,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;gBACvD,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,CAAC;wBACL,QAAQ,EAAE,QAAQ;wBAClB,OAAO,EAAE,uCAAuC;wBAChD,KAAK,EAAE,CAAC;qBACT,CAAC,CAAC;oBACH,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;wBAChB,OAAO,IAAI,CAAC,oBAAoB,CAAC;4BAC/B,OAAO;4BACP,MAAM;4BACN,OAAO,EAAE,OAAO,GAAG,CAAC;yBACrB,CAAC,CAAC;oBACL,CAAC;oBACD,MAAM,GAAG,CAAC;gBACZ,CAAC;gBAED,8EAA8E;gBAC9E,MAAM,gBAAgB,GAAG;oBACvB,IAAI,EAAE,UAAU;oBAChB,KAAK,EAAE,WAAW,CAAC,KAAK;iBACzB,CAAC;gBAEF,OAAO,gBAAqB,CAAC;YAC/B,CAAC;YAED,OAAO,WAAgB,CAAC;QAC1B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC;gBACL,QAAQ,EAAE,QAAQ;gBAClB,OAAO,EAAE,2CAA2C,KAAK,CAAC,OAAO,EAAE;gBACnE,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE;oBACT,YAAY,EAAE;wBACZ,KAAK,EAAE,YAAY,KAAK,CAAC,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;wBACjF,IAAI,EAAE,QAAQ;qBACf;oBACD,SAAS,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE;iBAChD;aACF,CAAC,CAAC;YAEH,oBAAoB;YACpB,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;gBAChB,MAAM,CAAC;oBACL,QAAQ,EAAE,QAAQ;oBAClB,OAAO,EAAE,gBAAgB,OAAO,iBAAiB;oBACjD,KAAK,EAAE,CAAC;iBACT,CAAC,CAAC;gBACH,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAC5B,UAAU,CAAC,OAAO,EAAE,IAAI,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAC1C,CAAC,CAAC,iBAAiB;gBACpB,OAAO,IAAI,CAAC,oBAAoB,CAAC;oBAC/B,OAAO;oBACP,MAAM;oBACN,OAAO,EAAE,OAAO,GAAG,CAAC;iBACrB,CAAC,CAAC;YACL,CAAC;YAED,sDAAsD;YACtD,IAAI,KAAK,YAAY,6BAAc,EAAE,CAAC;gBACpC,MAAM,KAAK,CAAC;YACd,CAAC;YACD,MAAM,IAAI,6BAAc,CACtB,iCAAiC,KAAK,CAAC,OAAO,EAAE,CACjD,CAAC;QACJ,CAAC;IACH,CAAC;CACF;AA9YD,oCA8YC","sourcesContent":["import {\n  GoogleGenAI,\n  HarmCategory,\n  HarmBlockThreshold,\n  Content,\n  Part,\n  Tool,\n  FunctionCall,\n  Schema,\n  Type,\n} from \"@google/genai\";\n\nimport { LogLine } from \"../types/public/logs.js\";\nimport { AvailableModel, ClientOptions } from \"../types/public/model.js\";\nimport {\n  validateZodSchema,\n  toGeminiSchema,\n  loadApiKeyFromEnv,\n} from \"../../utils.js\";\nimport {\n  ChatCompletionOptions,\n  ChatMessage,\n  CreateChatCompletionOptions,\n  LLMClient,\n  LLMResponse,\n  AnnotatedScreenshotText,\n} from \"./LLMClient.js\";\nimport {\n  CreateChatCompletionResponseError,\n  StagehandError,\n} from \"../types/public/sdkErrors.js\";\n\n// Mapping from generic roles to Gemini roles\nconst roleMap: { [key in ChatMessage[\"role\"]]: string } = {\n  user: \"user\",\n  assistant: \"model\",\n  system: \"user\", // Gemini API prefers system instructions either via system_instruction or at the start of 'user' content\n};\n\n// Basic safety settings - adjust as needed\nconst safetySettings = [\n  {\n    category: HarmCategory.HARM_CATEGORY_HARASSMENT,\n    threshold: HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,\n  },\n  {\n    category: HarmCategory.HARM_CATEGORY_HATE_SPEECH,\n    threshold: HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,\n  },\n  {\n    category: HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,\n    threshold: HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,\n  },\n  {\n    category: HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,\n    threshold: HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,\n  },\n];\n\nexport class GoogleClient extends LLMClient {\n  public type = \"google\" as const;\n  private client: GoogleGenAI;\n  declare public clientOptions: ClientOptions;\n  declare public hasVision: boolean;\n  private logger: (message: LogLine) => void;\n\n  constructor({\n    logger, // Added logger based on other clients\n    modelName,\n    clientOptions,\n  }: {\n    logger: (message: LogLine) => void; // Added logger type\n    modelName: AvailableModel;\n    clientOptions?: ClientOptions; // Expecting { apiKey: string } here\n  }) {\n    super(modelName);\n    if (!clientOptions?.apiKey) {\n      // Try to get the API key from the environment variable GOOGLE_API_KEY\n      clientOptions.apiKey = loadApiKeyFromEnv(\"google_legacy\", logger);\n    }\n    this.clientOptions = clientOptions;\n    this.client = new GoogleGenAI({ apiKey: clientOptions.apiKey });\n    this.modelName = modelName;\n    this.logger = logger;\n    // Determine vision capability based on model name (adjust as needed)\n    this.hasVision =\n      modelName.includes(\"vision\") || modelName.includes(\"gemini-1.5\"); // Example logic\n  }\n\n  // Helper to convert project's ChatMessage[] to Gemini's Content[]\n  private formatMessages(\n    messages: ChatMessage[],\n    image?: ChatCompletionOptions[\"image\"],\n  ): Content[] {\n    const contents: Content[] = [];\n    let systemInstruction: string | null = null;\n\n    messages.forEach((msg, index) => {\n      const role = roleMap[msg.role];\n      if (!role) {\n        this.logger({\n          category: \"google\",\n          message: `WARNING: Unsupported role: ${msg.role}`,\n          level: 1,\n        });\n        return; // Skip unsupported roles\n      }\n\n      // Handle system messages - prepend to the first user message or use system_instruction if available\n      if (msg.role === \"system\") {\n        if (typeof msg.content === \"string\") {\n          systemInstruction =\n            (systemInstruction ? systemInstruction + \"\\n\\n\" : \"\") + msg.content;\n        }\n        return; // Don't add system messages directly to contents yet\n      }\n\n      const parts: Part[] = [];\n\n      if (Array.isArray(msg.content)) {\n        msg.content.forEach((partContent) => {\n          if (partContent.type === \"text\") {\n            parts.push({ text: partContent.text });\n          } else if (partContent.type === \"image_url\") {\n            if (\"image_url\" in partContent && partContent.image_url?.url) {\n              // Assuming base64 data URI format: data:[<mediatype>];base64,<data>\n              const base64Data = partContent.image_url.url.split(\",\")[1];\n              const mimeTypeMatch = partContent.image_url.url.match(\n                /^data:(image\\/\\w+);base64,/,\n              );\n              if (base64Data && mimeTypeMatch) {\n                parts.push({\n                  inlineData: { mimeType: mimeTypeMatch[1], data: base64Data },\n                });\n              } else {\n                this.logger({\n                  category: \"google\",\n                  message: \"WARNING: Could not parse image data URI format\",\n                  level: 1,\n                });\n              }\n            }\n          }\n        });\n      } else if (typeof msg.content === \"string\") {\n        parts.push({ text: msg.content });\n      }\n\n      // Add image from options if this is the last message and it's a user message\n      if (image && index === messages.length - 1 && msg.role === \"user\") {\n        const imageDesc = image.description || AnnotatedScreenshotText;\n        parts.push({ text: imageDesc }); // Add description first\n        parts.push({\n          inlineData: {\n            mimeType: \"image/jpeg\", // Assuming JPEG, adjust if needed\n            data: image.buffer.toString(\"base64\"),\n          },\n        });\n      }\n\n      // Apply system instruction to the first non-system message if needed\n      if (systemInstruction && contents.length === 0 && role === \"user\") {\n        const firstPartText = parts.find((p) => \"text\" in p);\n        if (firstPartText && \"text\" in firstPartText) {\n          firstPartText.text = `${systemInstruction}\\n\\n${firstPartText.text}`;\n        } else {\n          parts.unshift({ text: systemInstruction });\n        }\n        systemInstruction = null; // Clear after applying\n      }\n\n      if (parts.length > 0) {\n        contents.push({ role, parts });\n      }\n    });\n\n    // If system instruction wasn't applied (e.g., no user messages followed it), add it as a final user message\n    if (systemInstruction) {\n      contents.unshift({ role: \"user\", parts: [{ text: systemInstruction }] });\n    }\n\n    return contents;\n  }\n\n  // Helper to convert LLMTool[] to Gemini's Tool[]\n  private formatTools(\n    tools?: ChatCompletionOptions[\"tools\"],\n  ): Tool[] | undefined {\n    if (!tools || tools.length === 0) {\n      return undefined;\n    }\n\n    return [\n      {\n        functionDeclarations: tools.map((tool) => {\n          let parameters: Schema | undefined = undefined;\n          if (tool.parameters) {\n            parameters = {\n              type: Type.OBJECT,\n              properties: tool.parameters.properties as {\n                [key: string]: Schema;\n              },\n              required: tool.parameters.required as string[] | undefined,\n            };\n          }\n          return {\n            name: tool.name,\n            description: tool.description,\n            parameters: parameters,\n          };\n        }),\n      },\n    ];\n  }\n\n  async createChatCompletion<T = LLMResponse>({\n    // Ensure LLMResponse is compatible\n    options,\n    logger,\n    retries = 3,\n  }: CreateChatCompletionOptions): Promise<T> {\n    const {\n      image,\n      requestId,\n      response_model,\n      tools,\n      temperature,\n      top_p,\n      maxOutputTokens,\n    } = options;\n\n    const formattedMessages = this.formatMessages(options.messages, image);\n    const formattedTools = this.formatTools(tools);\n\n    const generationConfig = {\n      maxOutputTokens: maxOutputTokens,\n      temperature: temperature,\n      topP: top_p,\n      responseMimeType: response_model ? \"application/json\" : undefined,\n      responseSchema: response_model\n        ? toGeminiSchema(response_model.schema)\n        : undefined,\n    };\n\n    logger({\n      category: \"google\",\n      message: \"creating chat completion\",\n      level: 2,\n      auxiliary: {\n        modelName: { value: this.modelName, type: \"string\" },\n        requestId: { value: requestId, type: \"string\" },\n        requestPayloadSummary: {\n          value: `Model: ${this.modelName}, Messages: ${formattedMessages.length}, Config Keys: ${Object.keys(generationConfig).join(\", \")}, Tools: ${formattedTools ? formattedTools.length : 0}, Safety Categories: ${safetySettings.map((s) => s.category).join(\", \")}`,\n          type: \"string\",\n        },\n      },\n    });\n\n    // Construct the full request object\n    const requestPayload = {\n      model: this.modelName,\n      contents: formattedMessages,\n      config: {\n        ...generationConfig,\n        safetySettings: safetySettings,\n        tools: formattedTools,\n      },\n    };\n\n    // Log the full payload safely\n    try {\n      logger({\n        category: \"google\",\n        message: \"Full request payload\",\n        level: 2,\n        auxiliary: {\n          requestId: { value: requestId, type: \"string\" },\n          fullPayload: {\n            value: JSON.stringify(requestPayload),\n            type: \"object\",\n          },\n        },\n      });\n    } catch (e) {\n      logger({\n        category: \"google\",\n        message: \"Failed to stringify full request payload for logging\",\n        level: 0,\n        auxiliary: {\n          requestId: { value: requestId, type: \"string\" },\n          error: { value: e.message, type: \"string\" },\n        },\n      });\n    }\n\n    try {\n      const result = await this.client.models.generateContent(requestPayload); // Pass the constructed payload\n\n      logger({\n        category: \"google\",\n        message: \"received response\",\n        level: 2,\n        auxiliary: {\n          requestId: { value: requestId, type: \"string\" },\n          response: {\n            value: JSON.stringify(result),\n            type: \"object\",\n          },\n        },\n      });\n\n      const finishReason = result.candidates?.[0]?.finishReason || \"unknown\";\n      const toolCalls = result.functionCalls?.map(\n        (fc: FunctionCall, index: number) => ({\n          id: `tool_call_${requestId}_${index}`,\n          type: \"function\" as const,\n          function: {\n            name: fc.name,\n            arguments: JSON.stringify(fc.args),\n          },\n        }),\n      );\n\n      let content: string | null = null;\n      try {\n        content = result.text;\n      } catch (e) {\n        logger({\n          category: \"google\",\n          message: `Could not extract text content: ${e.message}`,\n          level: 1,\n          auxiliary: { requestId: { value: requestId, type: \"string\" } },\n        });\n        content = null;\n      }\n\n      // Construct LLMResponse shape\n      const llmResponse: LLMResponse = {\n        id: result.candidates?.[0]?.index?.toString() || requestId,\n        object: \"chat.completion\",\n        created: Math.floor(Date.now() / 1000),\n        model: this.modelName,\n        choices: [\n          {\n            index: 0,\n            message: {\n              role: \"assistant\",\n              content: content,\n              tool_calls: toolCalls,\n            },\n            finish_reason: finishReason,\n          },\n        ],\n        usage: {\n          prompt_tokens: result.usageMetadata?.promptTokenCount || 0,\n          completion_tokens: result.usageMetadata?.candidatesTokenCount || 0,\n          total_tokens: result.usageMetadata?.totalTokenCount || 0,\n        },\n      };\n\n      // Validate schema if response_model was provided\n      if (response_model) {\n        let parsedData;\n        try {\n          // Need to handle potential markdown fences if the model didn't follow instructions perfectly\n          const potentialJson =\n            content?.trim().replace(/^```json\\n?|\\n?```$/g, \"\") || \"{}\";\n          parsedData = JSON.parse(potentialJson);\n        } catch (e) {\n          logger({\n            category: \"google\",\n            message: `Failed to parse JSON response: ${e.message}`,\n            level: 0,\n            auxiliary: {\n              content: { value: content || \"null\", type: \"string\" },\n            },\n          });\n          if (retries > 0) {\n            return this.createChatCompletion({\n              options,\n              logger,\n              retries: retries - 1,\n            });\n          }\n          throw new CreateChatCompletionResponseError(\n            `Failed to parse JSON response: ${e.message}`,\n          );\n        }\n\n        try {\n          validateZodSchema(response_model.schema, parsedData);\n        } catch (err) {\n          logger({\n            category: \"google\",\n            message: \"Response failed Zod schema validation\",\n            level: 0,\n          });\n          if (retries > 0) {\n            return this.createChatCompletion({\n              options,\n              logger,\n              retries: retries - 1,\n            });\n          }\n          throw err;\n        }\n\n        // If schema validation passes, structure the response for extraction use case\n        const extractionResult = {\n          data: parsedData,\n          usage: llmResponse.usage,\n        };\n\n        return extractionResult as T;\n      }\n\n      return llmResponse as T;\n    } catch (error) {\n      logger({\n        category: \"google\",\n        message: `Error during Google AI chat completion: ${error.message}`,\n        level: 0,\n        auxiliary: {\n          errorDetails: {\n            value: `Message: ${error.message}${error.stack ? \"\\nStack: \" + error.stack : \"\"}`,\n            type: \"string\",\n          },\n          requestId: { value: requestId, type: \"string\" },\n        },\n      });\n\n      // Basic retry logic\n      if (retries > 0) {\n        logger({\n          category: \"google\",\n          message: `Retrying... (${retries} attempts left)`,\n          level: 1,\n        });\n        await new Promise((resolve) =>\n          setTimeout(resolve, 1000 * (4 - retries)),\n        ); // Simple backoff\n        return this.createChatCompletion({\n          options,\n          logger,\n          retries: retries - 1,\n        });\n      }\n\n      // Re-throw specific Stagehand errors or a generic one\n      if (error instanceof StagehandError) {\n        throw error;\n      }\n      throw new StagehandError(\n        `Google AI API request failed: ${error.message}`,\n      );\n    }\n  }\n}\n"]}

package/dist/cjs/lib/v3/llm/GroqClient.d.ts

@@ -0,0 +1,17 @@
+import type { ClientOptions } from "openai";
+import { LogLine } from "../types/public/logs.js";
+import { AvailableModel } from "../types/public/model.js";
+import { CreateChatCompletionOptions, LLMClient, LLMResponse } from "./LLMClient.js";
+export declare class GroqClient extends LLMClient {
+    type: "groq";
+    private client;
+    clientOptions: ClientOptions;
+    hasVision: boolean;
+    constructor({ modelName, clientOptions, userProvidedInstructions, }: {
+        logger: (message: LogLine) => void;
+        modelName: AvailableModel;
+        clientOptions?: ClientOptions;
+        userProvidedInstructions?: string;
+    });
+    createChatCompletion<T = LLMResponse>({ options, retries, logger, }: CreateChatCompletionOptions): Promise<T>;
+}