@falai/agent 0.1.0-alpha2

package/dist/cjs/src/providers/AnthropicProvider.js

@@ -0,0 +1,377 @@
+"use strict";
+/**
+ * Anthropic (Claude) provider implementation with retry and backup models
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AnthropicProvider = void 0;
+const sdk_1 = __importDefault(require("@anthropic-ai/sdk"));
+const utils_1 = require("../utils");
+const DEFAULT_RETRY_CONFIG = {
+    timeout: 60000,
+    retries: 3,
+};
+/**
+ * Type guard to check if error is ErrorWithStatus
+ */
+function isErrorWithStatus(error) {
+    return (typeof error === "object" &&
+        error !== null &&
+        ("status" in error || "code" in error || "message" in error));
+}
+/**
+ * Safely extract error message
+ */
+function getErrorMessage(error) {
+    if (error instanceof Error) {
+        return error.message;
+    }
+    if (isErrorWithStatus(error) && error.message) {
+        return error.message;
+    }
+    return String(error);
+}
+/**
+ * Determines if an error should trigger backup model usage
+ */
+const shouldUseBackupModel = (error) => {
+    if (!isErrorWithStatus(error)) {
+        return false;
+    }
+    // Server errors
+    if (error.status === 500 || error.status === 503 || error.status === 529) {
+        return true;
+    }
+    // Rate limiting
+    if (error.status === 429) {
+        return true;
+    }
+    // Model overloaded or unavailable
+    if (error.type === "overloaded_error" ||
+        error.type === "api_error" ||
+        error.code === "overloaded") {
+        return true;
+    }
+    const message = getErrorMessage(error);
+    if (message.includes("overloaded") ||
+        message.includes("unavailable") ||
+        message.includes("internal error") ||
+        message.includes("Internal error")) {
+        return true;
+    }
+    return false;
+};
+/**
+ * Anthropic provider implementation with backup models and retry logic
+ */
+class AnthropicProvider {
+    constructor(options) {
+        this.name = "anthropic";
+        const { apiKey, model, backupModels = [], config, retryConfig } = options;
+        if (!apiKey) {
+            throw new Error("Anthropic API key is required");
+        }
+        if (!model) {
+            throw new Error("Model is required. Example: 'claude-sonnet-4-5'");
+        }
+        this.client = new sdk_1.default({
+            apiKey,
+        });
+        this.primaryModel = model;
+        this.backupModels = backupModels;
+        this.config = config;
+        this.retryConfig = {
+            timeout: retryConfig?.timeout || DEFAULT_RETRY_CONFIG.timeout,
+            retries: retryConfig?.retries || DEFAULT_RETRY_CONFIG.retries,
+        };
+    }
+    async generateMessage(input) {
+        return this.generateWithBackup(input);
+    }
+    async *generateMessageStream(input) {
+        yield* this.generateStreamWithBackup(input);
+    }
+    async generateWithBackup(input) {
+        // Try primary model first
+        try {
+            return await this.generateWithModel(this.primaryModel, input);
+        }
+        catch (primaryError) {
+            const primaryErrMsg = getErrorMessage(primaryError);
+            utils_1.logger.warn(`[ANTHROPIC] Primary model ${this.primaryModel} failed: ${primaryErrMsg}`);
+            if (!shouldUseBackupModel(primaryError)) {
+                throw primaryError;
+            }
+            utils_1.logger.debug(`[ANTHROPIC] Trying backup models`);
+            let lastBackupError = primaryError;
+            for (let i = 0; i < this.backupModels.length; i++) {
+                const backupModel = this.backupModels[i];
+                utils_1.logger.debug(`[ANTHROPIC] Trying backup model ${i + 1}/${this.backupModels.length}: ${backupModel}`);
+                try {
+                    const result = await this.generateWithModel(backupModel, input);
+                    utils_1.logger.debug(`[ANTHROPIC] Backup model ${backupModel} succeeded`);
+                    return result;
+                }
+                catch (backupError) {
+                    const backupErrMsg = getErrorMessage(backupError);
+                    utils_1.logger.warn(`[ANTHROPIC] Backup model ${backupModel} failed: ${backupErrMsg}`);
+                    lastBackupError = backupError;
+                    if (!shouldUseBackupModel(backupError) &&
+                        i < this.backupModels.length - 1) {
+                        utils_1.logger.debug(`[ANTHROPIC] Backup model error doesn't qualify for further attempts`);
+                        break;
+                    }
+                }
+            }
+            const lastBackupErrMsg = getErrorMessage(lastBackupError);
+            utils_1.logger.error(`[ANTHROPIC] All models failed. Primary: ${primaryErrMsg}, Last backup: ${lastBackupErrMsg}`);
+            throw lastBackupError;
+        }
+    }
+    async generateWithModel(model, input) {
+        const operation = async () => {
+            // Anthropic requires max_tokens to be specified
+            const maxTokens = input.parameters?.maxOutputTokens || 4096;
+            const params = {
+                model,
+                max_tokens: maxTokens,
+                messages: [
+                    {
+                        role: "user",
+                        content: input.prompt,
+                    },
+                ],
+                ...this.config,
+            };
+            // Add tools if provided
+            if (input.tools && input.tools.length > 0) {
+                params.tools = input.tools.map((tool) => ({
+                    name: tool.name || tool.id,
+                    description: tool.description || "",
+                    input_schema: tool.parameters, // JSON schema
+                }));
+            }
+            // Handle schema: Anthropic doesn't have a native schema mode, so embed constraints
+            if (input.parameters?.jsonSchema) {
+                const systemPrompt = "You must respond with valid JSON only and it MUST match the provided schema.";
+                // Merge with existing system if present
+                if (typeof this.config?.system === "string") {
+                    params.system = `${this.config.system}\n\n${systemPrompt}`;
+                }
+                else if (Array.isArray(this.config?.system)) {
+                    params.system = [
+                        ...this.config.system,
+                        {
+                            type: "text",
+                            text: systemPrompt,
+                        },
+                    ];
+                }
+                else {
+                    params.system = systemPrompt;
+                }
+            }
+            const response = await this.client.messages.create(params);
+            // Extract text and tool calls from response
+            const textContent = response.content.find((block) => block.type === "text");
+            const message = textContent?.type === "text" ? textContent.text : "";
+            if (!message) {
+                throw new Error("No response from Anthropic");
+            }
+            // Extract tool calls from response
+            const toolCalls = [];
+            // Check for tool_use content blocks
+            for (const block of response.content) {
+                if (block.type === "tool_use") {
+                    toolCalls.push({
+                        toolName: block.name,
+                        arguments: block.input,
+                    });
+                }
+            }
+            // Parse JSON response if schema was provided
+            let structured;
+            if (input.parameters?.jsonSchema) {
+                try {
+                    structured = JSON.parse(message);
+                }
+                catch (error) {
+                    utils_1.logger.warn("[ANTHROPIC] Failed to parse JSON response:", error);
+                    // Fall back to treating the message as plain text
+                }
+            }
+            // If tools were used, include them in structured response
+            if (toolCalls.length > 0) {
+                structured = {
+                    message,
+                    toolCalls,
+                    ...structured,
+                };
+            }
+            return {
+                message,
+                metadata: {
+                    model: response.model,
+                    stopReason: response.stop_reason,
+                    tokensUsed: response.usage.input_tokens + response.usage.output_tokens,
+                    promptTokens: response.usage.input_tokens,
+                    completionTokens: response.usage.output_tokens,
+                },
+                structured,
+            };
+        };
+        return (0, utils_1.withTimeoutAndRetry)(operation, this.retryConfig.timeout, this.retryConfig.retries, `Anthropic ${model}`);
+    }
+    async *generateStreamWithBackup(input) {
+        // Try primary model first
+        try {
+            yield* this.generateStreamWithModel(this.primaryModel, input);
+        }
+        catch (primaryError) {
+            const primaryErrMsg = getErrorMessage(primaryError);
+            utils_1.logger.warn(`[ANTHROPIC] Primary model ${this.primaryModel} failed: ${primaryErrMsg}`);
+            if (!shouldUseBackupModel(primaryError)) {
+                throw primaryError;
+            }
+            utils_1.logger.debug(`[ANTHROPIC] Trying backup models for streaming`);
+            let lastBackupError = primaryError;
+            for (let i = 0; i < this.backupModels.length; i++) {
+                const backupModel = this.backupModels[i];
+                utils_1.logger.debug(`[ANTHROPIC] Trying backup model ${i + 1}/${this.backupModels.length}: ${backupModel}`);
+                try {
+                    yield* this.generateStreamWithModel(backupModel, input);
+                    utils_1.logger.debug(`[ANTHROPIC] Backup model ${backupModel} succeeded`);
+                    return;
+                }
+                catch (backupError) {
+                    const backupErrMsg = getErrorMessage(backupError);
+                    utils_1.logger.warn(`[ANTHROPIC] Backup model ${backupModel} failed: ${backupErrMsg}`);
+                    lastBackupError = backupError;
+                    if (!shouldUseBackupModel(backupError) &&
+                        i < this.backupModels.length - 1) {
+                        utils_1.logger.debug(`[ANTHROPIC] Backup model error doesn't qualify for further attempts`);
+                        break;
+                    }
+                }
+            }
+            const lastBackupErrMsg = getErrorMessage(lastBackupError);
+            utils_1.logger.error(`[ANTHROPIC] All models failed. Primary: ${primaryErrMsg}, Last backup: ${lastBackupErrMsg}`);
+            throw lastBackupError;
+        }
+    }
+    async *generateStreamWithModel(model, input) {
+        // Anthropic requires max_tokens to be specified
+        const maxTokens = input.parameters?.maxOutputTokens || 4096;
+        const params = {
+            model,
+            max_tokens: maxTokens,
+            messages: [
+                {
+                    role: "user",
+                    content: input.prompt,
+                },
+            ],
+            stream: true,
+            ...this.config,
+        };
+        // Add tools if provided
+        if (input.tools && input.tools.length > 0) {
+            params.tools = input.tools.map((tool) => ({
+                name: tool.name || tool.id,
+                description: tool.description || "",
+                input_schema: tool.parameters, // JSON schema
+            }));
+        }
+        // Handle schema in streaming: embed constraint
+        if (input.parameters?.jsonSchema) {
+            const systemPrompt = "You must respond with valid JSON only and it MUST match the provided schema.";
+            if (typeof this.config?.system === "string") {
+                params.system = `${this.config.system}\n\n${systemPrompt}`;
+            }
+            else if (Array.isArray(this.config?.system)) {
+                params.system = [
+                    ...this.config.system,
+                    {
+                        type: "text",
+                        text: systemPrompt,
+                    },
+                ];
+            }
+            else {
+                params.system = systemPrompt;
+            }
+        }
+        const stream = this.client.messages.stream(params);
+        let accumulated = "";
+        let currentModel = model;
+        let stopReason;
+        let inputTokens = 0;
+        let outputTokens = 0;
+        const toolCalls = [];
+        for await (const chunk of stream) {
+            if (chunk.type === "message_start") {
+                currentModel = chunk.message.model;
+                inputTokens = chunk.message.usage.input_tokens;
+            }
+            else if (chunk.type === "content_block_start") {
+                if (chunk.content_block.type === "tool_use") {
+                    toolCalls.push({
+                        toolName: chunk.content_block.name,
+                        arguments: chunk.content_block.input,
+                    });
+                }
+            }
+            else if (chunk.type === "content_block_delta") {
+                if (chunk.delta.type === "text_delta") {
+                    const delta = chunk.delta.text;
+                    accumulated += delta;
+                    yield {
+                        delta,
+                        accumulated,
+                        done: false,
+                    };
+                }
+            }
+            else if (chunk.type === "message_delta") {
+                stopReason = chunk.delta.stop_reason || undefined;
+                outputTokens = chunk.usage.output_tokens;
+            }
+        }
+        // Parse JSON response if schema was provided
+        let structured;
+        if (input.parameters?.jsonSchema && accumulated) {
+            try {
+                structured = JSON.parse(accumulated);
+            }
+            catch (error) {
+                utils_1.logger.warn("[ANTHROPIC] Failed to parse JSON response in stream:", error);
+            }
+        }
+        // If tools were used, include them in structured response
+        if (toolCalls.length > 0) {
+            structured = {
+                message: accumulated,
+                toolCalls,
+                ...structured,
+            };
+        }
+        // Yield final chunk
+        yield {
+            delta: "",
+            accumulated,
+            done: true,
+            metadata: {
+                model: currentModel,
+                stopReason,
+                tokensUsed: inputTokens + outputTokens,
+                promptTokens: inputTokens,
+                completionTokens: outputTokens,
+            },
+            structured: structured,
+        };
+    }
+}
+exports.AnthropicProvider = AnthropicProvider;
+//# sourceMappingURL=AnthropicProvider.js.map

package/dist/cjs/src/providers/AnthropicProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AnthropicProvider.js","sourceRoot":"","sources":["../../../../src/providers/AnthropicProvider.ts"],"names":[],"mappings":";AAAA;;GAEG;;;;;;AAEH,4DAA0C;AAa1C,oCAAuD;AAEvD,MAAM,oBAAoB,GAAG;IAC3B,OAAO,EAAE,KAAK;IACd,OAAO,EAAE,CAAC;CACX,CAAC;AAgCF;;GAEG;AACH,SAAS,iBAAiB,CAAC,KAAc;IACvC,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,KAAK,IAAI;QACd,CAAC,QAAQ,IAAI,KAAK,IAAI,MAAM,IAAI,KAAK,IAAI,SAAS,IAAI,KAAK,CAAC,CAC7D,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,KAAc;IACrC,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC3B,OAAO,KAAK,CAAC,OAAO,CAAC;IACvB,CAAC;IACD,IAAI,iBAAiB,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;QAC9C,OAAO,KAAK,CAAC,OAAO,CAAC;IACvB,CAAC;IACD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;AACvB,CAAC;AAED;;GAEG;AACH,MAAM,oBAAoB,GAAG,CAAC,KAAc,EAAW,EAAE;IACvD,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,EAAE,CAAC;QAC9B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,gBAAgB;IAChB,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;QACzE,OAAO,IAAI,CAAC;IACd,CAAC;IAED,gBAAgB;IAChB,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,kCAAkC;IAClC,IACE,KAAK,CAAC,IAAI,KAAK,kBAAkB;QACjC,KAAK,CAAC,IAAI,KAAK,WAAW;QAC1B,KAAK,CAAC,IAAI,KAAK,YAAY,EAC3B,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IACvC,IACE,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC;QAC9B,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC;QAC/B,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAC;QAClC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAClC,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AAEF;;GAEG;AACH,MAAa,iBAAiB;IAU5B,YAAY,OAAiC;QAT7B,SAAI,GAAG,WAAW,CAAC;QAUjC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,YAAY,GAAG,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC;QAE1E,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;QACnD,CAAC;QAED,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACrE,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,IAAI,aAAS,CAAC;YAC1B,MAAM;SACP,CAAC,CAAC;QACH,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAC1B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,WAAW,GAAG;YACjB,OAAO,EAAE,WAAW,EAAE,OAAO,IAAI,oBAAoB,CAAC,OAAO;YAC7D,OAAO,EAAE,WAAW,EAAE,OAAO,IAAI,oBAAoB,CAAC,OAAO;SAC9D,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,eAAe,CAInB,KAAqC;QAErC,OAAO,IAAI,CAAC,kBAAkB,CAAwB,KAAK,CAAC,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,CAAC,qBAAqB,CAI1B,KAAqC;QAErC,KAAK,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAwB,KAAK,CAAC,CAAC;IACrE,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAI9B,KAAqC;QAErC,0BAA0B;QAC1B,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,iBAAiB,CACjC,IAAI,CAAC,YAAY,EACjB,KAAK,CACN,CAAC;QACJ,CAAC;QAAC,OAAO,YAAqB,EAAE,CAAC;YAC/B,MAAM,aAAa,GAAG,eAAe,CAAC,YAAY,CAAC,CAAC;YACpD,cAAM,CAAC,IAAI,CACT,6BAA6B,IAAI,CAAC,YAAY,YAAY,aAAa,EAAE,CAC1E,CAAC;YAEF,IAAI,CAAC,oBAAoB,CAAC,YAAY,CAAC,EAAE,CAAC;gBACxC,MAAM,YAAY,CAAC;YACrB,CAAC;YAED,cAAM,CAAC,KAAK,CAAC,kCAAkC,CAAC,CAAC;YAEjD,IAAI,eAAe,GAAY,YAAY,CAAC;YAE5C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAClD,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;gBACzC,cAAM,CAAC,KAAK,CACV,mCAAmC,CAAC,GAAG,CAAC,IACtC,IAAI,CAAC,YAAY,CAAC,MACpB,KAAK,WAAW,EAAE,CACnB,CAAC;gBAEF,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CACzC,WAAW,EACX,KAAK,CACN,CAAC;oBACF,cAAM,CAAC,KAAK,CAAC,4BAA4B,WAAW,YAAY,CAAC,CAAC;oBAClE,OAAO,MAAM,CAAC;gBAChB,CAAC;gBAAC,OAAO,WAAoB,EAAE,CAAC;oBAC9B,MAAM,YAAY,GAAG,eAAe,CAAC,WAAW,CAAC,CAAC;oBAClD,cAAM,CAAC,IAAI,CACT,4BAA4B,WAAW,YAAY,YAAY,EAAE,CAClE,CAAC;oBACF,eAAe,GAAG,WAAW,CAAC;oBAE9B,IACE,CAAC,oBAAoB,CAAC,WAAW,CAAC;wBAClC,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAChC,CAAC;wBACD,cAAM,CAAC,KAAK,CACV,qEAAqE,CACtE,CAAC;wBACF,MAAM;oBACR,CAAC;gBACH,CAAC;YACH,CAAC;YAED,MAAM,gBAAgB,GAAG,eAAe,CAAC,eAAe,CAAC,CAAC;YAC1D,cAAM,CAAC,KAAK,CACV,2CAA2C,aAAa,kBAAkB,gBAAgB,EAAE,CAC7F,CAAC;YACF,MAAM,eAAe,CAAC;QACxB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAI7B,KAAa,EACb,KAAqC;QAErC,MAAM,SAAS,GAAG,KAAK,IAAoC,EAAE;YAC3D,gDAAgD;YAChD,MAAM,SAAS,GAAG,KAAK,CAAC,UAAU,EAAE,eAAe,IAAI,IAAI,CAAC;YAE5D,MAAM,MAAM,GAAoC;gBAC9C,KAAK;gBACL,UAAU,EAAE,SAAS;gBACrB,QAAQ,EAAE;oBACR;wBACE,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,KAAK,CAAC,MAAM;qBACtB;iBACF;gBACD,GAAG,IAAI,CAAC,MAAM;aACf,CAAC;YAEF,wBAAwB;YACxB,IAAI,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1C,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;oBACxC,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,EAAE;oBAC1B,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;oBACnC,YAAY,EAAE,IAAI,CAAC,UAAkC,EAAE,cAAc;iBACtE,CAAC,CAAC,CAAC;YACN,CAAC;YAED,mFAAmF;YACnF,IAAI,KAAK,CAAC,UAAU,EAAE,UAAU,EAAE,CAAC;gBACjC,MAAM,YAAY,GAChB,8EAA8E,CAAC;gBAEjF,wCAAwC;gBACxC,IAAI,OAAO,IAAI,CAAC,MAAM,EAAE,MAAM,KAAK,QAAQ,EAAE,CAAC;oBAC5C,MAAM,CAAC,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,OAAO,YAAY,EAAE,CAAC;gBAC7D,CAAC;qBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC;oBAC9C,MAAM,CAAC,MAAM,GAAG;wBACd,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM;wBACrB;4BACE,IAAI,EAAE,MAAe;4BACrB,IAAI,EAAE,YAAY;yBACnB;qBACF,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,MAAM,CAAC,MAAM,GAAG,YAAY,CAAC;gBAC/B,CAAC;YACH,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAE3D,4CAA4C;YAC5C,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CACvC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CACjC,CAAC;YACF,MAAM,OAAO,GAAG,WAAW,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;YAErE,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;YAChD,CAAC;YAED,mCAAmC;YACnC,MAAM,SAAS,GAGV,EAAE,CAAC;YAER,oCAAoC;YACpC,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;gBACrC,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAC9B,SAAS,CAAC,IAAI,CAAC;wBACb,QAAQ,EAAE,KAAK,CAAC,IAAI;wBACpB,SAAS,EAAE,KAAK,CAAC,KAAgC;qBAClD,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAED,6CAA6C;YAC7C,IAAI,UAA+C,CAAC;YACpD,IAAI,KAAK,CAAC,UAAU,EAAE,UAAU,EAAE,CAAC;gBACjC,IAAI,CAAC;oBACH,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAA4B,CAAC;gBAC9D,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,cAAM,CAAC,IAAI,CAAC,4CAA4C,EAAE,KAAK,CAAC,CAAC;oBACjE,kDAAkD;gBACpD,CAAC;YACH,CAAC;YAED,0DAA0D;YAC1D,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzB,UAAU,GAAG;oBACX,OAAO;oBACP,SAAS;oBACT,GAAG,UAAU;iBACa,CAAC;YAC/B,CAAC;YAED,OAAO;gBACL,OAAO;gBACP,QAAQ,EAAE;oBACR,KAAK,EAAE,QAAQ,CAAC,KAAK;oBACrB,UAAU,EAAE,QAAQ,CAAC,WAAW;oBAChC,UAAU,EACR,QAAQ,CAAC,KAAK,CAAC,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,aAAa;oBAC5D,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;oBACzC,gBAAgB,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa;iBAC/C;gBACD,UAAU;aACX,CAAC;QACJ,CAAC,CAAC;QAEF,OAAO,IAAA,2BAAmB,EACxB,SAAS,EACT,IAAI,CAAC,WAAW,CAAC,OAAO,EACxB,IAAI,CAAC,WAAW,CAAC,OAAO,EACxB,aAAa,KAAK,EAAE,CAC0B,CAAC;IACnD,CAAC;IAEO,KAAK,CAAC,CAAC,wBAAwB,CAIrC,KAAqC;QAErC,0BAA0B;QAC1B,IAAI,CAAC;YACH,KAAK,CAAC,CAAC,IAAI,CAAC,uBAAuB,CACjC,IAAI,CAAC,YAAY,EACjB,KAAK,CACN,CAAC;QACJ,CAAC;QAAC,OAAO,YAAqB,EAAE,CAAC;YAC/B,MAAM,aAAa,GAAG,eAAe,CAAC,YAAY,CAAC,CAAC;YACpD,cAAM,CAAC,IAAI,CACT,6BAA6B,IAAI,CAAC,YAAY,YAAY,aAAa,EAAE,CAC1E,CAAC;YAEF,IAAI,CAAC,oBAAoB,CAAC,YAAY,CAAC,EAAE,CAAC;gBACxC,MAAM,YAAY,CAAC;YACrB,CAAC;YAED,cAAM,CAAC,KAAK,CAAC,gDAAgD,CAAC,CAAC;YAE/D,IAAI,eAAe,GAAY,YAAY,CAAC;YAE5C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAClD,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;gBACzC,cAAM,CAAC,KAAK,CACV,mCAAmC,CAAC,GAAG,CAAC,IACtC,IAAI,CAAC,YAAY,CAAC,MACpB,KAAK,WAAW,EAAE,CACnB,CAAC;gBAEF,IAAI,CAAC;oBACH,KAAK,CAAC,CAAC,IAAI,CAAC,uBAAuB,CACjC,WAAW,EACX,KAAK,CACN,CAAC;oBACF,cAAM,CAAC,KAAK,CAAC,4BAA4B,WAAW,YAAY,CAAC,CAAC;oBAClE,OAAO;gBACT,CAAC;gBAAC,OAAO,WAAoB,EAAE,CAAC;oBAC9B,MAAM,YAAY,GAAG,eAAe,CAAC,WAAW,CAAC,CAAC;oBAClD,cAAM,CAAC,IAAI,CACT,4BAA4B,WAAW,YAAY,YAAY,EAAE,CAClE,CAAC;oBACF,eAAe,GAAG,WAAW,CAAC;oBAE9B,IACE,CAAC,oBAAoB,CAAC,WAAW,CAAC;wBAClC,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAChC,CAAC;wBACD,cAAM,CAAC,KAAK,CACV,qEAAqE,CACtE,CAAC;wBACF,MAAM;oBACR,CAAC;gBACH,CAAC;YACH,CAAC;YAED,MAAM,gBAAgB,GAAG,eAAe,CAAC,eAAe,CAAC,CAAC;YAC1D,cAAM,CAAC,KAAK,CACV,2CAA2C,aAAa,kBAAkB,gBAAgB,EAAE,CAC7F,CAAC;YACF,MAAM,eAAe,CAAC;QACxB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,uBAAuB,CAIpC,KAAa,EACb,KAAqC;QAErC,gDAAgD;QAChD,MAAM,SAAS,GAAG,KAAK,CAAC,UAAU,EAAE,eAAe,IAAI,IAAI,CAAC;QAE5D,MAAM,MAAM,GAAG;YACb,KAAK;YACL,UAAU,EAAE,SAAS;YACrB,QAAQ,EAAE;gBACR;oBACE,IAAI,EAAE,MAAe;oBACrB,OAAO,EAAE,KAAK,CAAC,MAAM;iBACtB;aACF;YACD,MAAM,EAAE,IAAI;YACZ,GAAG,IAAI,CAAC,MAAM;SACf,CAAC;QAEF,wBAAwB;QACxB,IAAI,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1C,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACxC,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,EAAE;gBAC1B,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;gBACnC,YAAY,EAAE,IAAI,CAAC,UAAkC,EAAE,cAAc;aACtE,CAAC,CAAC,CAAC;QACN,CAAC;QAED,+CAA+C;QAC/C,IAAI,KAAK,CAAC,UAAU,EAAE,UAAU,EAAE,CAAC;YACjC,MAAM,YAAY,GAChB,8EAA8E,CAAC;YAEjF,IAAI,OAAO,IAAI,CAAC,MAAM,EAAE,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC5C,MAAM,CAAC,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,OAAO,YAAY,EAAE,CAAC;YAC7D,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC;gBAC9C,MAAM,CAAC,MAAM,GAAG;oBACd,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM;oBACrB;wBACE,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,YAAY;qBACnB;iBACF,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,MAAM,GAAG,YAAY,CAAC;YAC/B,CAAC;QACH,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAEnD,IAAI,WAAW,GAAG,EAAE,CAAC;QACrB,IAAI,YAAY,GAAG,KAAK,CAAC;QACzB,IAAI,UAA8B,CAAC;QACnC,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,MAAM,SAAS,GAGV,EAAE,CAAC;QAER,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,IAAI,KAAK,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;gBACnC,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;gBACnC,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC;YACjD,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,qBAAqB,EAAE,CAAC;gBAChD,IAAI,KAAK,CAAC,aAAa,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAC5C,SAAS,CAAC,IAAI,CAAC;wBACb,QAAQ,EAAE,KAAK,CAAC,aAAa,CAAC,IAAI;wBAClC,SAAS,EAAE,KAAK,CAAC,aAAa,CAAC,KAAgC;qBAChE,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,qBAAqB,EAAE,CAAC;gBAChD,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oBACtC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC;oBAC/B,WAAW,IAAI,KAAK,CAAC;oBACrB,MAAM;wBACJ,KAAK;wBACL,WAAW;wBACX,IAAI,EAAE,KAAK;qBAC+B,CAAC;gBAC/C,CAAC;YACH,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;gBAC1C,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,WAAW,IAAI,SAAS,CAAC;gBAClD,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC,aAAa,CAAC;YAC3C,CAAC;QACH,CAAC;QAED,6CAA6C;QAC7C,IAAI,UAA+C,CAAC;QACpD,IAAI,KAAK,CAAC,UAAU,EAAE,UAAU,IAAI,WAAW,EAAE,CAAC;YAChD,IAAI,CAAC;gBACH,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAA4B,CAAC;YAClE,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,cAAM,CAAC,IAAI,CACT,sDAAsD,EACtD,KAAK,CACN,CAAC;YACJ,CAAC;QACH,CAAC;QAED,0DAA0D;QAC1D,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,UAAU,GAAG;gBACX,OAAO,EAAE,WAAW;gBACpB,SAAS;gBACT,GAAG,UAAU;aACa,CAAC;QAC/B,CAAC;QAED,oBAAoB;QACpB,MAAM;YACJ,KAAK,EAAE,EAAE;YACT,WAAW;YACX,IAAI,EAAE,IAAI;YACV,QAAQ,EAAE;gBACR,KAAK,EAAE,YAAY;gBACnB,UAAU;gBACV,UAAU,EAAE,WAAW,GAAG,YAAY;gBACtC,YAAY,EAAE,WAAW;gBACzB,gBAAgB,EAAE,YAAY;aAC/B;YACD,UAAU,EAAE,UAAyB;SACtC,CAAC;IACJ,CAAC;CACF;AAtbD,8CAsbC"}

package/dist/cjs/src/providers/GeminiProvider.d.ts

@@ -0,0 +1,58 @@
+/**
+ * Google Gemini AI provider implementation with retry and backup models
+ */
+import type { GenerateContentConfig } from "@google/genai";
+import type { AiProvider, GenerateMessageInput, GenerateMessageOutput, GenerateMessageStreamChunk, AgentStructuredResponse } from "../types";
+/**
+ * Configuration options for Gemini provider
+ * Uses types from @google/genai package
+ */
+export interface GeminiProviderOptions {
+    /** Gemini API key */
+    apiKey: string;
+    /** Model to use (required) - e.g., "models/gemini-2.5-pro" */
+    model: string;
+    /** Backup models to try if primary fails (default: []) */
+    backupModels?: string[];
+    /** Default generation config - uses GenerateContentConfig from @google/genai */
+    config?: Partial<GenerateContentConfig>;
+    /** Retry configuration */
+    retryConfig?: {
+        timeout?: number;
+        retries?: number;
+    };
+}
+/**
+ * Gemini provider implementation with backup models and retry logic
+ */
+export declare class GeminiProvider implements AiProvider {
+    readonly name = "gemini";
+    private genAI;
+    private primaryModel;
+    private backupModels;
+    private config?;
+    private retryConfig;
+    constructor(options: GeminiProviderOptions);
+    /**
+     * Adapt common schema format to Gemini's specific requirements.
+     * Gemini has strict validation:
+     * - OBJECT types MUST have non-empty properties
+     * - Empty objects cause 400 Bad Request errors
+     * - Uses Gemini's Schema type with Type enums
+     *
+     * @private
+     */
+    private adaptSchemaForGemini;
+    /**
+     * Map common JSON Schema type strings to Gemini's Type enum
+     * @private
+     */
+    private mapToGeminiType;
+    generateMessage<TContext = unknown, TStructured = AgentStructuredResponse>(input: GenerateMessageInput<TContext>): Promise<GenerateMessageOutput<TStructured>>;
+    generateMessageStream<TContext = unknown, TStructured = AgentStructuredResponse>(input: GenerateMessageInput<TContext>): AsyncGenerator<GenerateMessageStreamChunk<TStructured>>;
+    private generateWithBackup;
+    private generateWithModel;
+    private generateStreamWithBackup;
+    private generateStreamWithModel;
+}
+//# sourceMappingURL=GeminiProvider.d.ts.map

package/dist/cjs/src/providers/GeminiProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"GeminiProvider.d.ts","sourceRoot":"","sources":["../../../../src/providers/GeminiProvider.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAEV,qBAAqB,EAItB,MAAM,eAAe,CAAC;AAGvB,OAAO,KAAK,EACV,UAAU,EACV,oBAAoB,EACpB,qBAAqB,EACrB,0BAA0B,EAC1B,uBAAuB,EAExB,MAAM,UAAU,CAAC;AAUlB;;;GAGG;AACH,MAAM,WAAW,qBAAqB;IACpC,qBAAqB;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,8DAA8D;IAC9D,KAAK,EAAE,MAAM,CAAC;IACd,0DAA0D;IAC1D,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,gFAAgF;IAChF,MAAM,CAAC,EAAE,OAAO,CAAC,qBAAqB,CAAC,CAAC;IACxC,0BAA0B;IAC1B,WAAW,CAAC,EAAE;QACZ,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,CAAC;CACH;AA0ED;;GAEG;AACH,qBAAa,cAAe,YAAW,UAAU;IAC/C,SAAgB,IAAI,YAAY;IAChC,OAAO,CAAC,KAAK,CAAkB;IAC/B,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,YAAY,CAAW;IAC/B,OAAO,CAAC,MAAM,CAAC,CAAiC;IAChD,OAAO,CAAC,WAAW,CAAuC;gBAE9C,OAAO,EAAE,qBAAqB;IAqB1C;;;;;;;;OAQG;IACH,OAAO,CAAC,oBAAoB;IAoE5B;;;OAGG;IACH,OAAO,CAAC,eAAe;IAsBjB,eAAe,CACnB,QAAQ,GAAG,OAAO,EAClB,WAAW,GAAG,uBAAuB,EAErC,KAAK,EAAE,oBAAoB,CAAC,QAAQ,CAAC,GACpC,OAAO,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC;IAIvC,qBAAqB,CAC1B,QAAQ,GAAG,OAAO,EAClB,WAAW,GAAG,uBAAuB,EAErC,KAAK,EAAE,oBAAoB,CAAC,QAAQ,CAAC,GACpC,cAAc,CAAC,0BAA0B,CAAC,WAAW,CAAC,CAAC;YAI5C,kBAAkB;YA6DlB,iBAAiB;YA8JhB,wBAAwB;YA6DxB,uBAAuB;CA+HvC"}