@browserbasehq/orca 3.2.1-preview.2 → 3.4.0-preview-1

package/dist/esm/lib/v3/handlers/extractHandler.js

@@ -41,7 +41,7 @@ export class ExtractHandler {
         this.onMetrics = onMetrics;
     }
     async extract(params) {
-        const { instruction, schema, page, selector, timeout, model } = params;
+        const { instruction, schema, page, selector, ignoreSelectors, timeout, model, } = params;
         const llmClient = this.resolveLlmClient(model);
         const ensureTimeRemaining = createTimeoutGuard(timeout, (ms) => new ExtractTimeoutError(ms));
         // No-args → page text (parity with v2)
@@ -52,6 +52,7 @@ export class ExtractHandler {
             const snap = await captureHybridSnapshot(page, {
                 experimental: this.experimental,
                 focusSelector: focusSelector || undefined,
+                ignoreSelectors,
             });
             ensureTimeRemaining();
             const result = { pageText: snap.combinedTree };
@@ -67,6 +68,7 @@ export class ExtractHandler {
         const { combinedTree, combinedUrlMap } = await captureHybridSnapshot(page, {
             experimental: this.experimental,
             focusSelector: focusSelector,
+            ignoreSelectors,
         });
         v3Logger({
             category: "extraction",

package/dist/esm/lib/v3/handlers/extractHandler.js.map

@@ -1 +1 @@
-{"version":3,"file":"extractHandler.js","sourceRoot":"","sources":["../../../../../lib/v3/handlers/extractHandler.ts"],"names":[],"mappings":"AAAA,oCAAoC;AACpC,OAAO,EAAE,OAAO,IAAI,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EACL,WAAW,EACX,UAAU,EACV,UAAU,EACV,eAAe,GAChB,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,qBAAqB,EAAE,MAAM,sCAAsC,CAAC;AAK7E,OAAO,EACL,oBAAoB,EACpB,cAAc,GACf,MAAM,4BAA4B,CAAC;AAMpC,OAAO,EACL,6BAA6B,EAC7B,mBAAmB,GACpB,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAOpE;;;;;;;;;GASG;AACH,MAAM,UAAU,+BAA+B,CAC7C,MAAS;IAET,MAAM,CAAC,WAAW,EAAE,QAAQ,CAAC,GAAG,eAAe,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IAC5D,OAAO,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;AACjC,CAAC;AAcD,MAAM,OAAO,cAAc;IACR,SAAS,CAAY;IACrB,gBAAgB,CAAiB;IACjC,oBAAoB,CAAgB;IACpC,gBAAgB,CAA4C;IAC5D,YAAY,CAAS;IACrB,kBAAkB,CAAU;IAC5B,YAAY,CAAU;IACtB,SAAS,CAOhB;IAEV,YACE,SAAoB,EACpB,gBAAgC,EAChC,oBAAmC,EACnC,gBAA2D,EAC3D,YAAqB,EACrB,kBAA4B,EAC5B,YAAsB,EACtB,SAOS;QAET,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,oBAAoB,GAAG,oBAAoB,CAAC;QACjD,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,YAAY,GAAG,YAAY,IAAI,EAAE,CAAC;QACvC,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,IAAI,KAAK,CAAC;QACtD,IAAI,CAAC,YAAY,GAAG,YAAY,IAAI,KAAK,CAAC;QAC1C,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,OAAO,CACX,MAA+B;QAE/B,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;QAEvE,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;QAE/C,MAAM,mBAAmB,GAAG,kBAAkB,CAC5C,OAAO,EACP,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,mBAAmB,CAAC,EAAE,CAAC,CACpC,CAAC;QAEF,uCAAuC;QACvC,MAAM,MAAM,GAAG,CAAC,WAAW,IAAI,CAAC,MAAM,CAAC;QACvC,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,aAAa,GAAG,QAAQ,EAAE,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;YAC9D,mBAAmB,EAAE,CAAC;YACtB,MAAM,IAAI,GAAG,MAAM,qBAAqB,CAAC,IAAI,EAAE;gBAC7C,YAAY,EAAE,IAAI,CAAC,YAAY;gBAC/B,aAAa,EAAE,aAAa,IAAI,SAAS;aAC1C,CAAC,CAAC;YACH,mBAAmB,EAAE,CAAC;YAEtB,MAAM,MAAM,GAAG,EAAE,QAAQ,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC;YAC/C,0CAA0C;YAC1C,OAAO,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACtC,CAAC;QAED,IAAI,CAAC,WAAW,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,IAAI,6BAA6B,CACrC,8DAA8D,CAC/D,CAAC;QACJ,CAAC;QAED,MAAM,aAAa,GAAG,QAAQ,EAAE,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;QAE7D,6EAA6E;QAC7E,mBAAmB,EAAE,CAAC;QACtB,MAAM,EAAE,YAAY,EAAE,cAAc,EAAE,GAAG,MAAM,qBAAqB,CAAC,IAAI,EAAE;YACzE,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,aAAa,EAAE,aAAa;SAC7B,CAAC,CAAC;QAEH,QAAQ,CAAC;YACP,QAAQ,EAAE,YAAY;YACtB,OAAO,EAAE,yCAAyC;YAClD,KAAK,EAAE,CAAC;YACR,SAAS,EAAE,WAAW;gBACpB,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;gBACzD,CAAC,CAAC,SAAS;SACd,CAAC,CAAC;QAEH,qFAAqF;QACrF,MAAM,UAAU,GAAuB,CAAC,MAAM;YAC5C,oBAAoB,CAAuB,CAAC;QAC9C,0EAA0E;QAC1E,MAAM,cAAc,GAAG,UAAU,CAAC,UAAU,CAAC,KAAK,QAAQ,CAAC;QAC3D,MAAM,QAAQ,GAAG,OAAgB,CAAC;QAClC,MAAM,OAAO,GAAG,WAAW,CAAC,UAAU,CAAC,CAAC;QACxC,MAAM,YAAY,GAAuB,cAAc;YACrD,CAAC,CAAE,UAAiC;YACpC,CAAC,CAAE,OAAO,CAAC,MAAM,CAAC;gBACd,CAAC,QAAQ,CAAC,EAAE,UAAwB;aACrC,CAAwB,CAAC;QAE9B,MAAM,CAAC,iBAAiB,EAAE,aAAa,CAAC,GACtC,+BAA+B,CAAC,YAAY,CAAC,CAAC;QAEhD,mBAAmB,EAAE,CAAC;QACtB,MAAM,kBAAkB,GACtB,MAAM,UAAU,CAAqB;YACnC,WAAW;YACX,WAAW,EAAE,YAAY;YACzB,MAAM,EAAE,iBAAuC;YAC/C,SAAS;YACT,wBAAwB,EAAE,IAAI,CAAC,YAAY;YAC3C,MAAM,EAAE,QAAQ;YAChB,kBAAkB,EAAE,IAAI,CAAC,kBAAkB;SAC5C,CAAC,CAAC;QAEL,MAAM,EACJ,QAAQ,EAAE,EAAE,SAAS,EAAE,EACvB,aAAa,EACb,iBAAiB,EACjB,gBAAgB,GAAG,CAAC,EACpB,mBAAmB,GAAG,CAAC,EACvB,iBAAiB,EACjB,GAAG,IAAI,EACR,GAAG,kBAAkB,CAAC;QACvB,IAAI,MAAM,GAAG,IAAgD,CAAC;QAE9D,4CAA4C;QAC5C,IAAI,CAAC,SAAS,EAAE,CACd,cAAc,CAAC,OAAO,EACtB,aAAa,EACb,iBAAiB,EACjB,gBAAgB,EAChB,mBAAmB,EACnB,iBAAiB,CAClB,CAAC;QAEF,2EAA2E;QAC3E,MAAM,OAAO,GAA8B,CAAC,cAAc,IAAI,EAAE,CAG/D,CAAC;QACF,KAAK,MAAM,EAAE,QAAQ,EAAE,IAAI,aAAa,EAAE,CAAC;YACzC,UAAU,CACR,MAAiC,EACjC,QAAQ,EACR,OAA4C,CAC7C,CAAC;QACJ,CAAC;QACD,sDAAsD;QACtD,IAAI,CAAC,cAAc,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC5D,MAAM,GAAI,MAAkC,CAAC,QAAQ,CAAC,CAAC;QACzD,CAAC;QAED,MAAM,mBAAmB,GAAG,GAAG,CAAC;QAChC,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,WAAW,CAAC;QAC3D,MAAM,aAAa,GACjB,YAAY,CAAC,MAAM,GAAG,mBAAmB;YACvC,CAAC,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,mBAAmB,CAAC,GAAG,KAAK;YACpD,CAAC,CAAC,YAAY,CAAC;QAEnB,QAAQ,CAAC;YACP,QAAQ,EAAE,YAAY;YACtB,OAAO,EAAE,SAAS;gBAChB,CAAC,CAAC,mCAAmC;gBACrC,CAAC,CAAC,iDAAiD;YACrD,KAAK,EAAE,CAAC;YACR,SAAS,EAAE;gBACT,aAAa,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC/D,iBAAiB,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACvE,iBAAiB,EAAE;oBACjB,KAAK,EAAE,MAAM,CAAC,iBAAiB,CAAC;oBAChC,IAAI,EAAE,QAAQ;iBACf;gBACD,MAAM,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE,QAAQ,EAAE;aACjD;SACF,CAAC,CAAC;QAEH,OAAO,MAAiC,CAAC;IAC3C,CAAC;CACF","sourcesContent":["// lib/v3/handlers/extractHandler.ts\nimport { extract as runExtract } from \"../../inference.js\";\nimport {\n  getZFactory,\n  getZodType,\n  injectUrls,\n  transformSchema,\n} from \"../../utils.js\";\nimport { v3Logger } from \"../logger.js\";\nimport { V3FunctionName } from \"../types/public/methods.js\";\nimport { captureHybridSnapshot } from \"../understudy/a11y/snapshot/index.js\";\nimport type { ZodTypeAny } from \"zod\";\nimport { LLMClient } from \"../llm/LLMClient.js\";\nimport { ExtractHandlerParams } from \"../types/private/handlers.js\";\nimport { EncodedId, ZodPathSegments } from \"../types/private/internal.js\";\nimport {\n  defaultExtractSchema,\n  pageTextSchema,\n} from \"../types/public/methods.js\";\nimport {\n  AvailableModel,\n  ClientOptions,\n  ModelConfiguration,\n} from \"../types/public/model.js\";\nimport {\n  StagehandInvalidArgumentError,\n  ExtractTimeoutError,\n} from \"../types/public/sdkErrors.js\";\nimport { createTimeoutGuard } from \"./handlerUtils/timeoutGuard.js\";\nimport type {\n  InferStagehandSchema,\n  StagehandZodObject,\n  StagehandZodSchema,\n} from \"../zodCompat.js\";\n\n/**\n * Scans the provided Zod schema for any `z.string().url()` fields and\n * replaces them with `z.number()`.\n *\n * @param schema - The Zod object schema to transform.\n * @returns A tuple containing:\n *   1. The transformed schema (or the original schema if no changes were needed).\n *   2. An array of {@link ZodPathSegments} objects representing all the replaced URL fields,\n *      with each path segment showing where in the schema the replacement occurred.\n */\nexport function transformUrlStringsToNumericIds<T extends StagehandZodSchema>(\n  schema: T,\n): [StagehandZodSchema, ZodPathSegments[]] {\n  const [finalSchema, urlPaths] = transformSchema(schema, []);\n  return [finalSchema, urlPaths];\n}\n\ninterface ExtractionResponseBase {\n  metadata: { completed: boolean };\n  prompt_tokens: number;\n  completion_tokens: number;\n  reasoning_tokens: number;\n  cached_input_tokens?: number;\n  inference_time_ms: number;\n}\n\ntype ExtractionResponse<T extends StagehandZodObject> = ExtractionResponseBase &\n  InferStagehandSchema<T>;\n\nexport class ExtractHandler {\n  private readonly llmClient: LLMClient;\n  private readonly defaultModelName: AvailableModel;\n  private readonly defaultClientOptions: ClientOptions;\n  private readonly resolveLlmClient: (model?: ModelConfiguration) => LLMClient;\n  private readonly systemPrompt: string;\n  private readonly logInferenceToFile: boolean;\n  private readonly experimental: boolean;\n  private readonly onMetrics?: (\n    functionName: V3FunctionName,\n    promptTokens: number,\n    completionTokens: number,\n    reasoningTokens: number,\n    cachedInputTokens: number,\n    inferenceTimeMs: number,\n  ) => void;\n\n  constructor(\n    llmClient: LLMClient,\n    defaultModelName: AvailableModel,\n    defaultClientOptions: ClientOptions,\n    resolveLlmClient: (model?: ModelConfiguration) => LLMClient,\n    systemPrompt?: string,\n    logInferenceToFile?: boolean,\n    experimental?: boolean,\n    onMetrics?: (\n      functionName: V3FunctionName,\n      promptTokens: number,\n      completionTokens: number,\n      reasoningTokens: number,\n      cachedInputTokens: number,\n      inferenceTimeMs: number,\n    ) => void,\n  ) {\n    this.llmClient = llmClient;\n    this.defaultModelName = defaultModelName;\n    this.defaultClientOptions = defaultClientOptions;\n    this.resolveLlmClient = resolveLlmClient;\n    this.systemPrompt = systemPrompt ?? \"\";\n    this.logInferenceToFile = logInferenceToFile ?? false;\n    this.experimental = experimental ?? false;\n    this.onMetrics = onMetrics;\n  }\n\n  async extract<T extends StagehandZodSchema>(\n    params: ExtractHandlerParams<T>,\n  ): Promise<InferStagehandSchema<T> | { pageText: string }> {\n    const { instruction, schema, page, selector, timeout, model } = params;\n\n    const llmClient = this.resolveLlmClient(model);\n\n    const ensureTimeRemaining = createTimeoutGuard(\n      timeout,\n      (ms) => new ExtractTimeoutError(ms),\n    );\n\n    // No-args → page text (parity with v2)\n    const noArgs = !instruction && !schema;\n    if (noArgs) {\n      const focusSelector = selector?.replace(/^xpath=/i, \"\") ?? \"\";\n      ensureTimeRemaining();\n      const snap = await captureHybridSnapshot(page, {\n        experimental: this.experimental,\n        focusSelector: focusSelector || undefined,\n      });\n      ensureTimeRemaining();\n\n      const result = { pageText: snap.combinedTree };\n      // Validate via the same schema used in v2\n      return pageTextSchema.parse(result);\n    }\n\n    if (!instruction && schema) {\n      throw new StagehandInvalidArgumentError(\n        \"extract() requires an instruction when a schema is provided.\",\n      );\n    }\n\n    const focusSelector = selector?.replace(/^xpath=/, \"\") ?? \"\";\n\n    // Build the hybrid snapshot (includes combinedTree; combinedUrlMap optional)\n    ensureTimeRemaining();\n    const { combinedTree, combinedUrlMap } = await captureHybridSnapshot(page, {\n      experimental: this.experimental,\n      focusSelector: focusSelector,\n    });\n\n    v3Logger({\n      category: \"extraction\",\n      message: \"Starting extraction using a11y snapshot\",\n      level: 1,\n      auxiliary: instruction\n        ? { instruction: { value: instruction, type: \"string\" } }\n        : undefined,\n    });\n\n    // Normalize schema: if instruction provided without schema, use defaultExtractSchema\n    const baseSchema: StagehandZodSchema = (schema ??\n      defaultExtractSchema) as StagehandZodSchema;\n    // Ensure we pass an object schema into inference; wrap non-object schemas\n    const isObjectSchema = getZodType(baseSchema) === \"object\";\n    const WRAP_KEY = \"value\" as const;\n    const factory = getZFactory(baseSchema);\n    const objectSchema: StagehandZodObject = isObjectSchema\n      ? (baseSchema as StagehandZodObject)\n      : (factory.object({\n          [WRAP_KEY]: baseSchema as ZodTypeAny,\n        }) as StagehandZodObject);\n\n    const [transformedSchema, urlFieldPaths] =\n      transformUrlStringsToNumericIds(objectSchema);\n\n    ensureTimeRemaining();\n    const extractionResponse: ExtractionResponse<StagehandZodObject> =\n      await runExtract<StagehandZodObject>({\n        instruction,\n        domElements: combinedTree,\n        schema: transformedSchema as StagehandZodObject,\n        llmClient,\n        userProvidedInstructions: this.systemPrompt,\n        logger: v3Logger,\n        logInferenceToFile: this.logInferenceToFile,\n      });\n\n    const {\n      metadata: { completed },\n      prompt_tokens,\n      completion_tokens,\n      reasoning_tokens = 0,\n      cached_input_tokens = 0,\n      inference_time_ms,\n      ...rest\n    } = extractionResponse;\n    let output = rest as InferStagehandSchema<StagehandZodObject>;\n\n    // Update EXTRACT metrics from the LLM calls\n    this.onMetrics?.(\n      V3FunctionName.EXTRACT,\n      prompt_tokens,\n      completion_tokens,\n      reasoning_tokens,\n      cached_input_tokens,\n      inference_time_ms,\n    );\n\n    // Re-inject URLs for any url() fields we temporarily converted to number()\n    const idToUrl: Record<EncodedId, string> = (combinedUrlMap ?? {}) as Record<\n      EncodedId,\n      string\n    >;\n    for (const { segments } of urlFieldPaths) {\n      injectUrls(\n        output as Record<string, unknown>,\n        segments,\n        idToUrl as unknown as Record<string, string>,\n      );\n    }\n    // If we wrapped a non-object schema, unwrap the value\n    if (!isObjectSchema && output && typeof output === \"object\") {\n      output = (output as Record<string, unknown>)[WRAP_KEY];\n    }\n\n    const resultPreviewLength = 200;\n    const resultString = JSON.stringify(output) ?? \"undefined\";\n    const resultPreview =\n      resultString.length > resultPreviewLength\n        ? resultString.slice(0, resultPreviewLength) + \"...\"\n        : resultString;\n\n    v3Logger({\n      category: \"extraction\",\n      message: completed\n        ? \"Extraction completed successfully\"\n        : \"Extraction incomplete after processing all data\",\n      level: 1,\n      auxiliary: {\n        prompt_tokens: { value: String(prompt_tokens), type: \"string\" },\n        completion_tokens: { value: String(completion_tokens), type: \"string\" },\n        inference_time_ms: {\n          value: String(inference_time_ms),\n          type: \"string\",\n        },\n        result: { value: resultPreview, type: \"string\" },\n      },\n    });\n\n    return output as InferStagehandSchema<T>;\n  }\n}\n"]}
+{"version":3,"file":"extractHandler.js","sourceRoot":"","sources":["../../../../../lib/v3/handlers/extractHandler.ts"],"names":[],"mappings":"AAAA,oCAAoC;AACpC,OAAO,EAAE,OAAO,IAAI,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EACL,WAAW,EACX,UAAU,EACV,UAAU,EACV,eAAe,GAChB,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,qBAAqB,EAAE,MAAM,sCAAsC,CAAC;AAK7E,OAAO,EACL,oBAAoB,EACpB,cAAc,GACf,MAAM,4BAA4B,CAAC;AAMpC,OAAO,EACL,6BAA6B,EAC7B,mBAAmB,GACpB,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAOpE;;;;;;;;;GASG;AACH,MAAM,UAAU,+BAA+B,CAC7C,MAAS;IAET,MAAM,CAAC,WAAW,EAAE,QAAQ,CAAC,GAAG,eAAe,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IAC5D,OAAO,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;AACjC,CAAC;AAcD,MAAM,OAAO,cAAc;IACR,SAAS,CAAY;IACrB,gBAAgB,CAAiB;IACjC,oBAAoB,CAAgB;IACpC,gBAAgB,CAA4C;IAC5D,YAAY,CAAS;IACrB,kBAAkB,CAAU;IAC5B,YAAY,CAAU;IACtB,SAAS,CAOhB;IAEV,YACE,SAAoB,EACpB,gBAAgC,EAChC,oBAAmC,EACnC,gBAA2D,EAC3D,YAAqB,EACrB,kBAA4B,EAC5B,YAAsB,EACtB,SAOS;QAET,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,oBAAoB,GAAG,oBAAoB,CAAC;QACjD,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,YAAY,GAAG,YAAY,IAAI,EAAE,CAAC;QACvC,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,IAAI,KAAK,CAAC;QACtD,IAAI,CAAC,YAAY,GAAG,YAAY,IAAI,KAAK,CAAC;QAC1C,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,OAAO,CACX,MAA+B;QAE/B,MAAM,EACJ,WAAW,EACX,MAAM,EACN,IAAI,EACJ,QAAQ,EACR,eAAe,EACf,OAAO,EACP,KAAK,GACN,GAAG,MAAM,CAAC;QAEX,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;QAE/C,MAAM,mBAAmB,GAAG,kBAAkB,CAC5C,OAAO,EACP,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,mBAAmB,CAAC,EAAE,CAAC,CACpC,CAAC;QAEF,uCAAuC;QACvC,MAAM,MAAM,GAAG,CAAC,WAAW,IAAI,CAAC,MAAM,CAAC;QACvC,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,aAAa,GAAG,QAAQ,EAAE,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;YAC9D,mBAAmB,EAAE,CAAC;YACtB,MAAM,IAAI,GAAG,MAAM,qBAAqB,CAAC,IAAI,EAAE;gBAC7C,YAAY,EAAE,IAAI,CAAC,YAAY;gBAC/B,aAAa,EAAE,aAAa,IAAI,SAAS;gBACzC,eAAe;aAChB,CAAC,CAAC;YACH,mBAAmB,EAAE,CAAC;YAEtB,MAAM,MAAM,GAAG,EAAE,QAAQ,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC;YAC/C,0CAA0C;YAC1C,OAAO,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACtC,CAAC;QAED,IAAI,CAAC,WAAW,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,IAAI,6BAA6B,CACrC,8DAA8D,CAC/D,CAAC;QACJ,CAAC;QAED,MAAM,aAAa,GAAG,QAAQ,EAAE,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;QAE7D,6EAA6E;QAC7E,mBAAmB,EAAE,CAAC;QACtB,MAAM,EAAE,YAAY,EAAE,cAAc,EAAE,GAAG,MAAM,qBAAqB,CAAC,IAAI,EAAE;YACzE,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,aAAa,EAAE,aAAa;YAC5B,eAAe;SAChB,CAAC,CAAC;QAEH,QAAQ,CAAC;YACP,QAAQ,EAAE,YAAY;YACtB,OAAO,EAAE,yCAAyC;YAClD,KAAK,EAAE,CAAC;YACR,SAAS,EAAE,WAAW;gBACpB,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;gBACzD,CAAC,CAAC,SAAS;SACd,CAAC,CAAC;QAEH,qFAAqF;QACrF,MAAM,UAAU,GAAuB,CAAC,MAAM;YAC5C,oBAAoB,CAAuB,CAAC;QAC9C,0EAA0E;QAC1E,MAAM,cAAc,GAAG,UAAU,CAAC,UAAU,CAAC,KAAK,QAAQ,CAAC;QAC3D,MAAM,QAAQ,GAAG,OAAgB,CAAC;QAClC,MAAM,OAAO,GAAG,WAAW,CAAC,UAAU,CAAC,CAAC;QACxC,MAAM,YAAY,GAAuB,cAAc;YACrD,CAAC,CAAE,UAAiC;YACpC,CAAC,CAAE,OAAO,CAAC,MAAM,CAAC;gBACd,CAAC,QAAQ,CAAC,EAAE,UAAwB;aACrC,CAAwB,CAAC;QAE9B,MAAM,CAAC,iBAAiB,EAAE,aAAa,CAAC,GACtC,+BAA+B,CAAC,YAAY,CAAC,CAAC;QAEhD,mBAAmB,EAAE,CAAC;QACtB,MAAM,kBAAkB,GACtB,MAAM,UAAU,CAAqB;YACnC,WAAW;YACX,WAAW,EAAE,YAAY;YACzB,MAAM,EAAE,iBAAuC;YAC/C,SAAS;YACT,wBAAwB,EAAE,IAAI,CAAC,YAAY;YAC3C,MAAM,EAAE,QAAQ;YAChB,kBAAkB,EAAE,IAAI,CAAC,kBAAkB;SAC5C,CAAC,CAAC;QAEL,MAAM,EACJ,QAAQ,EAAE,EAAE,SAAS,EAAE,EACvB,aAAa,EACb,iBAAiB,EACjB,gBAAgB,GAAG,CAAC,EACpB,mBAAmB,GAAG,CAAC,EACvB,iBAAiB,EACjB,GAAG,IAAI,EACR,GAAG,kBAAkB,CAAC;QACvB,IAAI,MAAM,GAAG,IAAgD,CAAC;QAE9D,4CAA4C;QAC5C,IAAI,CAAC,SAAS,EAAE,CACd,cAAc,CAAC,OAAO,EACtB,aAAa,EACb,iBAAiB,EACjB,gBAAgB,EAChB,mBAAmB,EACnB,iBAAiB,CAClB,CAAC;QAEF,2EAA2E;QAC3E,MAAM,OAAO,GAA8B,CAAC,cAAc,IAAI,EAAE,CAG/D,CAAC;QACF,KAAK,MAAM,EAAE,QAAQ,EAAE,IAAI,aAAa,EAAE,CAAC;YACzC,UAAU,CACR,MAAiC,EACjC,QAAQ,EACR,OAA4C,CAC7C,CAAC;QACJ,CAAC;QACD,sDAAsD;QACtD,IAAI,CAAC,cAAc,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC5D,MAAM,GAAI,MAAkC,CAAC,QAAQ,CAAC,CAAC;QACzD,CAAC;QAED,MAAM,mBAAmB,GAAG,GAAG,CAAC;QAChC,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,WAAW,CAAC;QAC3D,MAAM,aAAa,GACjB,YAAY,CAAC,MAAM,GAAG,mBAAmB;YACvC,CAAC,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,mBAAmB,CAAC,GAAG,KAAK;YACpD,CAAC,CAAC,YAAY,CAAC;QAEnB,QAAQ,CAAC;YACP,QAAQ,EAAE,YAAY;YACtB,OAAO,EAAE,SAAS;gBAChB,CAAC,CAAC,mCAAmC;gBACrC,CAAC,CAAC,iDAAiD;YACrD,KAAK,EAAE,CAAC;YACR,SAAS,EAAE;gBACT,aAAa,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC/D,iBAAiB,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACvE,iBAAiB,EAAE;oBACjB,KAAK,EAAE,MAAM,CAAC,iBAAiB,CAAC;oBAChC,IAAI,EAAE,QAAQ;iBACf;gBACD,MAAM,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE,QAAQ,EAAE;aACjD;SACF,CAAC,CAAC;QAEH,OAAO,MAAiC,CAAC;IAC3C,CAAC;CACF","sourcesContent":["// lib/v3/handlers/extractHandler.ts\nimport { extract as runExtract } from \"../../inference.js\";\nimport {\n  getZFactory,\n  getZodType,\n  injectUrls,\n  transformSchema,\n} from \"../../utils.js\";\nimport { v3Logger } from \"../logger.js\";\nimport { V3FunctionName } from \"../types/public/methods.js\";\nimport { captureHybridSnapshot } from \"../understudy/a11y/snapshot/index.js\";\nimport type { ZodTypeAny } from \"zod\";\nimport { LLMClient } from \"../llm/LLMClient.js\";\nimport { ExtractHandlerParams } from \"../types/private/handlers.js\";\nimport { EncodedId, ZodPathSegments } from \"../types/private/internal.js\";\nimport {\n  defaultExtractSchema,\n  pageTextSchema,\n} from \"../types/public/methods.js\";\nimport {\n  AvailableModel,\n  ClientOptions,\n  ModelConfiguration,\n} from \"../types/public/model.js\";\nimport {\n  StagehandInvalidArgumentError,\n  ExtractTimeoutError,\n} from \"../types/public/sdkErrors.js\";\nimport { createTimeoutGuard } from \"./handlerUtils/timeoutGuard.js\";\nimport type {\n  InferStagehandSchema,\n  StagehandZodObject,\n  StagehandZodSchema,\n} from \"../zodCompat.js\";\n\n/**\n * Scans the provided Zod schema for any `z.string().url()` fields and\n * replaces them with `z.number()`.\n *\n * @param schema - The Zod object schema to transform.\n * @returns A tuple containing:\n *   1. The transformed schema (or the original schema if no changes were needed).\n *   2. An array of {@link ZodPathSegments} objects representing all the replaced URL fields,\n *      with each path segment showing where in the schema the replacement occurred.\n */\nexport function transformUrlStringsToNumericIds<T extends StagehandZodSchema>(\n  schema: T,\n): [StagehandZodSchema, ZodPathSegments[]] {\n  const [finalSchema, urlPaths] = transformSchema(schema, []);\n  return [finalSchema, urlPaths];\n}\n\ninterface ExtractionResponseBase {\n  metadata: { completed: boolean };\n  prompt_tokens: number;\n  completion_tokens: number;\n  reasoning_tokens: number;\n  cached_input_tokens?: number;\n  inference_time_ms: number;\n}\n\ntype ExtractionResponse<T extends StagehandZodObject> = ExtractionResponseBase &\n  InferStagehandSchema<T>;\n\nexport class ExtractHandler {\n  private readonly llmClient: LLMClient;\n  private readonly defaultModelName: AvailableModel;\n  private readonly defaultClientOptions: ClientOptions;\n  private readonly resolveLlmClient: (model?: ModelConfiguration) => LLMClient;\n  private readonly systemPrompt: string;\n  private readonly logInferenceToFile: boolean;\n  private readonly experimental: boolean;\n  private readonly onMetrics?: (\n    functionName: V3FunctionName,\n    promptTokens: number,\n    completionTokens: number,\n    reasoningTokens: number,\n    cachedInputTokens: number,\n    inferenceTimeMs: number,\n  ) => void;\n\n  constructor(\n    llmClient: LLMClient,\n    defaultModelName: AvailableModel,\n    defaultClientOptions: ClientOptions,\n    resolveLlmClient: (model?: ModelConfiguration) => LLMClient,\n    systemPrompt?: string,\n    logInferenceToFile?: boolean,\n    experimental?: boolean,\n    onMetrics?: (\n      functionName: V3FunctionName,\n      promptTokens: number,\n      completionTokens: number,\n      reasoningTokens: number,\n      cachedInputTokens: number,\n      inferenceTimeMs: number,\n    ) => void,\n  ) {\n    this.llmClient = llmClient;\n    this.defaultModelName = defaultModelName;\n    this.defaultClientOptions = defaultClientOptions;\n    this.resolveLlmClient = resolveLlmClient;\n    this.systemPrompt = systemPrompt ?? \"\";\n    this.logInferenceToFile = logInferenceToFile ?? false;\n    this.experimental = experimental ?? false;\n    this.onMetrics = onMetrics;\n  }\n\n  async extract<T extends StagehandZodSchema>(\n    params: ExtractHandlerParams<T>,\n  ): Promise<InferStagehandSchema<T> | { pageText: string }> {\n    const {\n      instruction,\n      schema,\n      page,\n      selector,\n      ignoreSelectors,\n      timeout,\n      model,\n    } = params;\n\n    const llmClient = this.resolveLlmClient(model);\n\n    const ensureTimeRemaining = createTimeoutGuard(\n      timeout,\n      (ms) => new ExtractTimeoutError(ms),\n    );\n\n    // No-args → page text (parity with v2)\n    const noArgs = !instruction && !schema;\n    if (noArgs) {\n      const focusSelector = selector?.replace(/^xpath=/i, \"\") ?? \"\";\n      ensureTimeRemaining();\n      const snap = await captureHybridSnapshot(page, {\n        experimental: this.experimental,\n        focusSelector: focusSelector || undefined,\n        ignoreSelectors,\n      });\n      ensureTimeRemaining();\n\n      const result = { pageText: snap.combinedTree };\n      // Validate via the same schema used in v2\n      return pageTextSchema.parse(result);\n    }\n\n    if (!instruction && schema) {\n      throw new StagehandInvalidArgumentError(\n        \"extract() requires an instruction when a schema is provided.\",\n      );\n    }\n\n    const focusSelector = selector?.replace(/^xpath=/, \"\") ?? \"\";\n\n    // Build the hybrid snapshot (includes combinedTree; combinedUrlMap optional)\n    ensureTimeRemaining();\n    const { combinedTree, combinedUrlMap } = await captureHybridSnapshot(page, {\n      experimental: this.experimental,\n      focusSelector: focusSelector,\n      ignoreSelectors,\n    });\n\n    v3Logger({\n      category: \"extraction\",\n      message: \"Starting extraction using a11y snapshot\",\n      level: 1,\n      auxiliary: instruction\n        ? { instruction: { value: instruction, type: \"string\" } }\n        : undefined,\n    });\n\n    // Normalize schema: if instruction provided without schema, use defaultExtractSchema\n    const baseSchema: StagehandZodSchema = (schema ??\n      defaultExtractSchema) as StagehandZodSchema;\n    // Ensure we pass an object schema into inference; wrap non-object schemas\n    const isObjectSchema = getZodType(baseSchema) === \"object\";\n    const WRAP_KEY = \"value\" as const;\n    const factory = getZFactory(baseSchema);\n    const objectSchema: StagehandZodObject = isObjectSchema\n      ? (baseSchema as StagehandZodObject)\n      : (factory.object({\n          [WRAP_KEY]: baseSchema as ZodTypeAny,\n        }) as StagehandZodObject);\n\n    const [transformedSchema, urlFieldPaths] =\n      transformUrlStringsToNumericIds(objectSchema);\n\n    ensureTimeRemaining();\n    const extractionResponse: ExtractionResponse<StagehandZodObject> =\n      await runExtract<StagehandZodObject>({\n        instruction,\n        domElements: combinedTree,\n        schema: transformedSchema as StagehandZodObject,\n        llmClient,\n        userProvidedInstructions: this.systemPrompt,\n        logger: v3Logger,\n        logInferenceToFile: this.logInferenceToFile,\n      });\n\n    const {\n      metadata: { completed },\n      prompt_tokens,\n      completion_tokens,\n      reasoning_tokens = 0,\n      cached_input_tokens = 0,\n      inference_time_ms,\n      ...rest\n    } = extractionResponse;\n    let output = rest as InferStagehandSchema<StagehandZodObject>;\n\n    // Update EXTRACT metrics from the LLM calls\n    this.onMetrics?.(\n      V3FunctionName.EXTRACT,\n      prompt_tokens,\n      completion_tokens,\n      reasoning_tokens,\n      cached_input_tokens,\n      inference_time_ms,\n    );\n\n    // Re-inject URLs for any url() fields we temporarily converted to number()\n    const idToUrl: Record<EncodedId, string> = (combinedUrlMap ?? {}) as Record<\n      EncodedId,\n      string\n    >;\n    for (const { segments } of urlFieldPaths) {\n      injectUrls(\n        output as Record<string, unknown>,\n        segments,\n        idToUrl as unknown as Record<string, string>,\n      );\n    }\n    // If we wrapped a non-object schema, unwrap the value\n    if (!isObjectSchema && output && typeof output === \"object\") {\n      output = (output as Record<string, unknown>)[WRAP_KEY];\n    }\n\n    const resultPreviewLength = 200;\n    const resultString = JSON.stringify(output) ?? \"undefined\";\n    const resultPreview =\n      resultString.length > resultPreviewLength\n        ? resultString.slice(0, resultPreviewLength) + \"...\"\n        : resultString;\n\n    v3Logger({\n      category: \"extraction\",\n      message: completed\n        ? \"Extraction completed successfully\"\n        : \"Extraction incomplete after processing all data\",\n      level: 1,\n      auxiliary: {\n        prompt_tokens: { value: String(prompt_tokens), type: \"string\" },\n        completion_tokens: { value: String(completion_tokens), type: \"string\" },\n        inference_time_ms: {\n          value: String(inference_time_ms),\n          type: \"string\",\n        },\n        result: { value: resultPreview, type: \"string\" },\n      },\n    });\n\n    return output as InferStagehandSchema<T>;\n  }\n}\n"]}

package/dist/esm/lib/v3/handlers/observeHandler.js

@@ -27,7 +27,7 @@ export class ObserveHandler {
         this.onMetrics = onMetrics;
     }
     async observe(params) {
-        const { instruction, page, timeout, selector, model, variables } = params;
+        const { instruction, page, timeout, selector, ignoreSelectors, model, variables, } = params;
         const llmClient = this.resolveLlmClient(model);
         const ensureTimeRemaining = createTimeoutGuard(timeout, (ms) => new ObserveTimeoutError(ms));
         const effectiveInstruction = instruction ??
@@ -49,6 +49,7 @@ export class ObserveHandler {
         const snapshot = await captureHybridSnapshot(page, {
             experimental: this.experimental,
             focusSelector: focusSelector || undefined,
+            ignoreSelectors,
         });
         const combinedTree = snapshot.combinedTree;
         const combinedXpathMap = snapshot.combinedXpathMap ?? {};

package/dist/esm/lib/v3/handlers/observeHandler.js.map

@@ -1 +1 @@
-{"version":3,"file":"observeHandler.js","sourceRoot":"","sources":["../../../../../lib/v3/handlers/observeHandler.ts"],"names":[],"mappings":"AAAA,oCAAoC;AACpC,OAAO,EAAE,OAAO,IAAI,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AACtD,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,qBAAqB,EAAE,MAAM,sCAAsC,CAAC;AAE7E,OAAO,EAEL,yBAAyB,GAC1B,MAAM,8BAA8B,CAAC;AAQtC,OAAO,EAAE,mBAAmB,EAAE,MAAM,8BAA8B,CAAC;AACnE,OAAO,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAEpE,MAAM,OAAO,cAAc;IACR,SAAS,CAAY;IACrB,gBAAgB,CAAiB;IACjC,oBAAoB,CAAgB;IACpC,gBAAgB,CAA4C;IAC5D,YAAY,CAAS;IACrB,kBAAkB,CAAU;IAC5B,YAAY,CAAU;IACtB,SAAS,CAOhB;IAEV,YACE,SAAoB,EACpB,gBAAgC,EAChC,oBAAmC,EACnC,gBAA2D,EAC3D,YAAqB,EACrB,kBAA4B,EAC5B,YAAsB,EACtB,SAOS;QAET,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,oBAAoB,GAAG,oBAAoB,CAAC;QACjD,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,YAAY,GAAG,YAAY,IAAI,EAAE,CAAC;QACvC,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,IAAI,KAAK,CAAC;QACtD,IAAI,CAAC,YAAY,GAAG,YAAY,IAAI,KAAK,CAAC;QAC1C,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAA4B;QACxC,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,MAAM,CAAC;QAE1E,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;QAE/C,MAAM,mBAAmB,GAAG,kBAAkB,CAC5C,OAAO,EACP,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,mBAAmB,CAAC,EAAE,CAAC,CACpC,CAAC;QAEF,MAAM,oBAAoB,GACxB,WAAW;YACX,kSAAkS,CAAC;QAErS,QAAQ,CAAC;YACP,QAAQ,EAAE,aAAa;YACvB,OAAO,EAAE,sBAAsB;YAC/B,KAAK,EAAE,CAAC;YACR,SAAS,EAAE;gBACT,WAAW,EAAE;oBACX,KAAK,EAAE,oBAAoB;oBAC3B,IAAI,EAAE,QAAQ;iBACf;aACF;SACF,CAAC,CAAC;QAEH,mEAAmE;QACnE,MAAM,aAAa,GAAG,QAAQ,EAAE,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;QAC9D,mBAAmB,EAAE,CAAC;QACtB,MAAM,QAAQ,GAAG,MAAM,qBAAqB,CAAC,IAAI,EAAE;YACjD,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,aAAa,EAAE,aAAa,IAAI,SAAS;SAC1C,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,QAAQ,CAAC,YAAY,CAAC;QAC3C,MAAM,gBAAgB,GAAG,QAAQ,CAAC,gBAAgB,IAAI,EAAE,CAAC;QAEzD,QAAQ,CAAC;YACP,QAAQ,EAAE,aAAa;YACvB,OAAO,EAAE,6BAA6B;YACtC,KAAK,EAAE,CAAC;SACT,CAAC,CAAC;QAEH,8CAA8C;QAC9C,mBAAmB,EAAE,CAAC;QACtB,MAAM,mBAAmB,GAAG,MAAM,UAAU,CAAC;YAC3C,WAAW,EAAE,oBAAoB;YACjC,WAAW,EAAE,YAAY;YACzB,SAAS;YACT,wBAAwB,EAAE,IAAI,CAAC,YAAY;YAC3C,MAAM,EAAE,QAAQ;YAChB,kBAAkB,EAAE,IAAI,CAAC,kBAAkB;YAC3C,gBAAgB,EAAE,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC;YAC1D,SAAS;SACV,CAAC,CAAC;QAEH,MAAM,EACJ,aAAa,GAAG,CAAC,EACjB,iBAAiB,GAAG,CAAC,EACrB,gBAAgB,GAAG,CAAC,EACpB,mBAAmB,GAAG,CAAC,EACvB,iBAAiB,GAAG,CAAC,GACtB,GAAG,mBAAmB,CAAC;QAExB,uDAAuD;QACvD,IAAI,CAAC,SAAS,EAAE,CACd,cAAc,CAAC,OAAO,EACtB,aAAa,EACb,iBAAiB,EACjB,gBAAgB,EAChB,mBAAmB,EACnB,iBAAiB,CAClB,CAAC;QAEF,mDAAmD;QACnD,MAAM,qBAAqB,GAAG,CAC5B,MAAM,OAAO,CAAC,GAAG,CACf,mBAAmB,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;YACjD,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,CAAC,CAAC,4CAA4C;YACpF,IAAI,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC7D,MAAM,WAAW,GAAG,SAAsB,CAAC;gBAC3C,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAC;gBAC5C,MAAM,YAAY,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC;gBACjD,IAAI,CAAC,YAAY;oBAAE,OAAO,SAAS,CAAC;gBAEpC,6EAA6E;gBAC7E,IAAI,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC;gBAClC,IACE,IAAI,CAAC,MAAM,KAAK,aAAa;oBAC7B,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC;oBAC7B,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EACzB,CAAC;oBACD,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;oBACpC,4DAA4D;oBAC5D,IACE,OAAO,SAAS,KAAK,QAAQ;wBAC7B,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,EAC3B,CAAC;wBACD,MAAM,QAAQ,GAAG,gBAAgB,CAAC,SAAsB,CAAC,CAAC;wBAC1D,MAAM,eAAe,GAAG,oBAAoB,CAAC,QAAQ,CAAC,CAAC;wBACvD,IAAI,eAAe,EAAE,CAAC;4BACpB,YAAY,GAAG;gCACb,SAAS,eAAe,EAAE;gCAC1B,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;6BAC3B,CAAC;wBACJ,CAAC;6BAAM,CAAC;4BACN,uDAAuD;4BACvD,QAAQ,CAAC;gCACP,QAAQ,EAAE,aAAa;gCACvB,OAAO,EAAE,0CAA0C;gCACnD,KAAK,EAAE,CAAC;gCACR,SAAS,EAAE;oCACT,eAAe,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE;oCACrD,eAAe,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE;iCACtD;6BACF,CAAC,CAAC;4BACH,OAAO,SAAS,CAAC;wBACnB,CAAC;oBACH,CAAC;yBAAM,CAAC;wBACN,QAAQ,CAAC;4BACP,QAAQ,EAAE,aAAa;4BACvB,OAAO,EAAE,8CAA8C;4BACvD,KAAK,EAAE,CAAC;4BACR,SAAS,EAAE;gCACT,eAAe,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE;gCACrD,eAAe,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE;6BACtD;yBACF,CAAC,CAAC;wBACH,OAAO,SAAS,CAAC;oBACnB,CAAC;gBACH,CAAC;gBAED,OAAO;oBACL,GAAG,IAAI;oBACP,SAAS,EAAE,YAAY;oBACvB,QAAQ,EAAE,SAAS,YAAY,EAAE;iBAMlC,CAAC;YACJ,CAAC;YACD,wBAAwB;YACxB,OAAO;gBACL,WAAW,EAAE,gCAAgC;gBAC7C,MAAM,EAAE,eAAe;gBACvB,SAAS,EAAE,EAAE;gBACb,QAAQ,EAAE,eAAe;aAC1B,CAAC;QACJ,CAAC,CAAC,CACH,CACF,CAAC,MAAM,CAAC,CAAI,CAAgB,EAAU,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;QAE3D,QAAQ,CAAC;YACP,QAAQ,EAAE,aAAa;YACvB,OAAO,EAAE,gBAAgB;YACzB,KAAK,EAAE,CAAC;YACR,SAAS,EAAE;gBACT,QAAQ,EAAE;oBACR,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAAC;oBAC5C,IAAI,EAAE,QAAQ;iBACf;aACF;SACF,CAAC,CAAC;QAEH,OAAO,qBAAqB,CAAC;IAC/B,CAAC;CACF","sourcesContent":["// lib/v3/handlers/observeHandler.ts\nimport { observe as runObserve } from \"../../inference.js\";\nimport { trimTrailingTextNode } from \"../../utils.js\";\nimport { v3Logger } from \"../logger.js\";\nimport { V3FunctionName } from \"../types/public/methods.js\";\nimport { captureHybridSnapshot } from \"../understudy/a11y/snapshot/index.js\";\nimport { LLMClient } from \"../llm/LLMClient.js\";\nimport {\n  ObserveHandlerParams,\n  SupportedUnderstudyAction,\n} from \"../types/private/handlers.js\";\nimport { EncodedId } from \"../types/private/internal.js\";\nimport { Action } from \"../types/public/methods.js\";\nimport {\n  AvailableModel,\n  ClientOptions,\n  ModelConfiguration,\n} from \"../types/public/model.js\";\nimport { ObserveTimeoutError } from \"../types/public/sdkErrors.js\";\nimport { createTimeoutGuard } from \"./handlerUtils/timeoutGuard.js\";\n\nexport class ObserveHandler {\n  private readonly llmClient: LLMClient;\n  private readonly defaultModelName: AvailableModel;\n  private readonly defaultClientOptions: ClientOptions;\n  private readonly resolveLlmClient: (model?: ModelConfiguration) => LLMClient;\n  private readonly systemPrompt: string;\n  private readonly logInferenceToFile: boolean;\n  private readonly experimental: boolean;\n  private readonly onMetrics?: (\n    functionName: V3FunctionName,\n    promptTokens: number,\n    completionTokens: number,\n    reasoningTokens: number,\n    cachedInputTokens: number,\n    inferenceTimeMs: number,\n  ) => void;\n\n  constructor(\n    llmClient: LLMClient,\n    defaultModelName: AvailableModel,\n    defaultClientOptions: ClientOptions,\n    resolveLlmClient: (model?: ModelConfiguration) => LLMClient,\n    systemPrompt?: string,\n    logInferenceToFile?: boolean,\n    experimental?: boolean,\n    onMetrics?: (\n      functionName: V3FunctionName,\n      promptTokens: number,\n      completionTokens: number,\n      reasoningTokens: number,\n      cachedInputTokens: number,\n      inferenceTimeMs: number,\n    ) => void,\n  ) {\n    this.llmClient = llmClient;\n    this.defaultModelName = defaultModelName;\n    this.defaultClientOptions = defaultClientOptions;\n    this.resolveLlmClient = resolveLlmClient;\n    this.systemPrompt = systemPrompt ?? \"\";\n    this.logInferenceToFile = logInferenceToFile ?? false;\n    this.experimental = experimental ?? false;\n    this.onMetrics = onMetrics;\n  }\n\n  async observe(params: ObserveHandlerParams): Promise<Action[]> {\n    const { instruction, page, timeout, selector, model, variables } = params;\n\n    const llmClient = this.resolveLlmClient(model);\n\n    const ensureTimeRemaining = createTimeoutGuard(\n      timeout,\n      (ms) => new ObserveTimeoutError(ms),\n    );\n\n    const effectiveInstruction =\n      instruction ??\n      \"Find elements that can be used for any future actions in the page. These may be navigation links, related pages, section/subsection links, buttons, or other interactive elements. Be comprehensive: if there are multiple elements that may be relevant for future actions, return all of them.\";\n\n    v3Logger({\n      category: \"observation\",\n      message: \"starting observation\",\n      level: 1,\n      auxiliary: {\n        instruction: {\n          value: effectiveInstruction,\n          type: \"string\",\n        },\n      },\n    });\n\n    // Build the hybrid snapshot (a11y-centric text tree + lookup maps)\n    const focusSelector = selector?.replace(/^xpath=/i, \"\") ?? \"\";\n    ensureTimeRemaining();\n    const snapshot = await captureHybridSnapshot(page, {\n      experimental: this.experimental,\n      focusSelector: focusSelector || undefined,\n    });\n\n    const combinedTree = snapshot.combinedTree;\n    const combinedXpathMap = snapshot.combinedXpathMap ?? {};\n\n    v3Logger({\n      category: \"observation\",\n      message: \"Got accessibility tree data\",\n      level: 1,\n    });\n\n    // Call the LLM to propose actionable elements\n    ensureTimeRemaining();\n    const observationResponse = await runObserve({\n      instruction: effectiveInstruction,\n      domElements: combinedTree,\n      llmClient,\n      userProvidedInstructions: this.systemPrompt,\n      logger: v3Logger,\n      logInferenceToFile: this.logInferenceToFile,\n      supportedActions: Object.values(SupportedUnderstudyAction),\n      variables,\n    });\n\n    const {\n      prompt_tokens = 0,\n      completion_tokens = 0,\n      reasoning_tokens = 0,\n      cached_input_tokens = 0,\n      inference_time_ms = 0,\n    } = observationResponse;\n\n    // Update OBSERVE metrics from the LLM observation call\n    this.onMetrics?.(\n      V3FunctionName.OBSERVE,\n      prompt_tokens,\n      completion_tokens,\n      reasoning_tokens,\n      cached_input_tokens,\n      inference_time_ms,\n    );\n\n    // Map elementIds -> selectors via combinedXpathMap\n    const elementsWithSelectors = (\n      await Promise.all(\n        observationResponse.elements.map(async (element) => {\n          const { elementId, ...rest } = element; // rest may or may not have method/arguments\n          if (typeof elementId === \"string\" && elementId.includes(\"-\")) {\n            const lookUpIndex = elementId as EncodedId;\n            const xpath = combinedXpathMap[lookUpIndex];\n            const trimmedXpath = trimTrailingTextNode(xpath);\n            if (!trimmedXpath) return undefined;\n\n            // For dragAndDrop, convert element ID in arguments to xpath (target element)\n            let resolvedArgs = rest.arguments;\n            if (\n              rest.method === \"dragAndDrop\" &&\n              Array.isArray(rest.arguments) &&\n              rest.arguments.length > 0\n            ) {\n              const targetArg = rest.arguments[0];\n              // Check if argument looks like an element ID (e.g., \"1-67\")\n              if (\n                typeof targetArg === \"string\" &&\n                /^\\d+-\\d+$/.test(targetArg)\n              ) {\n                const argXpath = combinedXpathMap[targetArg as EncodedId];\n                const trimmedArgXpath = trimTrailingTextNode(argXpath);\n                if (trimmedArgXpath) {\n                  resolvedArgs = [\n                    `xpath=${trimmedArgXpath}`,\n                    ...rest.arguments.slice(1),\n                  ];\n                } else {\n                  // Target element lookup failed, filter out this action\n                  v3Logger({\n                    category: \"observation\",\n                    message: \"dragAndDrop target element lookup failed\",\n                    level: 0,\n                    auxiliary: {\n                      targetElementId: { value: targetArg, type: \"string\" },\n                      sourceElementId: { value: elementId, type: \"string\" },\n                    },\n                  });\n                  return undefined;\n                }\n              } else {\n                v3Logger({\n                  category: \"observation\",\n                  message: \"dragAndDrop target element invalid ID format\",\n                  level: 0,\n                  auxiliary: {\n                    targetElementId: { value: targetArg, type: \"string\" },\n                    sourceElementId: { value: elementId, type: \"string\" },\n                  },\n                });\n                return undefined;\n              }\n            }\n\n            return {\n              ...rest,\n              arguments: resolvedArgs,\n              selector: `xpath=${trimmedXpath}`,\n            } as {\n              description: string;\n              method?: string;\n              arguments?: string[];\n              selector: string;\n            };\n          }\n          // shadow-root fallback:\n          return {\n            description: \"an element inside a shadow DOM\",\n            method: \"not-supported\",\n            arguments: [],\n            selector: \"not-supported\",\n          };\n        }),\n      )\n    ).filter(<T>(e: T | undefined): e is T => e !== undefined);\n\n    v3Logger({\n      category: \"observation\",\n      message: \"found elements\",\n      level: 1,\n      auxiliary: {\n        elements: {\n          value: JSON.stringify(elementsWithSelectors),\n          type: \"object\",\n        },\n      },\n    });\n\n    return elementsWithSelectors;\n  }\n}\n"]}
+{"version":3,"file":"observeHandler.js","sourceRoot":"","sources":["../../../../../lib/v3/handlers/observeHandler.ts"],"names":[],"mappings":"AAAA,oCAAoC;AACpC,OAAO,EAAE,OAAO,IAAI,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AACtD,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,qBAAqB,EAAE,MAAM,sCAAsC,CAAC;AAE7E,OAAO,EAEL,yBAAyB,GAC1B,MAAM,8BAA8B,CAAC;AAQtC,OAAO,EAAE,mBAAmB,EAAE,MAAM,8BAA8B,CAAC;AACnE,OAAO,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAEpE,MAAM,OAAO,cAAc;IACR,SAAS,CAAY;IACrB,gBAAgB,CAAiB;IACjC,oBAAoB,CAAgB;IACpC,gBAAgB,CAA4C;IAC5D,YAAY,CAAS;IACrB,kBAAkB,CAAU;IAC5B,YAAY,CAAU;IACtB,SAAS,CAOhB;IAEV,YACE,SAAoB,EACpB,gBAAgC,EAChC,oBAAmC,EACnC,gBAA2D,EAC3D,YAAqB,EACrB,kBAA4B,EAC5B,YAAsB,EACtB,SAOS;QAET,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,oBAAoB,GAAG,oBAAoB,CAAC;QACjD,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,YAAY,GAAG,YAAY,IAAI,EAAE,CAAC;QACvC,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,IAAI,KAAK,CAAC;QACtD,IAAI,CAAC,YAAY,GAAG,YAAY,IAAI,KAAK,CAAC;QAC1C,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAA4B;QACxC,MAAM,EACJ,WAAW,EACX,IAAI,EACJ,OAAO,EACP,QAAQ,EACR,eAAe,EACf,KAAK,EACL,SAAS,GACV,GAAG,MAAM,CAAC;QAEX,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;QAE/C,MAAM,mBAAmB,GAAG,kBAAkB,CAC5C,OAAO,EACP,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,mBAAmB,CAAC,EAAE,CAAC,CACpC,CAAC;QAEF,MAAM,oBAAoB,GACxB,WAAW;YACX,kSAAkS,CAAC;QAErS,QAAQ,CAAC;YACP,QAAQ,EAAE,aAAa;YACvB,OAAO,EAAE,sBAAsB;YAC/B,KAAK,EAAE,CAAC;YACR,SAAS,EAAE;gBACT,WAAW,EAAE;oBACX,KAAK,EAAE,oBAAoB;oBAC3B,IAAI,EAAE,QAAQ;iBACf;aACF;SACF,CAAC,CAAC;QAEH,mEAAmE;QACnE,MAAM,aAAa,GAAG,QAAQ,EAAE,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;QAC9D,mBAAmB,EAAE,CAAC;QACtB,MAAM,QAAQ,GAAG,MAAM,qBAAqB,CAAC,IAAI,EAAE;YACjD,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,aAAa,EAAE,aAAa,IAAI,SAAS;YACzC,eAAe;SAChB,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,QAAQ,CAAC,YAAY,CAAC;QAC3C,MAAM,gBAAgB,GAAG,QAAQ,CAAC,gBAAgB,IAAI,EAAE,CAAC;QAEzD,QAAQ,CAAC;YACP,QAAQ,EAAE,aAAa;YACvB,OAAO,EAAE,6BAA6B;YACtC,KAAK,EAAE,CAAC;SACT,CAAC,CAAC;QAEH,8CAA8C;QAC9C,mBAAmB,EAAE,CAAC;QACtB,MAAM,mBAAmB,GAAG,MAAM,UAAU,CAAC;YAC3C,WAAW,EAAE,oBAAoB;YACjC,WAAW,EAAE,YAAY;YACzB,SAAS;YACT,wBAAwB,EAAE,IAAI,CAAC,YAAY;YAC3C,MAAM,EAAE,QAAQ;YAChB,kBAAkB,EAAE,IAAI,CAAC,kBAAkB;YAC3C,gBAAgB,EAAE,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC;YAC1D,SAAS;SACV,CAAC,CAAC;QAEH,MAAM,EACJ,aAAa,GAAG,CAAC,EACjB,iBAAiB,GAAG,CAAC,EACrB,gBAAgB,GAAG,CAAC,EACpB,mBAAmB,GAAG,CAAC,EACvB,iBAAiB,GAAG,CAAC,GACtB,GAAG,mBAAmB,CAAC;QAExB,uDAAuD;QACvD,IAAI,CAAC,SAAS,EAAE,CACd,cAAc,CAAC,OAAO,EACtB,aAAa,EACb,iBAAiB,EACjB,gBAAgB,EAChB,mBAAmB,EACnB,iBAAiB,CAClB,CAAC;QAEF,mDAAmD;QACnD,MAAM,qBAAqB,GAAG,CAC5B,MAAM,OAAO,CAAC,GAAG,CACf,mBAAmB,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;YACjD,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,CAAC,CAAC,4CAA4C;YACpF,IAAI,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC7D,MAAM,WAAW,GAAG,SAAsB,CAAC;gBAC3C,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAC;gBAC5C,MAAM,YAAY,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC;gBACjD,IAAI,CAAC,YAAY;oBAAE,OAAO,SAAS,CAAC;gBAEpC,6EAA6E;gBAC7E,IAAI,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC;gBAClC,IACE,IAAI,CAAC,MAAM,KAAK,aAAa;oBAC7B,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC;oBAC7B,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EACzB,CAAC;oBACD,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;oBACpC,4DAA4D;oBAC5D,IACE,OAAO,SAAS,KAAK,QAAQ;wBAC7B,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,EAC3B,CAAC;wBACD,MAAM,QAAQ,GAAG,gBAAgB,CAAC,SAAsB,CAAC,CAAC;wBAC1D,MAAM,eAAe,GAAG,oBAAoB,CAAC,QAAQ,CAAC,CAAC;wBACvD,IAAI,eAAe,EAAE,CAAC;4BACpB,YAAY,GAAG;gCACb,SAAS,eAAe,EAAE;gCAC1B,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;6BAC3B,CAAC;wBACJ,CAAC;6BAAM,CAAC;4BACN,uDAAuD;4BACvD,QAAQ,CAAC;gCACP,QAAQ,EAAE,aAAa;gCACvB,OAAO,EAAE,0CAA0C;gCACnD,KAAK,EAAE,CAAC;gCACR,SAAS,EAAE;oCACT,eAAe,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE;oCACrD,eAAe,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE;iCACtD;6BACF,CAAC,CAAC;4BACH,OAAO,SAAS,CAAC;wBACnB,CAAC;oBACH,CAAC;yBAAM,CAAC;wBACN,QAAQ,CAAC;4BACP,QAAQ,EAAE,aAAa;4BACvB,OAAO,EAAE,8CAA8C;4BACvD,KAAK,EAAE,CAAC;4BACR,SAAS,EAAE;gCACT,eAAe,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE;gCACrD,eAAe,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE;6BACtD;yBACF,CAAC,CAAC;wBACH,OAAO,SAAS,CAAC;oBACnB,CAAC;gBACH,CAAC;gBAED,OAAO;oBACL,GAAG,IAAI;oBACP,SAAS,EAAE,YAAY;oBACvB,QAAQ,EAAE,SAAS,YAAY,EAAE;iBAMlC,CAAC;YACJ,CAAC;YACD,wBAAwB;YACxB,OAAO;gBACL,WAAW,EAAE,gCAAgC;gBAC7C,MAAM,EAAE,eAAe;gBACvB,SAAS,EAAE,EAAE;gBACb,QAAQ,EAAE,eAAe;aAC1B,CAAC;QACJ,CAAC,CAAC,CACH,CACF,CAAC,MAAM,CAAC,CAAI,CAAgB,EAAU,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;QAE3D,QAAQ,CAAC;YACP,QAAQ,EAAE,aAAa;YACvB,OAAO,EAAE,gBAAgB;YACzB,KAAK,EAAE,CAAC;YACR,SAAS,EAAE;gBACT,QAAQ,EAAE;oBACR,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAAC;oBAC5C,IAAI,EAAE,QAAQ;iBACf;aACF;SACF,CAAC,CAAC;QAEH,OAAO,qBAAqB,CAAC;IAC/B,CAAC;CACF","sourcesContent":["// lib/v3/handlers/observeHandler.ts\nimport { observe as runObserve } from \"../../inference.js\";\nimport { trimTrailingTextNode } from \"../../utils.js\";\nimport { v3Logger } from \"../logger.js\";\nimport { V3FunctionName } from \"../types/public/methods.js\";\nimport { captureHybridSnapshot } from \"../understudy/a11y/snapshot/index.js\";\nimport { LLMClient } from \"../llm/LLMClient.js\";\nimport {\n  ObserveHandlerParams,\n  SupportedUnderstudyAction,\n} from \"../types/private/handlers.js\";\nimport { EncodedId } from \"../types/private/internal.js\";\nimport { Action } from \"../types/public/methods.js\";\nimport {\n  AvailableModel,\n  ClientOptions,\n  ModelConfiguration,\n} from \"../types/public/model.js\";\nimport { ObserveTimeoutError } from \"../types/public/sdkErrors.js\";\nimport { createTimeoutGuard } from \"./handlerUtils/timeoutGuard.js\";\n\nexport class ObserveHandler {\n  private readonly llmClient: LLMClient;\n  private readonly defaultModelName: AvailableModel;\n  private readonly defaultClientOptions: ClientOptions;\n  private readonly resolveLlmClient: (model?: ModelConfiguration) => LLMClient;\n  private readonly systemPrompt: string;\n  private readonly logInferenceToFile: boolean;\n  private readonly experimental: boolean;\n  private readonly onMetrics?: (\n    functionName: V3FunctionName,\n    promptTokens: number,\n    completionTokens: number,\n    reasoningTokens: number,\n    cachedInputTokens: number,\n    inferenceTimeMs: number,\n  ) => void;\n\n  constructor(\n    llmClient: LLMClient,\n    defaultModelName: AvailableModel,\n    defaultClientOptions: ClientOptions,\n    resolveLlmClient: (model?: ModelConfiguration) => LLMClient,\n    systemPrompt?: string,\n    logInferenceToFile?: boolean,\n    experimental?: boolean,\n    onMetrics?: (\n      functionName: V3FunctionName,\n      promptTokens: number,\n      completionTokens: number,\n      reasoningTokens: number,\n      cachedInputTokens: number,\n      inferenceTimeMs: number,\n    ) => void,\n  ) {\n    this.llmClient = llmClient;\n    this.defaultModelName = defaultModelName;\n    this.defaultClientOptions = defaultClientOptions;\n    this.resolveLlmClient = resolveLlmClient;\n    this.systemPrompt = systemPrompt ?? \"\";\n    this.logInferenceToFile = logInferenceToFile ?? false;\n    this.experimental = experimental ?? false;\n    this.onMetrics = onMetrics;\n  }\n\n  async observe(params: ObserveHandlerParams): Promise<Action[]> {\n    const {\n      instruction,\n      page,\n      timeout,\n      selector,\n      ignoreSelectors,\n      model,\n      variables,\n    } = params;\n\n    const llmClient = this.resolveLlmClient(model);\n\n    const ensureTimeRemaining = createTimeoutGuard(\n      timeout,\n      (ms) => new ObserveTimeoutError(ms),\n    );\n\n    const effectiveInstruction =\n      instruction ??\n      \"Find elements that can be used for any future actions in the page. These may be navigation links, related pages, section/subsection links, buttons, or other interactive elements. Be comprehensive: if there are multiple elements that may be relevant for future actions, return all of them.\";\n\n    v3Logger({\n      category: \"observation\",\n      message: \"starting observation\",\n      level: 1,\n      auxiliary: {\n        instruction: {\n          value: effectiveInstruction,\n          type: \"string\",\n        },\n      },\n    });\n\n    // Build the hybrid snapshot (a11y-centric text tree + lookup maps)\n    const focusSelector = selector?.replace(/^xpath=/i, \"\") ?? \"\";\n    ensureTimeRemaining();\n    const snapshot = await captureHybridSnapshot(page, {\n      experimental: this.experimental,\n      focusSelector: focusSelector || undefined,\n      ignoreSelectors,\n    });\n\n    const combinedTree = snapshot.combinedTree;\n    const combinedXpathMap = snapshot.combinedXpathMap ?? {};\n\n    v3Logger({\n      category: \"observation\",\n      message: \"Got accessibility tree data\",\n      level: 1,\n    });\n\n    // Call the LLM to propose actionable elements\n    ensureTimeRemaining();\n    const observationResponse = await runObserve({\n      instruction: effectiveInstruction,\n      domElements: combinedTree,\n      llmClient,\n      userProvidedInstructions: this.systemPrompt,\n      logger: v3Logger,\n      logInferenceToFile: this.logInferenceToFile,\n      supportedActions: Object.values(SupportedUnderstudyAction),\n      variables,\n    });\n\n    const {\n      prompt_tokens = 0,\n      completion_tokens = 0,\n      reasoning_tokens = 0,\n      cached_input_tokens = 0,\n      inference_time_ms = 0,\n    } = observationResponse;\n\n    // Update OBSERVE metrics from the LLM observation call\n    this.onMetrics?.(\n      V3FunctionName.OBSERVE,\n      prompt_tokens,\n      completion_tokens,\n      reasoning_tokens,\n      cached_input_tokens,\n      inference_time_ms,\n    );\n\n    // Map elementIds -> selectors via combinedXpathMap\n    const elementsWithSelectors = (\n      await Promise.all(\n        observationResponse.elements.map(async (element) => {\n          const { elementId, ...rest } = element; // rest may or may not have method/arguments\n          if (typeof elementId === \"string\" && elementId.includes(\"-\")) {\n            const lookUpIndex = elementId as EncodedId;\n            const xpath = combinedXpathMap[lookUpIndex];\n            const trimmedXpath = trimTrailingTextNode(xpath);\n            if (!trimmedXpath) return undefined;\n\n            // For dragAndDrop, convert element ID in arguments to xpath (target element)\n            let resolvedArgs = rest.arguments;\n            if (\n              rest.method === \"dragAndDrop\" &&\n              Array.isArray(rest.arguments) &&\n              rest.arguments.length > 0\n            ) {\n              const targetArg = rest.arguments[0];\n              // Check if argument looks like an element ID (e.g., \"1-67\")\n              if (\n                typeof targetArg === \"string\" &&\n                /^\\d+-\\d+$/.test(targetArg)\n              ) {\n                const argXpath = combinedXpathMap[targetArg as EncodedId];\n                const trimmedArgXpath = trimTrailingTextNode(argXpath);\n                if (trimmedArgXpath) {\n                  resolvedArgs = [\n                    `xpath=${trimmedArgXpath}`,\n                    ...rest.arguments.slice(1),\n                  ];\n                } else {\n                  // Target element lookup failed, filter out this action\n                  v3Logger({\n                    category: \"observation\",\n                    message: \"dragAndDrop target element lookup failed\",\n                    level: 0,\n                    auxiliary: {\n                      targetElementId: { value: targetArg, type: \"string\" },\n                      sourceElementId: { value: elementId, type: \"string\" },\n                    },\n                  });\n                  return undefined;\n                }\n              } else {\n                v3Logger({\n                  category: \"observation\",\n                  message: \"dragAndDrop target element invalid ID format\",\n                  level: 0,\n                  auxiliary: {\n                    targetElementId: { value: targetArg, type: \"string\" },\n                    sourceElementId: { value: elementId, type: \"string\" },\n                  },\n                });\n                return undefined;\n              }\n            }\n\n            return {\n              ...rest,\n              arguments: resolvedArgs,\n              selector: `xpath=${trimmedXpath}`,\n            } as {\n              description: string;\n              method?: string;\n              arguments?: string[];\n              selector: string;\n            };\n          }\n          // shadow-root fallback:\n          return {\n            description: \"an element inside a shadow DOM\",\n            method: \"not-supported\",\n            arguments: [],\n            selector: \"not-supported\",\n          };\n        }),\n      )\n    ).filter(<T>(e: T | undefined): e is T => e !== undefined);\n\n    v3Logger({\n      category: \"observation\",\n      message: \"found elements\",\n      level: 1,\n      auxiliary: {\n        elements: {\n          value: JSON.stringify(elementsWithSelectors),\n          type: \"object\",\n        },\n      },\n    });\n\n    return elementsWithSelectors;\n  }\n}\n"]}

package/dist/esm/lib/v3/handlers/v3AgentHandler.js

@@ -3,6 +3,7 @@ import { buildAgentSystemPrompt } from "../agent/prompts/agentSystemPrompt.js";
 import { wrapLanguageModel, stepCountIs, } from "ai";
 import { processMessages } from "../agent/utils/messageProcessing.js";
 import { FlowLogger } from "../flowlogger/FlowLogger.js";
+import { HYBRID_CAPABLE_MODEL_PATTERNS } from "../types/private/agent.js";
 import { V3FunctionName } from "../types/public/methods.js";
 import { mapToolResultToActions } from "../agent/utils/actionMapping.js";
 import { MissingLLMConfigurationError, MissingEnvironmentVariableError, StreamingCallbacksInNonStreamingModeError, AgentAbortError, } from "../types/public/sdkErrors.js";
@@ -90,9 +91,8 @@ export class V3AgentHandler {
                     ...FlowLogger.createLlmLoggingMiddleware(baseModel.modelId),
                 },
             });
-            if (this.mode === "hybrid" &&
-                !baseModel.modelId.includes("gemini-3-flash") &&
-                !baseModel.modelId.includes("claude")) {
+            const isHybridCapable = HYBRID_CAPABLE_MODEL_PATTERNS.some((pattern) => baseModel.modelId.includes(pattern));
+            if (this.mode === "hybrid" && !isHybridCapable) {
                 this.logger({
                     category: "agent",
                     message: `Warning: "${baseModel.modelId}" may not perform well in hybrid mode. See recommended models: https://docs.stagehand.dev/v3/basics/agent#hybrid-mode`,
@@ -256,7 +256,6 @@ export class V3AgentHandler {
                 messages: prependSystemMessage(systemPrompt, messages),
                 tools: allTools,
                 stopWhen: (result) => this.handleStop(result, maxSteps),
-                temperature: 1,
                 toolChoice: "auto",
                 prepareStep: this.createPrepareStep(callbacks?.prepareStep, captchaSolver),
                 onStepFinish: this.createStepHandler(state, callbacks?.onStepFinish),
@@ -348,7 +347,6 @@ export class V3AgentHandler {
                 messages: prependSystemMessage(systemPrompt, messages),
                 tools: allTools,
                 stopWhen: (result) => this.handleStop(result, maxSteps),
-                temperature: 1,
                 toolChoice: "auto",
                 prepareStep: this.createPrepareStep(callbacks?.prepareStep, captchaSolver),
                 onStepFinish: this.createStepHandler(state, callbacks?.onStepFinish),

package/dist/esm/lib/v3/handlers/v3AgentHandler.js.map

@@ -1 +1 @@
-{"version":3,"file":"v3AgentHandler.js","sourceRoot":"","sources":["../../../../../lib/v3/handlers/v3AgentHandler.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,EAAE,sBAAsB,EAAE,MAAM,uCAAuC,CAAC;AAG/E,OAAO,EAGL,iBAAiB,EACjB,WAAW,GAOZ,MAAM,IAAI,CAAC;AAEZ,OAAO,EAAE,eAAe,EAAE,MAAM,qCAAqC,CAAC;AAEtE,OAAO,EAAE,UAAU,EAAE,MAAM,6BAA6B,CAAC;AAczD,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,sBAAsB,EAAE,MAAM,iCAAiC,CAAC;AACzE,OAAO,EACL,4BAA4B,EAC5B,+BAA+B,EAC/B,yCAAyC,EACzC,eAAe,GAChB,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,kBAAkB,EAAE,MAAM,sCAAsC,CAAC;AAC1E,OAAO,EACL,aAAa,EACb,kBAAkB,EAClB,mBAAmB,GACpB,MAAM,iCAAiC,CAAC;AAEzC,SAAS,eAAe,CAAC,KAAc;IACrC,OAAO,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAChE,CAAC;AAED;;;GAGG;AACH,SAAS,oBAAoB,CAC3B,YAAoB,EACpB,QAAwB;IAExB,OAAO;QACL;YACE,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,YAAY;YACrB,eAAe,EAAE;gBACf,SAAS,EAAE;oBACT,YAAY,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;iBACpC;aACF;SACF;QACD,GAAG,QAAQ;KACZ,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,cAAc;IACjB,EAAE,CAAK;IACP,MAAM,CAA6B;IACnC,SAAS,CAAY;IACrB,cAAc,CAA6B;IAC3C,kBAAkB,CAAU;IAC5B,QAAQ,CAAW;IACnB,IAAI,CAAgB;IACpB,uBAAuB,CAAU;IAEzC,YACE,EAAM,EACN,MAAkC,EAClC,SAAoB,EACpB,cAA0C,EAC1C,kBAA2B,EAC3B,QAAkB,EAClB,IAAoB,EACpB,uBAAiC;QAEjC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;QAC7C,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,IAAI,GAAG,IAAI,IAAI,KAAK,CAAC;QAC1B,IAAI,CAAC,uBAAuB,GAAG,uBAAuB,IAAI,KAAK,CAAC;IAClE,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,oBAAsD;QAEtD,IAAI,CAAC;YACH,MAAM,OAAO,GACX,OAAO,oBAAoB,KAAK,QAAQ;gBACtC,CAAC,CAAC,EAAE,WAAW,EAAE,oBAAoB,EAAE;gBACvC,CAAC,CAAC,oBAAoB,CAAC;YAE3B,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC;YAExC,gEAAgE;YAChE,MAAM,cAAc,GAAG,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC;YAEvE,wDAAwD;YACxD,MAAM,YAAY,GAAG,sBAAsB,CAAC;gBAC1C,GAAG,EAAE,cAAc;gBACnB,oBAAoB,EAAE,OAAO,CAAC,WAAW;gBACzC,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,kBAAkB,EAAE,IAAI,CAAC,kBAAkB;gBAC3C,iBAAiB,EAAE,IAAI,CAAC,EAAE,CAAC,yBAAyB;gBACpD,YAAY,EAAE,OAAO,CAAC,YAAY;gBAClC,SAAS,EAAE,OAAO,CAAC,SAAS;gBAC5B,SAAS,EAAE,OAAO,CAAC,SAAS;aAC7B,CAAC,CAAC;YAEH,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;gBACtB,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,CAAC,iBAAiB,CAAC;gBAC3C,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACd,MAAM,IAAI,+BAA+B,CACvC,qBAAqB,EACrB,gCAAgC,CACjC,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAC5B,OAAO,CAAC,YAAY,EACpB,OAAO,CAAC,SAAS,EACjB,OAAO,CAAC,WAAW,EACnB,OAAO,CAAC,SAAS,CAClB,CAAC;YACF,MAAM,QAAQ,GAAY,EAAE,GAAG,KAAK,EAAE,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;YAEzD,8EAA8E;YAC9E,MAAM,QAAQ,GAAmB,OAAO,CAAC,QAAQ,EAAE,MAAM;gBACvD,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC;gBACvE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;YAErD,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,gBAAgB,EAAE,CAAC;gBACtC,MAAM,IAAI,4BAA4B,EAAE,CAAC;YAC3C,CAAC;YACD,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC;YACpD,kDAAkD;YAClD,MAAM,YAAY,GAAG,iBAAiB,CAAC;gBACrC,KAAK,EAAE,SAAS;gBAChB,UAAU,EAAE;oBACV,GAAG,UAAU,CAAC,0BAA0B,CAAC,SAAS,CAAC,OAAO,CAAC;iBAC5D;aACF,CAAC,CAAC;YAEH,IACE,IAAI,CAAC,IAAI,KAAK,QAAQ;gBACtB,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAC;gBAC7C,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,EACrC,CAAC;gBACD,IAAI,CAAC,MAAM,CAAC;oBACV,QAAQ,EAAE,OAAO;oBACjB,OAAO,EAAE,aAAa,SAAS,CAAC,OAAO,uHAAuH;oBAC9J,KAAK,EAAE,CAAC;iBACT,CAAC,CAAC;YACL,CAAC;YAED,OAAO;gBACL,OAAO;gBACP,QAAQ;gBACR,YAAY;gBACZ,QAAQ;gBACR,QAAQ;gBACR,YAAY;gBACZ,cAAc;aACf,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC;gBACV,QAAQ,EAAE,OAAO;gBACjB,OAAO,EAAE,4BAA4B,KAAK,EAAE;gBAC5C,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IACO,iBAAiB,CACvB,YAA2C,EAC3C,aAA6B;QAE7B,OAAO,KAAK,EAAE,OAAO,EAAE,EAAE;YACvB,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAClC,IAAI,aAAa,EAAE,CAAC;gBAClB,IAAI,aAAa,CAAC,SAAS,EAAE,EAAE,CAAC;oBAC9B,IAAI,CAAC,MAAM,CAAC;wBACV,QAAQ,EAAE,OAAO;wBACjB,OAAO,EACL,0EAA0E;wBAC5E,KAAK,EAAE,CAAC;qBACT,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM,aAAa,CAAC,aAAa,EAAE,CAAC;gBACpC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,aAAa,CAAC,kBAAkB,EAAE,CAAC;gBAC/D,IAAI,MAAM,EAAE,CAAC;oBACX,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;wBACpB,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,kBAAkB;qBAC5B,CAAC,CAAC;oBACH,IAAI,CAAC,MAAM,CAAC;wBACV,QAAQ,EAAE,OAAO;wBACjB,OAAO,EACL,kEAAkE;wBACpE,KAAK,EAAE,CAAC;qBACT,CAAC,CAAC;gBACL,CAAC;gBACD,IAAI,OAAO,EAAE,CAAC;oBACZ,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;wBACpB,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,mBAAmB;qBAC7B,CAAC,CAAC;oBACH,IAAI,CAAC,MAAM,CAAC;wBACV,QAAQ,EAAE,OAAO;wBACjB,OAAO,EACL,+EAA+E;wBACjF,KAAK,EAAE,CAAC;qBACT,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YACD,IAAI,YAAY,EAAE,CAAC;gBACjB,OAAO,YAAY,CAAC,OAAO,CAAC,CAAC;YAC/B,CAAC;YACD,OAAO,OAAO,CAAC;QACjB,CAAC,CAAC;IACJ,CAAC;IAEO,iBAAiB,CACvB,KAAiB,EACjB,YAE2C;QAE3C,OAAO,KAAK,EAAE,KAA0B,EAAE,EAAE;YAC1C,IAAI,CAAC,MAAM,CAAC;gBACV,QAAQ,EAAE,OAAO;gBACjB,OAAO,EAAE,kBAAkB,KAAK,CAAC,YAAY,EAAE;gBAC/C,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;YAEH,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAClD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oBAChD,MAAM,QAAQ,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;oBACpC,MAAM,IAAI,GAAG,QAAQ,CAAC,KAAK,CAAC;oBAC5B,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC;oBAE1C,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACxC,KAAK,CAAC,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBAC1C,IAAI,CAAC,MAAM,CAAC;4BACV,QAAQ,EAAE,OAAO;4BACjB,OAAO,EAAE,cAAc,KAAK,CAAC,IAAI,EAAE;4BACnC,KAAK,EAAE,CAAC;yBACT,CAAC,CAAC;oBACL,CAAC;oBAED,IAAI,QAAQ,CAAC,QAAQ,KAAK,MAAM,EAAE,CAAC;wBACjC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC;wBACvB,IAAI,IAAI,EAAE,YAAY,EAAE,CAAC;4BACvB,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC;4BACrC,MAAM,YAAY,GAAG,KAAK,CAAC,kBAAkB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;4BACxD,KAAK,CAAC,YAAY,GAAG,aAAa;gCAChC,CAAC,CAAC,GAAG,YAAY,IAAI,aAAa,EAAE,CAAC,IAAI,EAAE;gCAC3C,CAAC,CAAC,YAAY,IAAI,6BAA6B,CAAC;wBACpD,CAAC;oBACH,CAAC;oBACD,MAAM,aAAa,GAAG,sBAAsB,CAAC;wBAC3C,YAAY,EAAE,QAAQ,CAAC,QAAQ;wBAC/B,UAAU;wBACV,IAAI;wBACJ,SAAS,EAAE,KAAK,CAAC,IAAI,IAAI,SAAS;qBACnC,CAAC,CAAC;oBAEH,KAAK,MAAM,MAAM,IAAI,aAAa,EAAE,CAAC;wBACnC,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC,cAAc,CAAC;wBACtC,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;wBAC9B,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;oBAC7B,CAAC;gBACH,CAAC;gBACD,KAAK,CAAC,cAAc,GAAG,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC;YACzE,CAAC;YAED,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,YAAY,CAAC,KAAK,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,OAAO,CAClB,oBAAkD;QAElD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,OAAO,GACX,OAAO,oBAAoB,KAAK,QAAQ,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC;QACzE,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,CAAC;QAE/B,uEAAuE;QACvE,MAAM,qBAAqB,GACzB,OAAO,EAAE,eAAe,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC;QAErD,MAAM,KAAK,GAAe;YACxB,kBAAkB,EAAE,EAAE;YACtB,OAAO,EAAE,EAAE;YACX,YAAY,EAAE,EAAE;YAChB,SAAS,EAAE,KAAK;YAChB,cAAc,EAAE,EAAE;SACnB,CAAC;QAEF,IAAI,QAAQ,GAAmB,EAAE,CAAC;QAClC,IAAI,aAAwC,CAAC;QAE7C,IAAI,CAAC;YACH,MAAM,EACJ,OAAO,EAAE,eAAe,EACxB,QAAQ,EACR,YAAY,EACZ,QAAQ,EACR,QAAQ,EAAE,gBAAgB,EAC1B,YAAY,EACZ,cAAc,GACf,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,oBAAoB,CAAC,CAAC;YAElD,wEAAwE;YACxE,IAAI,qBAAqB,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACpD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC;gBACrD,MAAM,IAAI,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YACnD,CAAC;YAED,qDAAqD;YACrD,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;gBACjC,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;gBACpC,aAAa,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC;YAC9D,CAAC;YAED,QAAQ,GAAG,gBAAgB,CAAC;YAC5B,KAAK,CAAC,cAAc,GAAG,cAAc,CAAC;YAEtC,MAAM,SAAS,GAAI,oBAA4C,CAAC,SAAS,CAAC;YAE1E,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,sBAAsB,GAAG;oBAC7B,SAAS;oBACT,UAAU;oBACV,SAAS;oBACT,SAAS;iBACV,CAAC;gBACF,MAAM,gBAAgB,GAAG,sBAAsB,CAAC,MAAM,CACpD,CAAC,IAAI,EAAE,EAAE,CAAC,SAAS,CAAC,IAA8B,CAAC,IAAI,IAAI,CAC5D,CAAC;gBACF,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAChC,MAAM,IAAI,yCAAyC,CAAC,gBAAgB,CAAC,CAAC;gBACxE,CAAC;YACH,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC;gBAC/C,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,oBAAoB,CAAC,YAAY,EAAE,QAAQ,CAAC;gBACtD,KAAK,EAAE,QAAQ;gBACf,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,QAAQ,CAAC;gBACvD,WAAW,EAAE,CAAC;gBACd,UAAU,EAAE,MAAM;gBAElB,WAAW,EAAE,IAAI,CAAC,iBAAiB,CACjC,SAAS,EAAE,WAAW,EACtB,aAAa,CACd;gBACD,YAAY,EAAE,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,SAAS,EAAE,YAAY,CAAC;gBACpE,WAAW,EAAE,eAAe,CAAC,MAAM;gBACnC,eAAe,EAAE;oBACf,MAAM,EAAE,EAAE,eAAe,EAAE,uBAAuB,EAAE;oBACpD,MAAM,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE;iBACzB;aACF,CAAC,CAAC;YAEH,MAAM,WAAW,GAAG,CAAC,GAAG,QAAQ,EAAE,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,QAAQ,IAAI,EAAE,CAAC,CAAC,CAAC;YACxE,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,UAAU,CACtC,KAAK,EACL,YAAY,EACZ,WAAW,EACX,eAAe,CAAC,WAAW,EAC3B,eAAe,CAAC,MAAM,EACtB,IAAI,CAAC,MAAM,CACZ,CAAC;YAEF,OAAO,IAAI,CAAC,2BAA2B,CACrC,SAAS,EACT,KAAK,EACL,UAAU,CAAC,QAAQ,EACnB,MAAM,EACN,QAAQ,EACR,UAAU,CAAC,MAAM,CAClB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,iEAAiE;YACjE,IACE,KAAK,YAAY,yCAAyC;gBAC1D,KAAK,YAAY,+BAA+B,EAChD,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC;YAED,+EAA+E;YAC/E,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;gBACpB,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBACjE,MAAM,IAAI,eAAe,CAAC,MAAM,CAAC,CAAC;YACpC,CAAC;YAED,MAAM,YAAY,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;YAC5C,IAAI,CAAC,MAAM,CAAC;gBACV,QAAQ,EAAE,OAAO;gBACjB,OAAO,EAAE,+BAA+B,YAAY,EAAE;gBACtD,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;YAEH,oEAAoE;YACpE,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,OAAO,EAAE,2BAA2B,YAAY,EAAE;gBAClD,SAAS,EAAE,KAAK;gBAChB,QAAQ;aACT,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,aAAa,EAAE,OAAO,EAAE,CAAC;QAC3B,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,MAAM,CACjB,oBAAwD;QAExD,MAAM,aAAa,GACjB,OAAO,oBAAoB,KAAK,QAAQ,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC;QAEzE,uEAAuE;QACvE,MAAM,qBAAqB,GACzB,aAAa,EAAE,eAAe,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC;QAE3D,MAAM,EACJ,OAAO,EACP,QAAQ,EACR,YAAY,EACZ,QAAQ,EACR,QAAQ,EACR,YAAY,EACZ,cAAc,GACf,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,oBAAoB,CAAC,CAAC;QAElD,wEAAwE;QACxE,IAAI,qBAAqB,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACpD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC;YACrD,MAAM,IAAI,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACnD,CAAC;QAED,qDAAqD;QACrD,IAAI,aAAwC,CAAC;QAC7C,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;YACjC,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;YACpC,aAAa,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC;QAC9D,CAAC;QAED,MAAM,SAAS,GAAI,oBAAkD;aAClE,SAA6C,CAAC;QAEjD,MAAM,KAAK,GAAe;YACxB,kBAAkB,EAAE,EAAE;YACtB,OAAO,EAAE,EAAE;YACX,YAAY,EAAE,EAAE;YAChB,SAAS,EAAE,KAAK;YAChB,cAAc,EAAE,cAAc;SAC/B,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,aAAsE,CAAC;QAC3E,IAAI,YAAuC,CAAC;QAC5C,MAAM,aAAa,GAAG,IAAI,OAAO,CAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACjE,aAAa,GAAG,OAAO,CAAC;YACxB,YAAY,GAAG,MAAM,CAAC;QACxB,CAAC,CAAC,CAAC;QAEH,MAAM,WAAW,GAAG,CAAC,KAAc,EAAE,EAAE;YACrC,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACzD,IAAI,CAAC,MAAM,CAAC;gBACV,QAAQ,EAAE,OAAO;gBACjB,OAAO,EAAE,2BAA2B,YAAY,EAAE;gBAClD,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;YACH,YAAY,CAAC,KAAK,CAAC,CAAC;QACtB,CAAC,CAAC;QAEF,IAAI,YAA0D,CAAC;QAC/D,IAAI,CAAC;YACH,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC;gBACvC,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,oBAAoB,CAAC,YAAY,EAAE,QAAQ,CAAC;gBACtD,KAAK,EAAE,QAAQ;gBACf,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,QAAQ,CAAC;gBACvD,WAAW,EAAE,CAAC;gBACd,UAAU,EAAE,MAAM;gBAClB,WAAW,EAAE,IAAI,CAAC,iBAAiB,CACjC,SAAS,EAAE,WAAW,EACtB,aAAa,CACd;gBACD,YAAY,EAAE,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,SAAS,EAAE,YAAY,CAAC;gBACpE,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;oBACjB,aAAa,EAAE,OAAO,EAAE,CAAC;oBACzB,IAAI,SAAS,EAAE,OAAO,EAAE,CAAC;wBACvB,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;oBAC3B,CAAC;oBACD,WAAW,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBAC3B,CAAC;gBACD,OAAO,EAAE,SAAS,EAAE,OAAO;gBAC3B,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE;oBAClB,aAAa,EAAE,OAAO,EAAE,CAAC;oBACzB,IAAI,SAAS,EAAE,QAAQ,EAAE,CAAC;wBACxB,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;oBAC5B,CAAC;oBAED,MAAM,WAAW,GAAG;wBAClB,GAAG,QAAQ;wBACX,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,QAAQ,IAAI,EAAE,CAAC;qBACpC,CAAC;oBACF,IAAI,CAAC,UAAU,CACb,KAAK,EACL,YAAY,EACZ,WAAW,EACX,OAAO,CAAC,WAAW,EACnB,OAAO,CAAC,MAAM,EACd,IAAI,CAAC,MAAM,CACZ,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE,EAAE;wBACpB,MAAM,MAAM,GAAG,IAAI,CAAC,2BAA2B,CAC7C,SAAS,EACT,KAAK,EACL,UAAU,CAAC,QAAQ,EACnB,KAAK,EACL,QAAQ,EACR,UAAU,CAAC,MAAM,CAClB,CAAC;wBACF,aAAa,CAAC,MAAM,CAAC,CAAC;oBACxB,CAAC,CAAC,CAAC;gBACL,CAAC;gBACD,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;oBACjB,aAAa,EAAE,OAAO,EAAE,CAAC;oBACzB,IAAI,SAAS,EAAE,OAAO,EAAE,CAAC;wBACvB,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;oBAC3B,CAAC;oBACD,wEAAwE;oBACxE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,MAAM;wBACnC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC;wBAC/B,CAAC,CAAC,oBAAoB,CAAC;oBACzB,YAAY,CAAC,IAAI,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC;gBAC5C,CAAC;gBACD,WAAW,EAAE,OAAO,CAAC,MAAM;gBAC3B,eAAe,EAAE;oBACf,MAAM,EAAE,EAAE,eAAe,EAAE,uBAAuB,EAAE;oBACpD,MAAM,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE;iBACzB;aACF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,aAAa,EAAE,OAAO,EAAE,CAAC;YACzB,MAAM,KAAK,CAAC;QACd,CAAC;QAED,MAAM,iBAAiB,GAAG,YAAiC,CAAC;QAC5D,iBAAiB,CAAC,MAAM,GAAG,aAAa,CAAC;QACzC,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAEO,2BAA2B,CACjC,SAAiB,EACjB,KAAiB,EACjB,aAA6B,EAC7B,MAKC,EACD,QAAiB,EACjB,MAAgC;QAEhC,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;YACxB,MAAM,YAAY,GAAG,KAAK,CAAC,kBAAkB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;YAE/D,IAAI,CAAC,KAAK,CAAC,SAAS,IAAI,QAAQ,IAAI,MAAM,CAAC,KAAK,EAAE,MAAM,IAAI,QAAQ,EAAE,CAAC;gBACrE,IAAI,CAAC,MAAM,CAAC;oBACV,QAAQ,EAAE,OAAO;oBACjB,OAAO,EAAE,yCAAyC,QAAQ,GAAG;oBAC7D,KAAK,EAAE,CAAC;iBACT,CAAC,CAAC;gBACH,KAAK,CAAC,YAAY,GAAG,yCAAyC,QAAQ,GAAG,CAAC;YAC5E,CAAC;iBAAM,CAAC;gBACN,KAAK,CAAC,YAAY,GAAG,YAAY,IAAI,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;YACzD,CAAC;QACH,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC3B,MAAM,eAAe,GAAG,OAAO,GAAG,SAAS,CAAC;QAC5C,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;YACtB,IAAI,CAAC,EAAE,CAAC,aAAa,CACnB,cAAc,CAAC,KAAK,EACpB,MAAM,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC,EAClC,MAAM,CAAC,UAAU,CAAC,YAAY,IAAI,CAAC,EACnC,MAAM,CAAC,UAAU,CAAC,eAAe,IAAI,CAAC,EACtC,MAAM,CAAC,UAAU,CAAC,iBAAiB,IAAI,CAAC,EACxC,eAAe,CAChB,CAAC;QACJ,CAAC;QAED,OAAO;YACL,OAAO,EAAE,KAAK,CAAC,SAAS;YACxB,OAAO,EAAE,KAAK,CAAC,YAAY,IAAI,0BAA0B;YACzD,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,MAAM;YACN,KAAK,EAAE,MAAM,CAAC,UAAU;gBACtB,CAAC,CAAC;oBACE,YAAY,EAAE,MAAM,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC;oBAChD,aAAa,EAAE,MAAM,CAAC,UAAU,CAAC,YAAY,IAAI,CAAC;oBAClD,gBAAgB,EAAE,MAAM,CAAC,UAAU,CAAC,eAAe,IAAI,CAAC;oBACxD,mBAAmB,EAAE,MAAM,CAAC,UAAU,CAAC,iBAAiB,IAAI,CAAC;oBAC7D,iBAAiB,EAAE,eAAe;iBACnC;gBACH,CAAC,CAAC,SAAS;YACb,QAAQ,EAAE,aAAa;SACxB,CAAC;IACJ,CAAC;IAEO,WAAW,CACjB,YAAuB,EACvB,SAAqB,EACrB,WAAoB,EACpB,SAAmB;QAEnB,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,EAAE,gBAAgB,EAAE,EAAE,EAAE,QAAQ,CAAC;QAChE,OAAO,gBAAgB,CAAC,IAAI,CAAC,EAAE,EAAE;YAC/B,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,QAAQ;YACR,YAAY;YACZ,SAAS;YACT,WAAW;YACX,SAAS;YACT,iBAAiB,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC,SAAS;SACrE,CAAC,CAAC;IACL,CAAC;IAEO,UAAU,CAChB,MAAqD,EACrD,QAAgB;QAEhB,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACvD,IAAI,QAAQ,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,QAAQ,KAAK,MAAM,CAAC,EAAE,CAAC;YAC9D,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,WAAW,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,CAAC;IACvC,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,UAAU,CACtB,KAAiB,EACjB,KAAoB,EACpB,QAAwB,EACxB,WAAmB,EACnB,YAAiC,EACjC,MAAmC;QAEnC,IAAI,KAAK,CAAC,SAAS;YAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;QAEzC,MAAM,UAAU,GAAG,MAAM,kBAAkB,CAAC;YAC1C,KAAK;YACL,aAAa,EAAE,QAAQ;YACvB,WAAW;YACX,YAAY;YACZ,MAAM;SACP,CAAC,CAAC;QAEH,KAAK,CAAC,SAAS,GAAG,UAAU,CAAC,YAAY,CAAC;QAC1C,KAAK,CAAC,YAAY,GAAG,UAAU,CAAC,SAAS,CAAC;QAE1C,MAAM,UAAU,GAAG,sBAAsB,CAAC;YACxC,YAAY,EAAE,MAAM;YACpB,UAAU,EAAE;gBACV,OAAO,EAAE,IAAI;gBACb,SAAS,EAAE,UAAU,CAAC,SAAS;gBAC/B,YAAY,EAAE,UAAU,CAAC,YAAY;aACtC;YACD,IAAI,EAAE;gBACJ,SAAS,EAAE,UAAU,CAAC,SAAS;gBAC/B,YAAY,EAAE,UAAU,CAAC,YAAY;aACtC;YACD,SAAS,EAAE,UAAU,CAAC,SAAS;SAChC,CAAC,CAAC;QAEH,KAAK,MAAM,MAAM,IAAI,UAAU,EAAE,CAAC;YAChC,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC,cAAc,CAAC;YACtC,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC9B,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7B,CAAC;QAED,OAAO;YACL,QAAQ,EAAE,CAAC,GAAG,QAAQ,EAAE,GAAG,UAAU,CAAC,QAAQ,CAAC;YAC/C,MAAM,EAAE,UAAU,CAAC,MAAM;SAC1B,CAAC;IACJ,CAAC;CACF","sourcesContent":["import { createAgentTools } from \"../agent/tools/index.js\";\nimport { buildAgentSystemPrompt } from \"../agent/prompts/agentSystemPrompt.js\";\nimport { LogLine } from \"../types/public/logs.js\";\nimport { V3 } from \"../v3.js\";\nimport {\n  ModelMessage,\n  ToolSet,\n  wrapLanguageModel,\n  stepCountIs,\n  LanguageModel,\n  type LanguageModelUsage,\n  type StepResult,\n  type GenerateTextOnStepFinishCallback,\n  type StreamTextOnStepFinishCallback,\n  type PrepareStepFunction,\n} from \"ai\";\nimport { StagehandZodObject } from \"../zodCompat.js\";\nimport { processMessages } from \"../agent/utils/messageProcessing.js\";\nimport { LLMClient } from \"../llm/LLMClient.js\";\nimport { FlowLogger } from \"../flowlogger/FlowLogger.js\";\nimport {\n  AgentExecuteOptions,\n  AgentStreamExecuteOptions,\n  AgentExecuteOptionsBase,\n  AgentResult,\n  AgentContext,\n  AgentState,\n  AgentStreamResult,\n  AgentStreamCallbacks,\n  AgentToolMode,\n  AgentModelConfig,\n  Variables,\n} from \"../types/public/agent.js\";\nimport { V3FunctionName } from \"../types/public/methods.js\";\nimport { mapToolResultToActions } from \"../agent/utils/actionMapping.js\";\nimport {\n  MissingLLMConfigurationError,\n  MissingEnvironmentVariableError,\n  StreamingCallbacksInNonStreamingModeError,\n  AgentAbortError,\n} from \"../types/public/sdkErrors.js\";\nimport { handleDoneToolCall } from \"../agent/utils/handleDoneToolCall.js\";\nimport {\n  CaptchaSolver,\n  CAPTCHA_SOLVED_MSG,\n  CAPTCHA_ERRORED_MSG,\n} from \"../agent/utils/captchaSolver.js\";\n\nfunction getErrorMessage(error: unknown): string {\n  return error instanceof Error ? error.message : String(error);\n}\n\n/**\n * Prepends a system message with cache control to the messages array.\n * The cache control providerOptions are used by Anthropic and ignored by other providers.\n */\nfunction prependSystemMessage(\n  systemPrompt: string,\n  messages: ModelMessage[],\n): ModelMessage[] {\n  return [\n    {\n      role: \"system\",\n      content: systemPrompt,\n      providerOptions: {\n        anthropic: {\n          cacheControl: { type: \"ephemeral\" },\n        },\n      },\n    },\n    ...messages,\n  ];\n}\n\nexport class V3AgentHandler {\n  private v3: V3;\n  private logger: (message: LogLine) => void;\n  private llmClient: LLMClient;\n  private executionModel?: string | AgentModelConfig;\n  private systemInstructions?: string;\n  private mcpTools?: ToolSet;\n  private mode: AgentToolMode;\n  private captchaAutoSolveEnabled: boolean;\n\n  constructor(\n    v3: V3,\n    logger: (message: LogLine) => void,\n    llmClient: LLMClient,\n    executionModel?: string | AgentModelConfig,\n    systemInstructions?: string,\n    mcpTools?: ToolSet,\n    mode?: AgentToolMode,\n    captchaAutoSolveEnabled?: boolean,\n  ) {\n    this.v3 = v3;\n    this.logger = logger;\n    this.llmClient = llmClient;\n    this.executionModel = executionModel;\n    this.systemInstructions = systemInstructions;\n    this.mcpTools = mcpTools;\n    this.mode = mode ?? \"dom\";\n    this.captchaAutoSolveEnabled = captchaAutoSolveEnabled ?? false;\n  }\n\n  private async prepareAgent(\n    instructionOrOptions: string | AgentExecuteOptionsBase,\n  ): Promise<AgentContext> {\n    try {\n      const options =\n        typeof instructionOrOptions === \"string\"\n          ? { instruction: instructionOrOptions }\n          : instructionOrOptions;\n\n      const maxSteps = options.maxSteps || 20;\n\n      // Get the initial page URL first (needed for the system prompt)\n      const initialPageUrl = (await this.v3.context.awaitActivePage()).url();\n\n      // Build the system prompt with mode-aware tool guidance\n      const systemPrompt = buildAgentSystemPrompt({\n        url: initialPageUrl,\n        executionInstruction: options.instruction,\n        mode: this.mode,\n        systemInstructions: this.systemInstructions,\n        captchasAutoSolve: this.v3.isCaptchaAutoSolveEnabled,\n        excludeTools: options.excludeTools,\n        variables: options.variables,\n        useSearch: options.useSearch,\n      });\n\n      if (options.useSearch) {\n        const bbApiKey = this.v3.browserbaseApiKey;\n        if (!bbApiKey) {\n          throw new MissingEnvironmentVariableError(\n            \"BROWSERBASE_API_KEY\",\n            \"agent search (useSearch: true)\",\n          );\n        }\n      }\n\n      const tools = this.createTools(\n        options.excludeTools,\n        options.variables,\n        options.toolTimeout,\n        options.useSearch,\n      );\n      const allTools: ToolSet = { ...tools, ...this.mcpTools };\n\n      // Use provided messages for continuation, or start fresh with the instruction\n      const messages: ModelMessage[] = options.messages?.length\n        ? [...options.messages, { role: \"user\", content: options.instruction }]\n        : [{ role: \"user\", content: options.instruction }];\n\n      if (!this.llmClient?.getLanguageModel) {\n        throw new MissingLLMConfigurationError();\n      }\n      const baseModel = this.llmClient.getLanguageModel();\n      //to do - we likely do not need middleware anymore\n      const wrappedModel = wrapLanguageModel({\n        model: baseModel,\n        middleware: {\n          ...FlowLogger.createLlmLoggingMiddleware(baseModel.modelId),\n        },\n      });\n\n      if (\n        this.mode === \"hybrid\" &&\n        !baseModel.modelId.includes(\"gemini-3-flash\") &&\n        !baseModel.modelId.includes(\"claude\")\n      ) {\n        this.logger({\n          category: \"agent\",\n          message: `Warning: \"${baseModel.modelId}\" may not perform well in hybrid mode. See recommended models: https://docs.stagehand.dev/v3/basics/agent#hybrid-mode`,\n          level: 0,\n        });\n      }\n\n      return {\n        options,\n        maxSteps,\n        systemPrompt,\n        allTools,\n        messages,\n        wrappedModel,\n        initialPageUrl,\n      };\n    } catch (error) {\n      this.logger({\n        category: \"agent\",\n        message: `failed to prepare agent: ${error}`,\n        level: 0,\n      });\n      throw error;\n    }\n  }\n  private createPrepareStep(\n    userCallback?: PrepareStepFunction<ToolSet>,\n    captchaSolver?: CaptchaSolver,\n  ): PrepareStepFunction<ToolSet> {\n    return async (options) => {\n      processMessages(options.messages);\n      if (captchaSolver) {\n        if (captchaSolver.isSolving()) {\n          this.logger({\n            category: \"agent\",\n            message:\n              \"Captcha detected — waiting for Browserbase to solve it before continuing\",\n            level: 1,\n          });\n        }\n        await captchaSolver.waitIfSolving();\n        const { solved, errored } = captchaSolver.consumeSolveResult();\n        if (solved) {\n          options.messages.push({\n            role: \"user\",\n            content: CAPTCHA_SOLVED_MSG,\n          });\n          this.logger({\n            category: \"agent\",\n            message:\n              \"Captcha solved — injected notification into agent message stream\",\n            level: 1,\n          });\n        }\n        if (errored) {\n          options.messages.push({\n            role: \"user\",\n            content: CAPTCHA_ERRORED_MSG,\n          });\n          this.logger({\n            category: \"agent\",\n            message:\n              \"Captcha solver failed — injected error notification into agent message stream\",\n            level: 1,\n          });\n        }\n      }\n      if (userCallback) {\n        return userCallback(options);\n      }\n      return options;\n    };\n  }\n\n  private createStepHandler(\n    state: AgentState,\n    userCallback?:\n      | GenerateTextOnStepFinishCallback<ToolSet>\n      | StreamTextOnStepFinishCallback<ToolSet>,\n  ) {\n    return async (event: StepResult<ToolSet>) => {\n      this.logger({\n        category: \"agent\",\n        message: `Step finished: ${event.finishReason}`,\n        level: 2,\n      });\n\n      if (event.toolCalls && event.toolCalls.length > 0) {\n        for (let i = 0; i < event.toolCalls.length; i++) {\n          const toolCall = event.toolCalls[i];\n          const args = toolCall.input;\n          const toolResult = event.toolResults?.[i];\n\n          if (event.text && event.text.length > 0) {\n            state.collectedReasoning.push(event.text);\n            this.logger({\n              category: \"agent\",\n              message: `reasoning: ${event.text}`,\n              level: 1,\n            });\n          }\n\n          if (toolCall.toolName === \"done\") {\n            state.completed = true;\n            if (args?.taskComplete) {\n              const doneReasoning = args.reasoning;\n              const allReasoning = state.collectedReasoning.join(\" \");\n              state.finalMessage = doneReasoning\n                ? `${allReasoning} ${doneReasoning}`.trim()\n                : allReasoning || \"Task completed successfully\";\n            }\n          }\n          const mappedActions = mapToolResultToActions({\n            toolCallName: toolCall.toolName,\n            toolResult,\n            args,\n            reasoning: event.text || undefined,\n          });\n\n          for (const action of mappedActions) {\n            action.pageUrl = state.currentPageUrl;\n            action.timestamp = Date.now();\n            state.actions.push(action);\n          }\n        }\n        state.currentPageUrl = (await this.v3.context.awaitActivePage()).url();\n      }\n\n      if (userCallback) {\n        await userCallback(event);\n      }\n    };\n  }\n\n  public async execute(\n    instructionOrOptions: string | AgentExecuteOptions,\n  ): Promise<AgentResult> {\n    const startTime = Date.now();\n    const options =\n      typeof instructionOrOptions === \"object\" ? instructionOrOptions : null;\n    const signal = options?.signal;\n\n    // Highlight cursor defaults to true for hybrid mode, can be overridden\n    const shouldHighlightCursor =\n      options?.highlightCursor ?? this.mode === \"hybrid\";\n\n    const state: AgentState = {\n      collectedReasoning: [],\n      actions: [],\n      finalMessage: \"\",\n      completed: false,\n      currentPageUrl: \"\",\n    };\n\n    let messages: ModelMessage[] = [];\n    let captchaSolver: CaptchaSolver | undefined;\n\n    try {\n      const {\n        options: preparedOptions,\n        maxSteps,\n        systemPrompt,\n        allTools,\n        messages: preparedMessages,\n        wrappedModel,\n        initialPageUrl,\n      } = await this.prepareAgent(instructionOrOptions);\n\n      // Enable cursor overlay for hybrid mode (coordinate-based interactions)\n      if (shouldHighlightCursor && this.mode === \"hybrid\") {\n        const page = await this.v3.context.awaitActivePage();\n        await page.enableCursorOverlay().catch(() => {});\n      }\n\n      // Set up captcha solver for Browserbase environments\n      if (this.captchaAutoSolveEnabled) {\n        captchaSolver = new CaptchaSolver();\n        captchaSolver.init(() => this.v3.context.awaitActivePage());\n      }\n\n      messages = preparedMessages;\n      state.currentPageUrl = initialPageUrl;\n\n      const callbacks = (instructionOrOptions as AgentExecuteOptions).callbacks;\n\n      if (callbacks) {\n        const streamingOnlyCallbacks = [\n          \"onChunk\",\n          \"onFinish\",\n          \"onError\",\n          \"onAbort\",\n        ];\n        const invalidCallbacks = streamingOnlyCallbacks.filter(\n          (name) => callbacks[name as keyof typeof callbacks] != null,\n        );\n        if (invalidCallbacks.length > 0) {\n          throw new StreamingCallbacksInNonStreamingModeError(invalidCallbacks);\n        }\n      }\n\n      const result = await this.llmClient.generateText({\n        model: wrappedModel,\n        messages: prependSystemMessage(systemPrompt, messages),\n        tools: allTools,\n        stopWhen: (result) => this.handleStop(result, maxSteps),\n        temperature: 1,\n        toolChoice: \"auto\",\n\n        prepareStep: this.createPrepareStep(\n          callbacks?.prepareStep,\n          captchaSolver,\n        ),\n        onStepFinish: this.createStepHandler(state, callbacks?.onStepFinish),\n        abortSignal: preparedOptions.signal,\n        providerOptions: {\n          google: { mediaResolution: \"MEDIA_RESOLUTION_HIGH\" },\n          openai: { store: false },\n        },\n      });\n\n      const allMessages = [...messages, ...(result.response?.messages || [])];\n      const doneResult = await this.ensureDone(\n        state,\n        wrappedModel,\n        allMessages,\n        preparedOptions.instruction,\n        preparedOptions.output,\n        this.logger,\n      );\n\n      return this.consolidateMetricsAndResult(\n        startTime,\n        state,\n        doneResult.messages,\n        result,\n        maxSteps,\n        doneResult.output,\n      );\n    } catch (error) {\n      // Re-throw validation errors that should propagate to the caller\n      if (\n        error instanceof StreamingCallbacksInNonStreamingModeError ||\n        error instanceof MissingEnvironmentVariableError\n      ) {\n        throw error;\n      }\n\n      // Re-throw abort errors wrapped in AgentAbortError for consistent error typing\n      if (signal?.aborted) {\n        const reason = signal.reason ? String(signal.reason) : \"aborted\";\n        throw new AgentAbortError(reason);\n      }\n\n      const errorMessage = getErrorMessage(error);\n      this.logger({\n        category: \"agent\",\n        message: `Error executing agent task: ${errorMessage}`,\n        level: 0,\n      });\n\n      // For non-abort errors, return a failure result instead of throwing\n      return {\n        success: false,\n        actions: state.actions,\n        message: `Failed to execute task: ${errorMessage}`,\n        completed: false,\n        messages,\n      };\n    } finally {\n      captchaSolver?.dispose();\n    }\n  }\n\n  public async stream(\n    instructionOrOptions: string | AgentStreamExecuteOptions,\n  ): Promise<AgentStreamResult> {\n    const streamOptions =\n      typeof instructionOrOptions === \"object\" ? instructionOrOptions : null;\n\n    // Highlight cursor defaults to true for hybrid mode, can be overridden\n    const shouldHighlightCursor =\n      streamOptions?.highlightCursor ?? this.mode === \"hybrid\";\n\n    const {\n      options,\n      maxSteps,\n      systemPrompt,\n      allTools,\n      messages,\n      wrappedModel,\n      initialPageUrl,\n    } = await this.prepareAgent(instructionOrOptions);\n\n    // Enable cursor overlay for hybrid mode (coordinate-based interactions)\n    if (shouldHighlightCursor && this.mode === \"hybrid\") {\n      const page = await this.v3.context.awaitActivePage();\n      await page.enableCursorOverlay().catch(() => {});\n    }\n\n    // Set up captcha solver for Browserbase environments\n    let captchaSolver: CaptchaSolver | undefined;\n    if (this.captchaAutoSolveEnabled) {\n      captchaSolver = new CaptchaSolver();\n      captchaSolver.init(() => this.v3.context.awaitActivePage());\n    }\n\n    const callbacks = (instructionOrOptions as AgentStreamExecuteOptions)\n      .callbacks as AgentStreamCallbacks | undefined;\n\n    const state: AgentState = {\n      collectedReasoning: [],\n      actions: [],\n      finalMessage: \"\",\n      completed: false,\n      currentPageUrl: initialPageUrl,\n    };\n    const startTime = Date.now();\n\n    let resolveResult: (value: AgentResult | PromiseLike<AgentResult>) => void;\n    let rejectResult: (reason: unknown) => void;\n    const resultPromise = new Promise<AgentResult>((resolve, reject) => {\n      resolveResult = resolve;\n      rejectResult = reject;\n    });\n\n    const handleError = (error: unknown) => {\n      const errorMessage =\n        error instanceof Error ? error.message : String(error);\n      this.logger({\n        category: \"agent\",\n        message: `Error during streaming: ${errorMessage}`,\n        level: 0,\n      });\n      rejectResult(error);\n    };\n\n    let streamResult: ReturnType<typeof this.llmClient.streamText>;\n    try {\n      streamResult = this.llmClient.streamText({\n        model: wrappedModel,\n        messages: prependSystemMessage(systemPrompt, messages),\n        tools: allTools,\n        stopWhen: (result) => this.handleStop(result, maxSteps),\n        temperature: 1,\n        toolChoice: \"auto\",\n        prepareStep: this.createPrepareStep(\n          callbacks?.prepareStep,\n          captchaSolver,\n        ),\n        onStepFinish: this.createStepHandler(state, callbacks?.onStepFinish),\n        onError: (event) => {\n          captchaSolver?.dispose();\n          if (callbacks?.onError) {\n            callbacks.onError(event);\n          }\n          handleError(event.error);\n        },\n        onChunk: callbacks?.onChunk,\n        onFinish: (event) => {\n          captchaSolver?.dispose();\n          if (callbacks?.onFinish) {\n            callbacks.onFinish(event);\n          }\n\n          const allMessages = [\n            ...messages,\n            ...(event.response?.messages || []),\n          ];\n          this.ensureDone(\n            state,\n            wrappedModel,\n            allMessages,\n            options.instruction,\n            options.output,\n            this.logger,\n          ).then((doneResult) => {\n            const result = this.consolidateMetricsAndResult(\n              startTime,\n              state,\n              doneResult.messages,\n              event,\n              maxSteps,\n              doneResult.output,\n            );\n            resolveResult(result);\n          });\n        },\n        onAbort: (event) => {\n          captchaSolver?.dispose();\n          if (callbacks?.onAbort) {\n            callbacks.onAbort(event);\n          }\n          // Reject the result promise with AgentAbortError when stream is aborted\n          const reason = options.signal?.reason\n            ? String(options.signal.reason)\n            : \"Stream was aborted\";\n          rejectResult(new AgentAbortError(reason));\n        },\n        abortSignal: options.signal,\n        providerOptions: {\n          google: { mediaResolution: \"MEDIA_RESOLUTION_HIGH\" },\n          openai: { store: false },\n        },\n      });\n    } catch (error) {\n      captchaSolver?.dispose();\n      throw error;\n    }\n\n    const agentStreamResult = streamResult as AgentStreamResult;\n    agentStreamResult.result = resultPromise;\n    return agentStreamResult;\n  }\n\n  private consolidateMetricsAndResult(\n    startTime: number,\n    state: AgentState,\n    inputMessages: ModelMessage[],\n    result: {\n      text?: string;\n      totalUsage?: LanguageModelUsage;\n      response?: { messages?: ModelMessage[] };\n      steps?: StepResult<ToolSet>[];\n    },\n    maxSteps?: number,\n    output?: Record<string, unknown>,\n  ): AgentResult {\n    if (!state.finalMessage) {\n      const allReasoning = state.collectedReasoning.join(\" \").trim();\n\n      if (!state.completed && maxSteps && result.steps?.length >= maxSteps) {\n        this.logger({\n          category: \"agent\",\n          message: `Agent stopped: reached maximum steps (${maxSteps})`,\n          level: 1,\n        });\n        state.finalMessage = `Agent stopped: reached maximum steps (${maxSteps})`;\n      } else {\n        state.finalMessage = allReasoning || result.text || \"\";\n      }\n    }\n\n    const endTime = Date.now();\n    const inferenceTimeMs = endTime - startTime;\n    if (result.totalUsage) {\n      this.v3.updateMetrics(\n        V3FunctionName.AGENT,\n        result.totalUsage.inputTokens || 0,\n        result.totalUsage.outputTokens || 0,\n        result.totalUsage.reasoningTokens || 0,\n        result.totalUsage.cachedInputTokens || 0,\n        inferenceTimeMs,\n      );\n    }\n\n    return {\n      success: state.completed,\n      message: state.finalMessage || \"Task execution completed\",\n      actions: state.actions,\n      completed: state.completed,\n      output,\n      usage: result.totalUsage\n        ? {\n            input_tokens: result.totalUsage.inputTokens || 0,\n            output_tokens: result.totalUsage.outputTokens || 0,\n            reasoning_tokens: result.totalUsage.reasoningTokens || 0,\n            cached_input_tokens: result.totalUsage.cachedInputTokens || 0,\n            inference_time_ms: inferenceTimeMs,\n          }\n        : undefined,\n      messages: inputMessages,\n    };\n  }\n\n  private createTools(\n    excludeTools?: string[],\n    variables?: Variables,\n    toolTimeout?: number,\n    useSearch?: boolean,\n  ) {\n    const provider = this.llmClient?.getLanguageModel?.()?.provider;\n    return createAgentTools(this.v3, {\n      executionModel: this.executionModel,\n      logger: this.logger,\n      mode: this.mode,\n      provider,\n      excludeTools,\n      variables,\n      toolTimeout,\n      useSearch,\n      browserbaseApiKey: useSearch ? this.v3.browserbaseApiKey : undefined,\n    });\n  }\n\n  private handleStop(\n    result: Parameters<ReturnType<typeof stepCountIs>>[0],\n    maxSteps: number,\n  ): boolean | PromiseLike<boolean> {\n    const lastStep = result.steps[result.steps.length - 1];\n    if (lastStep?.toolCalls?.some((tc) => tc.toolName === \"done\")) {\n      return true;\n    }\n    return stepCountIs(maxSteps)(result);\n  }\n\n  /**\n   * Ensures the done tool is called at the end of agent execution.\n   * Returns the messages and any extracted output from the done call.\n   */\n  private async ensureDone(\n    state: AgentState,\n    model: LanguageModel,\n    messages: ModelMessage[],\n    instruction: string,\n    outputSchema?: StagehandZodObject,\n    logger?: (message: LogLine) => void,\n  ): Promise<{ messages: ModelMessage[]; output?: Record<string, unknown> }> {\n    if (state.completed) return { messages };\n\n    const doneResult = await handleDoneToolCall({\n      model,\n      inputMessages: messages,\n      instruction,\n      outputSchema,\n      logger,\n    });\n\n    state.completed = doneResult.taskComplete;\n    state.finalMessage = doneResult.reasoning;\n\n    const doneAction = mapToolResultToActions({\n      toolCallName: \"done\",\n      toolResult: {\n        success: true,\n        reasoning: doneResult.reasoning,\n        taskComplete: doneResult.taskComplete,\n      },\n      args: {\n        reasoning: doneResult.reasoning,\n        taskComplete: doneResult.taskComplete,\n      },\n      reasoning: doneResult.reasoning,\n    });\n\n    for (const action of doneAction) {\n      action.pageUrl = state.currentPageUrl;\n      action.timestamp = Date.now();\n      state.actions.push(action);\n    }\n\n    return {\n      messages: [...messages, ...doneResult.messages],\n      output: doneResult.output,\n    };\n  }\n}\n"]}
+{"version":3,"file":"v3AgentHandler.js","sourceRoot":"","sources":["../../../../../lib/v3/handlers/v3AgentHandler.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,EAAE,sBAAsB,EAAE,MAAM,uCAAuC,CAAC;AAG/E,OAAO,EAGL,iBAAiB,EACjB,WAAW,GAOZ,MAAM,IAAI,CAAC;AAEZ,OAAO,EAAE,eAAe,EAAE,MAAM,qCAAqC,CAAC;AAEtE,OAAO,EAAE,UAAU,EAAE,MAAM,6BAA6B,CAAC;AAczD,OAAO,EAAE,6BAA6B,EAAE,MAAM,2BAA2B,CAAC;AAC1E,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,sBAAsB,EAAE,MAAM,iCAAiC,CAAC;AACzE,OAAO,EACL,4BAA4B,EAC5B,+BAA+B,EAC/B,yCAAyC,EACzC,eAAe,GAChB,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,kBAAkB,EAAE,MAAM,sCAAsC,CAAC;AAC1E,OAAO,EACL,aAAa,EACb,kBAAkB,EAClB,mBAAmB,GACpB,MAAM,iCAAiC,CAAC;AAEzC,SAAS,eAAe,CAAC,KAAc;IACrC,OAAO,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAChE,CAAC;AAED;;;GAGG;AACH,SAAS,oBAAoB,CAC3B,YAAoB,EACpB,QAAwB;IAExB,OAAO;QACL;YACE,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,YAAY;YACrB,eAAe,EAAE;gBACf,SAAS,EAAE;oBACT,YAAY,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;iBACpC;aACF;SACF;QACD,GAAG,QAAQ;KACZ,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,cAAc;IACjB,EAAE,CAAK;IACP,MAAM,CAA6B;IACnC,SAAS,CAAY;IACrB,cAAc,CAA6B;IAC3C,kBAAkB,CAAU;IAC5B,QAAQ,CAAW;IACnB,IAAI,CAAgB;IACpB,uBAAuB,CAAU;IAEzC,YACE,EAAM,EACN,MAAkC,EAClC,SAAoB,EACpB,cAA0C,EAC1C,kBAA2B,EAC3B,QAAkB,EAClB,IAAoB,EACpB,uBAAiC;QAEjC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;QAC7C,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,IAAI,GAAG,IAAI,IAAI,KAAK,CAAC;QAC1B,IAAI,CAAC,uBAAuB,GAAG,uBAAuB,IAAI,KAAK,CAAC;IAClE,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,oBAAsD;QAEtD,IAAI,CAAC;YACH,MAAM,OAAO,GACX,OAAO,oBAAoB,KAAK,QAAQ;gBACtC,CAAC,CAAC,EAAE,WAAW,EAAE,oBAAoB,EAAE;gBACvC,CAAC,CAAC,oBAAoB,CAAC;YAE3B,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC;YAExC,gEAAgE;YAChE,MAAM,cAAc,GAAG,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC;YAEvE,wDAAwD;YACxD,MAAM,YAAY,GAAG,sBAAsB,CAAC;gBAC1C,GAAG,EAAE,cAAc;gBACnB,oBAAoB,EAAE,OAAO,CAAC,WAAW;gBACzC,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,kBAAkB,EAAE,IAAI,CAAC,kBAAkB;gBAC3C,iBAAiB,EAAE,IAAI,CAAC,EAAE,CAAC,yBAAyB;gBACpD,YAAY,EAAE,OAAO,CAAC,YAAY;gBAClC,SAAS,EAAE,OAAO,CAAC,SAAS;gBAC5B,SAAS,EAAE,OAAO,CAAC,SAAS;aAC7B,CAAC,CAAC;YAEH,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;gBACtB,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,CAAC,iBAAiB,CAAC;gBAC3C,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACd,MAAM,IAAI,+BAA+B,CACvC,qBAAqB,EACrB,gCAAgC,CACjC,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAC5B,OAAO,CAAC,YAAY,EACpB,OAAO,CAAC,SAAS,EACjB,OAAO,CAAC,WAAW,EACnB,OAAO,CAAC,SAAS,CAClB,CAAC;YACF,MAAM,QAAQ,GAAY,EAAE,GAAG,KAAK,EAAE,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;YAEzD,8EAA8E;YAC9E,MAAM,QAAQ,GAAmB,OAAO,CAAC,QAAQ,EAAE,MAAM;gBACvD,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC;gBACvE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;YAErD,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,gBAAgB,EAAE,CAAC;gBACtC,MAAM,IAAI,4BAA4B,EAAE,CAAC;YAC3C,CAAC;YACD,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC;YACpD,kDAAkD;YAClD,MAAM,YAAY,GAAG,iBAAiB,CAAC;gBACrC,KAAK,EAAE,SAAS;gBAChB,UAAU,EAAE;oBACV,GAAG,UAAU,CAAC,0BAA0B,CAAC,SAAS,CAAC,OAAO,CAAC;iBAC5D;aACF,CAAC,CAAC;YAEH,MAAM,eAAe,GAAG,6BAA6B,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CACrE,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CACpC,CAAC;YACF,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,eAAe,EAAE,CAAC;gBAC/C,IAAI,CAAC,MAAM,CAAC;oBACV,QAAQ,EAAE,OAAO;oBACjB,OAAO,EAAE,aAAa,SAAS,CAAC,OAAO,uHAAuH;oBAC9J,KAAK,EAAE,CAAC;iBACT,CAAC,CAAC;YACL,CAAC;YAED,OAAO;gBACL,OAAO;gBACP,QAAQ;gBACR,YAAY;gBACZ,QAAQ;gBACR,QAAQ;gBACR,YAAY;gBACZ,cAAc;aACf,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC;gBACV,QAAQ,EAAE,OAAO;gBACjB,OAAO,EAAE,4BAA4B,KAAK,EAAE;gBAC5C,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IACO,iBAAiB,CACvB,YAA2C,EAC3C,aAA6B;QAE7B,OAAO,KAAK,EAAE,OAAO,EAAE,EAAE;YACvB,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAClC,IAAI,aAAa,EAAE,CAAC;gBAClB,IAAI,aAAa,CAAC,SAAS,EAAE,EAAE,CAAC;oBAC9B,IAAI,CAAC,MAAM,CAAC;wBACV,QAAQ,EAAE,OAAO;wBACjB,OAAO,EACL,0EAA0E;wBAC5E,KAAK,EAAE,CAAC;qBACT,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM,aAAa,CAAC,aAAa,EAAE,CAAC;gBACpC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,aAAa,CAAC,kBAAkB,EAAE,CAAC;gBAC/D,IAAI,MAAM,EAAE,CAAC;oBACX,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;wBACpB,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,kBAAkB;qBAC5B,CAAC,CAAC;oBACH,IAAI,CAAC,MAAM,CAAC;wBACV,QAAQ,EAAE,OAAO;wBACjB,OAAO,EACL,kEAAkE;wBACpE,KAAK,EAAE,CAAC;qBACT,CAAC,CAAC;gBACL,CAAC;gBACD,IAAI,OAAO,EAAE,CAAC;oBACZ,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;wBACpB,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,mBAAmB;qBAC7B,CAAC,CAAC;oBACH,IAAI,CAAC,MAAM,CAAC;wBACV,QAAQ,EAAE,OAAO;wBACjB,OAAO,EACL,+EAA+E;wBACjF,KAAK,EAAE,CAAC;qBACT,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YACD,IAAI,YAAY,EAAE,CAAC;gBACjB,OAAO,YAAY,CAAC,OAAO,CAAC,CAAC;YAC/B,CAAC;YACD,OAAO,OAAO,CAAC;QACjB,CAAC,CAAC;IACJ,CAAC;IAEO,iBAAiB,CACvB,KAAiB,EACjB,YAE2C;QAE3C,OAAO,KAAK,EAAE,KAA0B,EAAE,EAAE;YAC1C,IAAI,CAAC,MAAM,CAAC;gBACV,QAAQ,EAAE,OAAO;gBACjB,OAAO,EAAE,kBAAkB,KAAK,CAAC,YAAY,EAAE;gBAC/C,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;YAEH,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAClD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oBAChD,MAAM,QAAQ,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;oBACpC,MAAM,IAAI,GAAG,QAAQ,CAAC,KAAK,CAAC;oBAC5B,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC;oBAE1C,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACxC,KAAK,CAAC,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBAC1C,IAAI,CAAC,MAAM,CAAC;4BACV,QAAQ,EAAE,OAAO;4BACjB,OAAO,EAAE,cAAc,KAAK,CAAC,IAAI,EAAE;4BACnC,KAAK,EAAE,CAAC;yBACT,CAAC,CAAC;oBACL,CAAC;oBAED,IAAI,QAAQ,CAAC,QAAQ,KAAK,MAAM,EAAE,CAAC;wBACjC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC;wBACvB,IAAI,IAAI,EAAE,YAAY,EAAE,CAAC;4BACvB,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC;4BACrC,MAAM,YAAY,GAAG,KAAK,CAAC,kBAAkB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;4BACxD,KAAK,CAAC,YAAY,GAAG,aAAa;gCAChC,CAAC,CAAC,GAAG,YAAY,IAAI,aAAa,EAAE,CAAC,IAAI,EAAE;gCAC3C,CAAC,CAAC,YAAY,IAAI,6BAA6B,CAAC;wBACpD,CAAC;oBACH,CAAC;oBACD,MAAM,aAAa,GAAG,sBAAsB,CAAC;wBAC3C,YAAY,EAAE,QAAQ,CAAC,QAAQ;wBAC/B,UAAU;wBACV,IAAI;wBACJ,SAAS,EAAE,KAAK,CAAC,IAAI,IAAI,SAAS;qBACnC,CAAC,CAAC;oBAEH,KAAK,MAAM,MAAM,IAAI,aAAa,EAAE,CAAC;wBACnC,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC,cAAc,CAAC;wBACtC,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;wBAC9B,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;oBAC7B,CAAC;gBACH,CAAC;gBACD,KAAK,CAAC,cAAc,GAAG,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC;YACzE,CAAC;YAED,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,YAAY,CAAC,KAAK,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,OAAO,CAClB,oBAAkD;QAElD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,OAAO,GACX,OAAO,oBAAoB,KAAK,QAAQ,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC;QACzE,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,CAAC;QAE/B,uEAAuE;QACvE,MAAM,qBAAqB,GACzB,OAAO,EAAE,eAAe,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC;QAErD,MAAM,KAAK,GAAe;YACxB,kBAAkB,EAAE,EAAE;YACtB,OAAO,EAAE,EAAE;YACX,YAAY,EAAE,EAAE;YAChB,SAAS,EAAE,KAAK;YAChB,cAAc,EAAE,EAAE;SACnB,CAAC;QAEF,IAAI,QAAQ,GAAmB,EAAE,CAAC;QAClC,IAAI,aAAwC,CAAC;QAE7C,IAAI,CAAC;YACH,MAAM,EACJ,OAAO,EAAE,eAAe,EACxB,QAAQ,EACR,YAAY,EACZ,QAAQ,EACR,QAAQ,EAAE,gBAAgB,EAC1B,YAAY,EACZ,cAAc,GACf,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,oBAAoB,CAAC,CAAC;YAElD,wEAAwE;YACxE,IAAI,qBAAqB,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACpD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC;gBACrD,MAAM,IAAI,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YACnD,CAAC;YAED,qDAAqD;YACrD,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;gBACjC,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;gBACpC,aAAa,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC;YAC9D,CAAC;YAED,QAAQ,GAAG,gBAAgB,CAAC;YAC5B,KAAK,CAAC,cAAc,GAAG,cAAc,CAAC;YAEtC,MAAM,SAAS,GAAI,oBAA4C,CAAC,SAAS,CAAC;YAE1E,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,sBAAsB,GAAG;oBAC7B,SAAS;oBACT,UAAU;oBACV,SAAS;oBACT,SAAS;iBACV,CAAC;gBACF,MAAM,gBAAgB,GAAG,sBAAsB,CAAC,MAAM,CACpD,CAAC,IAAI,EAAE,EAAE,CAAC,SAAS,CAAC,IAA8B,CAAC,IAAI,IAAI,CAC5D,CAAC;gBACF,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAChC,MAAM,IAAI,yCAAyC,CAAC,gBAAgB,CAAC,CAAC;gBACxE,CAAC;YACH,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC;gBAC/C,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,oBAAoB,CAAC,YAAY,EAAE,QAAQ,CAAC;gBACtD,KAAK,EAAE,QAAQ;gBACf,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,QAAQ,CAAC;gBACvD,UAAU,EAAE,MAAM;gBAElB,WAAW,EAAE,IAAI,CAAC,iBAAiB,CACjC,SAAS,EAAE,WAAW,EACtB,aAAa,CACd;gBACD,YAAY,EAAE,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,SAAS,EAAE,YAAY,CAAC;gBACpE,WAAW,EAAE,eAAe,CAAC,MAAM;gBACnC,eAAe,EAAE;oBACf,MAAM,EAAE,EAAE,eAAe,EAAE,uBAAuB,EAAE;oBACpD,MAAM,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE;iBACzB;aACF,CAAC,CAAC;YAEH,MAAM,WAAW,GAAG,CAAC,GAAG,QAAQ,EAAE,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,QAAQ,IAAI,EAAE,CAAC,CAAC,CAAC;YACxE,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,UAAU,CACtC,KAAK,EACL,YAAY,EACZ,WAAW,EACX,eAAe,CAAC,WAAW,EAC3B,eAAe,CAAC,MAAM,EACtB,IAAI,CAAC,MAAM,CACZ,CAAC;YAEF,OAAO,IAAI,CAAC,2BAA2B,CACrC,SAAS,EACT,KAAK,EACL,UAAU,CAAC,QAAQ,EACnB,MAAM,EACN,QAAQ,EACR,UAAU,CAAC,MAAM,CAClB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,iEAAiE;YACjE,IACE,KAAK,YAAY,yCAAyC;gBAC1D,KAAK,YAAY,+BAA+B,EAChD,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC;YAED,+EAA+E;YAC/E,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;gBACpB,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBACjE,MAAM,IAAI,eAAe,CAAC,MAAM,CAAC,CAAC;YACpC,CAAC;YAED,MAAM,YAAY,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;YAC5C,IAAI,CAAC,MAAM,CAAC;gBACV,QAAQ,EAAE,OAAO;gBACjB,OAAO,EAAE,+BAA+B,YAAY,EAAE;gBACtD,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;YAEH,oEAAoE;YACpE,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,OAAO,EAAE,2BAA2B,YAAY,EAAE;gBAClD,SAAS,EAAE,KAAK;gBAChB,QAAQ;aACT,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,aAAa,EAAE,OAAO,EAAE,CAAC;QAC3B,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,MAAM,CACjB,oBAAwD;QAExD,MAAM,aAAa,GACjB,OAAO,oBAAoB,KAAK,QAAQ,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC;QAEzE,uEAAuE;QACvE,MAAM,qBAAqB,GACzB,aAAa,EAAE,eAAe,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC;QAE3D,MAAM,EACJ,OAAO,EACP,QAAQ,EACR,YAAY,EACZ,QAAQ,EACR,QAAQ,EACR,YAAY,EACZ,cAAc,GACf,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,oBAAoB,CAAC,CAAC;QAElD,wEAAwE;QACxE,IAAI,qBAAqB,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACpD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC;YACrD,MAAM,IAAI,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACnD,CAAC;QAED,qDAAqD;QACrD,IAAI,aAAwC,CAAC;QAC7C,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;YACjC,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;YACpC,aAAa,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC;QAC9D,CAAC;QAED,MAAM,SAAS,GAAI,oBAAkD;aAClE,SAA6C,CAAC;QAEjD,MAAM,KAAK,GAAe;YACxB,kBAAkB,EAAE,EAAE;YACtB,OAAO,EAAE,EAAE;YACX,YAAY,EAAE,EAAE;YAChB,SAAS,EAAE,KAAK;YAChB,cAAc,EAAE,cAAc;SAC/B,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,aAAsE,CAAC;QAC3E,IAAI,YAAuC,CAAC;QAC5C,MAAM,aAAa,GAAG,IAAI,OAAO,CAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACjE,aAAa,GAAG,OAAO,CAAC;YACxB,YAAY,GAAG,MAAM,CAAC;QACxB,CAAC,CAAC,CAAC;QAEH,MAAM,WAAW,GAAG,CAAC,KAAc,EAAE,EAAE;YACrC,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACzD,IAAI,CAAC,MAAM,CAAC;gBACV,QAAQ,EAAE,OAAO;gBACjB,OAAO,EAAE,2BAA2B,YAAY,EAAE;gBAClD,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;YACH,YAAY,CAAC,KAAK,CAAC,CAAC;QACtB,CAAC,CAAC;QAEF,IAAI,YAA0D,CAAC;QAC/D,IAAI,CAAC;YACH,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC;gBACvC,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,oBAAoB,CAAC,YAAY,EAAE,QAAQ,CAAC;gBACtD,KAAK,EAAE,QAAQ;gBACf,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,QAAQ,CAAC;gBACvD,UAAU,EAAE,MAAM;gBAClB,WAAW,EAAE,IAAI,CAAC,iBAAiB,CACjC,SAAS,EAAE,WAAW,EACtB,aAAa,CACd;gBACD,YAAY,EAAE,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,SAAS,EAAE,YAAY,CAAC;gBACpE,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;oBACjB,aAAa,EAAE,OAAO,EAAE,CAAC;oBACzB,IAAI,SAAS,EAAE,OAAO,EAAE,CAAC;wBACvB,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;oBAC3B,CAAC;oBACD,WAAW,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBAC3B,CAAC;gBACD,OAAO,EAAE,SAAS,EAAE,OAAO;gBAC3B,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE;oBAClB,aAAa,EAAE,OAAO,EAAE,CAAC;oBACzB,IAAI,SAAS,EAAE,QAAQ,EAAE,CAAC;wBACxB,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;oBAC5B,CAAC;oBAED,MAAM,WAAW,GAAG;wBAClB,GAAG,QAAQ;wBACX,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,QAAQ,IAAI,EAAE,CAAC;qBACpC,CAAC;oBACF,IAAI,CAAC,UAAU,CACb,KAAK,EACL,YAAY,EACZ,WAAW,EACX,OAAO,CAAC,WAAW,EACnB,OAAO,CAAC,MAAM,EACd,IAAI,CAAC,MAAM,CACZ,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE,EAAE;wBACpB,MAAM,MAAM,GAAG,IAAI,CAAC,2BAA2B,CAC7C,SAAS,EACT,KAAK,EACL,UAAU,CAAC,QAAQ,EACnB,KAAK,EACL,QAAQ,EACR,UAAU,CAAC,MAAM,CAClB,CAAC;wBACF,aAAa,CAAC,MAAM,CAAC,CAAC;oBACxB,CAAC,CAAC,CAAC;gBACL,CAAC;gBACD,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;oBACjB,aAAa,EAAE,OAAO,EAAE,CAAC;oBACzB,IAAI,SAAS,EAAE,OAAO,EAAE,CAAC;wBACvB,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;oBAC3B,CAAC;oBACD,wEAAwE;oBACxE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,MAAM;wBACnC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC;wBAC/B,CAAC,CAAC,oBAAoB,CAAC;oBACzB,YAAY,CAAC,IAAI,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC;gBAC5C,CAAC;gBACD,WAAW,EAAE,OAAO,CAAC,MAAM;gBAC3B,eAAe,EAAE;oBACf,MAAM,EAAE,EAAE,eAAe,EAAE,uBAAuB,EAAE;oBACpD,MAAM,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE;iBACzB;aACF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,aAAa,EAAE,OAAO,EAAE,CAAC;YACzB,MAAM,KAAK,CAAC;QACd,CAAC;QAED,MAAM,iBAAiB,GAAG,YAAiC,CAAC;QAC5D,iBAAiB,CAAC,MAAM,GAAG,aAAa,CAAC;QACzC,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAEO,2BAA2B,CACjC,SAAiB,EACjB,KAAiB,EACjB,aAA6B,EAC7B,MAKC,EACD,QAAiB,EACjB,MAAgC;QAEhC,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;YACxB,MAAM,YAAY,GAAG,KAAK,CAAC,kBAAkB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;YAE/D,IAAI,CAAC,KAAK,CAAC,SAAS,IAAI,QAAQ,IAAI,MAAM,CAAC,KAAK,EAAE,MAAM,IAAI,QAAQ,EAAE,CAAC;gBACrE,IAAI,CAAC,MAAM,CAAC;oBACV,QAAQ,EAAE,OAAO;oBACjB,OAAO,EAAE,yCAAyC,QAAQ,GAAG;oBAC7D,KAAK,EAAE,CAAC;iBACT,CAAC,CAAC;gBACH,KAAK,CAAC,YAAY,GAAG,yCAAyC,QAAQ,GAAG,CAAC;YAC5E,CAAC;iBAAM,CAAC;gBACN,KAAK,CAAC,YAAY,GAAG,YAAY,IAAI,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;YACzD,CAAC;QACH,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC3B,MAAM,eAAe,GAAG,OAAO,GAAG,SAAS,CAAC;QAC5C,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;YACtB,IAAI,CAAC,EAAE,CAAC,aAAa,CACnB,cAAc,CAAC,KAAK,EACpB,MAAM,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC,EAClC,MAAM,CAAC,UAAU,CAAC,YAAY,IAAI,CAAC,EACnC,MAAM,CAAC,UAAU,CAAC,eAAe,IAAI,CAAC,EACtC,MAAM,CAAC,UAAU,CAAC,iBAAiB,IAAI,CAAC,EACxC,eAAe,CAChB,CAAC;QACJ,CAAC;QAED,OAAO;YACL,OAAO,EAAE,KAAK,CAAC,SAAS;YACxB,OAAO,EAAE,KAAK,CAAC,YAAY,IAAI,0BAA0B;YACzD,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,MAAM;YACN,KAAK,EAAE,MAAM,CAAC,UAAU;gBACtB,CAAC,CAAC;oBACE,YAAY,EAAE,MAAM,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC;oBAChD,aAAa,EAAE,MAAM,CAAC,UAAU,CAAC,YAAY,IAAI,CAAC;oBAClD,gBAAgB,EAAE,MAAM,CAAC,UAAU,CAAC,eAAe,IAAI,CAAC;oBACxD,mBAAmB,EAAE,MAAM,CAAC,UAAU,CAAC,iBAAiB,IAAI,CAAC;oBAC7D,iBAAiB,EAAE,eAAe;iBACnC;gBACH,CAAC,CAAC,SAAS;YACb,QAAQ,EAAE,aAAa;SACxB,CAAC;IACJ,CAAC;IAEO,WAAW,CACjB,YAAuB,EACvB,SAAqB,EACrB,WAAoB,EACpB,SAAmB;QAEnB,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,EAAE,gBAAgB,EAAE,EAAE,EAAE,QAAQ,CAAC;QAChE,OAAO,gBAAgB,CAAC,IAAI,CAAC,EAAE,EAAE;YAC/B,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,QAAQ;YACR,YAAY;YACZ,SAAS;YACT,WAAW;YACX,SAAS;YACT,iBAAiB,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC,SAAS;SACrE,CAAC,CAAC;IACL,CAAC;IAEO,UAAU,CAChB,MAAqD,EACrD,QAAgB;QAEhB,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACvD,IAAI,QAAQ,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,QAAQ,KAAK,MAAM,CAAC,EAAE,CAAC;YAC9D,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,WAAW,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,CAAC;IACvC,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,UAAU,CACtB,KAAiB,EACjB,KAAoB,EACpB,QAAwB,EACxB,WAAmB,EACnB,YAAiC,EACjC,MAAmC;QAEnC,IAAI,KAAK,CAAC,SAAS;YAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;QAEzC,MAAM,UAAU,GAAG,MAAM,kBAAkB,CAAC;YAC1C,KAAK;YACL,aAAa,EAAE,QAAQ;YACvB,WAAW;YACX,YAAY;YACZ,MAAM;SACP,CAAC,CAAC;QAEH,KAAK,CAAC,SAAS,GAAG,UAAU,CAAC,YAAY,CAAC;QAC1C,KAAK,CAAC,YAAY,GAAG,UAAU,CAAC,SAAS,CAAC;QAE1C,MAAM,UAAU,GAAG,sBAAsB,CAAC;YACxC,YAAY,EAAE,MAAM;YACpB,UAAU,EAAE;gBACV,OAAO,EAAE,IAAI;gBACb,SAAS,EAAE,UAAU,CAAC,SAAS;gBAC/B,YAAY,EAAE,UAAU,CAAC,YAAY;aACtC;YACD,IAAI,EAAE;gBACJ,SAAS,EAAE,UAAU,CAAC,SAAS;gBAC/B,YAAY,EAAE,UAAU,CAAC,YAAY;aACtC;YACD,SAAS,EAAE,UAAU,CAAC,SAAS;SAChC,CAAC,CAAC;QAEH,KAAK,MAAM,MAAM,IAAI,UAAU,EAAE,CAAC;YAChC,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC,cAAc,CAAC;YACtC,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC9B,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7B,CAAC;QAED,OAAO;YACL,QAAQ,EAAE,CAAC,GAAG,QAAQ,EAAE,GAAG,UAAU,CAAC,QAAQ,CAAC;YAC/C,MAAM,EAAE,UAAU,CAAC,MAAM;SAC1B,CAAC;IACJ,CAAC;CACF","sourcesContent":["import { createAgentTools } from \"../agent/tools/index.js\";\nimport { buildAgentSystemPrompt } from \"../agent/prompts/agentSystemPrompt.js\";\nimport { LogLine } from \"../types/public/logs.js\";\nimport { V3 } from \"../v3.js\";\nimport {\n  ModelMessage,\n  ToolSet,\n  wrapLanguageModel,\n  stepCountIs,\n  LanguageModel,\n  type LanguageModelUsage,\n  type StepResult,\n  type GenerateTextOnStepFinishCallback,\n  type StreamTextOnStepFinishCallback,\n  type PrepareStepFunction,\n} from \"ai\";\nimport { StagehandZodObject } from \"../zodCompat.js\";\nimport { processMessages } from \"../agent/utils/messageProcessing.js\";\nimport { LLMClient } from \"../llm/LLMClient.js\";\nimport { FlowLogger } from \"../flowlogger/FlowLogger.js\";\nimport {\n  AgentExecuteOptions,\n  AgentStreamExecuteOptions,\n  AgentExecuteOptionsBase,\n  AgentResult,\n  AgentContext,\n  AgentState,\n  AgentStreamResult,\n  AgentStreamCallbacks,\n  AgentToolMode,\n  AgentModelConfig,\n  Variables,\n} from \"../types/public/agent.js\";\nimport { HYBRID_CAPABLE_MODEL_PATTERNS } from \"../types/private/agent.js\";\nimport { V3FunctionName } from \"../types/public/methods.js\";\nimport { mapToolResultToActions } from \"../agent/utils/actionMapping.js\";\nimport {\n  MissingLLMConfigurationError,\n  MissingEnvironmentVariableError,\n  StreamingCallbacksInNonStreamingModeError,\n  AgentAbortError,\n} from \"../types/public/sdkErrors.js\";\nimport { handleDoneToolCall } from \"../agent/utils/handleDoneToolCall.js\";\nimport {\n  CaptchaSolver,\n  CAPTCHA_SOLVED_MSG,\n  CAPTCHA_ERRORED_MSG,\n} from \"../agent/utils/captchaSolver.js\";\n\nfunction getErrorMessage(error: unknown): string {\n  return error instanceof Error ? error.message : String(error);\n}\n\n/**\n * Prepends a system message with cache control to the messages array.\n * The cache control providerOptions are used by Anthropic and ignored by other providers.\n */\nfunction prependSystemMessage(\n  systemPrompt: string,\n  messages: ModelMessage[],\n): ModelMessage[] {\n  return [\n    {\n      role: \"system\",\n      content: systemPrompt,\n      providerOptions: {\n        anthropic: {\n          cacheControl: { type: \"ephemeral\" },\n        },\n      },\n    },\n    ...messages,\n  ];\n}\n\nexport class V3AgentHandler {\n  private v3: V3;\n  private logger: (message: LogLine) => void;\n  private llmClient: LLMClient;\n  private executionModel?: string | AgentModelConfig;\n  private systemInstructions?: string;\n  private mcpTools?: ToolSet;\n  private mode: AgentToolMode;\n  private captchaAutoSolveEnabled: boolean;\n\n  constructor(\n    v3: V3,\n    logger: (message: LogLine) => void,\n    llmClient: LLMClient,\n    executionModel?: string | AgentModelConfig,\n    systemInstructions?: string,\n    mcpTools?: ToolSet,\n    mode?: AgentToolMode,\n    captchaAutoSolveEnabled?: boolean,\n  ) {\n    this.v3 = v3;\n    this.logger = logger;\n    this.llmClient = llmClient;\n    this.executionModel = executionModel;\n    this.systemInstructions = systemInstructions;\n    this.mcpTools = mcpTools;\n    this.mode = mode ?? \"dom\";\n    this.captchaAutoSolveEnabled = captchaAutoSolveEnabled ?? false;\n  }\n\n  private async prepareAgent(\n    instructionOrOptions: string | AgentExecuteOptionsBase,\n  ): Promise<AgentContext> {\n    try {\n      const options =\n        typeof instructionOrOptions === \"string\"\n          ? { instruction: instructionOrOptions }\n          : instructionOrOptions;\n\n      const maxSteps = options.maxSteps || 20;\n\n      // Get the initial page URL first (needed for the system prompt)\n      const initialPageUrl = (await this.v3.context.awaitActivePage()).url();\n\n      // Build the system prompt with mode-aware tool guidance\n      const systemPrompt = buildAgentSystemPrompt({\n        url: initialPageUrl,\n        executionInstruction: options.instruction,\n        mode: this.mode,\n        systemInstructions: this.systemInstructions,\n        captchasAutoSolve: this.v3.isCaptchaAutoSolveEnabled,\n        excludeTools: options.excludeTools,\n        variables: options.variables,\n        useSearch: options.useSearch,\n      });\n\n      if (options.useSearch) {\n        const bbApiKey = this.v3.browserbaseApiKey;\n        if (!bbApiKey) {\n          throw new MissingEnvironmentVariableError(\n            \"BROWSERBASE_API_KEY\",\n            \"agent search (useSearch: true)\",\n          );\n        }\n      }\n\n      const tools = this.createTools(\n        options.excludeTools,\n        options.variables,\n        options.toolTimeout,\n        options.useSearch,\n      );\n      const allTools: ToolSet = { ...tools, ...this.mcpTools };\n\n      // Use provided messages for continuation, or start fresh with the instruction\n      const messages: ModelMessage[] = options.messages?.length\n        ? [...options.messages, { role: \"user\", content: options.instruction }]\n        : [{ role: \"user\", content: options.instruction }];\n\n      if (!this.llmClient?.getLanguageModel) {\n        throw new MissingLLMConfigurationError();\n      }\n      const baseModel = this.llmClient.getLanguageModel();\n      //to do - we likely do not need middleware anymore\n      const wrappedModel = wrapLanguageModel({\n        model: baseModel,\n        middleware: {\n          ...FlowLogger.createLlmLoggingMiddleware(baseModel.modelId),\n        },\n      });\n\n      const isHybridCapable = HYBRID_CAPABLE_MODEL_PATTERNS.some((pattern) =>\n        baseModel.modelId.includes(pattern),\n      );\n      if (this.mode === \"hybrid\" && !isHybridCapable) {\n        this.logger({\n          category: \"agent\",\n          message: `Warning: \"${baseModel.modelId}\" may not perform well in hybrid mode. See recommended models: https://docs.stagehand.dev/v3/basics/agent#hybrid-mode`,\n          level: 0,\n        });\n      }\n\n      return {\n        options,\n        maxSteps,\n        systemPrompt,\n        allTools,\n        messages,\n        wrappedModel,\n        initialPageUrl,\n      };\n    } catch (error) {\n      this.logger({\n        category: \"agent\",\n        message: `failed to prepare agent: ${error}`,\n        level: 0,\n      });\n      throw error;\n    }\n  }\n  private createPrepareStep(\n    userCallback?: PrepareStepFunction<ToolSet>,\n    captchaSolver?: CaptchaSolver,\n  ): PrepareStepFunction<ToolSet> {\n    return async (options) => {\n      processMessages(options.messages);\n      if (captchaSolver) {\n        if (captchaSolver.isSolving()) {\n          this.logger({\n            category: \"agent\",\n            message:\n              \"Captcha detected — waiting for Browserbase to solve it before continuing\",\n            level: 1,\n          });\n        }\n        await captchaSolver.waitIfSolving();\n        const { solved, errored } = captchaSolver.consumeSolveResult();\n        if (solved) {\n          options.messages.push({\n            role: \"user\",\n            content: CAPTCHA_SOLVED_MSG,\n          });\n          this.logger({\n            category: \"agent\",\n            message:\n              \"Captcha solved — injected notification into agent message stream\",\n            level: 1,\n          });\n        }\n        if (errored) {\n          options.messages.push({\n            role: \"user\",\n            content: CAPTCHA_ERRORED_MSG,\n          });\n          this.logger({\n            category: \"agent\",\n            message:\n              \"Captcha solver failed — injected error notification into agent message stream\",\n            level: 1,\n          });\n        }\n      }\n      if (userCallback) {\n        return userCallback(options);\n      }\n      return options;\n    };\n  }\n\n  private createStepHandler(\n    state: AgentState,\n    userCallback?:\n      | GenerateTextOnStepFinishCallback<ToolSet>\n      | StreamTextOnStepFinishCallback<ToolSet>,\n  ) {\n    return async (event: StepResult<ToolSet>) => {\n      this.logger({\n        category: \"agent\",\n        message: `Step finished: ${event.finishReason}`,\n        level: 2,\n      });\n\n      if (event.toolCalls && event.toolCalls.length > 0) {\n        for (let i = 0; i < event.toolCalls.length; i++) {\n          const toolCall = event.toolCalls[i];\n          const args = toolCall.input;\n          const toolResult = event.toolResults?.[i];\n\n          if (event.text && event.text.length > 0) {\n            state.collectedReasoning.push(event.text);\n            this.logger({\n              category: \"agent\",\n              message: `reasoning: ${event.text}`,\n              level: 1,\n            });\n          }\n\n          if (toolCall.toolName === \"done\") {\n            state.completed = true;\n            if (args?.taskComplete) {\n              const doneReasoning = args.reasoning;\n              const allReasoning = state.collectedReasoning.join(\" \");\n              state.finalMessage = doneReasoning\n                ? `${allReasoning} ${doneReasoning}`.trim()\n                : allReasoning || \"Task completed successfully\";\n            }\n          }\n          const mappedActions = mapToolResultToActions({\n            toolCallName: toolCall.toolName,\n            toolResult,\n            args,\n            reasoning: event.text || undefined,\n          });\n\n          for (const action of mappedActions) {\n            action.pageUrl = state.currentPageUrl;\n            action.timestamp = Date.now();\n            state.actions.push(action);\n          }\n        }\n        state.currentPageUrl = (await this.v3.context.awaitActivePage()).url();\n      }\n\n      if (userCallback) {\n        await userCallback(event);\n      }\n    };\n  }\n\n  public async execute(\n    instructionOrOptions: string | AgentExecuteOptions,\n  ): Promise<AgentResult> {\n    const startTime = Date.now();\n    const options =\n      typeof instructionOrOptions === \"object\" ? instructionOrOptions : null;\n    const signal = options?.signal;\n\n    // Highlight cursor defaults to true for hybrid mode, can be overridden\n    const shouldHighlightCursor =\n      options?.highlightCursor ?? this.mode === \"hybrid\";\n\n    const state: AgentState = {\n      collectedReasoning: [],\n      actions: [],\n      finalMessage: \"\",\n      completed: false,\n      currentPageUrl: \"\",\n    };\n\n    let messages: ModelMessage[] = [];\n    let captchaSolver: CaptchaSolver | undefined;\n\n    try {\n      const {\n        options: preparedOptions,\n        maxSteps,\n        systemPrompt,\n        allTools,\n        messages: preparedMessages,\n        wrappedModel,\n        initialPageUrl,\n      } = await this.prepareAgent(instructionOrOptions);\n\n      // Enable cursor overlay for hybrid mode (coordinate-based interactions)\n      if (shouldHighlightCursor && this.mode === \"hybrid\") {\n        const page = await this.v3.context.awaitActivePage();\n        await page.enableCursorOverlay().catch(() => {});\n      }\n\n      // Set up captcha solver for Browserbase environments\n      if (this.captchaAutoSolveEnabled) {\n        captchaSolver = new CaptchaSolver();\n        captchaSolver.init(() => this.v3.context.awaitActivePage());\n      }\n\n      messages = preparedMessages;\n      state.currentPageUrl = initialPageUrl;\n\n      const callbacks = (instructionOrOptions as AgentExecuteOptions).callbacks;\n\n      if (callbacks) {\n        const streamingOnlyCallbacks = [\n          \"onChunk\",\n          \"onFinish\",\n          \"onError\",\n          \"onAbort\",\n        ];\n        const invalidCallbacks = streamingOnlyCallbacks.filter(\n          (name) => callbacks[name as keyof typeof callbacks] != null,\n        );\n        if (invalidCallbacks.length > 0) {\n          throw new StreamingCallbacksInNonStreamingModeError(invalidCallbacks);\n        }\n      }\n\n      const result = await this.llmClient.generateText({\n        model: wrappedModel,\n        messages: prependSystemMessage(systemPrompt, messages),\n        tools: allTools,\n        stopWhen: (result) => this.handleStop(result, maxSteps),\n        toolChoice: \"auto\",\n\n        prepareStep: this.createPrepareStep(\n          callbacks?.prepareStep,\n          captchaSolver,\n        ),\n        onStepFinish: this.createStepHandler(state, callbacks?.onStepFinish),\n        abortSignal: preparedOptions.signal,\n        providerOptions: {\n          google: { mediaResolution: \"MEDIA_RESOLUTION_HIGH\" },\n          openai: { store: false },\n        },\n      });\n\n      const allMessages = [...messages, ...(result.response?.messages || [])];\n      const doneResult = await this.ensureDone(\n        state,\n        wrappedModel,\n        allMessages,\n        preparedOptions.instruction,\n        preparedOptions.output,\n        this.logger,\n      );\n\n      return this.consolidateMetricsAndResult(\n        startTime,\n        state,\n        doneResult.messages,\n        result,\n        maxSteps,\n        doneResult.output,\n      );\n    } catch (error) {\n      // Re-throw validation errors that should propagate to the caller\n      if (\n        error instanceof StreamingCallbacksInNonStreamingModeError ||\n        error instanceof MissingEnvironmentVariableError\n      ) {\n        throw error;\n      }\n\n      // Re-throw abort errors wrapped in AgentAbortError for consistent error typing\n      if (signal?.aborted) {\n        const reason = signal.reason ? String(signal.reason) : \"aborted\";\n        throw new AgentAbortError(reason);\n      }\n\n      const errorMessage = getErrorMessage(error);\n      this.logger({\n        category: \"agent\",\n        message: `Error executing agent task: ${errorMessage}`,\n        level: 0,\n      });\n\n      // For non-abort errors, return a failure result instead of throwing\n      return {\n        success: false,\n        actions: state.actions,\n        message: `Failed to execute task: ${errorMessage}`,\n        completed: false,\n        messages,\n      };\n    } finally {\n      captchaSolver?.dispose();\n    }\n  }\n\n  public async stream(\n    instructionOrOptions: string | AgentStreamExecuteOptions,\n  ): Promise<AgentStreamResult> {\n    const streamOptions =\n      typeof instructionOrOptions === \"object\" ? instructionOrOptions : null;\n\n    // Highlight cursor defaults to true for hybrid mode, can be overridden\n    const shouldHighlightCursor =\n      streamOptions?.highlightCursor ?? this.mode === \"hybrid\";\n\n    const {\n      options,\n      maxSteps,\n      systemPrompt,\n      allTools,\n      messages,\n      wrappedModel,\n      initialPageUrl,\n    } = await this.prepareAgent(instructionOrOptions);\n\n    // Enable cursor overlay for hybrid mode (coordinate-based interactions)\n    if (shouldHighlightCursor && this.mode === \"hybrid\") {\n      const page = await this.v3.context.awaitActivePage();\n      await page.enableCursorOverlay().catch(() => {});\n    }\n\n    // Set up captcha solver for Browserbase environments\n    let captchaSolver: CaptchaSolver | undefined;\n    if (this.captchaAutoSolveEnabled) {\n      captchaSolver = new CaptchaSolver();\n      captchaSolver.init(() => this.v3.context.awaitActivePage());\n    }\n\n    const callbacks = (instructionOrOptions as AgentStreamExecuteOptions)\n      .callbacks as AgentStreamCallbacks | undefined;\n\n    const state: AgentState = {\n      collectedReasoning: [],\n      actions: [],\n      finalMessage: \"\",\n      completed: false,\n      currentPageUrl: initialPageUrl,\n    };\n    const startTime = Date.now();\n\n    let resolveResult: (value: AgentResult | PromiseLike<AgentResult>) => void;\n    let rejectResult: (reason: unknown) => void;\n    const resultPromise = new Promise<AgentResult>((resolve, reject) => {\n      resolveResult = resolve;\n      rejectResult = reject;\n    });\n\n    const handleError = (error: unknown) => {\n      const errorMessage =\n        error instanceof Error ? error.message : String(error);\n      this.logger({\n        category: \"agent\",\n        message: `Error during streaming: ${errorMessage}`,\n        level: 0,\n      });\n      rejectResult(error);\n    };\n\n    let streamResult: ReturnType<typeof this.llmClient.streamText>;\n    try {\n      streamResult = this.llmClient.streamText({\n        model: wrappedModel,\n        messages: prependSystemMessage(systemPrompt, messages),\n        tools: allTools,\n        stopWhen: (result) => this.handleStop(result, maxSteps),\n        toolChoice: \"auto\",\n        prepareStep: this.createPrepareStep(\n          callbacks?.prepareStep,\n          captchaSolver,\n        ),\n        onStepFinish: this.createStepHandler(state, callbacks?.onStepFinish),\n        onError: (event) => {\n          captchaSolver?.dispose();\n          if (callbacks?.onError) {\n            callbacks.onError(event);\n          }\n          handleError(event.error);\n        },\n        onChunk: callbacks?.onChunk,\n        onFinish: (event) => {\n          captchaSolver?.dispose();\n          if (callbacks?.onFinish) {\n            callbacks.onFinish(event);\n          }\n\n          const allMessages = [\n            ...messages,\n            ...(event.response?.messages || []),\n          ];\n          this.ensureDone(\n            state,\n            wrappedModel,\n            allMessages,\n            options.instruction,\n            options.output,\n            this.logger,\n          ).then((doneResult) => {\n            const result = this.consolidateMetricsAndResult(\n              startTime,\n              state,\n              doneResult.messages,\n              event,\n              maxSteps,\n              doneResult.output,\n            );\n            resolveResult(result);\n          });\n        },\n        onAbort: (event) => {\n          captchaSolver?.dispose();\n          if (callbacks?.onAbort) {\n            callbacks.onAbort(event);\n          }\n          // Reject the result promise with AgentAbortError when stream is aborted\n          const reason = options.signal?.reason\n            ? String(options.signal.reason)\n            : \"Stream was aborted\";\n          rejectResult(new AgentAbortError(reason));\n        },\n        abortSignal: options.signal,\n        providerOptions: {\n          google: { mediaResolution: \"MEDIA_RESOLUTION_HIGH\" },\n          openai: { store: false },\n        },\n      });\n    } catch (error) {\n      captchaSolver?.dispose();\n      throw error;\n    }\n\n    const agentStreamResult = streamResult as AgentStreamResult;\n    agentStreamResult.result = resultPromise;\n    return agentStreamResult;\n  }\n\n  private consolidateMetricsAndResult(\n    startTime: number,\n    state: AgentState,\n    inputMessages: ModelMessage[],\n    result: {\n      text?: string;\n      totalUsage?: LanguageModelUsage;\n      response?: { messages?: ModelMessage[] };\n      steps?: StepResult<ToolSet>[];\n    },\n    maxSteps?: number,\n    output?: Record<string, unknown>,\n  ): AgentResult {\n    if (!state.finalMessage) {\n      const allReasoning = state.collectedReasoning.join(\" \").trim();\n\n      if (!state.completed && maxSteps && result.steps?.length >= maxSteps) {\n        this.logger({\n          category: \"agent\",\n          message: `Agent stopped: reached maximum steps (${maxSteps})`,\n          level: 1,\n        });\n        state.finalMessage = `Agent stopped: reached maximum steps (${maxSteps})`;\n      } else {\n        state.finalMessage = allReasoning || result.text || \"\";\n      }\n    }\n\n    const endTime = Date.now();\n    const inferenceTimeMs = endTime - startTime;\n    if (result.totalUsage) {\n      this.v3.updateMetrics(\n        V3FunctionName.AGENT,\n        result.totalUsage.inputTokens || 0,\n        result.totalUsage.outputTokens || 0,\n        result.totalUsage.reasoningTokens || 0,\n        result.totalUsage.cachedInputTokens || 0,\n        inferenceTimeMs,\n      );\n    }\n\n    return {\n      success: state.completed,\n      message: state.finalMessage || \"Task execution completed\",\n      actions: state.actions,\n      completed: state.completed,\n      output,\n      usage: result.totalUsage\n        ? {\n            input_tokens: result.totalUsage.inputTokens || 0,\n            output_tokens: result.totalUsage.outputTokens || 0,\n            reasoning_tokens: result.totalUsage.reasoningTokens || 0,\n            cached_input_tokens: result.totalUsage.cachedInputTokens || 0,\n            inference_time_ms: inferenceTimeMs,\n          }\n        : undefined,\n      messages: inputMessages,\n    };\n  }\n\n  private createTools(\n    excludeTools?: string[],\n    variables?: Variables,\n    toolTimeout?: number,\n    useSearch?: boolean,\n  ) {\n    const provider = this.llmClient?.getLanguageModel?.()?.provider;\n    return createAgentTools(this.v3, {\n      executionModel: this.executionModel,\n      logger: this.logger,\n      mode: this.mode,\n      provider,\n      excludeTools,\n      variables,\n      toolTimeout,\n      useSearch,\n      browserbaseApiKey: useSearch ? this.v3.browserbaseApiKey : undefined,\n    });\n  }\n\n  private handleStop(\n    result: Parameters<ReturnType<typeof stepCountIs>>[0],\n    maxSteps: number,\n  ): boolean | PromiseLike<boolean> {\n    const lastStep = result.steps[result.steps.length - 1];\n    if (lastStep?.toolCalls?.some((tc) => tc.toolName === \"done\")) {\n      return true;\n    }\n    return stepCountIs(maxSteps)(result);\n  }\n\n  /**\n   * Ensures the done tool is called at the end of agent execution.\n   * Returns the messages and any extracted output from the done call.\n   */\n  private async ensureDone(\n    state: AgentState,\n    model: LanguageModel,\n    messages: ModelMessage[],\n    instruction: string,\n    outputSchema?: StagehandZodObject,\n    logger?: (message: LogLine) => void,\n  ): Promise<{ messages: ModelMessage[]; output?: Record<string, unknown> }> {\n    if (state.completed) return { messages };\n\n    const doneResult = await handleDoneToolCall({\n      model,\n      inputMessages: messages,\n      instruction,\n      outputSchema,\n      logger,\n    });\n\n    state.completed = doneResult.taskComplete;\n    state.finalMessage = doneResult.reasoning;\n\n    const doneAction = mapToolResultToActions({\n      toolCallName: \"done\",\n      toolResult: {\n        success: true,\n        reasoning: doneResult.reasoning,\n        taskComplete: doneResult.taskComplete,\n      },\n      args: {\n        reasoning: doneResult.reasoning,\n        taskComplete: doneResult.taskComplete,\n      },\n      reasoning: doneResult.reasoning,\n    });\n\n    for (const action of doneAction) {\n      action.pageUrl = state.currentPageUrl;\n      action.timestamp = Date.now();\n      state.actions.push(action);\n    }\n\n    return {\n      messages: [...messages, ...doneResult.messages],\n      output: doneResult.output,\n    };\n  }\n}\n"]}

package/dist/esm/lib/v3/index.d.ts

@@ -57,7 +57,7 @@ declare const StagehandDefault: {
     Api: typeof PublicApi.Api;
     AISdkClient: typeof PublicApi.AISdkClient;
     CustomOpenAIClient: typeof PublicApi.CustomOpenAIClient;
-    AVAILABLE_CUA_MODELS: readonly ["openai/gpt-5.4", "openai/computer-use-preview", "openai/computer-use-preview-2025-03-11", "anthropic/claude-opus-4-5-20251101", "anthropic/claude-opus-4-6", "anthropic/claude-sonnet-4-6", "anthropic/claude-haiku-4-5-20251001", "anthropic/claude-sonnet-4-20250514", "anthropic/claude-sonnet-4-5-20250929", "google/gemini-2.5-computer-use-preview-10-2025", "google/gemini-3-flash-preview", "google/gemini-3-pro-preview", "microsoft/fara-7b"];
+    AVAILABLE_CUA_MODELS: readonly ["openai/gpt-5.4", "openai/gpt-5.4-mini", "openai/gpt-5.5", "openai/computer-use-preview", "openai/computer-use-preview-2025-03-11", "anthropic/claude-opus-4-5-20251101", "anthropic/claude-opus-4-6", "anthropic/claude-sonnet-4-6", "anthropic/claude-haiku-4-5", "anthropic/claude-haiku-4-5-20251001", "anthropic/claude-sonnet-4-20250514", "anthropic/claude-sonnet-4-5-20250929", "google/gemini-2.5-computer-use-preview-10-2025", "google/gemini-3-flash-preview", "google/gemini-3-pro-preview", "microsoft/fara-7b"];
     StagehandAPIError: typeof PublicApi.StagehandAPIError;
     StagehandAPIUnauthorizedError: typeof PublicApi.StagehandAPIUnauthorizedError;
     StagehandHttpError: typeof PublicApi.StagehandHttpError;

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

@@ -102,7 +102,7 @@ export class CerebrasClient extends LLMClient {
                         ]
                         : []),
                 ],
-                temperature: options.temperature || 0.7,
+                temperature: options.temperature ?? 0.7,
                 max_tokens: options.maxOutputTokens,
                 tools: tools,
                 tool_choice: options.tool_choice || "auto",

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

@@ -1 +1 @@
-{"version":3,"file":"CerebrasClient.js","sourceRoot":"","sources":["../../../../../lib/v3/llm/CerebrasClient.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAI5B,OAAO,EAGL,SAAS,GAEV,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,iCAAiC,EAAE,MAAM,8BAA8B,CAAC;AACjF,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAE/C,MAAM,OAAO,cAAe,SAAQ,SAAS;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,MAAM,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,YAAY,CAAC,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,iCAAiC,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","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"]}
+{"version":3,"file":"CerebrasClient.js","sourceRoot":"","sources":["../../../../../lib/v3/llm/CerebrasClient.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAI5B,OAAO,EAGL,SAAS,GAEV,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,iCAAiC,EAAE,MAAM,8BAA8B,CAAC;AACjF,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAE/C,MAAM,OAAO,cAAe,SAAQ,SAAS;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,MAAM,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,YAAY,CAAC,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,iCAAiC,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","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"]}

package/dist/esm/lib/v3/llm/GroqClient.js

@@ -102,7 +102,7 @@ export class GroqClient extends LLMClient {
                         ]
                         : []),
                 ],
-                temperature: options.temperature || 0.7,
+                temperature: options.temperature ?? 0.7,
                 max_tokens: options.maxOutputTokens,
                 tools: tools,
                 tool_choice: options.tool_choice || "auto",

package/dist/esm/lib/v3/llm/GroqClient.js.map

@@ -1 +1 @@
-{"version":3,"file":"GroqClient.js","sourceRoot":"","sources":["../../../../../lib/v3/llm/GroqClient.ts"],"names":[],"mappings":"AACA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAG5B,OAAO,EAGL,SAAS,GAEV,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,iCAAiC,EAAE,MAAM,8BAA8B,CAAC;AACjF,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAE/C,MAAM,OAAO,UAAW,SAAQ,SAAS;IAChC,IAAI,GAAG,MAAe,CAAC;IACtB,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,yDAAyD;QACzD,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC;YACvB,OAAO,EAAE,gCAAgC;YACzC,MAAM,EAAE,aAAa,EAAE,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY;YACzD,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,MAAM;YAChB,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,qDAAqD;QACrD,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,kDAAkD;YAClD,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,YAAY,CAAC,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,kCAAkC;YAClC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBAC5D,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBACvC,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,OAAO,CAAC,CAAC,CAAC,CAAC;gBACvC,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,MAAM;gBAChB,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,kEAAkE;YAClE,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;gBAC5B,OAAO,QAAa,CAAC;YACvB,CAAC;YAED,gEAAgE;YAChE,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,MAAM;wBAChB,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,uCAAuC;oBACvC,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,MAAM;wBAChB,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,iCAAiC,CAAC,yBAAyB,CAAC,CAAC;QACzE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC;gBACL,QAAQ,EAAE,MAAM;gBAChB,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","sourcesContent":["import type { ClientOptions } from \"openai\";\nimport OpenAI 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 GroqClient extends LLMClient {\n  public type = \"groq\" 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 Groq API\n    this.client = new OpenAI({\n      baseURL: \"https://api.groq.com/openai/v1\",\n      apiKey: clientOptions?.apiKey || process.env.GROQ_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: \"groq\",\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 Groq 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      // Groq 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 Groq API\n      const apiResponse = await this.client.chat.completions.create({\n        model: this.modelName.split(\"groq-\")[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(\"groq-\")[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: \"groq\",\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 there's no response model, return the entire response object\n      if (!options.response_model) {\n        return response as T;\n      }\n\n      // Otherwise, try parsing the JSON from the tool call 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: \"groq\",\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          // Try to extract JSON from the content\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: \"groq\",\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: \"groq\",\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"]}
+{"version":3,"file":"GroqClient.js","sourceRoot":"","sources":["../../../../../lib/v3/llm/GroqClient.ts"],"names":[],"mappings":"AACA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAG5B,OAAO,EAGL,SAAS,GAEV,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,iCAAiC,EAAE,MAAM,8BAA8B,CAAC;AACjF,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAE/C,MAAM,OAAO,UAAW,SAAQ,SAAS;IAChC,IAAI,GAAG,MAAe,CAAC;IACtB,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,yDAAyD;QACzD,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC;YACvB,OAAO,EAAE,gCAAgC;YACzC,MAAM,EAAE,aAAa,EAAE,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY;YACzD,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,MAAM;YAChB,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,qDAAqD;QACrD,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,kDAAkD;YAClD,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,YAAY,CAAC,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,kCAAkC;YAClC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBAC5D,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBACvC,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,OAAO,CAAC,CAAC,CAAC,CAAC;gBACvC,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,MAAM;gBAChB,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,kEAAkE;YAClE,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;gBAC5B,OAAO,QAAa,CAAC;YACvB,CAAC;YAED,gEAAgE;YAChE,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,MAAM;wBAChB,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,uCAAuC;oBACvC,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,MAAM;wBAChB,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,iCAAiC,CAAC,yBAAyB,CAAC,CAAC;QACzE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC;gBACL,QAAQ,EAAE,MAAM;gBAChB,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","sourcesContent":["import type { ClientOptions } from \"openai\";\nimport OpenAI 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 GroqClient extends LLMClient {\n  public type = \"groq\" 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 Groq API\n    this.client = new OpenAI({\n      baseURL: \"https://api.groq.com/openai/v1\",\n      apiKey: clientOptions?.apiKey || process.env.GROQ_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: \"groq\",\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 Groq 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      // Groq 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 Groq API\n      const apiResponse = await this.client.chat.completions.create({\n        model: this.modelName.split(\"groq-\")[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(\"groq-\")[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: \"groq\",\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 there's no response model, return the entire response object\n      if (!options.response_model) {\n        return response as T;\n      }\n\n      // Otherwise, try parsing the JSON from the tool call 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: \"groq\",\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          // Try to extract JSON from the content\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: \"groq\",\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: \"groq\",\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"]}