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

package/dist/cjs/lib/v3/external_clients/customOpenAI.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"customOpenAI.js","sourceRoot":"","sources":["../../../../../lib/v3/external_clients/customOpenAI.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAGH,sDAA6E;AAY7E,kDAA+C;AAC/C,6CAAmD;AACnD,+DAGsC;AAEtC,MAAa,kBAAmB,SAAQ,wBAAS;IACxC,IAAI,GAAG,QAAiB,CAAC;IACxB,MAAM,CAAS;IAEvB,YAAY,EAAE,SAAS,EAAE,MAAM,EAAyC;QACtE,KAAK,CAAC,SAA2B,CAAC,CAAC;QACnC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,SAAS,GAAG,SAA2B,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAqB,EAC7C,OAAO,EACP,OAAO,GAAG,CAAC,EACX,MAAM,GACsB;QAC5B,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,+BAA+B,EAAE,GAAG,OAAO,CAAC;QAEzE,iCAAiC;QACjC,IAAI,KAAK,EAAE,CAAC;YACV,OAAO,CAAC,IAAI,CACV,8DAA8D,CAC/D,CAAC;QACJ,CAAC;QAED,MAAM,CAAC;YACL,QAAQ,EAAE,QAAQ;YAClB,OAAO,EAAE,0BAA0B;YACnC,KAAK,EAAE,CAAC;YACR,SAAS,EAAE;gBACT,OAAO,EAAE;oBACP,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC;wBACpB,GAAG,+BAA+B;wBAClC,SAAS;qBACV,CAAC;oBACF,IAAI,EAAE,QAAQ;iBACf;gBACD,SAAS,EAAE;oBACT,KAAK,EAAE,IAAI,CAAC,SAAS;oBACrB,IAAI,EAAE,QAAQ;iBACf;aACF;SACF,CAAC,CAAC;QAEH,IAAI,cAES,CAAC;QACd,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;YAC3B,cAAc,GAAG;gBACf,IAAI,EAAE,aAAa;aACpB,CAAC;QACJ,CAAC;QAED,oBAAoB;QACpB,6BAA6B;QAC7B,MAAM,EAAE,cAAc,EAAE,GAAG,aAAa,EAAE,GAAG;YAC3C,GAAG,+BAA+B;YAClC,KAAK,EAAE,IAAI,CAAC,SAAS;SACtB,CAAC;QAEF,MAAM,CAAC;YACL,QAAQ,EAAE,QAAQ;YAClB,OAAO,EAAE,0BAA0B;YACnC,KAAK,EAAE,CAAC;YACR,SAAS,EAAE;gBACT,aAAa,EAAE;oBACb,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC;oBACpC,IAAI,EAAE,QAAQ;iBACf;aACF;SACF,CAAC,CAAC;QAEH,MAAM,iBAAiB,GACrB,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;YAC/B,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;gBACnC,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;oBACnD,IAAI,WAAW,IAAI,OAAO,EAAE,CAAC;wBAC3B,MAAM,YAAY,GAAmC;4BACnD,SAAS,EAAE;gCACT,GAAG,EAAE,OAAO,CAAC,SAAS,CAAC,GAAG;6BAC3B;4BACD,IAAI,EAAE,WAAW;yBAClB,CAAC;wBACF,OAAO,YAAY,CAAC;oBACtB,CAAC;yBAAM,CAAC;wBACN,MAAM,WAAW,GAAkC;4BACjD,IAAI,EAAE,OAAO,CAAC,IAAI;4BAClB,IAAI,EAAE,MAAM;yBACb,CAAC;wBACF,OAAO,WAAW,CAAC;oBACrB,CAAC;gBACH,CAAC,CAAC,CAAC;gBAEH,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;oBAC9B,MAAM,gBAAgB,GAAqC;wBACzD,GAAG,OAAO;wBACV,IAAI,EAAE,QAAQ;wBACd,OAAO,EAAE,YAAY,CAAC,MAAM,CAC1B,CAAC,OAAO,EAA4C,EAAE,CACpD,OAAO,CAAC,IAAI,KAAK,MAAM,CAC1B;qBACF,CAAC;oBACF,OAAO,gBAAgB,CAAC;gBAC1B,CAAC;qBAAM,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBACnC,MAAM,gBAAgB,GAAmC;wBACvD,GAAG,OAAO;wBACV,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,YAAY;qBACtB,CAAC;oBACF,OAAO,gBAAgB,CAAC;gBAC1B,CAAC;qBAAM,CAAC;oBACN,MAAM,gBAAgB,GAAwC;wBAC5D,GAAG,OAAO;wBACV,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,YAAY,CAAC,MAAM,CAC1B,CAAC,OAAO,EAA4C,EAAE,CACpD,OAAO,CAAC,IAAI,KAAK,MAAM,CAC1B;qBACF,CAAC;oBACF,OAAO,gBAAgB,CAAC;gBAC1B,CAAC;YACH,CAAC;YAED,OAAO;gBACL,GAAG,OAAO;gBACV,OAAO,EAAE,OAAO,CAAC,OAAO;aACK,CAAC;QAClC,CAAC,CAAC,CAAC;QAEL,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;YAC3B,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAC/B,IAAA,2BAAY,EAAC,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,EAC3C,IAAI,EACJ,CAAC,CACF,CAAC;YACF,iBAAiB,CAAC,IAAI,CAAC;gBACrB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,kDAAkD,UAAU,8JAA8J;aACpO,CAAC,CAAC;QACL,CAAC;QAED,MAAM,IAAI,GAA2C;YACnD,GAAG,aAAa;YAChB,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,QAAQ,EAAE,iBAAiB;YAC3B,eAAe,EAAE,cAAc;YAC/B,MAAM,EAAE,KAAK;YACb,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACnC,QAAQ,EAAE;oBACR,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,WAAW,EAAE,IAAI,CAAC,WAAW;oBAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;iBAC5B;gBACD,IAAI,EAAE,UAAU;aACjB,CAAC,CAAC;SACJ,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAEjE,MAAM,CAAC;YACL,QAAQ,EAAE,QAAQ;YAClB,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,SAAS;oBAChB,IAAI,EAAE,QAAQ;iBACf;aACF;SACF,CAAC,CAAC;QAEH,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;YAC3B,MAAM,aAAa,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;YAC1D,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,MAAM,IAAI,gDAAiC,CAAC,wBAAwB,CAAC,CAAC;YACxE,CAAC;YAED,IAAI,UAAmB,CAAC;YACxB,IAAI,CAAC;gBACH,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;gBACvC,IAAA,4BAAiB,EAAC,OAAO,CAAC,cAAc,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;YAC/D,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,MAAM,YAAY,GAAG,CAAC,YAAY,WAAW,CAAC;gBAC9C,MAAM,CAAC;oBACL,QAAQ,EAAE,QAAQ;oBAClB,OAAO,EAAE,YAAY;wBACnB,CAAC,CAAC,4BAA4B;wBAC9B,CAAC,CAAC,uCAAuC;oBAC3C,KAAK,EAAE,CAAC;iBACT,CAAC,CAAC;gBACH,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;oBAChB,OAAO,IAAI,CAAC,oBAAoB,CAAC;wBAC/B,OAAO;wBACP,MAAM;wBACN,OAAO,EAAE,OAAO,GAAG,CAAC;qBACrB,CAAC,CAAC;gBACL,CAAC;gBAED,IAAI,CAAC,YAAY,uCAAwB,EAAE,CAAC;oBAC1C,MAAM,CAAC;wBACL,QAAQ,EAAE,QAAQ;wBAClB,OAAO,EAAE,iCAAiC,CAAC,CAAC,OAAO,EAAE;wBACrD,KAAK,EAAE,CAAC;wBACR,SAAS,EAAE;4BACT,YAAY,EAAE;gCACZ,KAAK,EAAE,YAAY,CAAC,CAAC,OAAO,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;gCACrE,IAAI,EAAE,QAAQ;6BACf;4BACD,SAAS,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE;yBAChD;qBACF,CAAC,CAAC;oBACH,MAAM,IAAI,gDAAiC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBACzD,CAAC;gBACD,MAAM,IAAI,gDAAiC,CACzC,YAAY;oBACV,CAAC,CAAC,wCAAwC;oBAC1C,CAAC,CAAC,CAAC,YAAY,KAAK;wBAClB,CAAC,CAAC,CAAC,CAAC,OAAO;wBACX,CAAC,CAAC,0CAA0C,CACjD,CAAC;YACJ,CAAC;YAED,OAAO;gBACL,IAAI,EAAE,UAAU;gBAChB,KAAK,EAAE;oBACL,aAAa,EAAE,QAAQ,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;oBACjD,iBAAiB,EAAE,QAAQ,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC;oBACzD,YAAY,EAAE,QAAQ,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC;iBAChD;aACG,CAAC;QACT,CAAC;QAED,OAAO;YACL,IAAI,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO;YACzC,KAAK,EAAE;gBACL,aAAa,EAAE,QAAQ,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;gBACjD,iBAAiB,EAAE,QAAQ,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC;gBACzD,YAAY,EAAE,QAAQ,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC;aAChD;SACG,CAAC;IACT,CAAC;CACF;AApPD,gDAoPC","sourcesContent":["/**\n * Welcome to the Stagehand custom OpenAI client!\n *\n * This is a client for models that are compatible with the OpenAI API, like Ollama, Gemini, etc.\n * You can just pass in an OpenAI instance to the client and it will work.\n */\n\nimport type { AvailableModel } from \"../types/public/model.js\";\nimport { CreateChatCompletionOptions, LLMClient } from \"../llm/LLMClient.js\";\nimport OpenAI from \"openai\";\nimport type {\n  ChatCompletion,\n  ChatCompletionAssistantMessageParam,\n  ChatCompletionContentPartImage,\n  ChatCompletionContentPartText,\n  ChatCompletionCreateParamsNonStreaming,\n  ChatCompletionMessageParam,\n  ChatCompletionSystemMessageParam,\n  ChatCompletionUserMessageParam,\n} from \"openai/resources/chat/completions\";\nimport { toJsonSchema } from \"../zodCompat.js\";\nimport { validateZodSchema } from \"../../utils.js\";\nimport {\n  CreateChatCompletionResponseError,\n  ZodSchemaValidationError,\n} from \"../types/public/sdkErrors.js\";\n\nexport class CustomOpenAIClient extends LLMClient {\n  public type = \"openai\" as const;\n  private client: OpenAI;\n\n  constructor({ modelName, client }: { modelName: string; client: OpenAI }) {\n    super(modelName as AvailableModel);\n    this.client = client;\n    this.modelName = modelName as AvailableModel;\n  }\n\n  async createChatCompletion<T = ChatCompletion>({\n    options,\n    retries = 3,\n    logger,\n  }: CreateChatCompletionOptions): Promise<T> {\n    const { image, requestId, ...optionsWithoutImageAndRequestId } = options;\n\n    // TODO: Implement vision support\n    if (image) {\n      console.warn(\n        \"Image provided. Vision is not currently supported for openai\",\n      );\n    }\n\n    logger({\n      category: \"openai\",\n      message: \"creating chat completion\",\n      level: 1,\n      auxiliary: {\n        options: {\n          value: JSON.stringify({\n            ...optionsWithoutImageAndRequestId,\n            requestId,\n          }),\n          type: \"object\",\n        },\n        modelName: {\n          value: this.modelName,\n          type: \"string\",\n        },\n      },\n    });\n\n    let responseFormat:\n      | ChatCompletionCreateParamsNonStreaming[\"response_format\"]\n      | undefined;\n    if (options.response_model) {\n      responseFormat = {\n        type: \"json_object\",\n      };\n    }\n\n    /* eslint-disable */\n    // Remove unsupported options\n    const { response_model, ...openaiOptions } = {\n      ...optionsWithoutImageAndRequestId,\n      model: this.modelName,\n    };\n\n    logger({\n      category: \"openai\",\n      message: \"creating chat completion\",\n      level: 1,\n      auxiliary: {\n        openaiOptions: {\n          value: JSON.stringify(openaiOptions),\n          type: \"object\",\n        },\n      },\n    });\n\n    const formattedMessages: ChatCompletionMessageParam[] =\n      options.messages.map((message) => {\n        if (Array.isArray(message.content)) {\n          const contentParts = message.content.map((content) => {\n            if (\"image_url\" in content) {\n              const imageContent: ChatCompletionContentPartImage = {\n                image_url: {\n                  url: content.image_url.url,\n                },\n                type: \"image_url\",\n              };\n              return imageContent;\n            } else {\n              const textContent: ChatCompletionContentPartText = {\n                text: content.text,\n                type: \"text\",\n              };\n              return textContent;\n            }\n          });\n\n          if (message.role === \"system\") {\n            const formattedMessage: ChatCompletionSystemMessageParam = {\n              ...message,\n              role: \"system\",\n              content: contentParts.filter(\n                (content): content is ChatCompletionContentPartText =>\n                  content.type === \"text\",\n              ),\n            };\n            return formattedMessage;\n          } else if (message.role === \"user\") {\n            const formattedMessage: ChatCompletionUserMessageParam = {\n              ...message,\n              role: \"user\",\n              content: contentParts,\n            };\n            return formattedMessage;\n          } else {\n            const formattedMessage: ChatCompletionAssistantMessageParam = {\n              ...message,\n              role: \"assistant\",\n              content: contentParts.filter(\n                (content): content is ChatCompletionContentPartText =>\n                  content.type === \"text\",\n              ),\n            };\n            return formattedMessage;\n          }\n        }\n\n        return {\n          ...message,\n          content: message.content,\n        } as ChatCompletionMessageParam;\n      });\n\n    if (options.response_model) {\n      const schemaJson = JSON.stringify(\n        toJsonSchema(options.response_model.schema),\n        null,\n        2,\n      );\n      formattedMessages.push({\n        role: \"user\",\n        content: `Respond with valid JSON matching this schema:\\n${schemaJson}\\n\\nDo not include any other text, formatting or markdown in your output. Do not include \\`\\`\\` or \\`\\`\\`json in your response. Only the JSON object itself.`,\n      });\n    }\n\n    const body: ChatCompletionCreateParamsNonStreaming = {\n      ...openaiOptions,\n      model: this.modelName,\n      messages: formattedMessages,\n      response_format: responseFormat,\n      stream: false,\n      tools: options.tools?.map((tool) => ({\n        function: {\n          name: tool.name,\n          description: tool.description,\n          parameters: tool.parameters,\n        },\n        type: \"function\",\n      })),\n    };\n\n    const response = await this.client.chat.completions.create(body);\n\n    logger({\n      category: \"openai\",\n      message: \"response\",\n      level: 1,\n      auxiliary: {\n        response: {\n          value: JSON.stringify(response),\n          type: \"object\",\n        },\n        requestId: {\n          value: requestId,\n          type: \"string\",\n        },\n      },\n    });\n\n    if (options.response_model) {\n      const extractedData = response.choices[0].message.content;\n      if (!extractedData) {\n        throw new CreateChatCompletionResponseError(\"No content in response\");\n      }\n\n      let parsedData: unknown;\n      try {\n        parsedData = JSON.parse(extractedData);\n        validateZodSchema(options.response_model.schema, parsedData);\n      } catch (e) {\n        const isParseError = e instanceof SyntaxError;\n        logger({\n          category: \"openai\",\n          message: isParseError\n            ? \"Response is not valid JSON\"\n            : \"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\n        if (e instanceof ZodSchemaValidationError) {\n          logger({\n            category: \"openai\",\n            message: `Error during chat completion: ${e.message}`,\n            level: 0,\n            auxiliary: {\n              errorDetails: {\n                value: `Message: ${e.message}${e.stack ? \"\\nStack: \" + e.stack : \"\"}`,\n                type: \"string\",\n              },\n              requestId: { value: requestId, type: \"string\" },\n            },\n          });\n          throw new CreateChatCompletionResponseError(e.message);\n        }\n        throw new CreateChatCompletionResponseError(\n          isParseError\n            ? \"Failed to parse model response as JSON\"\n            : e instanceof Error\n              ? e.message\n              : \"Unknown error during response processing\",\n        );\n      }\n\n      return {\n        data: parsedData,\n        usage: {\n          prompt_tokens: response.usage?.prompt_tokens ?? 0,\n          completion_tokens: response.usage?.completion_tokens ?? 0,\n          total_tokens: response.usage?.total_tokens ?? 0,\n        },\n      } as T;\n    }\n\n    return {\n      data: response.choices[0].message.content,\n      usage: {\n        prompt_tokens: response.usage?.prompt_tokens ?? 0,\n        completion_tokens: response.usage?.completion_tokens ?? 0,\n        total_tokens: response.usage?.total_tokens ?? 0,\n      },\n    } as T;\n  }\n}\n"]}