@browserbasehq/orca 3.1.0-patch.4 → 3.2.0-middleware.1

package/dist/cjs/tests/integration/page-screenshot.spec.js.map

@@ -1 +1 @@
-{"version":3,"file":"page-screenshot.spec.js","sourceRoot":"","sources":["../../../../tests/integration/page-screenshot.spec.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,2CAAgD;AAChD,2BAAoC;AACpC,uCAAyB;AACzB,2CAA6B;AAC7B,8CAAwC;AACxC,iDAA8C;AAC9C,+DAAyD;AAEzD,MAAM,IAAI,GAAG,CAAC,EAAU,EAAE,EAAE,CAAC,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAE/E,WAAI,CAAC,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IAC5C,IAAI,EAAM,CAAC;IAEX,WAAI,CAAC,UAAU,CAAC,KAAK,IAAI,EAAE;QACzB,EAAE,GAAG,IAAI,UAAE,CAAC,2BAAY,CAAC,CAAC;QAC1B,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,WAAI,CAAC,SAAS,CAAC,KAAK,IAAI,EAAE;QACxB,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,IAAA,WAAI,EAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,IAAI,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;QAEjE,MAAM,IAAA,aAAM,EACV,IAAI,CAAC,UAAU,CAAC;YACd,QAAQ,EAAE,IAAI;YACd,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;SAC9C,CAAC,CACH,CAAC,OAAO,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,IAAA,WAAI,EAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,IAAI,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;QAEjE,MAAM,IAAA,aAAM,EACV,IAAI,CAAC,UAAU,CAAC;YACd,mEAAmE;YACnE,IAAI,EAAE,MAAM;SACb,CAAC,CACH,CAAC,OAAO,CAAC,OAAO,CAAC,wBAAwB,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,IAAA,WAAI,EAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,IAAI,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;QAEjE,MAAM,IAAA,aAAM,EAAC,IAAI,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACzE,8BAA8B,CAC/B,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,IAAA,WAAI,EAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,IAAI,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;QAEjE,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QACnC,MAAM,kBAAkB,GAAG,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAG9D,SAGD,CAAC,UAAU,GAAG,KAAK,IAAI,EAAE;YACxB,MAAM,IAAI,CAAC,EAAE,CAAC,CAAC;YACf,OAAO,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7B,CAAC,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,IAAA,aAAM,EAAC,IAAI,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC5E,CAAC;gBAAS,CAAC;YAEP,SAGD,CAAC,UAAU,GAAG,kBAAkB,CAAC;QACpC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAA,WAAI,EAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,iBAAiB,GAAG,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;QACxD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,OAAO,CAAC,GAAG,CACT,2BAA2B,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,YAAY,iBAAiB,EAAE,CAC5F,CAAC;QAEF,MAAM,IAAI,GAAG;;;;;;;;;;;;;;;;;;KAkBZ,CAAC;QAEF,MAAM,IAAI,CAAC,IAAI,CAAC,iBAAiB,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC,CAAC;QAC9D,OAAO,CAAC,GAAG,CAAC,oCAAoC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,IAAI,CAAC,CAAC;QAE5E,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CACxB,EAAE,CAAC,MAAM,EAAE,EACX,wBAAwB,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CACjF,CAAC;QACF,OAAO,CAAC,GAAG,CAAC,8BAA8B,QAAQ,EAAE,CAAC,CAAC;QAEtD,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;QACjC,MAAM,eAAe,GAGhB,EAAE,CAAC;QACR,MAAM,aAAa,GAA6C,EAAE,CAAC;QACnE,MAAM,kBAAkB,GAAG,gBAAK,CAAC,SAAS,CAAC,UAAU,CAAC;QACtD,MAAM,gBAAgB,GAAG,gBAAK,CAAC,SAAS,CAAC,QAAQ,CAAC;QAElD,4FAA4F;QAC5F,gBAAK,CAAC,SAAS,CAAC,UAAU,GAAG,KAAK,UAAU,aAAa,CAAC,OAAO;YAC/D,MAAM,KAAK,GAAG,IAAa,CAAC;YAC5B,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC9B,eAAe,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;gBAC1D,OAAO,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACnC,CAAC;YACD,OAAO,kBAAkB,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAChD,CAAC,CAAC;QAEF,2EAA2E;QAC3E,gBAAK,CAAC,SAAS,CAAC,QAAQ,GAAG,KAAK,UAAU,WAAW,CAAC,UAAU,EAAE,GAAI;YACpE,MAAM,KAAK,GAAG,IAAa,CAAC;YAC5B,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC9B,aAAa,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;YACtD,CAAC;YACD,OAAO,gBAAgB,CAAC,IAAI,CAAC,IAAI,EAAE,UAAmB,EAAE,GAAG,CAAC,CAAC;QAC/D,CAAsB,CAAC;QAEvB,MAAM,YAAY,GAAG,IAIpB,CAAC;QACF,MAAM,SAAS,GAA+C,EAAE,CAAC;QACjE,MAAM,YAAY,GAAG,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CACrD,YAAY,CAAC,WAAW,CACiC,CAAC;QAC5D,gFAAgF;QAChF,YAAY,CAAC,WAAW,CAAC,IAAI,GAAG,KAAK,EACnC,MAAc,EACd,MAAgB,EAChB,EAAE;YACF,SAAS,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;YACnC,OAAO,YAAY,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACtC,CAAC,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,MAAM,WAAW,CAAC,KAAK,EAAE,CAAC;YAC5C,OAAO,CAAC,GAAG,CAAC,uCAAuC,SAAS,EAAE,CAAC,CAAC;YAEhE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC;gBACnC,UAAU,EAAE,UAAU;gBACtB,KAAK,EAAE,MAAM;gBACb,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;gBAC7C,IAAI,EAAE,CAAC,WAAW,CAAC;gBACnB,SAAS,EAAE,sBAAsB;gBACjC,cAAc,EAAE,IAAI;gBACpB,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,EAAE;gBACX,KAAK,EAAE,KAAK;gBACZ,KAAK,EAAE,mCAAmC;gBAC1C,OAAO,EAAE,iBAAiB;gBAC1B,IAAI,EAAE,MAAM;aACb,CAAC,CAAC;YACH,OAAO,CAAC,GAAG,CACT,+CAA+C,MAAM,CAAC,MAAM,YAAY,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,IAAI,CACnG,CAAC;YAEF,IAAA,aAAM,EAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;YAC7C,IAAA,aAAM,EAAC,eAAe,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;YACzD,OAAO,CAAC,GAAG,CACT,qCAAqC,eAAe,CAAC,MAAM,kBAAkB,aAAa,CAAC,MAAM,cAAc,SAAS,CAAC,MAAM,EAAE,CAClI,CAAC;YACF,MAAM,QAAQ,GAAG,eAAe,CAAC,CAAC,CAAC,EAAE,OAAO,IAAI,EAAE,CAAC;YACnD,IAAA,aAAM,EAAC,QAAQ,CAAC,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;YAC9D,IAAA,aAAM,EAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,aAAa,CAAC;gBACnC,CAAC,EAAE,CAAC;gBACJ,CAAC,EAAE,CAAC;gBACJ,KAAK,EAAE,GAAG;gBACV,MAAM,EAAE,GAAG;aACZ,CAAC,CAAC;YACH,IAAI,OAAO,QAAQ,EAAE,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACxC,IAAA,aAAM,EAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;gBAC1C,IAAA,aAAM,EAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;YAChD,CAAC;YAED,MAAM,aAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAExB,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,QAAQ,CACnC,GAAG,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC,uBAAuB,CAAC,CAAC,MAAM,CAChE,CAAC;YACF,IAAA,aAAM,EAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAE1B,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,QAAQ,CACpC,GAAG,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC,wBAAwB,CAAC,CAAC,MAAM,CACjE,CAAC;YACF,IAAA,aAAM,EAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAE3B,MAAM,eAAe,GAAG,SAAS,CAAC,MAAM,CACtC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,6CAA6C,CACxE,CAAC;YACF,IAAA,aAAM,EAAC,eAAe,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAClD,IAAA,aAAM,EACJ,eAAe,CAAC,IAAI,CAClB,CAAC,IAAI,EAAE,EAAE,CACP,IAAI,CAAC,MAAM;gBACX,OAAO,IAAI,CAAC,MAAM,KAAK,QAAQ;gBAC/B,OAAO,IAAK,IAAI,CAAC,MAAkC,CACtD,CACF,CAAC,UAAU,EAAE,CAAC;YACf,IAAA,aAAM,EACJ,eAAe,CAAC,IAAI,CAClB,CAAC,IAAI,EAAE,EAAE,CACP,CAAC,IAAI,CAAC,MAAM;gBACZ,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAiC,CAAC,CAAC,MAAM,KAAK,CAAC,CACnE,CACF,CAAC,UAAU,EAAE,CAAC;YAEf,MAAM,OAAO,GAAG,aAAa;iBAC1B,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;gBACb,MAAM,KAAK,GAAG,KAAK,CAAC,GAAmC,CAAC;gBACxD,OAAO,KAAK,EAAE,GAAG,IAAI,IAAI,CAAC;YAC5B,CAAC,CAAC;iBACD,MAAM,CAAC,CAAC,GAAG,EAAiB,EAAE,CAAC,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC;YAE3D,MAAM,MAAM,GAAG,aAAa;iBACzB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;gBACb,MAAM,GAAG,GAAG,KAAK,CAAC,GAAqC,CAAC;gBACxD,OAAO,GAAG,EAAE,KAAK,IAAI,IAAI,CAAC;YAC5B,CAAC,CAAC;iBACD,MAAM,CAAC,CAAC,KAAK,EAAmB,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC;YAEjE,4EAA4E;YAC5E,IAAA,aAAM,EAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;YAC1E,IAAA,aAAM,EAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;YACrE,IAAA,aAAM,EAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;YACtE,sEAAsE;YACtE,IAAA,aAAM,EACJ,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,yBAAyB,CAAC,CAAC,CAC/D,CAAC,UAAU,EAAE,CAAC;YAEf,MAAM,SAAS,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC/C,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC;gBACtB,OAAO,CACL,GAAG;oBACH,OAAO,GAAG,KAAK,QAAQ;oBACvB,OAAO,IAAK,GAA+B,CAC5C,CAAC;YACJ,CAAC,CAAC,CAAC;YACH,IAAA,aAAM,EAAC,SAAS,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAC5C,MAAM,KAAK,GAAI,SAAS,CAAC,CAAC,CAAC,EAAE,GAAuC;gBAClE,EAAE,KAAK,CAAC;YACV,IAAA,aAAM,EAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;YAC1C,IAAA,aAAM,EAAE,KAAmB,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9C,CAAC;gBAAS,CAAC;YACT,gBAAK,CAAC,SAAS,CAAC,UAAU,GAAG,kBAAkB,CAAC;YAChD,gBAAK,CAAC,SAAS,CAAC,QAAQ,GAAG,gBAAgB,CAAC;YAC5C,YAAY,CAAC,WAAW,CAAC,IAAI,GAAG,YAAY,CAAC;YAC7C,MAAM,aAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAA,WAAI,EAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QAEnC,MAAM,IAAI,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KA8BZ,CAAC;QAEF,MAAM,IAAI,CAAC,IAAI,CAAC,iBAAiB,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC,CAAC;QAE9D,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;QACjC,MAAM,kBAAkB,GAAG,gBAAK,CAAC,SAAS,CAAC,UAAU,CAAC;QACtD,IAAI,eAAe,GAAG,CAAC,CAAC;QAExB,gBAAK,CAAC,SAAS,CAAC,UAAU,GAAG,KAAK,UAAU,aAAa,CAAC,OAAO;YAC/D,MAAM,KAAK,GAAG,IAAa,CAAC;YAC5B,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC9B,eAAe,GAAG,MAAM,KAAK,CAAC,QAAQ,CAAC,GAAG,EAAE;oBAC1C,MAAM,MAAM,GAAG,QAAQ,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC;oBACtD,IAAI,CAAC,MAAM;wBAAE,OAAO,CAAC,CAAC;oBACtB,OAAO,MAAM,CAAC,gBAAgB,CAAC,uBAAuB,CAAC,CAAC,MAAM,CAAC;gBACjE,CAAC,CAAC,CAAC;gBACH,OAAO,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACnC,CAAC;YACD,OAAO,kBAAkB,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAChD,CAAC,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,UAAU,CAAC;gBACpB,IAAI,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;aACtC,CAAC,CAAC;YACH,IAAA,aAAM,EAAC,eAAe,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC;gBAAS,CAAC;YACT,gBAAK,CAAC,SAAS,CAAC,UAAU,GAAG,kBAAkB,CAAC;QAClD,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { test, expect } from \"@playwright/test\";\nimport { promises as fs } from \"fs\";\nimport * as os from \"os\";\nimport * as path from \"path\";\nimport { V3 } from \"../../lib/v3/v3.js\";\nimport { v3TestConfig } from \"./v3.config.js\";\nimport { Frame } from \"../../lib/v3/understudy/frame.js\";\n\nconst wait = (ms: number) => new Promise((resolve) => setTimeout(resolve, ms));\n\ntest.describe(\"Page.screenshot options\", () => {\n  let v3: V3;\n\n  test.beforeEach(async () => {\n    v3 = new V3(v3TestConfig);\n    await v3.init();\n  });\n\n  test.afterEach(async () => {\n    await v3?.close?.().catch(() => {});\n  });\n\n  test(\"rejects clip combined with fullPage\", async () => {\n    const page = v3.context.pages()[0];\n    await page.goto(\"data:text/html,<html><body>test</body></html>\");\n\n    await expect(\n      page.screenshot({\n        fullPage: true,\n        clip: { x: 0, y: 0, width: 100, height: 100 },\n      }),\n    ).rejects.toThrow(/clip and fullPage/);\n  });\n\n  test(\"rejects unsupported image type\", async () => {\n    const page = v3.context.pages()[0];\n    await page.goto(\"data:text/html,<html><body>noop</body></html>\");\n\n    await expect(\n      page.screenshot({\n        // @ts-expect-error intentional invalid type for runtime validation\n        type: \"webp\",\n      }),\n    ).rejects.toThrow(/unsupported image type/);\n  });\n\n  test(\"rejects jpeg quality for png screenshots\", async () => {\n    const page = v3.context.pages()[0];\n    await page.goto(\"data:text/html,<html><body>noop</body></html>\");\n\n    await expect(page.screenshot({ type: \"png\", quality: 50 })).rejects.toThrow(\n      /quality option is only valid/,\n    );\n  });\n\n  test(\"honours timeout option\", async () => {\n    const page = v3.context.pages()[0];\n    await page.goto(\"data:text/html,<html><body>noop</body></html>\");\n\n    const mainFrame = page.mainFrame();\n    const originalScreenshot = mainFrame.screenshot.bind(mainFrame);\n\n    (\n      mainFrame as typeof mainFrame & {\n        screenshot: typeof mainFrame.screenshot;\n      }\n    ).screenshot = async () => {\n      await wait(50);\n      return Buffer.from(\"late\");\n    };\n\n    try {\n      await expect(page.screenshot({ timeout: 10 })).rejects.toThrow(/timeout/);\n    } finally {\n      (\n        mainFrame as typeof mainFrame & {\n          screenshot: typeof mainFrame.screenshot;\n        }\n      ).screenshot = originalScreenshot;\n    }\n  });\n\n  test(\"applies advanced options and cleans up overlays\", async () => {\n    const page = v3.context.pages()[0];\n    const screenshotTimeout = process.env.CI ? 15000 : 5000;\n    const testStart = Date.now();\n    console.log(\n      `[screenshot-test] start ${new Date(testStart).toISOString()} timeout=${screenshotTimeout}`,\n    );\n\n    const html = `\n      <!doctype html>\n      <html>\n        <head>\n          <meta charset=\"utf-8\" />\n          <style>\n            body { background: #aaccee; margin: 0; height: 100vh; display: flex; flex-direction: column; align-items: flex-start; }\n            .mask-target { width: 80px; height: 80px; margin: 40px; background: rgb(0, 180, 60); animation: pulse 1s infinite alternate; }\n            @keyframes pulse { from { transform: scale(1); } to { transform: scale(1.2); } }\n          </style>\n        </head>\n        <body>\n          <div class=\"mask-target\"></div>\n          <div class=\"mask-target\"></div>\n          <input id=\"focus-me\" value=\"focus\" />\n          <script>document.getElementById('focus-me').focus();</script>\n        </body>\n      </html>\n    `;\n\n    await page.goto(\"data:text/html,\" + encodeURIComponent(html));\n    console.log(`[screenshot-test] page loaded in ${Date.now() - testStart}ms`);\n\n    const maskLocator = page.locator(\".mask-target\");\n    const tempPath = path.join(\n      os.tmpdir(),\n      `stagehand-screenshot-${Date.now()}-${Math.random().toString(36).slice(2)}.jpeg`,\n    );\n    console.log(`[screenshot-test] tempPath=${tempPath}`);\n\n    const targetId = page.targetId();\n    const screenshotCalls: Array<{\n      frameId: string;\n      options: Parameters<Frame[\"screenshot\"]>[0];\n    }> = [];\n    const evaluateCalls: Array<{ frameId: string; arg: unknown }> = [];\n    const originalScreenshot = Frame.prototype.screenshot;\n    const originalEvaluate = Frame.prototype.evaluate;\n\n    // Hook Frame.screenshot so we can assert which options reach CDP without writing real data.\n    Frame.prototype.screenshot = async function screenshotSpy(options) {\n      const frame = this as Frame;\n      if (frame.pageId === targetId) {\n        screenshotCalls.push({ frameId: frame.frameId, options });\n        return Buffer.from(\"stub-image\");\n      }\n      return originalScreenshot.call(this, options);\n    };\n\n    // Spy on Frame.evaluate to capture the arguments used to inject CSS/masks.\n    Frame.prototype.evaluate = async function evaluateSpy(expression, arg?) {\n      const frame = this as Frame;\n      if (frame.pageId === targetId) {\n        evaluateCalls.push({ frameId: frame.frameId, arg });\n      }\n      return originalEvaluate.call(this, expression as never, arg);\n    } as Frame[\"evaluate\"];\n\n    const internalPage = page as unknown as {\n      mainSession: {\n        send: (method: string, params?: unknown) => Promise<unknown>;\n      };\n    };\n    const sendCalls: Array<{ method: string; params: unknown }> = [];\n    const originalSend = internalPage.mainSession.send.bind(\n      internalPage.mainSession,\n    ) as (method: string, params?: unknown) => Promise<unknown>;\n    // Capture background overrides so we can confirm omitBackground toggles on/off.\n    internalPage.mainSession.send = async (\n      method: string,\n      params?: unknown,\n    ) => {\n      sendCalls.push({ method, params });\n      return originalSend(method, params);\n    };\n\n    try {\n      const maskCount = await maskLocator.count();\n      console.log(`[screenshot-test] maskLocator.count=${maskCount}`);\n\n      const buffer = await page.screenshot({\n        animations: \"disabled\",\n        caret: \"hide\",\n        clip: { x: 0, y: 0, width: 200, height: 200 },\n        mask: [maskLocator],\n        maskColor: \"rgba(255, 0, 0, 0.4)\",\n        omitBackground: true,\n        path: tempPath,\n        quality: 80,\n        scale: \"css\",\n        style: \"body { border: 3px solid black; }\",\n        timeout: screenshotTimeout,\n        type: \"jpeg\",\n      });\n      console.log(\n        `[screenshot-test] screenshot returned bytes=${buffer.length} elapsed=${Date.now() - testStart}ms`,\n      );\n\n      expect(Buffer.isBuffer(buffer)).toBeTruthy();\n      expect(screenshotCalls.length).toBeGreaterThanOrEqual(1);\n      console.log(\n        `[screenshot-test] screenshotCalls=${screenshotCalls.length} evaluateCalls=${evaluateCalls.length} sendCalls=${sendCalls.length}`,\n      );\n      const recorded = screenshotCalls[0]?.options ?? {};\n      expect(recorded).toMatchObject({ type: \"jpeg\", quality: 80 });\n      expect(recorded?.clip).toMatchObject({\n        x: 0,\n        y: 0,\n        width: 200,\n        height: 200,\n      });\n      if (typeof recorded?.scale === \"number\") {\n        expect(recorded.scale).toBeGreaterThan(0);\n        expect(recorded.scale).toBeLessThanOrEqual(2);\n      }\n\n      await fs.stat(tempPath);\n\n      const maskNodes = await page.evaluate(\n        () => document.querySelectorAll(\"[data-stagehand-mask]\").length,\n      );\n      expect(maskNodes).toBe(0);\n\n      const styleNodes = await page.evaluate(\n        () => document.querySelectorAll(\"[data-stagehand-style]\").length,\n      );\n      expect(styleNodes).toBe(0);\n\n      const backgroundCalls = sendCalls.filter(\n        (call) => call.method === \"Emulation.setDefaultBackgroundColorOverride\",\n      );\n      expect(backgroundCalls.length).toBeGreaterThan(1);\n      expect(\n        backgroundCalls.some(\n          (call) =>\n            call.params &&\n            typeof call.params === \"object\" &&\n            \"color\" in (call.params as Record<string, unknown>),\n        ),\n      ).toBeTruthy();\n      expect(\n        backgroundCalls.some(\n          (call) =>\n            !call.params ||\n            Object.keys(call.params as Record<string, unknown>).length === 0,\n        ),\n      ).toBeTruthy();\n\n      const cssArgs = evaluateCalls\n        .map((entry) => {\n          const value = entry.arg as { css?: string } | undefined;\n          return value?.css ?? null;\n        })\n        .filter((css): css is string => typeof css === \"string\");\n\n      const tokens = evaluateCalls\n        .map((entry) => {\n          const arg = entry.arg as { token?: string } | undefined;\n          return arg?.token ?? null;\n        })\n        .filter((token): token is string => typeof token === \"string\");\n\n      // Tokens include which helper injected the style (animations/caret/custom).\n      expect(tokens.some((token) => token.includes(\"animations\"))).toBeTruthy();\n      expect(tokens.some((token) => token.includes(\"caret\"))).toBeTruthy();\n      expect(tokens.some((token) => token.includes(\"custom\"))).toBeTruthy();\n      // Custom style should bubble through so we check the actual CSS text.\n      expect(\n        cssArgs.some((css) => css.includes(\"border: 3px solid black\")),\n      ).toBeTruthy();\n\n      const maskCalls = evaluateCalls.filter((entry) => {\n        const arg = entry.arg;\n        return (\n          arg &&\n          typeof arg === \"object\" &&\n          \"rects\" in (arg as Record<string, unknown>)\n        );\n      });\n      expect(maskCalls.length).toBeGreaterThan(0);\n      const rects = (maskCalls[0]?.arg as { rects?: unknown } | undefined)\n        ?.rects;\n      expect(Array.isArray(rects)).toBeTruthy();\n      expect((rects as unknown[]).length).toBe(2);\n    } finally {\n      Frame.prototype.screenshot = originalScreenshot;\n      Frame.prototype.evaluate = originalEvaluate;\n      internalPage.mainSession.send = originalSend;\n      await fs.unlink(tempPath).catch(() => {});\n    }\n  });\n\n  test(\"masks elements inside dialog top layer\", async () => {\n    const page = v3.context.pages()[0];\n\n    const html = `\n      <!doctype html>\n      <html>\n        <head>\n          <meta charset=\"utf-8\" />\n          <style>\n            dialog { padding: 16px; border: 2px solid #444; }\n            #dialog-input { display: block; width: 160px; height: 32px; }\n          </style>\n        </head>\n        <body>\n          <dialog id=\"dialog\">\n            <label>Secret <input id=\"dialog-input\" value=\"top-layer\" /></label>\n          </dialog>\n          <script>\n            const dialog = document.getElementById(\"dialog\");\n            if (dialog) {\n              if (typeof dialog.showModal === \"function\") {\n                try {\n                  dialog.showModal();\n                } catch {\n                  dialog.setAttribute(\"open\", \"\");\n                }\n              } else {\n                dialog.setAttribute(\"open\", \"\");\n              }\n            }\n          </script>\n        </body>\n      </html>\n    `;\n\n    await page.goto(\"data:text/html,\" + encodeURIComponent(html));\n\n    const targetId = page.targetId();\n    const originalScreenshot = Frame.prototype.screenshot;\n    let dialogMaskCount = 0;\n\n    Frame.prototype.screenshot = async function screenshotSpy(options) {\n      const frame = this as Frame;\n      if (frame.pageId === targetId) {\n        dialogMaskCount = await frame.evaluate(() => {\n          const dialog = document.querySelector(\"dialog[open]\");\n          if (!dialog) return 0;\n          return dialog.querySelectorAll(\"[data-stagehand-mask]\").length;\n        });\n        return Buffer.from(\"stub-image\");\n      }\n      return originalScreenshot.call(this, options);\n    };\n\n    try {\n      await page.screenshot({\n        mask: [page.locator(\"#dialog-input\")],\n      });\n      expect(dialogMaskCount).toBeGreaterThan(0);\n    } finally {\n      Frame.prototype.screenshot = originalScreenshot;\n    }\n  });\n});\n"]}
+{"version":3,"file":"page-screenshot.spec.js","sourceRoot":"","sources":["../../../../tests/integration/page-screenshot.spec.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,2CAAgD;AAChD,2BAAoC;AACpC,uCAAyB;AACzB,2CAA6B;AAC7B,8CAAwC;AACxC,iDAA8C;AAC9C,+DAAyD;AAEzD,MAAM,IAAI,GAAG,CAAC,EAAU,EAAE,EAAE,CAAC,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAE/E,WAAI,CAAC,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IAC5C,IAAI,EAAM,CAAC;IAEX,WAAI,CAAC,UAAU,CAAC,KAAK,IAAI,EAAE;QACzB,EAAE,GAAG,IAAI,UAAE,CAAC,2BAAY,CAAC,CAAC;QAC1B,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,WAAI,CAAC,SAAS,CAAC,KAAK,IAAI,EAAE;QACxB,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,IAAA,WAAI,EAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,IAAI,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;QAEjE,MAAM,IAAA,aAAM,EACV,IAAI,CAAC,UAAU,CAAC;YACd,QAAQ,EAAE,IAAI;YACd,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;SAC9C,CAAC,CACH,CAAC,OAAO,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,IAAA,WAAI,EAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,IAAI,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;QAEjE,MAAM,IAAA,aAAM,EACV,IAAI,CAAC,UAAU,CAAC;YACd,mEAAmE;YACnE,IAAI,EAAE,MAAM;SACb,CAAC,CACH,CAAC,OAAO,CAAC,OAAO,CAAC,wBAAwB,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,IAAA,WAAI,EAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,IAAI,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;QAEjE,MAAM,IAAA,aAAM,EAAC,IAAI,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACzE,8BAA8B,CAC/B,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,IAAA,WAAI,EAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,IAAI,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;QAEjE,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QACnC,MAAM,kBAAkB,GAAG,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAG9D,SAGD,CAAC,UAAU,GAAG,KAAK,IAAI,EAAE;YACxB,MAAM,IAAI,CAAC,EAAE,CAAC,CAAC;YACf,OAAO,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7B,CAAC,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,IAAA,aAAM,EAAC,IAAI,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC5D,oBAAoB,CACrB,CAAC;QACJ,CAAC;gBAAS,CAAC;YAEP,SAGD,CAAC,UAAU,GAAG,kBAAkB,CAAC;QACpC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAA,WAAI,EAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,iBAAiB,GAAG,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;QACxD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,OAAO,CAAC,GAAG,CACT,2BAA2B,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,YAAY,iBAAiB,EAAE,CAC5F,CAAC;QAEF,MAAM,IAAI,GAAG;;;;;;;;;;;;;;;;;;KAkBZ,CAAC;QAEF,MAAM,IAAI,CAAC,IAAI,CAAC,iBAAiB,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC,CAAC;QAC9D,OAAO,CAAC,GAAG,CAAC,oCAAoC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,IAAI,CAAC,CAAC;QAE5E,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CACxB,EAAE,CAAC,MAAM,EAAE,EACX,wBAAwB,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CACjF,CAAC;QACF,OAAO,CAAC,GAAG,CAAC,8BAA8B,QAAQ,EAAE,CAAC,CAAC;QAEtD,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;QACjC,MAAM,eAAe,GAGhB,EAAE,CAAC;QACR,MAAM,aAAa,GAA6C,EAAE,CAAC;QACnE,MAAM,kBAAkB,GAAG,gBAAK,CAAC,SAAS,CAAC,UAAU,CAAC;QACtD,MAAM,gBAAgB,GAAG,gBAAK,CAAC,SAAS,CAAC,QAAQ,CAAC;QAElD,4FAA4F;QAC5F,gBAAK,CAAC,SAAS,CAAC,UAAU,GAAG,KAAK,UAAU,aAAa,CAAC,OAAO;YAC/D,MAAM,KAAK,GAAG,IAAa,CAAC;YAC5B,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC9B,eAAe,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;gBAC1D,OAAO,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACnC,CAAC;YACD,OAAO,kBAAkB,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAChD,CAAC,CAAC;QAEF,2EAA2E;QAC3E,gBAAK,CAAC,SAAS,CAAC,QAAQ,GAAG,KAAK,UAAU,WAAW,CAAC,UAAU,EAAE,GAAI;YACpE,MAAM,KAAK,GAAG,IAAa,CAAC;YAC5B,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC9B,aAAa,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;YACtD,CAAC;YACD,OAAO,gBAAgB,CAAC,IAAI,CAAC,IAAI,EAAE,UAAmB,EAAE,GAAG,CAAC,CAAC;QAC/D,CAAsB,CAAC;QAEvB,MAAM,YAAY,GAAG,IAIpB,CAAC;QACF,MAAM,SAAS,GAA+C,EAAE,CAAC;QACjE,MAAM,YAAY,GAAG,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CACrD,YAAY,CAAC,WAAW,CACiC,CAAC;QAC5D,gFAAgF;QAChF,YAAY,CAAC,WAAW,CAAC,IAAI,GAAG,KAAK,EACnC,MAAc,EACd,MAAgB,EAChB,EAAE;YACF,SAAS,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;YACnC,OAAO,YAAY,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACtC,CAAC,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,MAAM,WAAW,CAAC,KAAK,EAAE,CAAC;YAC5C,OAAO,CAAC,GAAG,CAAC,uCAAuC,SAAS,EAAE,CAAC,CAAC;YAEhE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC;gBACnC,UAAU,EAAE,UAAU;gBACtB,KAAK,EAAE,MAAM;gBACb,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;gBAC7C,IAAI,EAAE,CAAC,WAAW,CAAC;gBACnB,SAAS,EAAE,sBAAsB;gBACjC,cAAc,EAAE,IAAI;gBACpB,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,EAAE;gBACX,KAAK,EAAE,KAAK;gBACZ,KAAK,EAAE,mCAAmC;gBAC1C,OAAO,EAAE,iBAAiB;gBAC1B,IAAI,EAAE,MAAM;aACb,CAAC,CAAC;YACH,OAAO,CAAC,GAAG,CACT,+CAA+C,MAAM,CAAC,MAAM,YAAY,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,IAAI,CACnG,CAAC;YAEF,IAAA,aAAM,EAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;YAC7C,IAAA,aAAM,EAAC,eAAe,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;YACzD,OAAO,CAAC,GAAG,CACT,qCAAqC,eAAe,CAAC,MAAM,kBAAkB,aAAa,CAAC,MAAM,cAAc,SAAS,CAAC,MAAM,EAAE,CAClI,CAAC;YACF,MAAM,QAAQ,GAAG,eAAe,CAAC,CAAC,CAAC,EAAE,OAAO,IAAI,EAAE,CAAC;YACnD,IAAA,aAAM,EAAC,QAAQ,CAAC,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;YAC9D,IAAA,aAAM,EAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,aAAa,CAAC;gBACnC,CAAC,EAAE,CAAC;gBACJ,CAAC,EAAE,CAAC;gBACJ,KAAK,EAAE,GAAG;gBACV,MAAM,EAAE,GAAG;aACZ,CAAC,CAAC;YACH,IAAI,OAAO,QAAQ,EAAE,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACxC,IAAA,aAAM,EAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;gBAC1C,IAAA,aAAM,EAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;YAChD,CAAC;YAED,MAAM,aAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAExB,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,QAAQ,CACnC,GAAG,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC,uBAAuB,CAAC,CAAC,MAAM,CAChE,CAAC;YACF,IAAA,aAAM,EAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAE1B,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,QAAQ,CACpC,GAAG,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC,wBAAwB,CAAC,CAAC,MAAM,CACjE,CAAC;YACF,IAAA,aAAM,EAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAE3B,MAAM,eAAe,GAAG,SAAS,CAAC,MAAM,CACtC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,6CAA6C,CACxE,CAAC;YACF,IAAA,aAAM,EAAC,eAAe,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAClD,IAAA,aAAM,EACJ,eAAe,CAAC,IAAI,CAClB,CAAC,IAAI,EAAE,EAAE,CACP,IAAI,CAAC,MAAM;gBACX,OAAO,IAAI,CAAC,MAAM,KAAK,QAAQ;gBAC/B,OAAO,IAAK,IAAI,CAAC,MAAkC,CACtD,CACF,CAAC,UAAU,EAAE,CAAC;YACf,IAAA,aAAM,EACJ,eAAe,CAAC,IAAI,CAClB,CAAC,IAAI,EAAE,EAAE,CACP,CAAC,IAAI,CAAC,MAAM;gBACZ,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAiC,CAAC,CAAC,MAAM,KAAK,CAAC,CACnE,CACF,CAAC,UAAU,EAAE,CAAC;YAEf,MAAM,OAAO,GAAG,aAAa;iBAC1B,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;gBACb,MAAM,KAAK,GAAG,KAAK,CAAC,GAAmC,CAAC;gBACxD,OAAO,KAAK,EAAE,GAAG,IAAI,IAAI,CAAC;YAC5B,CAAC,CAAC;iBACD,MAAM,CAAC,CAAC,GAAG,EAAiB,EAAE,CAAC,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC;YAE3D,MAAM,MAAM,GAAG,aAAa;iBACzB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;gBACb,MAAM,GAAG,GAAG,KAAK,CAAC,GAAqC,CAAC;gBACxD,OAAO,GAAG,EAAE,KAAK,IAAI,IAAI,CAAC;YAC5B,CAAC,CAAC;iBACD,MAAM,CAAC,CAAC,KAAK,EAAmB,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC;YAEjE,4EAA4E;YAC5E,IAAA,aAAM,EAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;YAC1E,IAAA,aAAM,EAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;YACrE,IAAA,aAAM,EAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;YACtE,sEAAsE;YACtE,IAAA,aAAM,EACJ,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,yBAAyB,CAAC,CAAC,CAC/D,CAAC,UAAU,EAAE,CAAC;YAEf,MAAM,SAAS,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC/C,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC;gBACtB,OAAO,CACL,GAAG;oBACH,OAAO,GAAG,KAAK,QAAQ;oBACvB,OAAO,IAAK,GAA+B,CAC5C,CAAC;YACJ,CAAC,CAAC,CAAC;YACH,IAAA,aAAM,EAAC,SAAS,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAC5C,MAAM,KAAK,GAAI,SAAS,CAAC,CAAC,CAAC,EAAE,GAAuC;gBAClE,EAAE,KAAK,CAAC;YACV,IAAA,aAAM,EAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;YAC1C,IAAA,aAAM,EAAE,KAAmB,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9C,CAAC;gBAAS,CAAC;YACT,gBAAK,CAAC,SAAS,CAAC,UAAU,GAAG,kBAAkB,CAAC;YAChD,gBAAK,CAAC,SAAS,CAAC,QAAQ,GAAG,gBAAgB,CAAC;YAC5C,YAAY,CAAC,WAAW,CAAC,IAAI,GAAG,YAAY,CAAC;YAC7C,MAAM,aAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAA,WAAI,EAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QAEnC,MAAM,IAAI,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KA8BZ,CAAC;QAEF,MAAM,IAAI,CAAC,IAAI,CAAC,iBAAiB,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC,CAAC;QAE9D,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;QACjC,MAAM,kBAAkB,GAAG,gBAAK,CAAC,SAAS,CAAC,UAAU,CAAC;QACtD,IAAI,eAAe,GAAG,CAAC,CAAC;QAExB,gBAAK,CAAC,SAAS,CAAC,UAAU,GAAG,KAAK,UAAU,aAAa,CAAC,OAAO;YAC/D,MAAM,KAAK,GAAG,IAAa,CAAC;YAC5B,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC9B,eAAe,GAAG,MAAM,KAAK,CAAC,QAAQ,CAAC,GAAG,EAAE;oBAC1C,MAAM,MAAM,GAAG,QAAQ,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC;oBACtD,IAAI,CAAC,MAAM;wBAAE,OAAO,CAAC,CAAC;oBACtB,OAAO,MAAM,CAAC,gBAAgB,CAAC,uBAAuB,CAAC,CAAC,MAAM,CAAC;gBACjE,CAAC,CAAC,CAAC;gBACH,OAAO,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACnC,CAAC;YACD,OAAO,kBAAkB,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAChD,CAAC,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,UAAU,CAAC;gBACpB,IAAI,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;aACtC,CAAC,CAAC;YACH,IAAA,aAAM,EAAC,eAAe,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC;gBAAS,CAAC;YACT,gBAAK,CAAC,SAAS,CAAC,UAAU,GAAG,kBAAkB,CAAC;QAClD,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { test, expect } from \"@playwright/test\";\nimport { promises as fs } from \"fs\";\nimport * as os from \"os\";\nimport * as path from \"path\";\nimport { V3 } from \"../../lib/v3/v3.js\";\nimport { v3TestConfig } from \"./v3.config.js\";\nimport { Frame } from \"../../lib/v3/understudy/frame.js\";\n\nconst wait = (ms: number) => new Promise((resolve) => setTimeout(resolve, ms));\n\ntest.describe(\"Page.screenshot options\", () => {\n  let v3: V3;\n\n  test.beforeEach(async () => {\n    v3 = new V3(v3TestConfig);\n    await v3.init();\n  });\n\n  test.afterEach(async () => {\n    await v3?.close?.().catch(() => {});\n  });\n\n  test(\"rejects clip combined with fullPage\", async () => {\n    const page = v3.context.pages()[0];\n    await page.goto(\"data:text/html,<html><body>test</body></html>\");\n\n    await expect(\n      page.screenshot({\n        fullPage: true,\n        clip: { x: 0, y: 0, width: 100, height: 100 },\n      }),\n    ).rejects.toThrow(/clip and fullPage/);\n  });\n\n  test(\"rejects unsupported image type\", async () => {\n    const page = v3.context.pages()[0];\n    await page.goto(\"data:text/html,<html><body>noop</body></html>\");\n\n    await expect(\n      page.screenshot({\n        // @ts-expect-error intentional invalid type for runtime validation\n        type: \"webp\",\n      }),\n    ).rejects.toThrow(/unsupported image type/);\n  });\n\n  test(\"rejects jpeg quality for png screenshots\", async () => {\n    const page = v3.context.pages()[0];\n    await page.goto(\"data:text/html,<html><body>noop</body></html>\");\n\n    await expect(page.screenshot({ type: \"png\", quality: 50 })).rejects.toThrow(\n      /quality option is only valid/,\n    );\n  });\n\n  test(\"honours timeout option\", async () => {\n    const page = v3.context.pages()[0];\n    await page.goto(\"data:text/html,<html><body>noop</body></html>\");\n\n    const mainFrame = page.mainFrame();\n    const originalScreenshot = mainFrame.screenshot.bind(mainFrame);\n\n    (\n      mainFrame as typeof mainFrame & {\n        screenshot: typeof mainFrame.screenshot;\n      }\n    ).screenshot = async () => {\n      await wait(50);\n      return Buffer.from(\"late\");\n    };\n\n    try {\n      await expect(page.screenshot({ timeout: 10 })).rejects.toThrow(\n        /timed out|timeout/i,\n      );\n    } finally {\n      (\n        mainFrame as typeof mainFrame & {\n          screenshot: typeof mainFrame.screenshot;\n        }\n      ).screenshot = originalScreenshot;\n    }\n  });\n\n  test(\"applies advanced options and cleans up overlays\", async () => {\n    const page = v3.context.pages()[0];\n    const screenshotTimeout = process.env.CI ? 15000 : 5000;\n    const testStart = Date.now();\n    console.log(\n      `[screenshot-test] start ${new Date(testStart).toISOString()} timeout=${screenshotTimeout}`,\n    );\n\n    const html = `\n      <!doctype html>\n      <html>\n        <head>\n          <meta charset=\"utf-8\" />\n          <style>\n            body { background: #aaccee; margin: 0; height: 100vh; display: flex; flex-direction: column; align-items: flex-start; }\n            .mask-target { width: 80px; height: 80px; margin: 40px; background: rgb(0, 180, 60); animation: pulse 1s infinite alternate; }\n            @keyframes pulse { from { transform: scale(1); } to { transform: scale(1.2); } }\n          </style>\n        </head>\n        <body>\n          <div class=\"mask-target\"></div>\n          <div class=\"mask-target\"></div>\n          <input id=\"focus-me\" value=\"focus\" />\n          <script>document.getElementById('focus-me').focus();</script>\n        </body>\n      </html>\n    `;\n\n    await page.goto(\"data:text/html,\" + encodeURIComponent(html));\n    console.log(`[screenshot-test] page loaded in ${Date.now() - testStart}ms`);\n\n    const maskLocator = page.locator(\".mask-target\");\n    const tempPath = path.join(\n      os.tmpdir(),\n      `stagehand-screenshot-${Date.now()}-${Math.random().toString(36).slice(2)}.jpeg`,\n    );\n    console.log(`[screenshot-test] tempPath=${tempPath}`);\n\n    const targetId = page.targetId();\n    const screenshotCalls: Array<{\n      frameId: string;\n      options: Parameters<Frame[\"screenshot\"]>[0];\n    }> = [];\n    const evaluateCalls: Array<{ frameId: string; arg: unknown }> = [];\n    const originalScreenshot = Frame.prototype.screenshot;\n    const originalEvaluate = Frame.prototype.evaluate;\n\n    // Hook Frame.screenshot so we can assert which options reach CDP without writing real data.\n    Frame.prototype.screenshot = async function screenshotSpy(options) {\n      const frame = this as Frame;\n      if (frame.pageId === targetId) {\n        screenshotCalls.push({ frameId: frame.frameId, options });\n        return Buffer.from(\"stub-image\");\n      }\n      return originalScreenshot.call(this, options);\n    };\n\n    // Spy on Frame.evaluate to capture the arguments used to inject CSS/masks.\n    Frame.prototype.evaluate = async function evaluateSpy(expression, arg?) {\n      const frame = this as Frame;\n      if (frame.pageId === targetId) {\n        evaluateCalls.push({ frameId: frame.frameId, arg });\n      }\n      return originalEvaluate.call(this, expression as never, arg);\n    } as Frame[\"evaluate\"];\n\n    const internalPage = page as unknown as {\n      mainSession: {\n        send: (method: string, params?: unknown) => Promise<unknown>;\n      };\n    };\n    const sendCalls: Array<{ method: string; params: unknown }> = [];\n    const originalSend = internalPage.mainSession.send.bind(\n      internalPage.mainSession,\n    ) as (method: string, params?: unknown) => Promise<unknown>;\n    // Capture background overrides so we can confirm omitBackground toggles on/off.\n    internalPage.mainSession.send = async (\n      method: string,\n      params?: unknown,\n    ) => {\n      sendCalls.push({ method, params });\n      return originalSend(method, params);\n    };\n\n    try {\n      const maskCount = await maskLocator.count();\n      console.log(`[screenshot-test] maskLocator.count=${maskCount}`);\n\n      const buffer = await page.screenshot({\n        animations: \"disabled\",\n        caret: \"hide\",\n        clip: { x: 0, y: 0, width: 200, height: 200 },\n        mask: [maskLocator],\n        maskColor: \"rgba(255, 0, 0, 0.4)\",\n        omitBackground: true,\n        path: tempPath,\n        quality: 80,\n        scale: \"css\",\n        style: \"body { border: 3px solid black; }\",\n        timeout: screenshotTimeout,\n        type: \"jpeg\",\n      });\n      console.log(\n        `[screenshot-test] screenshot returned bytes=${buffer.length} elapsed=${Date.now() - testStart}ms`,\n      );\n\n      expect(Buffer.isBuffer(buffer)).toBeTruthy();\n      expect(screenshotCalls.length).toBeGreaterThanOrEqual(1);\n      console.log(\n        `[screenshot-test] screenshotCalls=${screenshotCalls.length} evaluateCalls=${evaluateCalls.length} sendCalls=${sendCalls.length}`,\n      );\n      const recorded = screenshotCalls[0]?.options ?? {};\n      expect(recorded).toMatchObject({ type: \"jpeg\", quality: 80 });\n      expect(recorded?.clip).toMatchObject({\n        x: 0,\n        y: 0,\n        width: 200,\n        height: 200,\n      });\n      if (typeof recorded?.scale === \"number\") {\n        expect(recorded.scale).toBeGreaterThan(0);\n        expect(recorded.scale).toBeLessThanOrEqual(2);\n      }\n\n      await fs.stat(tempPath);\n\n      const maskNodes = await page.evaluate(\n        () => document.querySelectorAll(\"[data-stagehand-mask]\").length,\n      );\n      expect(maskNodes).toBe(0);\n\n      const styleNodes = await page.evaluate(\n        () => document.querySelectorAll(\"[data-stagehand-style]\").length,\n      );\n      expect(styleNodes).toBe(0);\n\n      const backgroundCalls = sendCalls.filter(\n        (call) => call.method === \"Emulation.setDefaultBackgroundColorOverride\",\n      );\n      expect(backgroundCalls.length).toBeGreaterThan(1);\n      expect(\n        backgroundCalls.some(\n          (call) =>\n            call.params &&\n            typeof call.params === \"object\" &&\n            \"color\" in (call.params as Record<string, unknown>),\n        ),\n      ).toBeTruthy();\n      expect(\n        backgroundCalls.some(\n          (call) =>\n            !call.params ||\n            Object.keys(call.params as Record<string, unknown>).length === 0,\n        ),\n      ).toBeTruthy();\n\n      const cssArgs = evaluateCalls\n        .map((entry) => {\n          const value = entry.arg as { css?: string } | undefined;\n          return value?.css ?? null;\n        })\n        .filter((css): css is string => typeof css === \"string\");\n\n      const tokens = evaluateCalls\n        .map((entry) => {\n          const arg = entry.arg as { token?: string } | undefined;\n          return arg?.token ?? null;\n        })\n        .filter((token): token is string => typeof token === \"string\");\n\n      // Tokens include which helper injected the style (animations/caret/custom).\n      expect(tokens.some((token) => token.includes(\"animations\"))).toBeTruthy();\n      expect(tokens.some((token) => token.includes(\"caret\"))).toBeTruthy();\n      expect(tokens.some((token) => token.includes(\"custom\"))).toBeTruthy();\n      // Custom style should bubble through so we check the actual CSS text.\n      expect(\n        cssArgs.some((css) => css.includes(\"border: 3px solid black\")),\n      ).toBeTruthy();\n\n      const maskCalls = evaluateCalls.filter((entry) => {\n        const arg = entry.arg;\n        return (\n          arg &&\n          typeof arg === \"object\" &&\n          \"rects\" in (arg as Record<string, unknown>)\n        );\n      });\n      expect(maskCalls.length).toBeGreaterThan(0);\n      const rects = (maskCalls[0]?.arg as { rects?: unknown } | undefined)\n        ?.rects;\n      expect(Array.isArray(rects)).toBeTruthy();\n      expect((rects as unknown[]).length).toBe(2);\n    } finally {\n      Frame.prototype.screenshot = originalScreenshot;\n      Frame.prototype.evaluate = originalEvaluate;\n      internalPage.mainSession.send = originalSend;\n      await fs.unlink(tempPath).catch(() => {});\n    }\n  });\n\n  test(\"masks elements inside dialog top layer\", async () => {\n    const page = v3.context.pages()[0];\n\n    const html = `\n      <!doctype html>\n      <html>\n        <head>\n          <meta charset=\"utf-8\" />\n          <style>\n            dialog { padding: 16px; border: 2px solid #444; }\n            #dialog-input { display: block; width: 160px; height: 32px; }\n          </style>\n        </head>\n        <body>\n          <dialog id=\"dialog\">\n            <label>Secret <input id=\"dialog-input\" value=\"top-layer\" /></label>\n          </dialog>\n          <script>\n            const dialog = document.getElementById(\"dialog\");\n            if (dialog) {\n              if (typeof dialog.showModal === \"function\") {\n                try {\n                  dialog.showModal();\n                } catch {\n                  dialog.setAttribute(\"open\", \"\");\n                }\n              } else {\n                dialog.setAttribute(\"open\", \"\");\n              }\n            }\n          </script>\n        </body>\n      </html>\n    `;\n\n    await page.goto(\"data:text/html,\" + encodeURIComponent(html));\n\n    const targetId = page.targetId();\n    const originalScreenshot = Frame.prototype.screenshot;\n    let dialogMaskCount = 0;\n\n    Frame.prototype.screenshot = async function screenshotSpy(options) {\n      const frame = this as Frame;\n      if (frame.pageId === targetId) {\n        dialogMaskCount = await frame.evaluate(() => {\n          const dialog = document.querySelector(\"dialog[open]\");\n          if (!dialog) return 0;\n          return dialog.querySelectorAll(\"[data-stagehand-mask]\").length;\n        });\n        return Buffer.from(\"stub-image\");\n      }\n      return originalScreenshot.call(this, options);\n    };\n\n    try {\n      await page.screenshot({\n        mask: [page.locator(\"#dialog-input\")],\n      });\n      expect(dialogMaskCount).toBeGreaterThan(0);\n    } finally {\n      Frame.prototype.screenshot = originalScreenshot;\n    }\n  });\n});\n"]}

package/dist/cjs/tests/integration/testUtils.d.ts

@@ -1,7 +1,40 @@
 import type { V3 } from "../../lib/v3/v3.js";
+import type { LanguageModelV2CallOptions, LanguageModelV2Content, LanguageModelV2FinishReason, LanguageModelV2Usage } from "@ai-sdk/provider";
+import { AISdkClient } from "../../lib/v3/llm/aisdk.js";
 /**
  * Races a promise against a timeout.
  * Resolves to the promise value or "timeout" if the deadline expires.
  */
 export declare function raceTimeout<T>(promise: Promise<T>, ms: number): Promise<T | "timeout">;
 export declare function closeV3(v3?: V3 | null): Promise<void>;
+type JsonResponseKey = "act" | "Observation" | "Metadata" | "Extraction" | "default";
+type JsonResponseValue = Record<string, unknown> | ((options: LanguageModelV2CallOptions) => Record<string, unknown>);
+type JsonResponseScript = JsonResponseValue | JsonResponseValue[];
+type GenerateResponseValue = {
+    content: LanguageModelV2Content[];
+    finishReason?: LanguageModelV2FinishReason;
+    usage?: Partial<LanguageModelV2Usage>;
+} | ((options: LanguageModelV2CallOptions) => {
+    content: LanguageModelV2Content[];
+    finishReason?: LanguageModelV2FinishReason;
+    usage?: Partial<LanguageModelV2Usage>;
+});
+export declare function promptToText(prompt: LanguageModelV2CallOptions["prompt"]): string;
+export declare function findEncodedIdForText(options: LanguageModelV2CallOptions, text: string): string;
+export declare function findLastEncodedId(options: LanguageModelV2CallOptions): string;
+export declare function toolCallResponse(toolName: string, input: Record<string, unknown>, toolCallId?: string): {
+    content: LanguageModelV2Content[];
+    finishReason: LanguageModelV2FinishReason;
+    usage: LanguageModelV2Usage;
+};
+export declare function doneToolResponse(reasoning?: string, taskComplete?: boolean, toolCallId?: string): {
+    content: LanguageModelV2Content[];
+    finishReason: LanguageModelV2FinishReason;
+    usage: LanguageModelV2Usage;
+};
+export declare function createScriptedAisdkTestLlmClient(options?: {
+    modelId?: string;
+    jsonResponses?: Partial<Record<JsonResponseKey, JsonResponseScript>>;
+    generateResponses?: GenerateResponseValue[];
+}): AISdkClient;
+export {};

package/dist/cjs/tests/integration/testUtils.js

@@ -2,6 +2,13 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.raceTimeout = raceTimeout;
 exports.closeV3 = closeV3;
+exports.promptToText = promptToText;
+exports.findEncodedIdForText = findEncodedIdForText;
+exports.findLastEncodedId = findLastEncodedId;
+exports.toolCallResponse = toolCallResponse;
+exports.doneToolResponse = doneToolResponse;
+exports.createScriptedAisdkTestLlmClient = createScriptedAisdkTestLlmClient;
+const aisdk_js_1 = require("../../lib/v3/llm/aisdk.js");
 /**
  * Races a promise against a timeout.
  * Resolves to the promise value or "timeout" if the deadline expires.
@@ -41,4 +48,141 @@ async function closeV3(v3) {
     }
     await settleWithTimeout(v3.close(), CLOSE_TIMEOUT_MS);
 }
+const DEFAULT_USAGE = {
+    inputTokens: 1,
+    outputTokens: 1,
+    totalTokens: 2,
+    reasoningTokens: 0,
+    cachedInputTokens: 0,
+};
+const mergeUsage = (usage) => ({
+    ...DEFAULT_USAGE,
+    ...(usage ?? {}),
+});
+function consumeScriptValue(value, fallback) {
+    if (!Array.isArray(value)) {
+        return value ?? fallback;
+    }
+    if (value.length <= 1) {
+        return value[0] ?? fallback;
+    }
+    return value.shift() ?? fallback;
+}
+function resolveJsonResponseKey(options) {
+    const responseFormat = options.responseFormat;
+    if (!responseFormat || responseFormat.type !== "json") {
+        return "default";
+    }
+    const schema = responseFormat.schema;
+    const properties = schema?.properties ?? {};
+    if ("elementId" in properties && "twoStep" in properties) {
+        return "act";
+    }
+    if ("elements" in properties) {
+        return "Observation";
+    }
+    if ("completed" in properties && "progress" in properties) {
+        return "Metadata";
+    }
+    return "Extraction";
+}
+function promptToText(prompt) {
+    return (prompt ?? [])
+        .flatMap((message) => {
+        if (typeof message.content === "string") {
+            return [message.content];
+        }
+        return (message.content ?? [])
+            .map((part) => (part.type === "text" ? part.text : ""))
+            .filter((text) => text.length > 0);
+    })
+        .join("\n");
+}
+function findEncodedIds(options) {
+    return [...promptToText(options.prompt).matchAll(/\b\d+-\d+\b/g)].map((match) => match[0]);
+}
+function findEncodedIdForText(options, text) {
+    const promptText = promptToText(options.prompt);
+    const lines = promptText.split("\n");
+    const line = lines.find((entry) => entry.includes(text));
+    const match = line?.match(/\b\d+-\d+\b/);
+    if (!match) {
+        throw new Error(`Could not find encoded id for text: ${text}`);
+    }
+    return match[0];
+}
+function findLastEncodedId(options) {
+    const matches = findEncodedIds(options);
+    if (matches.length === 0) {
+        throw new Error("Could not find any encoded ids in the prompt.");
+    }
+    return matches[matches.length - 1];
+}
+function toolCallResponse(toolName, input, toolCallId = `${toolName}-1`) {
+    return {
+        content: [
+            {
+                type: "tool-call",
+                toolCallId,
+                toolName,
+                input: JSON.stringify(input),
+            },
+        ],
+        finishReason: "tool-calls",
+        usage: DEFAULT_USAGE,
+    };
+}
+function doneToolResponse(reasoning = "done", taskComplete = true, toolCallId = "done-1") {
+    return toolCallResponse("done", { reasoning, taskComplete }, toolCallId);
+}
+function createGenerateResult(result) {
+    return {
+        content: result.content,
+        finishReason: result.finishReason ?? "stop",
+        usage: mergeUsage(result.usage),
+        warnings: [],
+    };
+}
+function createScriptedAisdkTestLlmClient(options) {
+    const jsonResponses = Object.fromEntries(Object.entries(options?.jsonResponses ?? {}).map(([key, value]) => [
+        key,
+        Array.isArray(value) ? [...value] : value,
+    ]));
+    const generateResponses = [...(options?.generateResponses ?? [])];
+    const model = {
+        provider: "mock",
+        modelId: options?.modelId ?? "mock/stagehand-flow-logger",
+        specificationVersion: "v2",
+        supportedUrls: {},
+        doGenerateCalls: [],
+        doGenerate: async (callOptions) => {
+            model.doGenerateCalls.push(callOptions);
+            if (callOptions.responseFormat?.type === "json") {
+                const key = resolveJsonResponseKey(callOptions);
+                const responseScripts = consumeScriptValue(jsonResponses[key], jsonResponses.default);
+                const responseScript = consumeScriptValue(responseScripts, undefined);
+                const response = typeof responseScript === "function"
+                    ? responseScript(callOptions)
+                    : (responseScript ?? {});
+                return createGenerateResult({
+                    content: [{ type: "text", text: JSON.stringify(response) }],
+                });
+            }
+            const responseScript = consumeScriptValue(generateResponses, undefined);
+            if (!responseScript) {
+                return createGenerateResult({
+                    content: [{ type: "text", text: "done" }],
+                });
+            }
+            const response = typeof responseScript === "function"
+                ? responseScript(callOptions)
+                : responseScript;
+            return createGenerateResult(response);
+        },
+        doStream: async () => {
+            throw new Error("Streaming is not implemented for this test model.");
+        },
+    };
+    return new aisdk_js_1.AISdkClient({ model });
+}
 //# sourceMappingURL=testUtils.js.map

package/dist/cjs/tests/integration/testUtils.js.map

@@ -1 +1 @@
-{"version":3,"file":"testUtils.js","sourceRoot":"","sources":["../../../../tests/integration/testUtils.ts"],"names":[],"mappings":";;AAMA,kCASC;AAmBD,0BAeC;AA/CD;;;GAGG;AACH,SAAgB,WAAW,CACzB,OAAmB,EACnB,EAAU;IAEV,IAAI,KAAoC,CAAC;IACzC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAY,CAAC,OAAO,EAAE,EAAE;QACjD,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IACH,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC;AAC7E,CAAC;AAED,MAAM,gBAAgB,GAAG,KAAK,CAAC;AAE/B,KAAK,UAAU,iBAAiB,CAC9B,OAAyB,EACzB,SAAiB;IAEjB,IAAI,SAAqC,CAAC;IAC1C,MAAM,OAAO,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;QAC5C,SAAS,GAAG,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IACH,IAAI,CAAC;QACH,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;IACzD,CAAC;YAAS,CAAC;QACT,IAAI,SAAS;YAAE,YAAY,CAAC,SAAS,CAAC,CAAC;IACzC,CAAC;AACH,CAAC;AAEM,KAAK,UAAU,OAAO,CAAC,EAAc;IAC1C,IAAI,CAAC,EAAE;QAAE,OAAO;IAChB,MAAM,aAAa,GAAG,EAAE,CAAC,aAAa,CAAC;IACvC,IAAI,aAAa,EAAE,CAAC;QAClB,IAAI,CAAC;YACH,MAAM,iBAAiB,CACrB,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,EACrC,gBAAgB,CACjB,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,sBAAsB;QACxB,CAAC;IACH,CAAC;IAED,MAAM,iBAAiB,CAAC,EAAE,CAAC,KAAK,EAAE,EAAE,gBAAgB,CAAC,CAAC;AACxD,CAAC","sourcesContent":["import type { V3 } from \"../../lib/v3/v3.js\";\n\n/**\n * Races a promise against a timeout.\n * Resolves to the promise value or \"timeout\" if the deadline expires.\n */\nexport function raceTimeout<T>(\n  promise: Promise<T>,\n  ms: number,\n): Promise<T | \"timeout\"> {\n  let timer: ReturnType<typeof setTimeout>;\n  const timeout = new Promise<\"timeout\">((resolve) => {\n    timer = setTimeout(() => resolve(\"timeout\"), ms);\n  });\n  return Promise.race([promise, timeout]).finally(() => clearTimeout(timer));\n}\n\nconst CLOSE_TIMEOUT_MS = 5_000;\n\nasync function settleWithTimeout(\n  promise: Promise<unknown>,\n  timeoutMs: number,\n): Promise<void> {\n  let timeoutId: NodeJS.Timeout | undefined;\n  const timeout = new Promise<void>((resolve) => {\n    timeoutId = setTimeout(resolve, timeoutMs);\n  });\n  try {\n    await Promise.race([promise.catch(() => {}), timeout]);\n  } finally {\n    if (timeoutId) clearTimeout(timeoutId);\n  }\n}\n\nexport async function closeV3(v3?: V3 | null): Promise<void> {\n  if (!v3) return;\n  const isBrowserbase = v3.isBrowserbase;\n  if (isBrowserbase) {\n    try {\n      await settleWithTimeout(\n        v3.context.conn.send(\"Browser.close\"),\n        CLOSE_TIMEOUT_MS,\n      );\n    } catch {\n      // best-effort cleanup\n    }\n  }\n\n  await settleWithTimeout(v3.close(), CLOSE_TIMEOUT_MS);\n}\n"]}
+{"version":3,"file":"testUtils.js","sourceRoot":"","sources":["../../../../tests/integration/testUtils.ts"],"names":[],"mappings":";;AAcA,kCASC;AAmBD,0BAeC;AA6FD,oCAcC;AAQD,oDAcC;AAED,8CAOC;AAED,4CAqBC;AAED,4CAUC;AAgBD,4EA+DC;AA7SD,wDAAwD;AAExD;;;GAGG;AACH,SAAgB,WAAW,CACzB,OAAmB,EACnB,EAAU;IAEV,IAAI,KAAoC,CAAC;IACzC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAY,CAAC,OAAO,EAAE,EAAE;QACjD,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IACH,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC;AAC7E,CAAC;AAED,MAAM,gBAAgB,GAAG,KAAK,CAAC;AAE/B,KAAK,UAAU,iBAAiB,CAC9B,OAAyB,EACzB,SAAiB;IAEjB,IAAI,SAAqC,CAAC;IAC1C,MAAM,OAAO,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;QAC5C,SAAS,GAAG,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IACH,IAAI,CAAC;QACH,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;IACzD,CAAC;YAAS,CAAC;QACT,IAAI,SAAS;YAAE,YAAY,CAAC,SAAS,CAAC,CAAC;IACzC,CAAC;AACH,CAAC;AAEM,KAAK,UAAU,OAAO,CAAC,EAAc;IAC1C,IAAI,CAAC,EAAE;QAAE,OAAO;IAChB,MAAM,aAAa,GAAG,EAAE,CAAC,aAAa,CAAC;IACvC,IAAI,aAAa,EAAE,CAAC;QAClB,IAAI,CAAC;YACH,MAAM,iBAAiB,CACrB,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,EACrC,gBAAgB,CACjB,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,sBAAsB;QACxB,CAAC;IACH,CAAC;IAED,MAAM,iBAAiB,CAAC,EAAE,CAAC,KAAK,EAAE,EAAE,gBAAgB,CAAC,CAAC;AACxD,CAAC;AAqCD,MAAM,aAAa,GAAyB;IAC1C,WAAW,EAAE,CAAC;IACd,YAAY,EAAE,CAAC;IACf,WAAW,EAAE,CAAC;IACd,eAAe,EAAE,CAAC;IAClB,iBAAiB,EAAE,CAAC;CACrB,CAAC;AAEF,MAAM,UAAU,GAAG,CACjB,KAAqC,EACf,EAAE,CAAC,CAAC;IAC1B,GAAG,aAAa;IAChB,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC;CACjB,CAAC,CAAC;AAEH,SAAS,kBAAkB,CAAI,KAA0B,EAAE,QAAW;IACpE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,KAAK,IAAI,QAAQ,CAAC;IAC3B,CAAC;IAED,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;QACtB,OAAO,KAAK,CAAC,CAAC,CAAC,IAAI,QAAQ,CAAC;IAC9B,CAAC;IAED,OAAO,KAAK,CAAC,KAAK,EAAE,IAAI,QAAQ,CAAC;AACnC,CAAC;AAED,SAAS,sBAAsB,CAC7B,OAAmC;IAEnC,MAAM,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC;IAC9C,IAAI,CAAC,cAAc,IAAI,cAAc,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QACtD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,MAAM,GAAG,cAAc,CAAC,MAG7B,CAAC;IACF,MAAM,UAAU,GAAG,MAAM,EAAE,UAAU,IAAI,EAAE,CAAC;IAE5C,IAAI,WAAW,IAAI,UAAU,IAAI,SAAS,IAAI,UAAU,EAAE,CAAC;QACzD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,UAAU,IAAI,UAAU,EAAE,CAAC;QAC7B,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,IAAI,WAAW,IAAI,UAAU,IAAI,UAAU,IAAI,UAAU,EAAE,CAAC;QAC1D,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,SAAgB,YAAY,CAC1B,MAA4C;IAE5C,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC;SAClB,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QACnB,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YACxC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC3B,CAAC;QAED,OAAO,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;aAC3B,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;aACtD,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACvD,CAAC,CAAC;SACD,IAAI,CAAC,IAAI,CAAC,CAAC;AAChB,CAAC;AAED,SAAS,cAAc,CAAC,OAAmC;IACzD,OAAO,CAAC,GAAG,YAAY,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,CAAC,GAAG,CACnE,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CACpB,CAAC;AACJ,CAAC;AAED,SAAgB,oBAAoB,CAClC,OAAmC,EACnC,IAAY;IAEZ,MAAM,UAAU,GAAG,YAAY,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAChD,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACrC,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;IACzD,MAAM,KAAK,GAAG,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,CAAC;IAEzC,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CAAC,uCAAuC,IAAI,EAAE,CAAC,CAAC;IACjE,CAAC;IAED,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC;AAED,SAAgB,iBAAiB,CAAC,OAAmC;IACnE,MAAM,OAAO,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;IACxC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;IACnE,CAAC;IAED,OAAO,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AACrC,CAAC;AAED,SAAgB,gBAAgB,CAC9B,QAAgB,EAChB,KAA8B,EAC9B,UAAU,GAAG,GAAG,QAAQ,IAAI;IAM5B,OAAO;QACL,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,WAAW;gBACjB,UAAU;gBACV,QAAQ;gBACR,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;aAC7B;SACF;QACD,YAAY,EAAE,YAAY;QAC1B,KAAK,EAAE,aAAa;KACrB,CAAC;AACJ,CAAC;AAED,SAAgB,gBAAgB,CAC9B,SAAS,GAAG,MAAM,EAClB,YAAY,GAAG,IAAI,EACnB,UAAU,GAAG,QAAQ;IAMrB,OAAO,gBAAgB,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,YAAY,EAAE,EAAE,UAAU,CAAC,CAAC;AAC3E,CAAC;AAED,SAAS,oBAAoB,CAAC,MAA8B;IAM1D,OAAO;QACL,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,YAAY,EAAE,MAAM,CAAC,YAAY,IAAI,MAAM;QAC3C,KAAK,EAAE,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC;QAC/B,QAAQ,EAAE,EAAE;KACb,CAAC;AACJ,CAAC;AAED,SAAgB,gCAAgC,CAAC,OAIhD;IACC,MAAM,aAAa,GAAG,MAAM,CAAC,WAAW,CACtC,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,aAAa,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC;QACjE,GAAG;QACH,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK;KAC1C,CAAC,CACqD,CAAC;IAC1D,MAAM,iBAAiB,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,iBAAiB,IAAI,EAAE,CAAC,CAAC,CAAC;IAElE,MAAM,KAAK,GAA0B;QACnC,QAAQ,EAAE,MAAM;QAChB,OAAO,EAAE,OAAO,EAAE,OAAO,IAAI,4BAA4B;QACzD,oBAAoB,EAAE,IAAI;QAC1B,aAAa,EAAE,EAAE;QACjB,eAAe,EAAE,EAAE;QACnB,UAAU,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE;YAChC,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAExC,IAAI,WAAW,CAAC,cAAc,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;gBAChD,MAAM,GAAG,GAAG,sBAAsB,CAAC,WAAW,CAAC,CAAC;gBAChD,MAAM,eAAe,GAAG,kBAAkB,CAExC,aAAa,CAAC,GAAG,CAAC,EAAE,aAAa,CAAC,OAAO,CAAC,CAAC;gBAC7C,MAAM,cAAc,GAAG,kBAAkB,CAEvC,eAAe,EAAE,SAAS,CAAC,CAAC;gBAC9B,MAAM,QAAQ,GACZ,OAAO,cAAc,KAAK,UAAU;oBAClC,CAAC,CAAC,cAAc,CAAC,WAAW,CAAC;oBAC7B,CAAC,CAAC,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC;gBAE7B,OAAO,oBAAoB,CAAC;oBAC1B,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC;iBAC5D,CAAC,CAAC;YACL,CAAC;YAED,MAAM,cAAc,GAAG,kBAAkB,CAEvC,iBAAiB,EAAE,SAAS,CAAC,CAAC;YAEhC,IAAI,CAAC,cAAc,EAAE,CAAC;gBACpB,OAAO,oBAAoB,CAAC;oBAC1B,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;iBAC1C,CAAC,CAAC;YACL,CAAC;YAED,MAAM,QAAQ,GACZ,OAAO,cAAc,KAAK,UAAU;gBAClC,CAAC,CAAC,cAAc,CAAC,WAAW,CAAC;gBAC7B,CAAC,CAAC,cAAc,CAAC;YAErB,OAAO,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QACxC,CAAC;QACD,QAAQ,EAAE,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;QACvE,CAAC;KACF,CAAC;IAEF,OAAO,IAAI,sBAAW,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;AACpC,CAAC","sourcesContent":["import type { V3 } from \"../../lib/v3/v3.js\";\nimport type {\n  LanguageModelV2,\n  LanguageModelV2CallOptions,\n  LanguageModelV2Content,\n  LanguageModelV2FinishReason,\n  LanguageModelV2Usage,\n} from \"@ai-sdk/provider\";\nimport { AISdkClient } from \"../../lib/v3/llm/aisdk.js\";\n\n/**\n * Races a promise against a timeout.\n * Resolves to the promise value or \"timeout\" if the deadline expires.\n */\nexport function raceTimeout<T>(\n  promise: Promise<T>,\n  ms: number,\n): Promise<T | \"timeout\"> {\n  let timer: ReturnType<typeof setTimeout>;\n  const timeout = new Promise<\"timeout\">((resolve) => {\n    timer = setTimeout(() => resolve(\"timeout\"), ms);\n  });\n  return Promise.race([promise, timeout]).finally(() => clearTimeout(timer));\n}\n\nconst CLOSE_TIMEOUT_MS = 5_000;\n\nasync function settleWithTimeout(\n  promise: Promise<unknown>,\n  timeoutMs: number,\n): Promise<void> {\n  let timeoutId: NodeJS.Timeout | undefined;\n  const timeout = new Promise<void>((resolve) => {\n    timeoutId = setTimeout(resolve, timeoutMs);\n  });\n  try {\n    await Promise.race([promise.catch(() => {}), timeout]);\n  } finally {\n    if (timeoutId) clearTimeout(timeoutId);\n  }\n}\n\nexport async function closeV3(v3?: V3 | null): Promise<void> {\n  if (!v3) return;\n  const isBrowserbase = v3.isBrowserbase;\n  if (isBrowserbase) {\n    try {\n      await settleWithTimeout(\n        v3.context.conn.send(\"Browser.close\"),\n        CLOSE_TIMEOUT_MS,\n      );\n    } catch {\n      // best-effort cleanup\n    }\n  }\n\n  await settleWithTimeout(v3.close(), CLOSE_TIMEOUT_MS);\n}\n\ntype JsonResponseKey =\n  | \"act\"\n  | \"Observation\"\n  | \"Metadata\"\n  | \"Extraction\"\n  | \"default\";\n\ntype JsonResponseValue =\n  | Record<string, unknown>\n  | ((options: LanguageModelV2CallOptions) => Record<string, unknown>);\n\ntype JsonResponseScript = JsonResponseValue | JsonResponseValue[];\n\ntype GenerateResponseValue =\n  | {\n      content: LanguageModelV2Content[];\n      finishReason?: LanguageModelV2FinishReason;\n      usage?: Partial<LanguageModelV2Usage>;\n    }\n  | ((options: LanguageModelV2CallOptions) => {\n      content: LanguageModelV2Content[];\n      finishReason?: LanguageModelV2FinishReason;\n      usage?: Partial<LanguageModelV2Usage>;\n    });\n\ntype ScriptedLanguageModel = LanguageModelV2 & {\n  doGenerateCalls: LanguageModelV2CallOptions[];\n};\n\ntype ScriptedGenerateResult = {\n  content: LanguageModelV2Content[];\n  finishReason?: LanguageModelV2FinishReason;\n  usage?: Partial<LanguageModelV2Usage>;\n};\n\nconst DEFAULT_USAGE: LanguageModelV2Usage = {\n  inputTokens: 1,\n  outputTokens: 1,\n  totalTokens: 2,\n  reasoningTokens: 0,\n  cachedInputTokens: 0,\n};\n\nconst mergeUsage = (\n  usage?: Partial<LanguageModelV2Usage>,\n): LanguageModelV2Usage => ({\n  ...DEFAULT_USAGE,\n  ...(usage ?? {}),\n});\n\nfunction consumeScriptValue<T>(value: T | T[] | undefined, fallback: T): T {\n  if (!Array.isArray(value)) {\n    return value ?? fallback;\n  }\n\n  if (value.length <= 1) {\n    return value[0] ?? fallback;\n  }\n\n  return value.shift() ?? fallback;\n}\n\nfunction resolveJsonResponseKey(\n  options: LanguageModelV2CallOptions,\n): JsonResponseKey {\n  const responseFormat = options.responseFormat;\n  if (!responseFormat || responseFormat.type !== \"json\") {\n    return \"default\";\n  }\n\n  const schema = responseFormat.schema as {\n    type?: string;\n    properties?: Record<string, unknown>;\n  };\n  const properties = schema?.properties ?? {};\n\n  if (\"elementId\" in properties && \"twoStep\" in properties) {\n    return \"act\";\n  }\n\n  if (\"elements\" in properties) {\n    return \"Observation\";\n  }\n\n  if (\"completed\" in properties && \"progress\" in properties) {\n    return \"Metadata\";\n  }\n\n  return \"Extraction\";\n}\n\nexport function promptToText(\n  prompt: LanguageModelV2CallOptions[\"prompt\"],\n): string {\n  return (prompt ?? [])\n    .flatMap((message) => {\n      if (typeof message.content === \"string\") {\n        return [message.content];\n      }\n\n      return (message.content ?? [])\n        .map((part) => (part.type === \"text\" ? part.text : \"\"))\n        .filter((text): text is string => text.length > 0);\n    })\n    .join(\"\\n\");\n}\n\nfunction findEncodedIds(options: LanguageModelV2CallOptions): string[] {\n  return [...promptToText(options.prompt).matchAll(/\\b\\d+-\\d+\\b/g)].map(\n    (match) => match[0],\n  );\n}\n\nexport function findEncodedIdForText(\n  options: LanguageModelV2CallOptions,\n  text: string,\n): string {\n  const promptText = promptToText(options.prompt);\n  const lines = promptText.split(\"\\n\");\n  const line = lines.find((entry) => entry.includes(text));\n  const match = line?.match(/\\b\\d+-\\d+\\b/);\n\n  if (!match) {\n    throw new Error(`Could not find encoded id for text: ${text}`);\n  }\n\n  return match[0];\n}\n\nexport function findLastEncodedId(options: LanguageModelV2CallOptions): string {\n  const matches = findEncodedIds(options);\n  if (matches.length === 0) {\n    throw new Error(\"Could not find any encoded ids in the prompt.\");\n  }\n\n  return matches[matches.length - 1];\n}\n\nexport function toolCallResponse(\n  toolName: string,\n  input: Record<string, unknown>,\n  toolCallId = `${toolName}-1`,\n): {\n  content: LanguageModelV2Content[];\n  finishReason: LanguageModelV2FinishReason;\n  usage: LanguageModelV2Usage;\n} {\n  return {\n    content: [\n      {\n        type: \"tool-call\",\n        toolCallId,\n        toolName,\n        input: JSON.stringify(input),\n      },\n    ],\n    finishReason: \"tool-calls\",\n    usage: DEFAULT_USAGE,\n  };\n}\n\nexport function doneToolResponse(\n  reasoning = \"done\",\n  taskComplete = true,\n  toolCallId = \"done-1\",\n): {\n  content: LanguageModelV2Content[];\n  finishReason: LanguageModelV2FinishReason;\n  usage: LanguageModelV2Usage;\n} {\n  return toolCallResponse(\"done\", { reasoning, taskComplete }, toolCallId);\n}\n\nfunction createGenerateResult(result: ScriptedGenerateResult): {\n  content: LanguageModelV2Content[];\n  finishReason: LanguageModelV2FinishReason;\n  usage: LanguageModelV2Usage;\n  warnings: [];\n} {\n  return {\n    content: result.content,\n    finishReason: result.finishReason ?? \"stop\",\n    usage: mergeUsage(result.usage),\n    warnings: [],\n  };\n}\n\nexport function createScriptedAisdkTestLlmClient(options?: {\n  modelId?: string;\n  jsonResponses?: Partial<Record<JsonResponseKey, JsonResponseScript>>;\n  generateResponses?: GenerateResponseValue[];\n}): AISdkClient {\n  const jsonResponses = Object.fromEntries(\n    Object.entries(options?.jsonResponses ?? {}).map(([key, value]) => [\n      key,\n      Array.isArray(value) ? [...value] : value,\n    ]),\n  ) as Partial<Record<JsonResponseKey, JsonResponseScript>>;\n  const generateResponses = [...(options?.generateResponses ?? [])];\n\n  const model: ScriptedLanguageModel = {\n    provider: \"mock\",\n    modelId: options?.modelId ?? \"mock/stagehand-flow-logger\",\n    specificationVersion: \"v2\",\n    supportedUrls: {},\n    doGenerateCalls: [],\n    doGenerate: async (callOptions) => {\n      model.doGenerateCalls.push(callOptions);\n\n      if (callOptions.responseFormat?.type === \"json\") {\n        const key = resolveJsonResponseKey(callOptions);\n        const responseScripts = consumeScriptValue<\n          JsonResponseScript | undefined\n        >(jsonResponses[key], jsonResponses.default);\n        const responseScript = consumeScriptValue<\n          JsonResponseValue | undefined\n        >(responseScripts, undefined);\n        const response =\n          typeof responseScript === \"function\"\n            ? responseScript(callOptions)\n            : (responseScript ?? {});\n\n        return createGenerateResult({\n          content: [{ type: \"text\", text: JSON.stringify(response) }],\n        });\n      }\n\n      const responseScript = consumeScriptValue<\n        GenerateResponseValue | undefined\n      >(generateResponses, undefined);\n\n      if (!responseScript) {\n        return createGenerateResult({\n          content: [{ type: \"text\", text: \"done\" }],\n        });\n      }\n\n      const response =\n        typeof responseScript === \"function\"\n          ? responseScript(callOptions)\n          : responseScript;\n\n      return createGenerateResult(response);\n    },\n    doStream: async () => {\n      throw new Error(\"Streaming is not implemented for this test model.\");\n    },\n  };\n\n  return new AISdkClient({ model });\n}\n"]}

package/dist/cjs/tests/integration/timeouts.spec.js

@@ -5,6 +5,145 @@ const v3_js_1 = require("../../lib/v3/v3.js");
 const v3_dynamic_config_js_1 = require("./v3.dynamic.config.js");
 const zod_1 = require("zod");
 const testUtils_js_1 = require("./testUtils.js");
+const ai_1 = require("ai");
+function createToolTimeoutTestLlmClient(toolName, toolInput) {
+    const usage = {
+        prompt_tokens: 0,
+        completion_tokens: 0,
+        reasoning_tokens: 0,
+        cached_input_tokens: 0,
+        total_tokens: 0,
+    };
+    let generateCallCount = 0;
+    const model = {
+        provider: "mock",
+        modelId: "mock/tool-timeout-test",
+        specificationVersion: "v2",
+        supportedUrls: {},
+        doGenerate: async () => {
+            generateCallCount += 1;
+            if (generateCallCount === 1) {
+                return {
+                    content: [
+                        {
+                            type: "tool-call",
+                            toolCallId: "tool-1",
+                            toolName,
+                            input: JSON.stringify(toolInput),
+                        },
+                    ],
+                    finishReason: "tool-calls",
+                    usage: { inputTokens: 0, outputTokens: 0, totalTokens: 0 },
+                    warnings: [],
+                };
+            }
+            return {
+                content: [
+                    {
+                        type: "tool-call",
+                        toolCallId: "done-1",
+                        toolName: "done",
+                        input: JSON.stringify({ reasoning: "done", taskComplete: true }),
+                    },
+                ],
+                finishReason: "tool-calls",
+                usage: { inputTokens: 0, outputTokens: 0, totalTokens: 0 },
+                warnings: [],
+            };
+        },
+        doStream: async () => {
+            throw new Error("doStream not implemented in timeout test model");
+        },
+    };
+    const llm = {
+        type: "openai",
+        modelName: "openai/gpt-4.1-mini",
+        hasVision: false,
+        clientOptions: {},
+        model,
+        getLanguageModel: () => model,
+        generateText: ai_1.generateText,
+        createChatCompletion: async (options) => {
+            const responseModelName = options?.options?.response_model?.name;
+            if (responseModelName === "act") {
+                return {
+                    data: {
+                        elementId: "1-0",
+                        description: "click body",
+                        method: "click",
+                        arguments: [],
+                        twoStep: false,
+                    },
+                    usage,
+                };
+            }
+            if (responseModelName === "Observation") {
+                return { data: { elements: [] }, usage };
+            }
+            if (responseModelName === "Extraction") {
+                return { data: {}, usage };
+            }
+            if (responseModelName === "Metadata") {
+                return { data: { completed: true, progress: "" }, usage };
+            }
+            return { data: {}, usage };
+        },
+    };
+    return llm;
+}
+function findToolOutput(stepEvents, toolName) {
+    for (const event of stepEvents) {
+        if (!event.toolCalls || !event.toolResults)
+            continue;
+        const toolIndex = event.toolCalls.findIndex((tc) => tc.toolName === toolName);
+        if (toolIndex !== -1) {
+            return event.toolResults[toolIndex]?.output;
+        }
+    }
+    return undefined;
+}
+async function runAgentToolTimeoutScenario(toolName, toolInput, options) {
+    const llmClient = createToolTimeoutTestLlmClient(toolName, toolInput);
+    const stepEvents = [];
+    const v3 = new v3_js_1.V3({
+        ...v3_dynamic_config_js_1.v3DynamicTestConfig,
+        experimental: true,
+        llmClient,
+    });
+    await v3.init();
+    try {
+        const page = v3.context.pages()[0];
+        await page.goto("https://example.com");
+        const agent = v3.agent({
+            ...(options?.mode ? { mode: options.mode } : {}),
+        });
+        await agent.execute({
+            instruction: `Use ${toolName} and then finish`,
+            maxSteps: 2,
+            toolTimeout: 1,
+            callbacks: {
+                onStepFinish: (event) => {
+                    stepEvents.push({
+                        toolCalls: event.toolCalls?.map((tc) => ({
+                            toolName: tc.toolName,
+                        })),
+                        toolResults: event.toolResults?.map((tr) => ({
+                            output: tr.output,
+                        })),
+                    });
+                },
+            },
+        });
+        const toolOutput = findToolOutput(stepEvents, toolName);
+        if (!toolOutput) {
+            throw new Error(`No tool output captured for ${toolName}`);
+        }
+        return { toolOutput };
+    }
+    finally {
+        await (0, testUtils_js_1.closeV3)(v3);
+    }
+}
 test_1.test.describe("V3 hard timeouts", () => {
     let v3;
     test_1.test.beforeEach(async () => {
@@ -25,5 +164,144 @@ test_1.test.describe("V3 hard timeouts", () => {
     (0, test_1.test)("act() enforces timeoutMs", async () => {
         await (0, test_1.expect)(v3.act("do nothing", { timeout: 5 })).rejects.toThrow(/timed out/i);
     });
+    (0, test_1.test)("agent toolTimeout enforces timeout for act tool", async () => {
+        const { toolOutput } = await runAgentToolTimeoutScenario("act", {
+            action: "click somewhere",
+        });
+        const output = toolOutput;
+        (0, test_1.expect)(output.success).toBe(false);
+        (0, test_1.expect)(output.error).toContain("TimeoutError");
+        (0, test_1.expect)(output.error).toContain("1ms");
+    });
+    (0, test_1.test)("agent toolTimeout enforces timeout for extract tool", async () => {
+        const { toolOutput } = await runAgentToolTimeoutScenario("extract", {
+            instruction: "extract the page title",
+            schema: { type: "object", properties: { title: { type: "string" } } },
+        });
+        const output = toolOutput;
+        (0, test_1.expect)(output.success).toBe(false);
+        (0, test_1.expect)(output.error).toContain("TimeoutError");
+        (0, test_1.expect)(output.error).toContain("1ms");
+    });
+    (0, test_1.test)("agent toolTimeout enforces timeout for fillForm tool", async () => {
+        const { toolOutput } = await runAgentToolTimeoutScenario("fillForm", {
+            fields: [{ action: "type hello into name" }],
+        });
+        const output = toolOutput;
+        (0, test_1.expect)(output.success).toBe(false);
+        (0, test_1.expect)(output.error).toContain("TimeoutError");
+        (0, test_1.expect)(output.error).toContain("1ms");
+    });
+    (0, test_1.test)("agent toolTimeout enforces timeout for ariaTree", async () => {
+        const { toolOutput } = await runAgentToolTimeoutScenario("ariaTree", {});
+        const output = toolOutput;
+        (0, test_1.expect)(output.success).toBe(false);
+        (0, test_1.expect)(output.error).toContain("TimeoutError");
+        (0, test_1.expect)(output.error).toContain("1ms");
+    });
+    (0, test_1.test)("agent toolTimeout enforces timeout for goto tool", async () => {
+        const { toolOutput } = await runAgentToolTimeoutScenario("goto", {
+            url: "https://example.com/slow",
+        });
+        const output = toolOutput;
+        (0, test_1.expect)(output.success).toBe(false);
+        (0, test_1.expect)(output.error).toContain("TimeoutError");
+        (0, test_1.expect)(output.error).toContain("1ms");
+    });
+    (0, test_1.test)("agent toolTimeout enforces timeout for navback tool", async () => {
+        const { toolOutput } = await runAgentToolTimeoutScenario("navback", {
+            reasoningText: "going back",
+        });
+        const output = toolOutput;
+        (0, test_1.expect)(output.success).toBe(false);
+        (0, test_1.expect)(output.error).toContain("TimeoutError");
+        (0, test_1.expect)(output.error).toContain("1ms");
+    });
+    (0, test_1.test)("agent toolTimeout enforces timeout for screenshot tool", async () => {
+        const { toolOutput } = await runAgentToolTimeoutScenario("screenshot", {});
+        const output = toolOutput;
+        (0, test_1.expect)(output.success).toBe(false);
+        (0, test_1.expect)(output.error).toContain("TimeoutError");
+        (0, test_1.expect)(output.error).toContain("1ms");
+    });
+    (0, test_1.test)("agent toolTimeout enforces timeout for scroll tool", async () => {
+        const { toolOutput } = await runAgentToolTimeoutScenario("scroll", {
+            direction: "down",
+        });
+        const output = toolOutput;
+        (0, test_1.expect)(output.success).toBe(false);
+        (0, test_1.expect)(output.error).toContain("TimeoutError");
+        (0, test_1.expect)(output.error).toContain("1ms");
+    });
+    (0, test_1.test)("agent toolTimeout enforces timeout for keys tool", async () => {
+        const { toolOutput } = await runAgentToolTimeoutScenario("keys", {
+            method: "press",
+            value: "Enter",
+        });
+        const output = toolOutput;
+        (0, test_1.expect)(output.success).toBe(false);
+        (0, test_1.expect)(output.error).toContain("TimeoutError");
+        (0, test_1.expect)(output.error).toContain("1ms");
+    });
+    (0, test_1.test)("agent toolTimeout enforces timeout for click tool (hybrid)", async () => {
+        const { toolOutput } = await runAgentToolTimeoutScenario("click", { describe: "click element", coordinates: [100, 100] }, { mode: "hybrid" });
+        const output = toolOutput;
+        (0, test_1.expect)(output.success).toBe(false);
+        (0, test_1.expect)(output.error).toContain("TimeoutError");
+        (0, test_1.expect)(output.error).toContain("1ms");
+    });
+    (0, test_1.test)("agent toolTimeout enforces timeout for type tool (hybrid)", async () => {
+        const { toolOutput } = await runAgentToolTimeoutScenario("type", {
+            describe: "type into field",
+            text: "hello",
+            coordinates: [100, 100],
+        }, { mode: "hybrid" });
+        const output = toolOutput;
+        (0, test_1.expect)(output.success).toBe(false);
+        (0, test_1.expect)(output.error).toContain("TimeoutError");
+        (0, test_1.expect)(output.error).toContain("1ms");
+    });
+    (0, test_1.test)("agent toolTimeout enforces timeout for dragAndDrop tool (hybrid)", async () => {
+        const { toolOutput } = await runAgentToolTimeoutScenario("dragAndDrop", {
+            describe: "drag element",
+            startCoordinates: [100, 100],
+            endCoordinates: [200, 200],
+        }, { mode: "hybrid" });
+        const output = toolOutput;
+        (0, test_1.expect)(output.success).toBe(false);
+        (0, test_1.expect)(output.error).toContain("TimeoutError");
+        (0, test_1.expect)(output.error).toContain("1ms");
+    });
+    (0, test_1.test)("agent toolTimeout enforces timeout for clickAndHold tool (hybrid)", async () => {
+        const { toolOutput } = await runAgentToolTimeoutScenario("clickAndHold", {
+            describe: "hold element",
+            coordinates: [100, 100],
+            duration: 1000,
+        }, { mode: "hybrid" });
+        const output = toolOutput;
+        (0, test_1.expect)(output.success).toBe(false);
+        (0, test_1.expect)(output.error).toContain("TimeoutError");
+        (0, test_1.expect)(output.error).toContain("1ms");
+    });
+    (0, test_1.test)("agent toolTimeout enforces timeout for fillFormVision tool (hybrid)", async () => {
+        const { toolOutput } = await runAgentToolTimeoutScenario("fillFormVision", {
+            fields: [
+                {
+                    action: "type hello into name",
+                    value: "hello",
+                    coordinates: { x: 100, y: 100 },
+                },
+                {
+                    action: "type world into email",
+                    value: "world",
+                    coordinates: { x: 100, y: 200 },
+                },
+            ],
+        }, { mode: "hybrid" });
+        const output = toolOutput;
+        (0, test_1.expect)(output.success).toBe(false);
+        (0, test_1.expect)(output.error).toContain("TimeoutError");
+        (0, test_1.expect)(output.error).toContain("1ms");
+    });
 });
 //# sourceMappingURL=timeouts.spec.js.map