@browserbasehq/orca 3.1.0-patch.3 → 3.2.0-preview.1

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

@@ -0,0 +1,246 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AnthropicClient = void 0;
+const sdk_1 = __importDefault(require("@anthropic-ai/sdk"));
+const LLMClient_js_1 = require("./LLMClient.js");
+const sdkErrors_js_1 = require("../types/public/sdkErrors.js");
+const zodCompat_js_1 = require("../zodCompat.js");
+class AnthropicClient extends LLMClient_js_1.LLMClient {
+    type = "anthropic";
+    client;
+    constructor({ modelName, clientOptions, userProvidedInstructions, }) {
+        super(modelName);
+        this.client = new sdk_1.default(clientOptions);
+        this.modelName = modelName;
+        this.clientOptions = clientOptions;
+        this.userProvidedInstructions = userProvidedInstructions;
+    }
+    async createChatCompletion({ options, retries, logger, }) {
+        const optionsWithoutImage = { ...options };
+        delete optionsWithoutImage.image;
+        logger({
+            category: "anthropic",
+            message: "creating chat completion",
+            level: 2,
+            auxiliary: {
+                options: {
+                    value: JSON.stringify(optionsWithoutImage),
+                    type: "object",
+                },
+            },
+        });
+        const systemMessage = options.messages.find((msg) => {
+            if (msg.role === "system") {
+                if (typeof msg.content === "string") {
+                    return true;
+                }
+                else if (Array.isArray(msg.content)) {
+                    return msg.content.every((content) => content.type !== "image_url");
+                }
+            }
+            return false;
+        });
+        const userMessages = options.messages.filter((msg) => msg.role !== "system");
+        const formattedMessages = userMessages.map((msg) => {
+            if (typeof msg.content === "string") {
+                return {
+                    role: msg.role, // ensure its not checking for system types
+                    content: msg.content,
+                };
+            }
+            else {
+                return {
+                    role: msg.role,
+                    content: msg.content.map((content) => {
+                        if ("image_url" in content) {
+                            const formattedContent = {
+                                type: "image",
+                                source: {
+                                    type: "base64",
+                                    media_type: "image/jpeg",
+                                    data: content.image_url.url,
+                                },
+                            };
+                            return formattedContent;
+                        }
+                        else {
+                            return { type: "text", text: content.text };
+                        }
+                    }),
+                };
+            }
+        });
+        if (options.image) {
+            const screenshotMessage = {
+                role: "user",
+                content: [
+                    {
+                        type: "image",
+                        source: {
+                            type: "base64",
+                            media_type: "image/jpeg",
+                            data: options.image.buffer.toString("base64"),
+                        },
+                    },
+                ],
+            };
+            if (options.image.description &&
+                Array.isArray(screenshotMessage.content)) {
+                screenshotMessage.content.push({
+                    type: "text",
+                    text: options.image.description,
+                });
+            }
+            formattedMessages.push(screenshotMessage);
+        }
+        let anthropicTools = options.tools?.map((tool) => {
+            return {
+                name: tool.name,
+                description: tool.description,
+                input_schema: {
+                    type: "object",
+                    properties: tool.parameters.properties,
+                    required: tool.parameters.required,
+                },
+            };
+        });
+        let toolDefinition;
+        if (options.response_model) {
+            const jsonSchema = (0, zodCompat_js_1.toJsonSchema)(options.response_model.schema);
+            const { properties: schemaProperties, required: schemaRequired } = extractSchemaProperties(jsonSchema);
+            toolDefinition = {
+                name: "print_extracted_data",
+                description: "Prints the extracted data based on the provided schema.",
+                input_schema: {
+                    type: "object",
+                    properties: schemaProperties,
+                    required: schemaRequired,
+                },
+            };
+        }
+        if (toolDefinition) {
+            anthropicTools = anthropicTools ?? [];
+            anthropicTools.push(toolDefinition);
+        }
+        const response = await this.client.messages.create({
+            model: this.modelName,
+            max_tokens: options.maxOutputTokens || 8192,
+            messages: formattedMessages,
+            tools: anthropicTools,
+            system: systemMessage
+                ? systemMessage.content // we can cast because we already filtered out image content
+                : undefined,
+            temperature: options.temperature,
+        });
+        logger({
+            category: "anthropic",
+            message: "response",
+            level: 2,
+            auxiliary: {
+                response: {
+                    value: JSON.stringify(response),
+                    type: "object",
+                },
+                requestId: {
+                    value: options.requestId,
+                    type: "string",
+                },
+            },
+        });
+        // We'll compute usage data from the response
+        const usageData = {
+            prompt_tokens: response.usage.input_tokens,
+            completion_tokens: response.usage.output_tokens,
+            total_tokens: response.usage.input_tokens + response.usage.output_tokens,
+        };
+        const transformedResponse = {
+            id: response.id,
+            object: "chat.completion",
+            created: Date.now(),
+            model: response.model,
+            choices: [
+                {
+                    index: 0,
+                    message: {
+                        role: "assistant",
+                        content: response.content.find((c) => c.type === "text")?.text || null,
+                        tool_calls: response.content
+                            .filter((c) => c.type === "tool_use")
+                            .map((toolUse) => ({
+                            id: toolUse.id,
+                            type: "function",
+                            function: {
+                                name: toolUse.name,
+                                arguments: JSON.stringify(toolUse.input),
+                            },
+                        })),
+                    },
+                    finish_reason: response.stop_reason,
+                },
+            ],
+            usage: usageData,
+        };
+        logger({
+            category: "anthropic",
+            message: "transformed response",
+            level: 2,
+            auxiliary: {
+                transformedResponse: {
+                    value: JSON.stringify(transformedResponse),
+                    type: "object",
+                },
+                requestId: {
+                    value: options.requestId,
+                    type: "string",
+                },
+            },
+        });
+        if (options.response_model) {
+            const toolUse = response.content.find((c) => c.type === "tool_use");
+            if (toolUse && "input" in toolUse) {
+                const result = toolUse.input;
+                const finalParsedResponse = {
+                    data: result,
+                    usage: usageData,
+                };
+                return finalParsedResponse;
+            }
+            else {
+                if (!retries || retries < 5) {
+                    return this.createChatCompletion({
+                        options,
+                        logger,
+                        retries: (retries ?? 0) + 1,
+                    });
+                }
+                logger({
+                    category: "anthropic",
+                    message: "error creating chat completion",
+                    level: 0,
+                    auxiliary: {
+                        requestId: {
+                            value: options.requestId,
+                            type: "string",
+                        },
+                    },
+                });
+                throw new sdkErrors_js_1.CreateChatCompletionResponseError("No tool use with input in response");
+            }
+        }
+        // if the function was called with a response model, it would have returned earlier
+        // so we can safely cast here to T, which defaults to AnthropicTransformedResponse
+        return transformedResponse;
+    }
+}
+exports.AnthropicClient = AnthropicClient;
+const extractSchemaProperties = (jsonSchema) => {
+    const schemaRoot = jsonSchema.definitions?.MySchema || jsonSchema;
+    return {
+        properties: schemaRoot.properties,
+        required: schemaRoot.required,
+    };
+};
+//# sourceMappingURL=AnthropicClient.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AnthropicClient.js","sourceRoot":"","sources":["../../../../../lib/v3/llm/AnthropicClient.ts"],"names":[],"mappings":";;;;;;AAAA,4DAA6D;AAY7D,iDAIwB;AACxB,+DAAiF;AACjF,kDAA+C;AAE/C,MAAa,eAAgB,SAAQ,wBAAS;IACrC,IAAI,GAAG,WAAoB,CAAC;IAC3B,MAAM,CAAY;IAG1B,YAAY,EACV,SAAS,EACT,aAAa,EACb,wBAAwB,GAMzB;QACC,KAAK,CAAC,SAAS,CAAC,CAAC;QACjB,IAAI,CAAC,MAAM,GAAG,IAAI,aAAS,CAAC,aAAa,CAAC,CAAC;QAC3C,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,wBAAwB,GAAG,wBAAwB,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAkB,EAC1C,OAAO,EACP,OAAO,EACP,MAAM,GACsB;QAC5B,MAAM,mBAAmB,GAAG,EAAE,GAAG,OAAO,EAAE,CAAC;QAC3C,OAAO,mBAAmB,CAAC,KAAK,CAAC;QAEjC,MAAM,CAAC;YACL,QAAQ,EAAE,WAAW;YACrB,OAAO,EAAE,0BAA0B;YACnC,KAAK,EAAE,CAAC;YACR,SAAS,EAAE;gBACT,OAAO,EAAE;oBACP,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC;oBAC1C,IAAI,EAAE,QAAQ;iBACf;aACF;SACF,CAAC,CAAC;QAEH,MAAM,aAAa,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE;YAClD,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC1B,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;oBACpC,OAAO,IAAI,CAAC;gBACd,CAAC;qBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBACtC,OAAO,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;gBACtE,CAAC;YACH,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAC1C,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,QAAQ,CAC/B,CAAC;QAEF,MAAM,iBAAiB,GAAmB,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACjE,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;gBACpC,OAAO;oBACL,IAAI,EAAE,GAAG,CAAC,IAA4B,EAAE,2CAA2C;oBACnF,OAAO,EAAE,GAAG,CAAC,OAAO;iBACrB,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,OAAO;oBACL,IAAI,EAAE,GAAG,CAAC,IAA4B;oBACtC,OAAO,EAAE,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;wBACnC,IAAI,WAAW,IAAI,OAAO,EAAE,CAAC;4BAC3B,MAAM,gBAAgB,GAAoB;gCACxC,IAAI,EAAE,OAAO;gCACb,MAAM,EAAE;oCACN,IAAI,EAAE,QAAQ;oCACd,UAAU,EAAE,YAAY;oCACxB,IAAI,EAAE,OAAO,CAAC,SAAS,CAAC,GAAG;iCAC5B;6BACF,CAAC;4BAEF,OAAO,gBAAgB,CAAC;wBAC1B,CAAC;6BAAM,CAAC;4BACN,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,CAAC;wBAC9C,CAAC;oBACH,CAAC,CAAC;iBACH,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,MAAM,iBAAiB,GAAiB;gBACtC,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,OAAO;wBACb,MAAM,EAAE;4BACN,IAAI,EAAE,QAAQ;4BACd,UAAU,EAAE,YAAY;4BACxB,IAAI,EAAE,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC;yBAC9C;qBACF;iBACF;aACF,CAAC;YACF,IACE,OAAO,CAAC,KAAK,CAAC,WAAW;gBACzB,KAAK,CAAC,OAAO,CAAC,iBAAiB,CAAC,OAAO,CAAC,EACxC,CAAC;gBACD,iBAAiB,CAAC,OAAO,CAAC,IAAI,CAAC;oBAC7B,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,OAAO,CAAC,KAAK,CAAC,WAAW;iBAChC,CAAC,CAAC;YACL,CAAC;YAED,iBAAiB,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC5C,CAAC;QAED,IAAI,cAAc,GAAW,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACvD,OAAO;gBACL,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,YAAY,EAAE;oBACZ,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,UAAU;oBACtC,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,QAAQ;iBACnC;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,cAAgC,CAAC;QACrC,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;YAC3B,MAAM,UAAU,GAAG,IAAA,2BAAY,EAAC,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;YAC/D,MAAM,EAAE,UAAU,EAAE,gBAAgB,EAAE,QAAQ,EAAE,cAAc,EAAE,GAC9D,uBAAuB,CAAC,UAAU,CAAC,CAAC;YAEtC,cAAc,GAAG;gBACf,IAAI,EAAE,sBAAsB;gBAC5B,WAAW,EAAE,yDAAyD;gBACtE,YAAY,EAAE;oBACZ,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE,gBAAgB;oBAC5B,QAAQ,EAAE,cAAc;iBACzB;aACF,CAAC;QACJ,CAAC;QAED,IAAI,cAAc,EAAE,CAAC;YACnB,cAAc,GAAG,cAAc,IAAI,EAAE,CAAC;YACtC,cAAc,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACtC,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;YACjD,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,UAAU,EAAE,OAAO,CAAC,eAAe,IAAI,IAAI;YAC3C,QAAQ,EAAE,iBAAiB;YAC3B,KAAK,EAAE,cAAc;YACrB,MAAM,EAAE,aAAa;gBACnB,CAAC,CAAE,aAAa,CAAC,OAAqC,CAAC,4DAA4D;gBACnH,CAAC,CAAC,SAAS;YACb,WAAW,EAAE,OAAO,CAAC,WAAW;SACjC,CAAC,CAAC;QAEH,MAAM,CAAC;YACL,QAAQ,EAAE,WAAW;YACrB,OAAO,EAAE,UAAU;YACnB,KAAK,EAAE,CAAC;YACR,SAAS,EAAE;gBACT,QAAQ,EAAE;oBACR,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC;oBAC/B,IAAI,EAAE,QAAQ;iBACf;gBACD,SAAS,EAAE;oBACT,KAAK,EAAE,OAAO,CAAC,SAAS;oBACxB,IAAI,EAAE,QAAQ;iBACf;aACF;SACF,CAAC,CAAC;QAEH,6CAA6C;QAC7C,MAAM,SAAS,GAAG;YAChB,aAAa,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;YAC1C,iBAAiB,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa;YAC/C,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,aAAa;SACzE,CAAC;QAEF,MAAM,mBAAmB,GAAgB;YACvC,EAAE,EAAE,QAAQ,CAAC,EAAE;YACf,MAAM,EAAE,iBAAiB;YACzB,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE;YACnB,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,OAAO,EAAE;gBACP;oBACE,KAAK,EAAE,CAAC;oBACR,OAAO,EAAE;wBACP,IAAI,EAAE,WAAW;wBACjB,OAAO,EACL,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,IAAI,IAAI,IAAI;wBAC/D,UAAU,EAAE,QAAQ,CAAC,OAAO;6BACzB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC;6BACpC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;4BACjB,EAAE,EAAE,OAAO,CAAC,EAAE;4BACd,IAAI,EAAE,UAAU;4BAChB,QAAQ,EAAE;gCACR,IAAI,EAAE,OAAO,CAAC,IAAI;gCAClB,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC;6BACzC;yBACF,CAAC,CAAC;qBACN;oBACD,aAAa,EAAE,QAAQ,CAAC,WAAW;iBACpC;aACF;YACD,KAAK,EAAE,SAAS;SACjB,CAAC;QAEF,MAAM,CAAC;YACL,QAAQ,EAAE,WAAW;YACrB,OAAO,EAAE,sBAAsB;YAC/B,KAAK,EAAE,CAAC;YACR,SAAS,EAAE;gBACT,mBAAmB,EAAE;oBACnB,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC;oBAC1C,IAAI,EAAE,QAAQ;iBACf;gBACD,SAAS,EAAE;oBACT,KAAK,EAAE,OAAO,CAAC,SAAS;oBACxB,IAAI,EAAE,QAAQ;iBACf;aACF;SACF,CAAC,CAAC;QAEH,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;YAC3B,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;YACpE,IAAI,OAAO,IAAI,OAAO,IAAI,OAAO,EAAE,CAAC;gBAClC,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;gBAE7B,MAAM,mBAAmB,GAAG;oBAC1B,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,SAAS;iBACD,CAAC;gBAElB,OAAO,mBAAmB,CAAC;YAC7B,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,OAAO,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;oBAC5B,OAAO,IAAI,CAAC,oBAAoB,CAAC;wBAC/B,OAAO;wBACP,MAAM;wBACN,OAAO,EAAE,CAAC,OAAO,IAAI,CAAC,CAAC,GAAG,CAAC;qBAC5B,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM,CAAC;oBACL,QAAQ,EAAE,WAAW;oBACrB,OAAO,EAAE,gCAAgC;oBACzC,KAAK,EAAE,CAAC;oBACR,SAAS,EAAE;wBACT,SAAS,EAAE;4BACT,KAAK,EAAE,OAAO,CAAC,SAAS;4BACxB,IAAI,EAAE,QAAQ;yBACf;qBACF;iBACF,CAAC,CAAC;gBACH,MAAM,IAAI,gDAAiC,CACzC,oCAAoC,CACrC,CAAC;YACJ,CAAC;QACH,CAAC;QAED,mFAAmF;QACnF,kFAAkF;QAClF,OAAO,mBAAwB,CAAC;IAClC,CAAC;CACF;AA1QD,0CA0QC;AAED,MAAM,uBAAuB,GAAG,CAAC,UAAqC,EAAE,EAAE;IACxE,MAAM,UAAU,GAAG,UAAU,CAAC,WAAW,EAAE,QAAQ,IAAI,UAAU,CAAC;IAElE,OAAO;QACL,UAAU,EAAE,UAAU,CAAC,UAAU;QACjC,QAAQ,EAAE,UAAU,CAAC,QAAQ;KAC9B,CAAC;AACJ,CAAC,CAAC","sourcesContent":["import Anthropic, { ClientOptions } from \"@anthropic-ai/sdk\";\nimport {\n  ImageBlockParam,\n  MessageParam,\n  TextBlockParam,\n  Tool,\n} from \"@anthropic-ai/sdk/resources\";\nimport { LogLine } from \"../types/public/logs.js\";\nimport {\n  AnthropicJsonSchemaObject,\n  AvailableModel,\n} from \"../types/public/model.js\";\nimport {\n  CreateChatCompletionOptions,\n  LLMClient,\n  LLMResponse,\n} from \"./LLMClient.js\";\nimport { CreateChatCompletionResponseError } from \"../types/public/sdkErrors.js\";\nimport { toJsonSchema } from \"../zodCompat.js\";\n\nexport class AnthropicClient extends LLMClient {\n  public type = \"anthropic\" as const;\n  private client: Anthropic;\n  declare public clientOptions: ClientOptions;\n\n  constructor({\n    modelName,\n    clientOptions,\n    userProvidedInstructions,\n  }: {\n    logger: (message: LogLine) => void;\n    modelName: AvailableModel;\n    clientOptions?: ClientOptions;\n    userProvidedInstructions?: string;\n  }) {\n    super(modelName);\n    this.client = new Anthropic(clientOptions);\n    this.modelName = modelName;\n    this.clientOptions = clientOptions;\n    this.userProvidedInstructions = userProvidedInstructions;\n  }\n\n  async createChatCompletion<T = LLMResponse>({\n    options,\n    retries,\n    logger,\n  }: CreateChatCompletionOptions): Promise<T> {\n    const optionsWithoutImage = { ...options };\n    delete optionsWithoutImage.image;\n\n    logger({\n      category: \"anthropic\",\n      message: \"creating chat completion\",\n      level: 2,\n      auxiliary: {\n        options: {\n          value: JSON.stringify(optionsWithoutImage),\n          type: \"object\",\n        },\n      },\n    });\n\n    const systemMessage = options.messages.find((msg) => {\n      if (msg.role === \"system\") {\n        if (typeof msg.content === \"string\") {\n          return true;\n        } else if (Array.isArray(msg.content)) {\n          return msg.content.every((content) => content.type !== \"image_url\");\n        }\n      }\n      return false;\n    });\n\n    const userMessages = options.messages.filter(\n      (msg) => msg.role !== \"system\",\n    );\n\n    const formattedMessages: MessageParam[] = userMessages.map((msg) => {\n      if (typeof msg.content === \"string\") {\n        return {\n          role: msg.role as \"user\" | \"assistant\", // ensure its not checking for system types\n          content: msg.content,\n        };\n      } else {\n        return {\n          role: msg.role as \"user\" | \"assistant\",\n          content: msg.content.map((content) => {\n            if (\"image_url\" in content) {\n              const formattedContent: ImageBlockParam = {\n                type: \"image\",\n                source: {\n                  type: \"base64\",\n                  media_type: \"image/jpeg\",\n                  data: content.image_url.url,\n                },\n              };\n\n              return formattedContent;\n            } else {\n              return { type: \"text\", text: content.text };\n            }\n          }),\n        };\n      }\n    });\n\n    if (options.image) {\n      const screenshotMessage: MessageParam = {\n        role: \"user\",\n        content: [\n          {\n            type: \"image\",\n            source: {\n              type: \"base64\",\n              media_type: \"image/jpeg\",\n              data: options.image.buffer.toString(\"base64\"),\n            },\n          },\n        ],\n      };\n      if (\n        options.image.description &&\n        Array.isArray(screenshotMessage.content)\n      ) {\n        screenshotMessage.content.push({\n          type: \"text\",\n          text: options.image.description,\n        });\n      }\n\n      formattedMessages.push(screenshotMessage);\n    }\n\n    let anthropicTools: Tool[] = options.tools?.map((tool) => {\n      return {\n        name: tool.name,\n        description: tool.description,\n        input_schema: {\n          type: \"object\",\n          properties: tool.parameters.properties,\n          required: tool.parameters.required,\n        },\n      };\n    });\n\n    let toolDefinition: Tool | undefined;\n    if (options.response_model) {\n      const jsonSchema = toJsonSchema(options.response_model.schema);\n      const { properties: schemaProperties, required: schemaRequired } =\n        extractSchemaProperties(jsonSchema);\n\n      toolDefinition = {\n        name: \"print_extracted_data\",\n        description: \"Prints the extracted data based on the provided schema.\",\n        input_schema: {\n          type: \"object\",\n          properties: schemaProperties,\n          required: schemaRequired,\n        },\n      };\n    }\n\n    if (toolDefinition) {\n      anthropicTools = anthropicTools ?? [];\n      anthropicTools.push(toolDefinition);\n    }\n\n    const response = await this.client.messages.create({\n      model: this.modelName,\n      max_tokens: options.maxOutputTokens || 8192,\n      messages: formattedMessages,\n      tools: anthropicTools,\n      system: systemMessage\n        ? (systemMessage.content as string | TextBlockParam[]) // we can cast because we already filtered out image content\n        : undefined,\n      temperature: options.temperature,\n    });\n\n    logger({\n      category: \"anthropic\",\n      message: \"response\",\n      level: 2,\n      auxiliary: {\n        response: {\n          value: JSON.stringify(response),\n          type: \"object\",\n        },\n        requestId: {\n          value: options.requestId,\n          type: \"string\",\n        },\n      },\n    });\n\n    // We'll compute usage data from the response\n    const usageData = {\n      prompt_tokens: response.usage.input_tokens,\n      completion_tokens: response.usage.output_tokens,\n      total_tokens: response.usage.input_tokens + response.usage.output_tokens,\n    };\n\n    const transformedResponse: LLMResponse = {\n      id: response.id,\n      object: \"chat.completion\",\n      created: Date.now(),\n      model: response.model,\n      choices: [\n        {\n          index: 0,\n          message: {\n            role: \"assistant\",\n            content:\n              response.content.find((c) => c.type === \"text\")?.text || null,\n            tool_calls: response.content\n              .filter((c) => c.type === \"tool_use\")\n              .map((toolUse) => ({\n                id: toolUse.id,\n                type: \"function\",\n                function: {\n                  name: toolUse.name,\n                  arguments: JSON.stringify(toolUse.input),\n                },\n              })),\n          },\n          finish_reason: response.stop_reason,\n        },\n      ],\n      usage: usageData,\n    };\n\n    logger({\n      category: \"anthropic\",\n      message: \"transformed response\",\n      level: 2,\n      auxiliary: {\n        transformedResponse: {\n          value: JSON.stringify(transformedResponse),\n          type: \"object\",\n        },\n        requestId: {\n          value: options.requestId,\n          type: \"string\",\n        },\n      },\n    });\n\n    if (options.response_model) {\n      const toolUse = response.content.find((c) => c.type === \"tool_use\");\n      if (toolUse && \"input\" in toolUse) {\n        const result = toolUse.input;\n\n        const finalParsedResponse = {\n          data: result,\n          usage: usageData,\n        } as unknown as T;\n\n        return finalParsedResponse;\n      } else {\n        if (!retries || retries < 5) {\n          return this.createChatCompletion({\n            options,\n            logger,\n            retries: (retries ?? 0) + 1,\n          });\n        }\n        logger({\n          category: \"anthropic\",\n          message: \"error creating chat completion\",\n          level: 0,\n          auxiliary: {\n            requestId: {\n              value: options.requestId,\n              type: \"string\",\n            },\n          },\n        });\n        throw new CreateChatCompletionResponseError(\n          \"No tool use with input in response\",\n        );\n      }\n    }\n\n    // if the function was called with a response model, it would have returned earlier\n    // so we can safely cast here to T, which defaults to AnthropicTransformedResponse\n    return transformedResponse as T;\n  }\n}\n\nconst extractSchemaProperties = (jsonSchema: AnthropicJsonSchemaObject) => {\n  const schemaRoot = jsonSchema.definitions?.MySchema || jsonSchema;\n\n  return {\n    properties: schemaRoot.properties,\n    required: schemaRoot.required,\n  };\n};\n"]}

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

@@ -0,0 +1,242 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CerebrasClient = void 0;
+const openai_1 = __importDefault(require("openai"));
+const LLMClient_js_1 = require("./LLMClient.js");
+const sdkErrors_js_1 = require("../types/public/sdkErrors.js");
+const zodCompat_js_1 = require("../zodCompat.js");
+class CerebrasClient extends LLMClient_js_1.LLMClient {
+    type = "cerebras";
+    client;
+    hasVision = false;
+    constructor({ modelName, clientOptions, userProvidedInstructions, }) {
+        super(modelName, userProvidedInstructions);
+        // Create OpenAI client with the base URL set to Cerebras API
+        this.client = new openai_1.default({
+            baseURL: "https://api.cerebras.ai/v1",
+            apiKey: clientOptions?.apiKey || process.env.CEREBRAS_API_KEY,
+            ...clientOptions,
+        });
+        this.modelName = modelName;
+        this.clientOptions = clientOptions;
+    }
+    async createChatCompletion({ options, retries, logger, }) {
+        const optionsWithoutImage = { ...options };
+        delete optionsWithoutImage.image;
+        logger({
+            category: "cerebras",
+            message: "creating chat completion",
+            level: 2,
+            auxiliary: {
+                options: {
+                    value: JSON.stringify(optionsWithoutImage),
+                    type: "object",
+                },
+            },
+        });
+        // Format messages for Cerebras API (using OpenAI format)
+        const formattedMessages = options.messages.map((msg) => {
+            const baseMessage = {
+                content: typeof msg.content === "string"
+                    ? msg.content
+                    : Array.isArray(msg.content) &&
+                        msg.content.length > 0 &&
+                        "text" in msg.content[0]
+                        ? msg.content[0].text
+                        : "",
+            };
+            // Cerebras only supports system, user, and assistant roles
+            if (msg.role === "system") {
+                return { ...baseMessage, role: "system" };
+            }
+            else if (msg.role === "assistant") {
+                return { ...baseMessage, role: "assistant" };
+            }
+            else {
+                // Default to user for any other role
+                return { ...baseMessage, role: "user" };
+            }
+        });
+        // Format tools if provided
+        let tools = options.tools?.map((tool) => ({
+            type: "function",
+            function: {
+                name: tool.name,
+                description: tool.description,
+                parameters: {
+                    type: "object",
+                    properties: tool.parameters.properties,
+                    required: tool.parameters.required,
+                },
+            },
+        }));
+        // Add response model as a tool if provided
+        if (options.response_model) {
+            const jsonSchema = (0, zodCompat_js_1.toJsonSchema)(options.response_model.schema);
+            const schemaProperties = jsonSchema.properties || {};
+            const schemaRequired = jsonSchema.required || [];
+            const responseTool = {
+                type: "function",
+                function: {
+                    name: "print_extracted_data",
+                    description: "Prints the extracted data based on the provided schema.",
+                    parameters: {
+                        type: "object",
+                        properties: schemaProperties,
+                        required: schemaRequired,
+                    },
+                },
+            };
+            tools = tools ? [...tools, responseTool] : [responseTool];
+        }
+        try {
+            // Use OpenAI client with Cerebras API
+            const apiResponse = await this.client.chat.completions.create({
+                model: this.modelName.split("cerebras-")[1],
+                messages: [
+                    ...formattedMessages,
+                    // Add explicit instruction to return JSON if we have a response model
+                    ...(options.response_model
+                        ? [
+                            {
+                                role: "system",
+                                content: `IMPORTANT: Your response must be valid JSON that matches this schema: ${JSON.stringify(options.response_model.schema)}`,
+                            },
+                        ]
+                        : []),
+                ],
+                temperature: options.temperature || 0.7,
+                max_tokens: options.maxOutputTokens,
+                tools: tools,
+                tool_choice: options.tool_choice || "auto",
+            });
+            // Format the response to match the expected LLMResponse format
+            const response = {
+                id: apiResponse.id,
+                object: "chat.completion",
+                created: Date.now(),
+                model: this.modelName.split("cerebras-")[1],
+                choices: [
+                    {
+                        index: 0,
+                        message: {
+                            role: "assistant",
+                            content: apiResponse.choices[0]?.message?.content || null,
+                            tool_calls: apiResponse.choices[0]?.message?.tool_calls || [],
+                        },
+                        finish_reason: apiResponse.choices[0]?.finish_reason || "stop",
+                    },
+                ],
+                usage: {
+                    prompt_tokens: apiResponse.usage?.prompt_tokens || 0,
+                    completion_tokens: apiResponse.usage?.completion_tokens || 0,
+                    total_tokens: apiResponse.usage?.total_tokens || 0,
+                },
+            };
+            logger({
+                category: "cerebras",
+                message: "response",
+                level: 2,
+                auxiliary: {
+                    response: {
+                        value: JSON.stringify(response),
+                        type: "object",
+                    },
+                    requestId: {
+                        value: options.requestId,
+                        type: "string",
+                    },
+                },
+            });
+            // If we have no response model, just return the entire LLMResponse
+            if (!options.response_model) {
+                return response;
+            }
+            // If we have a response model, parse JSON from tool calls or content
+            const toolCall = response.choices[0]?.message?.tool_calls?.[0];
+            if (toolCall?.function?.arguments) {
+                try {
+                    const result = JSON.parse(toolCall.function.arguments);
+                    const finalResponse = {
+                        data: result,
+                        usage: response.usage,
+                    };
+                    return finalResponse;
+                }
+                catch (e) {
+                    logger({
+                        category: "cerebras",
+                        message: "failed to parse tool call arguments as JSON, retrying",
+                        level: 0,
+                        auxiliary: {
+                            error: {
+                                value: e.message,
+                                type: "string",
+                            },
+                        },
+                    });
+                }
+            }
+            // If we have content but no tool calls, try to parse the content as JSON
+            const content = response.choices[0]?.message?.content;
+            if (content) {
+                try {
+                    const jsonMatch = content.match(/\{[\s\S]*\}/);
+                    if (jsonMatch) {
+                        const result = JSON.parse(jsonMatch[0]);
+                        const finalResponse = {
+                            data: result,
+                            usage: response.usage,
+                        };
+                        return finalResponse;
+                    }
+                }
+                catch (e) {
+                    logger({
+                        category: "cerebras",
+                        message: "failed to parse content as JSON",
+                        level: 0,
+                        auxiliary: {
+                            error: {
+                                value: e.message,
+                                type: "string",
+                            },
+                        },
+                    });
+                }
+            }
+            // If we still haven't found valid JSON and have retries left, try again
+            if (!retries || retries < 5) {
+                return this.createChatCompletion({
+                    options,
+                    logger,
+                    retries: (retries ?? 0) + 1,
+                });
+            }
+            throw new sdkErrors_js_1.CreateChatCompletionResponseError("Invalid response schema");
+        }
+        catch (error) {
+            logger({
+                category: "cerebras",
+                message: "error creating chat completion",
+                level: 0,
+                auxiliary: {
+                    error: {
+                        value: error.message,
+                        type: "string",
+                    },
+                    requestId: {
+                        value: options.requestId,
+                        type: "string",
+                    },
+                },
+            });
+            throw error;
+        }
+    }
+}
+exports.CerebrasClient = CerebrasClient;
+//# sourceMappingURL=CerebrasClient.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"CerebrasClient.js","sourceRoot":"","sources":["../../../../../lib/v3/llm/CerebrasClient.ts"],"names":[],"mappings":";;;;;;AAAA,oDAA4B;AAI5B,iDAKwB;AACxB,+DAAiF;AACjF,kDAA+C;AAE/C,MAAa,cAAe,SAAQ,wBAAS;IACpC,IAAI,GAAG,UAAmB,CAAC;IAC1B,MAAM,CAAS;IAEhB,SAAS,GAAG,KAAK,CAAC;IAEzB,YAAY,EACV,SAAS,EACT,aAAa,EACb,wBAAwB,GAMzB;QACC,KAAK,CAAC,SAAS,EAAE,wBAAwB,CAAC,CAAC;QAE3C,6DAA6D;QAC7D,IAAI,CAAC,MAAM,GAAG,IAAI,gBAAM,CAAC;YACvB,OAAO,EAAE,4BAA4B;YACrC,MAAM,EAAE,aAAa,EAAE,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB;YAC7D,GAAG,aAAa;SACjB,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAkB,EAC1C,OAAO,EACP,OAAO,EACP,MAAM,GACsB;QAC5B,MAAM,mBAAmB,GAAG,EAAE,GAAG,OAAO,EAAE,CAAC;QAC3C,OAAO,mBAAmB,CAAC,KAAK,CAAC;QAEjC,MAAM,CAAC;YACL,QAAQ,EAAE,UAAU;YACpB,OAAO,EAAE,0BAA0B;YACnC,KAAK,EAAE,CAAC;YACR,SAAS,EAAE;gBACT,OAAO,EAAE;oBACP,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC;oBAC1C,IAAI,EAAE,QAAQ;iBACf;aACF;SACF,CAAC,CAAC;QAEH,yDAAyD;QACzD,MAAM,iBAAiB,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAgB,EAAE,EAAE;YAClE,MAAM,WAAW,GAAG;gBAClB,OAAO,EACL,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ;oBAC7B,CAAC,CAAC,GAAG,CAAC,OAAO;oBACb,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC;wBACxB,GAAG,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC;wBACtB,MAAM,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;wBAC1B,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI;wBACrB,CAAC,CAAC,EAAE;aACX,CAAC;YAEF,2DAA2D;YAC3D,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC1B,OAAO,EAAE,GAAG,WAAW,EAAE,IAAI,EAAE,QAAiB,EAAE,CAAC;YACrD,CAAC;iBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBACpC,OAAO,EAAE,GAAG,WAAW,EAAE,IAAI,EAAE,WAAoB,EAAE,CAAC;YACxD,CAAC;iBAAM,CAAC;gBACN,qCAAqC;gBACrC,OAAO,EAAE,GAAG,WAAW,EAAE,IAAI,EAAE,MAAe,EAAE,CAAC;YACnD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,2BAA2B;QAC3B,IAAI,KAAK,GAAG,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACxC,IAAI,EAAE,UAAmB;YACzB,QAAQ,EAAE;gBACR,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,UAAU;oBACtC,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,QAAQ;iBACnC;aACF;SACF,CAAC,CAAC,CAAC;QAEJ,2CAA2C;QAC3C,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;YAC3B,MAAM,UAAU,GAAG,IAAA,2BAAY,EAAC,OAAO,CAAC,cAAc,CAAC,MAAM,CAG5D,CAAC;YACF,MAAM,gBAAgB,GAAG,UAAU,CAAC,UAAU,IAAI,EAAE,CAAC;YACrD,MAAM,cAAc,GAAG,UAAU,CAAC,QAAQ,IAAI,EAAE,CAAC;YAEjD,MAAM,YAAY,GAAG;gBACnB,IAAI,EAAE,UAAmB;gBACzB,QAAQ,EAAE;oBACR,IAAI,EAAE,sBAAsB;oBAC5B,WAAW,EACT,yDAAyD;oBAC3D,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,gBAAgB;wBAC5B,QAAQ,EAAE,cAAc;qBACzB;iBACF;aACF,CAAC;YAEF,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,KAAK,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC;QAC5D,CAAC;QAED,IAAI,CAAC;YACH,sCAAsC;YACtC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBAC5D,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;gBAC3C,QAAQ,EAAE;oBACR,GAAG,iBAAiB;oBACpB,sEAAsE;oBACtE,GAAG,CAAC,OAAO,CAAC,cAAc;wBACxB,CAAC,CAAC;4BACE;gCACE,IAAI,EAAE,QAAiB;gCACvB,OAAO,EAAE,yEAAyE,IAAI,CAAC,SAAS,CAC9F,OAAO,CAAC,cAAc,CAAC,MAAM,CAC9B,EAAE;6BACJ;yBACF;wBACH,CAAC,CAAC,EAAE,CAAC;iBACR;gBACD,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;gBACvC,UAAU,EAAE,OAAO,CAAC,eAAe;gBACnC,KAAK,EAAE,KAAK;gBACZ,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,MAAM;aAC3C,CAAC,CAAC;YAEH,+DAA+D;YAC/D,MAAM,QAAQ,GAAgB;gBAC5B,EAAE,EAAE,WAAW,CAAC,EAAE;gBAClB,MAAM,EAAE,iBAAiB;gBACzB,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE;gBACnB,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;gBAC3C,OAAO,EAAE;oBACP;wBACE,KAAK,EAAE,CAAC;wBACR,OAAO,EAAE;4BACP,IAAI,EAAE,WAAW;4BACjB,OAAO,EAAE,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,IAAI;4BACzD,UAAU,EAAE,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,IAAI,EAAE;yBAC9D;wBACD,aAAa,EAAE,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,IAAI,MAAM;qBAC/D;iBACF;gBACD,KAAK,EAAE;oBACL,aAAa,EAAE,WAAW,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;oBACpD,iBAAiB,EAAE,WAAW,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC;oBAC5D,YAAY,EAAE,WAAW,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC;iBACnD;aACF,CAAC;YAEF,MAAM,CAAC;gBACL,QAAQ,EAAE,UAAU;gBACpB,OAAO,EAAE,UAAU;gBACnB,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE;oBACT,QAAQ,EAAE;wBACR,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC;wBAC/B,IAAI,EAAE,QAAQ;qBACf;oBACD,SAAS,EAAE;wBACT,KAAK,EAAE,OAAO,CAAC,SAAS;wBACxB,IAAI,EAAE,QAAQ;qBACf;iBACF;aACF,CAAC,CAAC;YAEH,mEAAmE;YACnE,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;gBAC5B,OAAO,QAAa,CAAC;YACvB,CAAC;YAED,qEAAqE;YACrE,MAAM,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;YAC/D,IAAI,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC;gBAClC,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;oBACvD,MAAM,aAAa,GAAG;wBACpB,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,QAAQ,CAAC,KAAK;qBACtB,CAAC;oBACF,OAAO,aAAkB,CAAC;gBAC5B,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,MAAM,CAAC;wBACL,QAAQ,EAAE,UAAU;wBACpB,OAAO,EAAE,uDAAuD;wBAChE,KAAK,EAAE,CAAC;wBACR,SAAS,EAAE;4BACT,KAAK,EAAE;gCACL,KAAK,EAAE,CAAC,CAAC,OAAO;gCAChB,IAAI,EAAE,QAAQ;6BACf;yBACF;qBACF,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAED,yEAAyE;YACzE,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC;YACtD,IAAI,OAAO,EAAE,CAAC;gBACZ,IAAI,CAAC;oBACH,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;oBAC/C,IAAI,SAAS,EAAE,CAAC;wBACd,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;wBACxC,MAAM,aAAa,GAAG;4BACpB,IAAI,EAAE,MAAM;4BACZ,KAAK,EAAE,QAAQ,CAAC,KAAK;yBACtB,CAAC;wBACF,OAAO,aAAkB,CAAC;oBAC5B,CAAC;gBACH,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,MAAM,CAAC;wBACL,QAAQ,EAAE,UAAU;wBACpB,OAAO,EAAE,iCAAiC;wBAC1C,KAAK,EAAE,CAAC;wBACR,SAAS,EAAE;4BACT,KAAK,EAAE;gCACL,KAAK,EAAE,CAAC,CAAC,OAAO;gCAChB,IAAI,EAAE,QAAQ;6BACf;yBACF;qBACF,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAED,wEAAwE;YACxE,IAAI,CAAC,OAAO,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;gBAC5B,OAAO,IAAI,CAAC,oBAAoB,CAAC;oBAC/B,OAAO;oBACP,MAAM;oBACN,OAAO,EAAE,CAAC,OAAO,IAAI,CAAC,CAAC,GAAG,CAAC;iBAC5B,CAAC,CAAC;YACL,CAAC;YAED,MAAM,IAAI,gDAAiC,CAAC,yBAAyB,CAAC,CAAC;QACzE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC;gBACL,QAAQ,EAAE,UAAU;gBACpB,OAAO,EAAE,gCAAgC;gBACzC,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE;oBACT,KAAK,EAAE;wBACL,KAAK,EAAE,KAAK,CAAC,OAAO;wBACpB,IAAI,EAAE,QAAQ;qBACf;oBACD,SAAS,EAAE;wBACT,KAAK,EAAE,OAAO,CAAC,SAAS;wBACxB,IAAI,EAAE,QAAQ;qBACf;iBACF;aACF,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF;AAxQD,wCAwQC","sourcesContent":["import OpenAI from \"openai\";\nimport type { ClientOptions } from \"openai\";\nimport { LogLine } from \"../types/public/logs.js\";\nimport { AvailableModel } from \"../types/public/model.js\";\nimport {\n  ChatMessage,\n  CreateChatCompletionOptions,\n  LLMClient,\n  LLMResponse,\n} from \"./LLMClient.js\";\nimport { CreateChatCompletionResponseError } from \"../types/public/sdkErrors.js\";\nimport { toJsonSchema } from \"../zodCompat.js\";\n\nexport class CerebrasClient extends LLMClient {\n  public type = \"cerebras\" as const;\n  private client: OpenAI;\n  declare public clientOptions: ClientOptions;\n  public hasVision = false;\n\n  constructor({\n    modelName,\n    clientOptions,\n    userProvidedInstructions,\n  }: {\n    logger: (message: LogLine) => void;\n    modelName: AvailableModel;\n    clientOptions?: ClientOptions;\n    userProvidedInstructions?: string;\n  }) {\n    super(modelName, userProvidedInstructions);\n\n    // Create OpenAI client with the base URL set to Cerebras API\n    this.client = new OpenAI({\n      baseURL: \"https://api.cerebras.ai/v1\",\n      apiKey: clientOptions?.apiKey || process.env.CEREBRAS_API_KEY,\n      ...clientOptions,\n    });\n\n    this.modelName = modelName;\n    this.clientOptions = clientOptions;\n  }\n\n  async createChatCompletion<T = LLMResponse>({\n    options,\n    retries,\n    logger,\n  }: CreateChatCompletionOptions): Promise<T> {\n    const optionsWithoutImage = { ...options };\n    delete optionsWithoutImage.image;\n\n    logger({\n      category: \"cerebras\",\n      message: \"creating chat completion\",\n      level: 2,\n      auxiliary: {\n        options: {\n          value: JSON.stringify(optionsWithoutImage),\n          type: \"object\",\n        },\n      },\n    });\n\n    // Format messages for Cerebras API (using OpenAI format)\n    const formattedMessages = options.messages.map((msg: ChatMessage) => {\n      const baseMessage = {\n        content:\n          typeof msg.content === \"string\"\n            ? msg.content\n            : Array.isArray(msg.content) &&\n                msg.content.length > 0 &&\n                \"text\" in msg.content[0]\n              ? msg.content[0].text\n              : \"\",\n      };\n\n      // Cerebras only supports system, user, and assistant roles\n      if (msg.role === \"system\") {\n        return { ...baseMessage, role: \"system\" as const };\n      } else if (msg.role === \"assistant\") {\n        return { ...baseMessage, role: \"assistant\" as const };\n      } else {\n        // Default to user for any other role\n        return { ...baseMessage, role: \"user\" as const };\n      }\n    });\n\n    // Format tools if provided\n    let tools = options.tools?.map((tool) => ({\n      type: \"function\" as const,\n      function: {\n        name: tool.name,\n        description: tool.description,\n        parameters: {\n          type: \"object\",\n          properties: tool.parameters.properties,\n          required: tool.parameters.required,\n        },\n      },\n    }));\n\n    // Add response model as a tool if provided\n    if (options.response_model) {\n      const jsonSchema = toJsonSchema(options.response_model.schema) as {\n        properties?: Record<string, unknown>;\n        required?: string[];\n      };\n      const schemaProperties = jsonSchema.properties || {};\n      const schemaRequired = jsonSchema.required || [];\n\n      const responseTool = {\n        type: \"function\" as const,\n        function: {\n          name: \"print_extracted_data\",\n          description:\n            \"Prints the extracted data based on the provided schema.\",\n          parameters: {\n            type: \"object\",\n            properties: schemaProperties,\n            required: schemaRequired,\n          },\n        },\n      };\n\n      tools = tools ? [...tools, responseTool] : [responseTool];\n    }\n\n    try {\n      // Use OpenAI client with Cerebras API\n      const apiResponse = await this.client.chat.completions.create({\n        model: this.modelName.split(\"cerebras-\")[1],\n        messages: [\n          ...formattedMessages,\n          // Add explicit instruction to return JSON if we have a response model\n          ...(options.response_model\n            ? [\n                {\n                  role: \"system\" as const,\n                  content: `IMPORTANT: Your response must be valid JSON that matches this schema: ${JSON.stringify(\n                    options.response_model.schema,\n                  )}`,\n                },\n              ]\n            : []),\n        ],\n        temperature: options.temperature || 0.7,\n        max_tokens: options.maxOutputTokens,\n        tools: tools,\n        tool_choice: options.tool_choice || \"auto\",\n      });\n\n      // Format the response to match the expected LLMResponse format\n      const response: LLMResponse = {\n        id: apiResponse.id,\n        object: \"chat.completion\",\n        created: Date.now(),\n        model: this.modelName.split(\"cerebras-\")[1],\n        choices: [\n          {\n            index: 0,\n            message: {\n              role: \"assistant\",\n              content: apiResponse.choices[0]?.message?.content || null,\n              tool_calls: apiResponse.choices[0]?.message?.tool_calls || [],\n            },\n            finish_reason: apiResponse.choices[0]?.finish_reason || \"stop\",\n          },\n        ],\n        usage: {\n          prompt_tokens: apiResponse.usage?.prompt_tokens || 0,\n          completion_tokens: apiResponse.usage?.completion_tokens || 0,\n          total_tokens: apiResponse.usage?.total_tokens || 0,\n        },\n      };\n\n      logger({\n        category: \"cerebras\",\n        message: \"response\",\n        level: 2,\n        auxiliary: {\n          response: {\n            value: JSON.stringify(response),\n            type: \"object\",\n          },\n          requestId: {\n            value: options.requestId,\n            type: \"string\",\n          },\n        },\n      });\n\n      // If we have no response model, just return the entire LLMResponse\n      if (!options.response_model) {\n        return response as T;\n      }\n\n      // If we have a response model, parse JSON from tool calls or content\n      const toolCall = response.choices[0]?.message?.tool_calls?.[0];\n      if (toolCall?.function?.arguments) {\n        try {\n          const result = JSON.parse(toolCall.function.arguments);\n          const finalResponse = {\n            data: result,\n            usage: response.usage,\n          };\n          return finalResponse as T;\n        } catch (e) {\n          logger({\n            category: \"cerebras\",\n            message: \"failed to parse tool call arguments as JSON, retrying\",\n            level: 0,\n            auxiliary: {\n              error: {\n                value: e.message,\n                type: \"string\",\n              },\n            },\n          });\n        }\n      }\n\n      // If we have content but no tool calls, try to parse the content as JSON\n      const content = response.choices[0]?.message?.content;\n      if (content) {\n        try {\n          const jsonMatch = content.match(/\\{[\\s\\S]*\\}/);\n          if (jsonMatch) {\n            const result = JSON.parse(jsonMatch[0]);\n            const finalResponse = {\n              data: result,\n              usage: response.usage,\n            };\n            return finalResponse as T;\n          }\n        } catch (e) {\n          logger({\n            category: \"cerebras\",\n            message: \"failed to parse content as JSON\",\n            level: 0,\n            auxiliary: {\n              error: {\n                value: e.message,\n                type: \"string\",\n              },\n            },\n          });\n        }\n      }\n\n      // If we still haven't found valid JSON and have retries left, try again\n      if (!retries || retries < 5) {\n        return this.createChatCompletion({\n          options,\n          logger,\n          retries: (retries ?? 0) + 1,\n        });\n      }\n\n      throw new CreateChatCompletionResponseError(\"Invalid response schema\");\n    } catch (error) {\n      logger({\n        category: \"cerebras\",\n        message: \"error creating chat completion\",\n        level: 0,\n        auxiliary: {\n          error: {\n            value: error.message,\n            type: \"string\",\n          },\n          requestId: {\n            value: options.requestId,\n            type: \"string\",\n          },\n        },\n      });\n      throw error;\n    }\n  }\n}\n"]}