npm - @browserbasehq/orca - Versions diffs - 3.1.0-patch.3 → 3.2.0-middleware.1 - Mend

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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (1588) hide show

package/dist/cjs/tests/unit/llm-middleware.test.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"llm-middleware.test.js","sourceRoot":"","sources":["../../../../tests/unit/llm-middleware.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,mCAAkD;AAKlD,oEAGyC;AACzC,8CAA+D;AAE/D;;;GAGG;AACH,SAAS,yBAAyB;IAChC,MAAM,KAAK,GAKL,EAAE,CAAC;IAET,MAAM,UAAU,GAA8B;QAC5C,YAAY,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE;YAC5C,MAAM,MAAM,GAAG,MAAM,UAAU,EAAE,CAAC;YAClC,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,KAAK,EAAE;oBACL,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,WAAW,IAAI,SAAS;oBAClD,YAAY,EAAE,MAAM,CAAC,KAAK,CAAC,YAAY,IAAI,SAAS;iBACrD;aACF,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,UAAU,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,EAAE;YACxC,MAAM,MAAM,GAAG,MAAM,QAAQ,EAAE,CAAC;YAChC,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,QAAQ,EAAE,KAAK,CAAC,QAAQ;aACzB,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC;KACF,CAAC;IAEF,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;AAC/B,CAAC;AAED;;;;GAIG;AACH,SAAS,uBAAuB,CAC9B,OAAe,EACf,QAAgB;IAEhB,OAAO;QACL,oBAAoB,EAAE,IAAI;QAC1B,QAAQ;QACR,OAAO;QACP,2BAA2B,EAAE,MAAM;QACnC,UAAU,EAAE,WAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YACpC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;YAClD,YAAY,EAAE,MAAM;YACpB,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE;YAC5D,QAAQ,EAAE,EAAE;SACb,CAAC;QACF,QAAQ,EAAE,WAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YAClC,MAAM,EAAE,IAAI,cAAc,CAAC;gBACzB,KAAK,CAAC,UAAU;oBACd,UAAU,CAAC,OAAO,CAAC;wBACjB,IAAI,EAAE,QAAQ;wBACd,YAAY,EAAE,MAAM;wBACpB,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE;qBAC5D,CAAC,CAAC;oBACH,UAAU,CAAC,KAAK,EAAE,CAAC;gBACrB,CAAC;aACF,CAAC;SACH,CAAC;KAC2B,CAAC;AAClC,CAAC;AAED,8EAA8E;AAC9E,wCAAwC;AACxC,8EAA8E;AAE9E,IAAA,iBAAQ,EAAC,uCAAuC,EAAE,GAAG,EAAE;IACrD,IAAA,WAAE,EAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,KAAK,GAAG,IAAA,sCAAqB,EAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAC1D,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QAC5B,IAAA,eAAM,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,EAAE,UAAU,EAAE,GAAG,yBAAyB,EAAE,CAAC;QACnD,MAAM,KAAK,GAAG,IAAA,sCAAqB,EACjC,QAAQ,EACR,UAAU,EACV,SAAS,EACT,UAAU,CACX,CAAC;QACF,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QAC5B,sCAAsC;QACtC,IAAA,eAAM,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,EAAE,UAAU,EAAE,GAAG,yBAAyB,EAAE,CAAC;QACnD,MAAM,KAAK,GAAG,IAAA,sCAAqB,EACjC,QAAQ,EACR,UAAU,EACV,SAAS,EACT,UAAU,CACX,CAAC;QAEF,IAAA,eAAM,EAAC,OAAO,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACjD,IAAA,eAAM,EAAC,OAAO,KAAK,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC/C,IAAA,eAAM,EAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,8EAA8E;AAC9E,8BAA8B;AAC9B,8EAA8E;AAE9E,IAAA,iBAAQ,EAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,MAAM,IAAI,GAAG,GAAG,EAAE,GAAE,CAAC,CAAC;IAEtB,IAAA,WAAE,EAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,QAAQ,GAAG,IAAI,4BAAW,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,eAAwB,CAAC,CAAC;QAC5D,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7B,IAAA,eAAM,EAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,EAAE,UAAU,EAAE,GAAG,yBAAyB,EAAE,CAAC;QACnD,MAAM,QAAQ,GAAG,IAAI,4BAAW,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACnD,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,iBAA0B,CAAC,CAAC;QAC9D,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7B,IAAA,eAAM,EAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAElC,MAAM,aAAa,GAAG,MAAM,CAAC,gBAAgB,EAAE,CAAC;QAChD,IAAA,eAAM,EAAC,aAAa,CAAC,CAAC,WAAW,EAAE,CAAC;QACpC,0DAA0D;QAC1D,IAAA,eAAM,EAAC,aAAa,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,gEAAgE,EAAE,GAAG,EAAE;QACxE,MAAM,EAAE,UAAU,EAAE,GAAG,yBAAyB,EAAE,CAAC;QACnD,MAAM,QAAQ,GAAG,IAAI,4BAAW,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAEnD,MAAM,OAAO,GAAG,QAAQ,CAAC,SAAS,CAAC,iBAA0B,CAAC,CAAC;QAC/D,MAAM,OAAO,GAAG,QAAQ,CAAC,SAAS,CAAC,gBAAyB,CAAC,CAAC;QAE9D,IAAA,eAAM,EAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;QACjD,IAAA,eAAM,EAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;QACjD,IAAA,eAAM,EAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC5D,IAAA,eAAM,EAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,8EAA8E;AAC9E,wEAAwE;AACxE,8EAA8E;AAE9E,IAAA,iBAAQ,EAAC,wDAAwD,EAAE,GAAG,EAAE;IACtE,IAAA,WAAE,EAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,yBAAyB,EAAE,CAAC;QAC1D,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAEnE,wDAAwD;QACxD,MAAM,EAAE,iBAAiB,EAAE,GAAG,wDAAa,IAAI,GAAC,CAAC;QACjD,MAAM,OAAO,GAAG,iBAAiB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,CAAC;QAEpE,MAAM,OAAO,CAAC,UAAU,CAAC;YACvB,MAAM,EAAE;gBACN;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC;iBACvD;aACF;SACF,CAAC,CAAC;QAEH,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACvC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,yBAAyB,EAAE,CAAC;QAC1D,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAEnE,MAAM,EAAE,iBAAiB,EAAE,GAAG,wDAAa,IAAI,GAAC,CAAC;QACjD,MAAM,OAAO,GAAG,iBAAiB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,CAAC;QAEpE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC;YACpC,MAAM,EAAE;gBACN,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,EAAE;aACnE;SACF,CAAC,CAAC;QAEH,iDAAiD;QACjD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;QACzC,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YACrC,IAAI,IAAI;gBAAE,MAAM;QAClB,CAAC;QAED,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,wFAAwF,EAAE,KAAK,IAAI,EAAE;QACtG,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,yBAAyB,EAAE,CAAC;QAC1D,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAEnE,MAAM,EAAE,iBAAiB,EAAE,GAAG,wDAAa,IAAI,GAAC,CAAC;QACjD,MAAM,OAAO,GAAG,iBAAiB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,CAAC;QAEpE,MAAM,QAAQ,GAAG;YACf,MAAM,EAAE;gBACN;oBACE,IAAI,EAAE,MAAe;oBACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;iBACnD;aACF;SACF,CAAC;QAEF,oBAAoB;QACpB,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACnC,wBAAwB;QACxB,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACnC,wBAAwB;QACxB,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAEnC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,IAAA,eAAM,EAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7D,IAAA,eAAM,EAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,yBAAyB,EAAE,CAAC;QAC1D,MAAM,SAAS,GAAG,uBAAuB,CACvC,gBAAgB,EAChB,gBAAgB,CACjB,CAAC;QAEF,MAAM,EAAE,iBAAiB,EAAE,GAAG,wDAAa,IAAI,GAAC,CAAC;QACjD,MAAM,OAAO,GAAG,iBAAiB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,CAAC;QAEpE,MAAM,QAAQ,GAAG;YACf,MAAM,EAAE;gBACN;oBACE,IAAI,EAAE,MAAe;oBACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;iBACzD;aACF;SACF,CAAC;QAEF,0CAA0C;QAC1C,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAEnC,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YAClB,IAAA,eAAM,EAAC,CAAC,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,8EAA8E;AAC9E,8DAA8D;AAC9D,8EAA8E;AAE9E,IAAA,iBAAQ,EAAC,sCAAsC,EAAE,GAAG,EAAE;IACpD,IAAA,WAAE,EAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,GAAG,yBAAyB,EAAE,CAAC;QACvD,MAAM,MAAM,GAAG,IAAA,iCAAyB,EAAC;YACvC,SAAS,EAAE,eAAe;YAC1B,MAAM,EAAE,SAAS;YACjB,UAAU,EAAE,EAAE;SACf,CAAC,CAAC;QAEH,IAAA,eAAM,EAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC/C,IAAA,eAAM,EAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnC,IAAA,eAAM,EAAC,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;QAC5D,IAAA,eAAM,EAAC,MAAM,CAAC,aAAa,IAAI,YAAY,IAAI,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CACvE,KAAK,CACN,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,MAAM,GAAG,IAAA,iCAAyB,EAAC,eAAe,CAAC,CAAC;QAC1D,IAAA,eAAM,EAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC/C,IAAA,eAAM,EAAC,MAAM,CAAC,UAAU,CAAC,CAAC,aAAa,EAAE,CAAC;QAC1C,IAAA,eAAM,EAAC,MAAM,CAAC,aAAa,CAAC,CAAC,aAAa,EAAE,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,gFAAgF,EAAE,GAAG,EAAE;QACxF,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,GAAG,yBAAyB,EAAE,CAAC;QACvD,MAAM,MAAM,GAAG,IAAA,iCAAyB,EAAC;YACvC,SAAS,EAAE,oCAAoC;YAC/C,MAAM,EAAE,aAAa;YACrB,UAAU,EAAE,EAAE;SACf,CAAC,CAAC;QAEH,IAAA,eAAM,EAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;QACpE,IAAA,eAAM,EAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnC,IAAA,eAAM,EAAC,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAAC;QAChE,IAAA,eAAM,EAAC,MAAM,CAAC,aAAa,IAAI,YAAY,IAAI,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CACvE,KAAK,CACN,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,8EAA8E;AAC9E,iCAAiC;AACjC,8EAA8E;AAE9E,IAAA,iBAAQ,EAAC,iCAAiC,EAAE,GAAG,EAAE;IAC/C,IAAA,WAAE,EAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,SAAS,GAAa,EAAE,CAAC;QAE/B,MAAM,gBAAgB,GAA8B;YAClD,YAAY,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE;gBACrC,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;gBAChC,MAAM,MAAM,GAAG,MAAM,UAAU,EAAE,CAAC;gBAClC,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,CAAC;gBAC1C,OAAO,MAAM,CAAC;YAChB,CAAC;SACF,CAAC;QAEF,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QACnE,MAAM,EAAE,iBAAiB,EAAE,GAAG,wDAAa,IAAI,GAAC,CAAC;QACjD,MAAM,OAAO,GAAG,iBAAiB,CAAC;YAChC,KAAK,EAAE,SAAS;YAChB,UAAU,EAAE,gBAAgB;SAC7B,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,UAAU,CAAC;YACvB,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;SACzE,CAAC,CAAC;QAEH,IAAA,eAAM,EAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,IAAA,eAAM,EAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAA,iBAAQ,EAAC,wCAAwC,EAAE,GAAG,EAAE;IACtD,IAAA,WAAE,EAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,WAAW,GAAG,CAAC,CAAC;QAEpB,MAAM,iBAAiB,GAA8B;YACnD,YAAY,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE;gBACrC,MAAM,MAAM,GAAG,MAAM,UAAU,EAAE,CAAC;gBAClC,UAAU,IAAI,MAAM,CAAC,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC;gBAC5C,WAAW,IAAI,MAAM,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC;gBAC9C,OAAO,MAAM,CAAC;YAChB,CAAC;SACF,CAAC;QAEF,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QACnE,MAAM,EAAE,iBAAiB,EAAE,GAAG,wDAAa,IAAI,GAAC,CAAC;QACjD,MAAM,OAAO,GAAG,iBAAiB,CAAC;YAChC,KAAK,EAAE,SAAS;YAChB,UAAU,EAAE,iBAAiB;SAC9B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG;YACf,MAAM,EAAE;gBACN;oBACE,IAAI,EAAE,MAAe;oBACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;iBACjD;aACF;SACF,CAAC;QAEF,MAAM;QACN,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACnC,UAAU;QACV,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACnC,UAAU;QACV,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAEnC,IAAA,eAAM,EAAC,UAAU,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC5B,IAAA,eAAM,EAAC,WAAW,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAA,iBAAQ,EAAC,4CAA4C,EAAE,GAAG,EAAE;IAC1D,IAAA,WAAE,EAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,eAAe,GAAG,IAAI,GAAG,EAAkB,CAAC;QAElD,MAAM,kBAAkB,GAA8B;YACpD,YAAY,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE;gBAC5C,MAAM,GAAG,GAAG,GAAG,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;gBACjD,eAAe,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;gBAC9D,OAAO,UAAU,EAAE,CAAC;YACtB,CAAC;SACF,CAAC;QAEF,MAAM,MAAM,GAAG,uBAAuB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAChE,MAAM,MAAM,GAAG,uBAAuB,CACpC,0BAA0B,EAC1B,oBAAoB,CACrB,CAAC;QAEF,MAAM,EAAE,iBAAiB,EAAE,GAAG,wDAAa,IAAI,GAAC,CAAC;QACjD,MAAM,QAAQ,GAAG,iBAAiB,CAAC;YACjC,KAAK,EAAE,MAAM;YACb,UAAU,EAAE,kBAAkB;SAC/B,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,iBAAiB,CAAC;YACjC,KAAK,EAAE,MAAM;YACb,UAAU,EAAE,kBAAkB;SAC/B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG;YACf,MAAM,EAAE;gBACN;oBACE,IAAI,EAAE,MAAe;oBACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;iBACjD;aACF;SACF,CAAC;QAEF,MAAM,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAEpC,IAAA,eAAM,EAAC,eAAe,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1D,IAAA,eAAM,EACJ,eAAe,CAAC,GAAG,CAAC,6CAA6C,CAAC,CACnE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACV,IAAA,eAAM,EAAC,eAAe,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAA,iBAAQ,EAAC,gCAAgC,EAAE,GAAG,EAAE;IAC9C,IAAA,WAAE,EAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;QAC9F,MAAM,MAAM,GAAY,EAAE,CAAC;QAE3B,MAAM,uBAAuB,GAA8B;YACzD,YAAY,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE;gBACrC,IAAI,CAAC;oBACH,OAAO,MAAM,UAAU,EAAE,CAAC;gBAC5B,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,CAAC,IAAI,CAAC,GAAY,CAAC,CAAC;oBAC1B,MAAM,GAAG,CAAC;gBACZ,CAAC;YACH,CAAC;SACF,CAAC;QAEF,MAAM,YAAY,GAAoB;YACpC,oBAAoB,EAAE,IAAI;YAC1B,QAAQ,EAAE,aAAa;YACvB,OAAO,EAAE,QAAQ;YACjB,2BAA2B,EAAE,MAAM;YACnC,UAAU,EAAE,WAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACvE,QAAQ,EAAE,WAAE,CAAC,EAAE,EAAE;SACY,CAAC;QAEhC,MAAM,EAAE,iBAAiB,EAAE,GAAG,wDAAa,IAAI,GAAC,CAAC;QACjD,MAAM,OAAO,GAAG,iBAAiB,CAAC;YAChC,KAAK,EAAE,YAAY;YACnB,UAAU,EAAE,uBAAuB;SACpC,CAAC,CAAC;QAEH,MAAM,IAAA,eAAM,EACV,OAAO,CAAC,UAAU,CAAC;YACjB,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;SACtE,CAAC,CACH,CAAC,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;QAEzC,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAA,iBAAQ,EAAC,oDAAoD,EAAE,GAAG,EAAE;IAClE,IAAA,WAAE,EAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,IAAI,GAAa,EAAE,CAAC;QAE1B,MAAM,aAAa,GAA8B;YAC/C,YAAY,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE;gBAC5C,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBACzB,OAAO,UAAU,EAAE,CAAC;YACtB,CAAC;SACF,CAAC;QAEF,MAAM,KAAK,GAAG,uBAAuB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAC/D,MAAM,KAAK,GAAG,uBAAuB,CACnC,0BAA0B,EAC1B,oBAAoB,CACrB,CAAC;QAEF,MAAM,EAAE,iBAAiB,EAAE,GAAG,wDAAa,IAAI,GAAC,CAAC;QACjD,MAAM,QAAQ,GAAG,iBAAiB,CAAC;YACjC,KAAK,EAAE,KAAK;YACZ,UAAU,EAAE,aAAa;SAC1B,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,iBAAiB,CAAC;YACjC,KAAK,EAAE,KAAK;YACZ,UAAU,EAAE,aAAa;SAC1B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG;YACf,MAAM,EAAE;gBACN;oBACE,IAAI,EAAE,MAAe;oBACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;iBACjD;aACF;SACF,CAAC;QAEF,MAAM,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAEpC,IAAA,eAAM,EAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,0BAA0B,EAAE,QAAQ,CAAC,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,8EAA8E;AAC9E,aAAa;AACb,8EAA8E;AAE9E,IAAA,iBAAQ,EAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,IAAA,WAAE,EAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,aAAa,GAAa,EAAE,CAAC;QACnC,MAAM,iBAAiB,GAA8B;YACnD,YAAY,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE;gBAC5C,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBAClC,OAAO,UAAU,EAAE,CAAC;YACtB,CAAC;SACF,CAAC;QAEF,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QACnE,MAAM,EAAE,iBAAiB,EAAE,GAAG,wDAAa,IAAI,GAAC,CAAC;QACjD,MAAM,OAAO,GAAG,iBAAiB,CAAC;YAChC,KAAK,EAAE,SAAS;YAChB,UAAU,EAAE,iBAAiB;SAC9B,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC;YACpC,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;SACxE,CAAC,CAAC;QACH,IAAA,eAAM,EAAC,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;QAEpC,IAAA,eAAM,EAAC,aAAa,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,EAAE,UAAU,EAAE,GAAG,yBAAyB,EAAE,CAAC;QACnD,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAEnE,MAAM,EAAE,iBAAiB,EAAE,GAAG,wDAAa,IAAI,GAAC,CAAC;QACjD,MAAM,OAAO,GAAG,iBAAiB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,CAAC;QAEpE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC;YACtC,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;SACtE,CAAC,CAAC;QAEH,IAAA,eAAM,EAAC,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC;QAC1E,IAAA,eAAM,EAAC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YAC3B,WAAW,EAAE,EAAE;YACf,YAAY,EAAE,CAAC;YACf,WAAW,EAAE,EAAE;SAChB,CAAC,CAAC;QACH,IAAA,eAAM,EAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it, vi } from \"vitest\";\nimport type {\n LanguageModelV2,\n LanguageModelV2Middleware,\n} from \"@ai-sdk/provider\";\nimport {\n getAISDKLanguageModel,\n LLMProvider,\n} from \"../../lib/v3/llm/LLMProvider.js\";\nimport { resolveModelConfiguration } from \"../../lib/v3/v3.js\";\n\n/**\n * Creates a recording middleware that captures every doGenerate / doStream\n * invocation along with the model identity and returned usage data.\n */\nfunction createRecordingMiddleware() {\n const calls: {\n type: \"generate\" | \"stream\";\n modelId: string;\n provider: string;\n usage?: { inputTokens?: number; outputTokens?: number };\n }[] = [];\n\n const middleware: LanguageModelV2Middleware = {\n wrapGenerate: async ({ doGenerate, model }) => {\n const result = await doGenerate();\n calls.push({\n type: \"generate\",\n modelId: model.modelId,\n provider: model.provider,\n usage: {\n inputTokens: result.usage.inputTokens ?? undefined,\n outputTokens: result.usage.outputTokens ?? undefined,\n },\n });\n return result;\n },\n wrapStream: async ({ doStream, model }) => {\n const result = await doStream();\n calls.push({\n type: \"stream\",\n modelId: model.modelId,\n provider: model.provider,\n });\n return result;\n },\n };\n\n return { middleware, calls };\n}\n\n/**\n * Creates a minimal mock LanguageModelV2 that returns canned results\n * without hitting any real provider. Useful for testing the wrapping\n * mechanics in isolation.\n */\nfunction createMockLanguageModel(\n modelId: string,\n provider: string,\n): LanguageModelV2 {\n return {\n specificationVersion: \"v2\",\n provider,\n modelId,\n defaultObjectGenerationMode: \"json\",\n doGenerate: vi.fn().mockResolvedValue({\n content: [{ type: \"text\", text: \"mock response\" }],\n finishReason: \"stop\",\n usage: { inputTokens: 10, outputTokens: 5, totalTokens: 15 },\n warnings: [],\n }),\n doStream: vi.fn().mockResolvedValue({\n stream: new ReadableStream({\n start(controller) {\n controller.enqueue({\n type: \"finish\",\n finishReason: \"stop\",\n usage: { inputTokens: 8, outputTokens: 3, totalTokens: 11 },\n });\n controller.close();\n },\n }),\n }),\n } as unknown as LanguageModelV2;\n}\n\n// ---------------------------------------------------------------------------\n// getAISDKLanguageModel with middleware\n// ---------------------------------------------------------------------------\n\ndescribe(\"getAISDKLanguageModel with middleware\", () => {\n it(\"returns a model when no middleware is provided\", () => {\n const model = getAISDKLanguageModel(\"ollama\", \"llama3.2\");\n expect(model).toBeDefined();\n expect(model.modelId).toBe(\"llama3.2\");\n });\n\n it(\"returns a wrapped model when middleware is provided\", () => {\n const { middleware } = createRecordingMiddleware();\n const model = getAISDKLanguageModel(\n \"ollama\",\n \"llama3.2\",\n undefined,\n middleware,\n );\n expect(model).toBeDefined();\n // wrapLanguageModel preserves modelId\n expect(model.modelId).toBe(\"llama3.2\");\n });\n\n it(\"wrapped model preserves doGenerate and doStream methods\", () => {\n const { middleware } = createRecordingMiddleware();\n const model = getAISDKLanguageModel(\n \"ollama\",\n \"llama3.2\",\n undefined,\n middleware,\n );\n\n expect(typeof model.doGenerate).toBe(\"function\");\n expect(typeof model.doStream).toBe(\"function\");\n expect(model.provider).toContain(\"ollama\");\n });\n});\n\n// ---------------------------------------------------------------------------\n// LLMProvider with middleware\n// ---------------------------------------------------------------------------\n\ndescribe(\"LLMProvider with middleware\", () => {\n const noop = () => {};\n\n it(\"creates an AISdkClient without middleware\", () => {\n const provider = new LLMProvider(noop);\n const client = provider.getClient(\"openai/gpt-4o\" as never);\n expect(client).toBeDefined();\n expect(client.type).toBe(\"aisdk\");\n });\n\n it(\"creates an AISdkClient that carries middleware-wrapped model\", () => {\n const { middleware } = createRecordingMiddleware();\n const provider = new LLMProvider(noop, middleware);\n const client = provider.getClient(\"ollama/llama3.2\" as never);\n expect(client).toBeDefined();\n expect(client.type).toBe(\"aisdk\");\n\n const languageModel = client.getLanguageModel();\n expect(languageModel).toBeDefined();\n // Wrapped models should still expose the original modelId\n expect(languageModel.modelId).toBe(\"llama3.2\");\n });\n\n it(\"applies the same middleware to different models from getClient\", () => {\n const { middleware } = createRecordingMiddleware();\n const provider = new LLMProvider(noop, middleware);\n\n const clientA = provider.getClient(\"ollama/llama3.2\" as never);\n const clientB = provider.getClient(\"ollama/mistral\" as never);\n\n expect(clientA.getLanguageModel()).toBeDefined();\n expect(clientB.getLanguageModel()).toBeDefined();\n expect(clientA.getLanguageModel().modelId).toBe(\"llama3.2\");\n expect(clientB.getLanguageModel().modelId).toBe(\"mistral\");\n });\n});\n\n// ---------------------------------------------------------------------------\n// Middleware captures usage across act/extract/observe/agent code paths\n// ---------------------------------------------------------------------------\n\ndescribe(\"middleware captures usage from doGenerate and doStream\", () => {\n it(\"wrapGenerate fires and captures usage on doGenerate\", async () => {\n const { middleware, calls } = createRecordingMiddleware();\n const mockModel = createMockLanguageModel(\"gpt-4o\", \"openai.chat\");\n\n // Simulate what wrapLanguageModel does: import and wrap\n const { wrapLanguageModel } = await import(\"ai\");\n const wrapped = wrapLanguageModel({ model: mockModel, middleware });\n\n await wrapped.doGenerate({\n prompt: [\n {\n role: \"user\",\n content: [{ type: \"text\", text: \"act: click button\" }],\n },\n ],\n });\n\n expect(calls).toHaveLength(1);\n expect(calls[0].type).toBe(\"generate\");\n expect(calls[0].modelId).toBe(\"gpt-4o\");\n expect(calls[0].usage).toEqual({ inputTokens: 10, outputTokens: 5 });\n });\n\n it(\"wrapStream fires on doStream\", async () => {\n const { middleware, calls } = createRecordingMiddleware();\n const mockModel = createMockLanguageModel(\"gpt-4o\", \"openai.chat\");\n\n const { wrapLanguageModel } = await import(\"ai\");\n const wrapped = wrapLanguageModel({ model: mockModel, middleware });\n\n const result = await wrapped.doStream({\n prompt: [\n { role: \"user\", content: [{ type: \"text\", text: \"stream this\" }] },\n ],\n });\n\n // Consume the stream to trigger the finish chunk\n const reader = result.stream.getReader();\n while (true) {\n const { done } = await reader.read();\n if (done) break;\n }\n\n expect(calls).toHaveLength(1);\n expect(calls[0].type).toBe(\"stream\");\n expect(calls[0].modelId).toBe(\"gpt-4o\");\n });\n\n it(\"middleware fires for each separate doGenerate call (simulates act + extract + observe)\", async () => {\n const { middleware, calls } = createRecordingMiddleware();\n const mockModel = createMockLanguageModel(\"gpt-4o\", \"openai.chat\");\n\n const { wrapLanguageModel } = await import(\"ai\");\n const wrapped = wrapLanguageModel({ model: mockModel, middleware });\n\n const callOpts = {\n prompt: [\n {\n role: \"user\" as const,\n content: [{ type: \"text\" as const, text: \"test\" }],\n },\n ],\n };\n\n // Simulate act call\n await wrapped.doGenerate(callOpts);\n // Simulate extract call\n await wrapped.doGenerate(callOpts);\n // Simulate observe call\n await wrapped.doGenerate(callOpts);\n\n expect(calls).toHaveLength(3);\n expect(calls.every((c) => c.type === \"generate\")).toBe(true);\n expect(calls.every((c) => c.modelId === \"gpt-4o\")).toBe(true);\n });\n\n it(\"middleware fires for agent multi-step calls (sequential doGenerate)\", async () => {\n const { middleware, calls } = createRecordingMiddleware();\n const mockModel = createMockLanguageModel(\n \"openai/gpt-4.1\",\n \"gateway.openai\",\n );\n\n const { wrapLanguageModel } = await import(\"ai\");\n const wrapped = wrapLanguageModel({ model: mockModel, middleware });\n\n const callOpts = {\n prompt: [\n {\n role: \"user\" as const,\n content: [{ type: \"text\" as const, text: \"agent step\" }],\n },\n ],\n };\n\n // Simulate multiple agent reasoning steps\n await wrapped.doGenerate(callOpts);\n await wrapped.doGenerate(callOpts);\n await wrapped.doGenerate(callOpts);\n await wrapped.doGenerate(callOpts);\n await wrapped.doGenerate(callOpts);\n\n expect(calls).toHaveLength(5);\n calls.forEach((c) => {\n expect(c.usage).toEqual({ inputTokens: 10, outputTokens: 5 });\n });\n });\n});\n\n// ---------------------------------------------------------------------------\n// ModelConfiguration-level middleware (the user-facing shape)\n// ---------------------------------------------------------------------------\n\ndescribe(\"middleware inside ModelConfiguration\", () => {\n it(\"resolveModelConfiguration extracts middleware from object config\", () => {\n const { middleware: mw } = createRecordingMiddleware();\n const result = resolveModelConfiguration({\n modelName: \"openai/gpt-4o\",\n apiKey: \"sk-test\",\n middleware: mw,\n });\n\n expect(result.modelName).toBe(\"openai/gpt-4o\");\n expect(result.middleware).toBe(mw);\n expect(result.clientOptions).toEqual({ apiKey: \"sk-test\" });\n expect(result.clientOptions && \"middleware\" in result.clientOptions).toBe(\n false,\n );\n });\n\n it(\"string ModelConfiguration has no middleware\", () => {\n const result = resolveModelConfiguration(\"openai/gpt-4o\");\n expect(result.modelName).toBe(\"openai/gpt-4o\");\n expect(result.middleware).toBeUndefined();\n expect(result.clientOptions).toBeUndefined();\n });\n\n it(\"middleware is separated from clientOptions when resolving per-method overrides\", () => {\n const { middleware: mw } = createRecordingMiddleware();\n const result = resolveModelConfiguration({\n modelName: \"anthropic/claude-sonnet-4-20250514\",\n apiKey: \"sk-ant-test\",\n middleware: mw,\n });\n\n expect(result.modelName).toBe(\"anthropic/claude-sonnet-4-20250514\");\n expect(result.middleware).toBe(mw);\n expect(result.clientOptions).toEqual({ apiKey: \"sk-ant-test\" });\n expect(result.clientOptions && \"middleware\" in result.clientOptions).toBe(\n false,\n );\n });\n});\n\n// ---------------------------------------------------------------------------\n// Practical middleware behaviors\n// ---------------------------------------------------------------------------\n\ndescribe(\"middleware that tracks duration\", () => {\n it(\"measures wall-clock time of doGenerate\", async () => {\n const durations: number[] = [];\n\n const timingMiddleware: LanguageModelV2Middleware = {\n wrapGenerate: async ({ doGenerate }) => {\n const start = performance.now();\n const result = await doGenerate();\n durations.push(performance.now() - start);\n return result;\n },\n };\n\n const mockModel = createMockLanguageModel(\"gpt-4o\", \"openai.chat\");\n const { wrapLanguageModel } = await import(\"ai\");\n const wrapped = wrapLanguageModel({\n model: mockModel,\n middleware: timingMiddleware,\n });\n\n await wrapped.doGenerate({\n prompt: [{ role: \"user\", content: [{ type: \"text\", text: \"time me\" }] }],\n });\n\n expect(durations).toHaveLength(1);\n expect(durations[0]).toBeGreaterThanOrEqual(0);\n });\n});\n\ndescribe(\"middleware that aggregates token usage\", () => {\n it(\"sums tokens across multiple calls like a billing tracker\", async () => {\n let totalInput = 0;\n let totalOutput = 0;\n\n const billingMiddleware: LanguageModelV2Middleware = {\n wrapGenerate: async ({ doGenerate }) => {\n const result = await doGenerate();\n totalInput += result.usage.inputTokens ?? 0;\n totalOutput += result.usage.outputTokens ?? 0;\n return result;\n },\n };\n\n const mockModel = createMockLanguageModel(\"gpt-4o\", \"openai.chat\");\n const { wrapLanguageModel } = await import(\"ai\");\n const wrapped = wrapLanguageModel({\n model: mockModel,\n middleware: billingMiddleware,\n });\n\n const callOpts = {\n prompt: [\n {\n role: \"user\" as const,\n content: [{ type: \"text\" as const, text: \"go\" }],\n },\n ],\n };\n\n // act\n await wrapped.doGenerate(callOpts);\n // extract\n await wrapped.doGenerate(callOpts);\n // observe\n await wrapped.doGenerate(callOpts);\n\n expect(totalInput).toBe(30);\n expect(totalOutput).toBe(15);\n });\n});\n\ndescribe(\"middleware that logs per-model call counts\", () => {\n it(\"tracks which models were called and how many times\", async () => {\n const modelCallCounts = new Map<string, number>();\n\n const countingMiddleware: LanguageModelV2Middleware = {\n wrapGenerate: async ({ doGenerate, model }) => {\n const key = `${model.provider}/${model.modelId}`;\n modelCallCounts.set(key, (modelCallCounts.get(key) ?? 0) + 1);\n return doGenerate();\n },\n };\n\n const modelA = createMockLanguageModel(\"gpt-4o\", \"openai.chat\");\n const modelB = createMockLanguageModel(\n \"claude-sonnet-4-20250514\",\n \"anthropic.messages\",\n );\n\n const { wrapLanguageModel } = await import(\"ai\");\n const wrappedA = wrapLanguageModel({\n model: modelA,\n middleware: countingMiddleware,\n });\n const wrappedB = wrapLanguageModel({\n model: modelB,\n middleware: countingMiddleware,\n });\n\n const callOpts = {\n prompt: [\n {\n role: \"user\" as const,\n content: [{ type: \"text\" as const, text: \"go\" }],\n },\n ],\n };\n\n await wrappedA.doGenerate(callOpts);\n await wrappedA.doGenerate(callOpts);\n await wrappedB.doGenerate(callOpts);\n\n expect(modelCallCounts.get(\"openai.chat/gpt-4o\")).toBe(2);\n expect(\n modelCallCounts.get(\"anthropic.messages/claude-sonnet-4-20250514\"),\n ).toBe(1);\n expect(modelCallCounts.size).toBe(2);\n });\n});\n\ndescribe(\"middleware that detects errors\", () => {\n it(\"catches and re-throws errors from doGenerate while still recording the failure\", async () => {\n const errors: Error[] = [];\n\n const errorTrackingMiddleware: LanguageModelV2Middleware = {\n wrapGenerate: async ({ doGenerate }) => {\n try {\n return await doGenerate();\n } catch (err) {\n errors.push(err as Error);\n throw err;\n }\n },\n };\n\n const failingModel: LanguageModelV2 = {\n specificationVersion: \"v2\",\n provider: \"openai.chat\",\n modelId: \"gpt-4o\",\n defaultObjectGenerationMode: \"json\",\n doGenerate: vi.fn().mockRejectedValue(new Error(\"rate limit exceeded\")),\n doStream: vi.fn(),\n } as unknown as LanguageModelV2;\n\n const { wrapLanguageModel } = await import(\"ai\");\n const wrapped = wrapLanguageModel({\n model: failingModel,\n middleware: errorTrackingMiddleware,\n });\n\n await expect(\n wrapped.doGenerate({\n prompt: [{ role: \"user\", content: [{ type: \"text\", text: \"fail\" }] }],\n }),\n ).rejects.toThrow(\"rate limit exceeded\");\n\n expect(errors).toHaveLength(1);\n expect(errors[0].message).toBe(\"rate limit exceeded\");\n });\n});\n\ndescribe(\"chaining middleware across multiple wrapped models\", () => {\n it(\"same middleware instance sees calls from different models\", async () => {\n const seen: string[] = [];\n\n const spyMiddleware: LanguageModelV2Middleware = {\n wrapGenerate: async ({ doGenerate, model }) => {\n seen.push(model.modelId);\n return doGenerate();\n },\n };\n\n const mockA = createMockLanguageModel(\"gpt-4o\", \"openai.chat\");\n const mockB = createMockLanguageModel(\n \"claude-sonnet-4-20250514\",\n \"anthropic.messages\",\n );\n\n const { wrapLanguageModel } = await import(\"ai\");\n const wrappedA = wrapLanguageModel({\n model: mockA,\n middleware: spyMiddleware,\n });\n const wrappedB = wrapLanguageModel({\n model: mockB,\n middleware: spyMiddleware,\n });\n\n const callOpts = {\n prompt: [\n {\n role: \"user\" as const,\n content: [{ type: \"text\" as const, text: \"go\" }],\n },\n ],\n };\n\n await wrappedA.doGenerate(callOpts);\n await wrappedB.doGenerate(callOpts);\n await wrappedA.doGenerate(callOpts);\n\n expect(seen).toEqual([\"gpt-4o\", \"claude-sonnet-4-20250514\", \"gpt-4o\"]);\n });\n});\n\n// ---------------------------------------------------------------------------\n// Edge cases\n// ---------------------------------------------------------------------------\n\ndescribe(\"middleware edge cases\", () => {\n it(\"middleware that only implements wrapGenerate still works for doStream\", async () => {\n const generateCalls: string[] = [];\n const partialMiddleware: LanguageModelV2Middleware = {\n wrapGenerate: async ({ doGenerate, model }) => {\n generateCalls.push(model.modelId);\n return doGenerate();\n },\n };\n\n const mockModel = createMockLanguageModel(\"gpt-4o\", \"openai.chat\");\n const { wrapLanguageModel } = await import(\"ai\");\n const wrapped = wrapLanguageModel({\n model: mockModel,\n middleware: partialMiddleware,\n });\n\n const result = await wrapped.doStream({\n prompt: [{ role: \"user\", content: [{ type: \"text\", text: \"stream\" }] }],\n });\n expect(result.stream).toBeDefined();\n\n expect(generateCalls).toHaveLength(0);\n });\n\n it(\"middleware does not alter the response data\", async () => {\n const { middleware } = createRecordingMiddleware();\n const mockModel = createMockLanguageModel(\"gpt-4o\", \"openai.chat\");\n\n const { wrapLanguageModel } = await import(\"ai\");\n const wrapped = wrapLanguageModel({ model: mockModel, middleware });\n\n const result = await wrapped.doGenerate({\n prompt: [{ role: \"user\", content: [{ type: \"text\", text: \"test\" }] }],\n });\n\n expect(result.content).toEqual([{ type: \"text\", text: \"mock response\" }]);\n expect(result.usage).toEqual({\n inputTokens: 10,\n outputTokens: 5,\n totalTokens: 15,\n });\n expect(result.finishReason).toBe(\"stop\");\n });\n});\n"]}

package/dist/cjs/tests/unit/llm-provider.test.d.ts ADDED Viewed

	@@ -0,0 +1 @@
1	+ export {};

package/dist/cjs/tests/unit/llm-provider.test.js ADDED Viewed

@@ -0,0 +1,64 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const LLMProvider_js_1 = require("../../lib/v3/llm/LLMProvider.js");
+(0, vitest_1.describe)("getAISDKLanguageModel", () => {
+    (0, vitest_1.describe)("ollama provider", () => {
+        (0, vitest_1.it)("works without clientOptions", () => {
+            const model = (0, LLMProvider_js_1.getAISDKLanguageModel)("ollama", "llama3.2");
+            (0, vitest_1.expect)(model).toBeDefined();
+        });
+        (0, vitest_1.it)("works with empty clientOptions", () => {
+            const model = (0, LLMProvider_js_1.getAISDKLanguageModel)("ollama", "llama3.2", {});
+            (0, vitest_1.expect)(model).toBeDefined();
+        });
+        (0, vitest_1.it)("works with clientOptions containing only undefined values", () => {
+            const model = (0, LLMProvider_js_1.getAISDKLanguageModel)("ollama", "llama3.2", {
+                apiKey: undefined,
+            });
+            (0, vitest_1.expect)(model).toBeDefined();
+        });
+        (0, vitest_1.it)("works with clientOptions containing only null values", () => {
+            const model = (0, LLMProvider_js_1.getAISDKLanguageModel)("ollama", "llama3.2", {
+                apiKey: null,
+            });
+            (0, vitest_1.expect)(model).toBeDefined();
+        });
+        (0, vitest_1.it)("works with custom baseURL", () => {
+            const model = (0, LLMProvider_js_1.getAISDKLanguageModel)("ollama", "llama3.2", {
+                baseURL: "http://custom-ollama:11434",
+            });
+            (0, vitest_1.expect)(model).toBeDefined();
+        });
+        (0, vitest_1.it)("works even when apiKey is mistakenly provided", () => {
+            // Ollama doesn't need an API key, but users might set one anyway
+            const model = (0, LLMProvider_js_1.getAISDKLanguageModel)("ollama", "llama3.2", {
+                apiKey: "unnecessary-key",
+            });
+            (0, vitest_1.expect)(model).toBeDefined();
+        });
+    });
+    (0, vitest_1.describe)("providers with API keys", () => {
+        (0, vitest_1.it)("openai requires valid clientOptions for custom configuration", () => {
+            // Without clientOptions, uses default provider
+            const defaultModel = (0, LLMProvider_js_1.getAISDKLanguageModel)("openai", "gpt-4o");
+            (0, vitest_1.expect)(defaultModel).toBeDefined();
+            // With valid apiKey, uses custom provider
+            const customModel = (0, LLMProvider_js_1.getAISDKLanguageModel)("openai", "gpt-4o", {
+                apiKey: "test-key",
+            });
+            (0, vitest_1.expect)(customModel).toBeDefined();
+        });
+    });
+    (0, vitest_1.describe)("hasValidOptions logic", () => {
+        (0, vitest_1.it)("treats undefined apiKey as no options", () => {
+            // This should use the default provider path (AISDKProviders)
+            // not the custom provider path (AISDKProvidersWithAPIKey)
+            const model = (0, LLMProvider_js_1.getAISDKLanguageModel)("ollama", "llama3.2", {
+                apiKey: undefined,
+            });
+            (0, vitest_1.expect)(model).toBeDefined();
+        });
+    });
+});
+//# sourceMappingURL=llm-provider.test.js.map

package/dist/cjs/tests/unit/llm-provider.test.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"llm-provider.test.js","sourceRoot":"","sources":["../../../../tests/unit/llm-provider.test.ts"],"names":[],"mappings":";;AAAA,mCAA8C;AAC9C,oEAAwE;AAExE,IAAA,iBAAQ,EAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,IAAA,iBAAQ,EAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,IAAA,WAAE,EAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,KAAK,GAAG,IAAA,sCAAqB,EAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YAC1D,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,KAAK,GAAG,IAAA,sCAAqB,EAAC,QAAQ,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;YAC9D,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,2DAA2D,EAAE,GAAG,EAAE;YACnE,MAAM,KAAK,GAAG,IAAA,sCAAqB,EAAC,QAAQ,EAAE,UAAU,EAAE;gBACxD,MAAM,EAAE,SAAS;aAClB,CAAC,CAAC;YACH,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,KAAK,GAAG,IAAA,sCAAqB,EAAC,QAAQ,EAAE,UAAU,EAAE;gBACxD,MAAM,EAAE,IAAyB;aAClC,CAAC,CAAC;YACH,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,2BAA2B,EAAE,GAAG,EAAE;YACnC,MAAM,KAAK,GAAG,IAAA,sCAAqB,EAAC,QAAQ,EAAE,UAAU,EAAE;gBACxD,OAAO,EAAE,4BAA4B;aACtC,CAAC,CAAC;YACH,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,iEAAiE;YACjE,MAAM,KAAK,GAAG,IAAA,sCAAqB,EAAC,QAAQ,EAAE,UAAU,EAAE;gBACxD,MAAM,EAAE,iBAAiB;aAC1B,CAAC,CAAC;YACH,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAA,iBAAQ,EAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,IAAA,WAAE,EAAC,8DAA8D,EAAE,GAAG,EAAE;YACtE,+CAA+C;YAC/C,MAAM,YAAY,GAAG,IAAA,sCAAqB,EAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAC/D,IAAA,eAAM,EAAC,YAAY,CAAC,CAAC,WAAW,EAAE,CAAC;YAEnC,0CAA0C;YAC1C,MAAM,WAAW,GAAG,IAAA,sCAAqB,EAAC,QAAQ,EAAE,QAAQ,EAAE;gBAC5D,MAAM,EAAE,UAAU;aACnB,CAAC,CAAC;YACH,IAAA,eAAM,EAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;QACpC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAA,iBAAQ,EAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,IAAA,WAAE,EAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,6DAA6D;YAC7D,0DAA0D;YAC1D,MAAM,KAAK,GAAG,IAAA,sCAAqB,EAAC,QAAQ,EAAE,UAAU,EAAE;gBACxD,MAAM,EAAE,SAAS;aAClB,CAAC,CAAC;YACH,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it } from \"vitest\";\nimport { getAISDKLanguageModel } from \"../../lib/v3/llm/LLMProvider.js\";\n\ndescribe(\"getAISDKLanguageModel\", () => {\n describe(\"ollama provider\", () => {\n it(\"works without clientOptions\", () => {\n const model = getAISDKLanguageModel(\"ollama\", \"llama3.2\");\n expect(model).toBeDefined();\n });\n\n it(\"works with empty clientOptions\", () => {\n const model = getAISDKLanguageModel(\"ollama\", \"llama3.2\", {});\n expect(model).toBeDefined();\n });\n\n it(\"works with clientOptions containing only undefined values\", () => {\n const model = getAISDKLanguageModel(\"ollama\", \"llama3.2\", {\n apiKey: undefined,\n });\n expect(model).toBeDefined();\n });\n\n it(\"works with clientOptions containing only null values\", () => {\n const model = getAISDKLanguageModel(\"ollama\", \"llama3.2\", {\n apiKey: null as unknown as string,\n });\n expect(model).toBeDefined();\n });\n\n it(\"works with custom baseURL\", () => {\n const model = getAISDKLanguageModel(\"ollama\", \"llama3.2\", {\n baseURL: \"http://custom-ollama:11434\",\n });\n expect(model).toBeDefined();\n });\n\n it(\"works even when apiKey is mistakenly provided\", () => {\n // Ollama doesn't need an API key, but users might set one anyway\n const model = getAISDKLanguageModel(\"ollama\", \"llama3.2\", {\n apiKey: \"unnecessary-key\",\n });\n expect(model).toBeDefined();\n });\n });\n\n describe(\"providers with API keys\", () => {\n it(\"openai requires valid clientOptions for custom configuration\", () => {\n // Without clientOptions, uses default provider\n const defaultModel = getAISDKLanguageModel(\"openai\", \"gpt-4o\");\n expect(defaultModel).toBeDefined();\n\n // With valid apiKey, uses custom provider\n const customModel = getAISDKLanguageModel(\"openai\", \"gpt-4o\", {\n apiKey: \"test-key\",\n });\n expect(customModel).toBeDefined();\n });\n });\n\n describe(\"hasValidOptions logic\", () => {\n it(\"treats undefined apiKey as no options\", () => {\n // This should use the default provider path (AISDKProviders)\n // not the custom provider path (AISDKProvidersWithAPIKey)\n const model = getAISDKLanguageModel(\"ollama\", \"llama3.2\", {\n apiKey: undefined,\n });\n expect(model).toBeDefined();\n });\n });\n});\n"]}

package/dist/cjs/tests/unit/model-deprecation.test.d.ts ADDED Viewed

	@@ -0,0 +1 @@
1	+ export {};

package/dist/cjs/tests/unit/model-deprecation.test.js ADDED Viewed

@@ -0,0 +1,142 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const LLMProvider_js_1 = require("../../lib/v3/llm/LLMProvider.js");
+const sdkErrors_js_1 = require("../../lib/v3/types/public/sdkErrors.js");
+// Mock client options with fake API keys for testing
+const mockClientOptions = { apiKey: "test-api-key-for-testing" };
+(0, vitest_1.describe)("Model format deprecation", () => {
+    (0, vitest_1.describe)("UnsupportedModelError", () => {
+        (0, vitest_1.it)("includes guidance to use provider/model format for unknown model names", () => {
+            const error = new sdkErrors_js_1.UnsupportedModelError(["gpt-4o", "gemini-2.0-flash"]);
+            // Should mention the new format
+            (0, vitest_1.expect)(error.message).toContain("provider/model");
+            // Should include link to docs
+            (0, vitest_1.expect)(error.message).toContain("https://docs.stagehand.dev/v3/configuration/models");
+        });
+        (0, vitest_1.it)("includes example of provider/model format", () => {
+            const error = new sdkErrors_js_1.UnsupportedModelError(["gpt-4o"]);
+            // Should provide examples like openai/gpt-4o
+            (0, vitest_1.expect)(error.message).toContain("openai/gpt-4o");
+            (0, vitest_1.expect)(error.message).toContain("anthropic/claude-sonnet-4");
+        });
+        (0, vitest_1.it)("works with feature parameter", () => {
+            const error = new sdkErrors_js_1.UnsupportedModelError(["gpt-4o"], "extract");
+            (0, vitest_1.expect)(error.message).toContain("extract");
+            (0, vitest_1.expect)(error.message).toContain("provider/model");
+            (0, vitest_1.expect)(error.message).toContain("https://docs.stagehand.dev/v3/configuration/models");
+        });
+    });
+    (0, vitest_1.describe)("LLMProvider.getClient deprecation warning", () => {
+        (0, vitest_1.it)("logs deprecation warning for legacy model names", () => {
+            const logs = [];
+            const logger = (line) => logs.push(line);
+            const provider = new LLMProvider_js_1.LLMProvider(logger);
+            // Using a legacy model name like "gpt-4o" instead of "openai/gpt-4o"
+            // Should not throw, but should log a deprecation warning
+            const client = provider.getClient("gpt-4o", mockClientOptions);
+            // Should return a client (not throw)
+            (0, vitest_1.expect)(client).toBeDefined();
+            // Should have logged a deprecation warning at level 0
+            const deprecationWarning = logs.find((log) => log.message.toLowerCase().includes("deprecated") ||
+                log.message.toLowerCase().includes("deprecation"));
+            (0, vitest_1.expect)(deprecationWarning).toBeDefined();
+            (0, vitest_1.expect)(deprecationWarning.level).toBe(0);
+        });
+        (0, vitest_1.it)("deprecation warning mentions provider/model format", () => {
+            const logs = [];
+            const logger = (line) => logs.push(line);
+            const provider = new LLMProvider_js_1.LLMProvider(logger);
+            provider.getClient("gpt-4o", mockClientOptions);
+            const deprecationWarning = logs.find((log) => log.message.toLowerCase().includes("deprecated") ||
+                log.message.toLowerCase().includes("deprecation"));
+            (0, vitest_1.expect)(deprecationWarning).toBeDefined();
+            const message = deprecationWarning.message;
+            // Should mention the provider/model format
+            (0, vitest_1.expect)(message).toContain("provider/model");
+            // Should give an example
+            (0, vitest_1.expect)(message).toContain("openai/gpt-5");
+        });
+        (0, vitest_1.it)("returns OpenAIClient for legacy OpenAI model names", () => {
+            const logs = [];
+            const logger = (line) => logs.push(line);
+            const provider = new LLMProvider_js_1.LLMProvider(logger);
+            const client = provider.getClient("gpt-4o", mockClientOptions);
+            // Should return a client
+            (0, vitest_1.expect)(client).toBeDefined();
+            // The client should be an OpenAIClient (check constructor name)
+            (0, vitest_1.expect)(client.constructor.name).toBe("OpenAIClient");
+        });
+        (0, vitest_1.it)("returns GoogleClient for legacy Google model names", () => {
+            const logs = [];
+            const logger = (line) => logs.push(line);
+            const provider = new LLMProvider_js_1.LLMProvider(logger);
+            const client = provider.getClient("gemini-2.0-flash", mockClientOptions);
+            // Should return a client
+            (0, vitest_1.expect)(client).toBeDefined();
+            // The client should be a GoogleClient
+            (0, vitest_1.expect)(client.constructor.name).toBe("GoogleClient");
+        });
+    });
+    (0, vitest_1.describe)("LLMProvider.getClient error handling", () => {
+        (0, vitest_1.it)("throws UnsupportedModelError for unknown model without slash", () => {
+            const logs = [];
+            const logger = (line) => logs.push(line);
+            const provider = new LLMProvider_js_1.LLMProvider(logger);
+            // Unknown model without slash should throw UnsupportedModelError
+            (0, vitest_1.expect)(() => {
+                provider.getClient("some-unknown-model", mockClientOptions);
+            }).toThrow(sdkErrors_js_1.UnsupportedModelError);
+        });
+        (0, vitest_1.it)("UnsupportedModelError includes provider/model format guidance", () => {
+            const logs = [];
+            const logger = (line) => logs.push(line);
+            const provider = new LLMProvider_js_1.LLMProvider(logger);
+            try {
+                provider.getClient("some-unknown-model", mockClientOptions);
+            }
+            catch (error) {
+                (0, vitest_1.expect)(error.message).toContain("provider/model");
+            }
+        });
+        (0, vitest_1.it)("throws UnsupportedAISDKModelProviderError for invalid provider in provider/model format", () => {
+            const logs = [];
+            const logger = (line) => logs.push(line);
+            const provider = new LLMProvider_js_1.LLMProvider(logger);
+            // Invalid provider but correct format
+            (0, vitest_1.expect)(() => {
+                provider.getClient("invalid-provider/some-model", mockClientOptions);
+            }).toThrow(sdkErrors_js_1.UnsupportedAISDKModelProviderError);
+        });
+        (0, vitest_1.it)("UnsupportedAISDKModelProviderError lists valid providers", () => {
+            const logs = [];
+            const logger = (line) => logs.push(line);
+            const provider = new LLMProvider_js_1.LLMProvider(logger);
+            try {
+                provider.getClient("invalid-provider/some-model", mockClientOptions);
+            }
+            catch (error) {
+                const message = error.message;
+                // Should list valid providers
+                (0, vitest_1.expect)(message).toContain("openai");
+                (0, vitest_1.expect)(message).toContain("anthropic");
+                (0, vitest_1.expect)(message).toContain("google");
+            }
+        });
+    });
+    (0, vitest_1.describe)("new provider/model format", () => {
+        (0, vitest_1.it)("does not log deprecation warning for provider/model format", () => {
+            const logs = [];
+            const logger = (line) => logs.push(line);
+            const provider = new LLMProvider_js_1.LLMProvider(logger);
+            // Using the new format
+            const client = provider.getClient("openai/gpt-4o", mockClientOptions);
+            (0, vitest_1.expect)(client).toBeDefined();
+            // Should NOT have a deprecation warning
+            const deprecationWarning = logs.find((log) => log.message.toLowerCase().includes("deprecated") ||
+                log.message.toLowerCase().includes("deprecation"));
+            (0, vitest_1.expect)(deprecationWarning).toBeUndefined();
+        });
+    });
+});
+//# sourceMappingURL=model-deprecation.test.js.map

package/dist/cjs/tests/unit/model-deprecation.test.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"model-deprecation.test.js","sourceRoot":"","sources":["../../../../tests/unit/model-deprecation.test.ts"],"names":[],"mappings":";;AAAA,mCAA8C;AAC9C,oEAA8D;AAC9D,yEAGgD;AAGhD,qDAAqD;AACrD,MAAM,iBAAiB,GAAG,EAAE,MAAM,EAAE,0BAA0B,EAAE,CAAC;AAEjE,IAAA,iBAAQ,EAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,IAAA,iBAAQ,EAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,IAAA,WAAE,EAAC,wEAAwE,EAAE,GAAG,EAAE;YAChF,MAAM,KAAK,GAAG,IAAI,oCAAqB,CAAC,CAAC,QAAQ,EAAE,kBAAkB,CAAC,CAAC,CAAC;YAExE,gCAAgC;YAChC,IAAA,eAAM,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAClD,8BAA8B;YAC9B,IAAA,eAAM,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAC7B,oDAAoD,CACrD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,KAAK,GAAG,IAAI,oCAAqB,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;YAEpD,6CAA6C;YAC7C,IAAA,eAAM,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;YACjD,IAAA,eAAM,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,2BAA2B,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,KAAK,GAAG,IAAI,oCAAqB,CAAC,CAAC,QAAQ,CAAC,EAAE,SAAS,CAAC,CAAC;YAE/D,IAAA,eAAM,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YAC3C,IAAA,eAAM,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAClD,IAAA,eAAM,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAC7B,oDAAoD,CACrD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAA,iBAAQ,EAAC,2CAA2C,EAAE,GAAG,EAAE;QACzD,IAAA,WAAE,EAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,4BAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,qEAAqE;YACrE,yDAAyD;YACzD,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC;YAE/D,qCAAqC;YACrC,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAE7B,sDAAsD;YACtD,MAAM,kBAAkB,GAAG,IAAI,CAAC,IAAI,CAClC,CAAC,GAAG,EAAE,EAAE,CACN,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC;gBAChD,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,CACpD,CAAC;YACF,IAAA,eAAM,EAAC,kBAAkB,CAAC,CAAC,WAAW,EAAE,CAAC;YACzC,IAAA,eAAM,EAAC,kBAAmB,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,4BAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,QAAQ,CAAC,SAAS,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC;YAEhD,MAAM,kBAAkB,GAAG,IAAI,CAAC,IAAI,CAClC,CAAC,GAAG,EAAE,EAAE,CACN,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC;gBAChD,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,CACpD,CAAC;YAEF,IAAA,eAAM,EAAC,kBAAkB,CAAC,CAAC,WAAW,EAAE,CAAC;YACzC,MAAM,OAAO,GAAG,kBAAmB,CAAC,OAAO,CAAC;YAC5C,2CAA2C;YAC3C,IAAA,eAAM,EAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAC5C,yBAAyB;YACzB,IAAA,eAAM,EAAC,OAAO,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,4BAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC;YAE/D,yBAAyB;YACzB,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAC7B,gEAAgE;YAChE,IAAA,eAAM,EAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,4BAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,kBAAkB,EAAE,iBAAiB,CAAC,CAAC;YAEzE,yBAAyB;YACzB,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAC7B,sCAAsC;YACtC,IAAA,eAAM,EAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAA,iBAAQ,EAAC,sCAAsC,EAAE,GAAG,EAAE;QACpD,IAAA,WAAE,EAAC,8DAA8D,EAAE,GAAG,EAAE;YACtE,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,4BAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,iEAAiE;YACjE,IAAA,eAAM,EAAC,GAAG,EAAE;gBACV,QAAQ,CAAC,SAAS,CAAC,oBAAoB,EAAE,iBAAiB,CAAC,CAAC;YAC9D,CAAC,CAAC,CAAC,OAAO,CAAC,oCAAqB,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,4BAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,IAAI,CAAC;gBACH,QAAQ,CAAC,SAAS,CAAC,oBAAoB,EAAE,iBAAiB,CAAC,CAAC;YAC9D,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAA,eAAM,EAAE,KAAe,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,yFAAyF,EAAE,GAAG,EAAE;YACjG,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,4BAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,sCAAsC;YACtC,IAAA,eAAM,EAAC,GAAG,EAAE;gBACV,QAAQ,CAAC,SAAS,CAAC,6BAA6B,EAAE,iBAAiB,CAAC,CAAC;YACvE,CAAC,CAAC,CAAC,OAAO,CAAC,iDAAkC,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,IAAA,WAAE,EAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,4BAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,IAAI,CAAC;gBACH,QAAQ,CAAC,SAAS,CAAC,6BAA6B,EAAE,iBAAiB,CAAC,CAAC;YACvE,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,OAAO,GAAI,KAAe,CAAC,OAAO,CAAC;gBACzC,8BAA8B;gBAC9B,IAAA,eAAM,EAAC,OAAO,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;gBACpC,IAAA,eAAM,EAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;gBACvC,IAAA,eAAM,EAAC,OAAO,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YACtC,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAA,iBAAQ,EAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,IAAA,WAAE,EAAC,4DAA4D,EAAE,GAAG,EAAE;YACpE,MAAM,IAAI,GAAc,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,4BAAW,CAAC,MAAM,CAAC,CAAC;YAEzC,uBAAuB;YACvB,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,eAAe,EAAE,iBAAiB,CAAC,CAAC;YAEtE,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAE7B,wCAAwC;YACxC,MAAM,kBAAkB,GAAG,IAAI,CAAC,IAAI,CAClC,CAAC,GAAG,EAAE,EAAE,CACN,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC;gBAChD,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,CACpD,CAAC;YACF,IAAA,eAAM,EAAC,kBAAkB,CAAC,CAAC,aAAa,EAAE,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it } from \"vitest\";\nimport { LLMProvider } from \"../../lib/v3/llm/LLMProvider.js\";\nimport {\n UnsupportedModelError,\n UnsupportedAISDKModelProviderError,\n} from \"../../lib/v3/types/public/sdkErrors.js\";\nimport type { LogLine } from \"../../lib/v3/types/public/logs.js\";\n\n// Mock client options with fake API keys for testing\nconst mockClientOptions = { apiKey: \"test-api-key-for-testing\" };\n\ndescribe(\"Model format deprecation\", () => {\n describe(\"UnsupportedModelError\", () => {\n it(\"includes guidance to use provider/model format for unknown model names\", () => {\n const error = new UnsupportedModelError([\"gpt-4o\", \"gemini-2.0-flash\"]);\n\n // Should mention the new format\n expect(error.message).toContain(\"provider/model\");\n // Should include link to docs\n expect(error.message).toContain(\n \"https://docs.stagehand.dev/v3/configuration/models\",\n );\n });\n\n it(\"includes example of provider/model format\", () => {\n const error = new UnsupportedModelError([\"gpt-4o\"]);\n\n // Should provide examples like openai/gpt-4o\n expect(error.message).toContain(\"openai/gpt-4o\");\n expect(error.message).toContain(\"anthropic/claude-sonnet-4\");\n });\n\n it(\"works with feature parameter\", () => {\n const error = new UnsupportedModelError([\"gpt-4o\"], \"extract\");\n\n expect(error.message).toContain(\"extract\");\n expect(error.message).toContain(\"provider/model\");\n expect(error.message).toContain(\n \"https://docs.stagehand.dev/v3/configuration/models\",\n );\n });\n });\n\n describe(\"LLMProvider.getClient deprecation warning\", () => {\n it(\"logs deprecation warning for legacy model names\", () => {\n const logs: LogLine[] = [];\n const logger = (line: LogLine) => logs.push(line);\n const provider = new LLMProvider(logger);\n\n // Using a legacy model name like \"gpt-4o\" instead of \"openai/gpt-4o\"\n // Should not throw, but should log a deprecation warning\n const client = provider.getClient(\"gpt-4o\", mockClientOptions);\n\n // Should return a client (not throw)\n expect(client).toBeDefined();\n\n // Should have logged a deprecation warning at level 0\n const deprecationWarning = logs.find(\n (log) =>\n log.message.toLowerCase().includes(\"deprecated\") ||\n log.message.toLowerCase().includes(\"deprecation\"),\n );\n expect(deprecationWarning).toBeDefined();\n expect(deprecationWarning!.level).toBe(0);\n });\n\n it(\"deprecation warning mentions provider/model format\", () => {\n const logs: LogLine[] = [];\n const logger = (line: LogLine) => logs.push(line);\n const provider = new LLMProvider(logger);\n\n provider.getClient(\"gpt-4o\", mockClientOptions);\n\n const deprecationWarning = logs.find(\n (log) =>\n log.message.toLowerCase().includes(\"deprecated\") ||\n log.message.toLowerCase().includes(\"deprecation\"),\n );\n\n expect(deprecationWarning).toBeDefined();\n const message = deprecationWarning!.message;\n // Should mention the provider/model format\n expect(message).toContain(\"provider/model\");\n // Should give an example\n expect(message).toContain(\"openai/gpt-5\");\n });\n\n it(\"returns OpenAIClient for legacy OpenAI model names\", () => {\n const logs: LogLine[] = [];\n const logger = (line: LogLine) => logs.push(line);\n const provider = new LLMProvider(logger);\n\n const client = provider.getClient(\"gpt-4o\", mockClientOptions);\n\n // Should return a client\n expect(client).toBeDefined();\n // The client should be an OpenAIClient (check constructor name)\n expect(client.constructor.name).toBe(\"OpenAIClient\");\n });\n\n it(\"returns GoogleClient for legacy Google model names\", () => {\n const logs: LogLine[] = [];\n const logger = (line: LogLine) => logs.push(line);\n const provider = new LLMProvider(logger);\n\n const client = provider.getClient(\"gemini-2.0-flash\", mockClientOptions);\n\n // Should return a client\n expect(client).toBeDefined();\n // The client should be a GoogleClient\n expect(client.constructor.name).toBe(\"GoogleClient\");\n });\n });\n\n describe(\"LLMProvider.getClient error handling\", () => {\n it(\"throws UnsupportedModelError for unknown model without slash\", () => {\n const logs: LogLine[] = [];\n const logger = (line: LogLine) => logs.push(line);\n const provider = new LLMProvider(logger);\n\n // Unknown model without slash should throw UnsupportedModelError\n expect(() => {\n provider.getClient(\"some-unknown-model\", mockClientOptions);\n }).toThrow(UnsupportedModelError);\n });\n\n it(\"UnsupportedModelError includes provider/model format guidance\", () => {\n const logs: LogLine[] = [];\n const logger = (line: LogLine) => logs.push(line);\n const provider = new LLMProvider(logger);\n\n try {\n provider.getClient(\"some-unknown-model\", mockClientOptions);\n } catch (error) {\n expect((error as Error).message).toContain(\"provider/model\");\n }\n });\n\n it(\"throws UnsupportedAISDKModelProviderError for invalid provider in provider/model format\", () => {\n const logs: LogLine[] = [];\n const logger = (line: LogLine) => logs.push(line);\n const provider = new LLMProvider(logger);\n\n // Invalid provider but correct format\n expect(() => {\n provider.getClient(\"invalid-provider/some-model\", mockClientOptions);\n }).toThrow(UnsupportedAISDKModelProviderError);\n });\n\n it(\"UnsupportedAISDKModelProviderError lists valid providers\", () => {\n const logs: LogLine[] = [];\n const logger = (line: LogLine) => logs.push(line);\n const provider = new LLMProvider(logger);\n\n try {\n provider.getClient(\"invalid-provider/some-model\", mockClientOptions);\n } catch (error) {\n const message = (error as Error).message;\n // Should list valid providers\n expect(message).toContain(\"openai\");\n expect(message).toContain(\"anthropic\");\n expect(message).toContain(\"google\");\n }\n });\n });\n\n describe(\"new provider/model format\", () => {\n it(\"does not log deprecation warning for provider/model format\", () => {\n const logs: LogLine[] = [];\n const logger = (line: LogLine) => logs.push(line);\n const provider = new LLMProvider(logger);\n\n // Using the new format\n const client = provider.getClient(\"openai/gpt-4o\", mockClientOptions);\n\n expect(client).toBeDefined();\n\n // Should NOT have a deprecation warning\n const deprecationWarning = logs.find(\n (log) =>\n log.message.toLowerCase().includes(\"deprecated\") ||\n log.message.toLowerCase().includes(\"deprecation\"),\n );\n expect(deprecationWarning).toBeUndefined();\n });\n });\n});\n"]}

package/dist/cjs/tests/unit/model-utils.test.d.ts ADDED Viewed

	@@ -0,0 +1 @@
1	+ export {};

package/dist/cjs/tests/unit/model-utils.test.js ADDED Viewed

@@ -0,0 +1,42 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const modelUtils_js_1 = require("../../lib/modelUtils.js");
+(0, vitest_1.describe)("extractModelName", () => {
+    (0, vitest_1.it)("returns undefined for undefined input", () => {
+        (0, vitest_1.expect)((0, modelUtils_js_1.extractModelName)(undefined)).toBeUndefined();
+    });
+    (0, vitest_1.it)("returns the string as-is for a string input", () => {
+        (0, vitest_1.expect)((0, modelUtils_js_1.extractModelName)("openai/gpt-4o")).toBe("openai/gpt-4o");
+    });
+    (0, vitest_1.it)("returns modelName from an object input", () => {
+        (0, vitest_1.expect)((0, modelUtils_js_1.extractModelName)({ modelName: "anthropic/claude-sonnet-4-20250514" })).toBe("anthropic/claude-sonnet-4-20250514");
+    });
+    (0, vitest_1.it)("returns modelName from an object with extra properties", () => {
+        (0, vitest_1.expect)((0, modelUtils_js_1.extractModelName)({
+            modelName: "openai/gpt-4o-mini",
+            apiKey: "sk-test",
+            baseURL: "https://custom.endpoint",
+        })).toBe("openai/gpt-4o-mini");
+    });
+});
+(0, vitest_1.describe)("resolveModel", () => {
+    (0, vitest_1.it)("extracts provider and modelName from a string", () => {
+        const result = (0, modelUtils_js_1.resolveModel)("openai/gpt-4o");
+        (0, vitest_1.expect)(result.provider).toBe("openai");
+        (0, vitest_1.expect)(result.modelName).toBe("gpt-4o");
+        (0, vitest_1.expect)(result.clientOptions).toEqual({});
+    });
+    (0, vitest_1.it)("extracts clientOptions from an object config", () => {
+        const result = (0, modelUtils_js_1.resolveModel)({
+            modelName: "openai/gpt-4o",
+            apiKey: "sk-test",
+        });
+        (0, vitest_1.expect)(result.provider).toBe("openai");
+        (0, vitest_1.expect)(result.modelName).toBe("gpt-4o");
+        (0, vitest_1.expect)(result.clientOptions).toMatchObject({ apiKey: "sk-test" });
+        // modelName should not leak into clientOptions
+        (0, vitest_1.expect)(result.clientOptions).not.toHaveProperty("modelName");
+    });
+});
+//# sourceMappingURL=model-utils.test.js.map

package/dist/cjs/tests/unit/model-utils.test.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"model-utils.test.js","sourceRoot":"","sources":["../../../../tests/unit/model-utils.test.ts"],"names":[],"mappings":";;AAAA,mCAA8C;AAC9C,2DAAyE;AAEzE,IAAA,iBAAQ,EAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAA,WAAE,EAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,IAAA,eAAM,EAAC,IAAA,gCAAgB,EAAC,SAAS,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,IAAA,eAAM,EAAC,IAAA,gCAAgB,EAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,IAAA,eAAM,EACJ,IAAA,gCAAgB,EAAC,EAAE,SAAS,EAAE,oCAAoC,EAAE,CAAC,CACtE,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,IAAA,eAAM,EACJ,IAAA,gCAAgB,EAAC;YACf,SAAS,EAAE,oBAAoB;YAC/B,MAAM,EAAE,SAAS;YACjB,OAAO,EAAE,yBAAyB;SACnC,CAAC,CACH,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAA,iBAAQ,EAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,IAAA,WAAE,EAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,MAAM,GAAG,IAAA,4BAAY,EAAC,eAAe,CAAC,CAAC;QAC7C,IAAA,eAAM,EAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvC,IAAA,eAAM,EAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAA,eAAM,EAAC,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,MAAM,GAAG,IAAA,4BAAY,EAAC;YAC1B,SAAS,EAAE,eAAwB;YACnC,MAAM,EAAE,SAAS;SAClB,CAAC,CAAC;QACH,IAAA,eAAM,EAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvC,IAAA,eAAM,EAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAA,eAAM,EAAC,MAAM,CAAC,aAAa,CAAC,CAAC,aAAa,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;QAClE,+CAA+C;QAC/C,IAAA,eAAM,EAAC,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it } from \"vitest\";\nimport { extractModelName, resolveModel } from \"../../lib/modelUtils.js\";\n\ndescribe(\"extractModelName\", () => {\n it(\"returns undefined for undefined input\", () => {\n expect(extractModelName(undefined)).toBeUndefined();\n });\n\n it(\"returns the string as-is for a string input\", () => {\n expect(extractModelName(\"openai/gpt-4o\")).toBe(\"openai/gpt-4o\");\n });\n\n it(\"returns modelName from an object input\", () => {\n expect(\n extractModelName({ modelName: \"anthropic/claude-sonnet-4-20250514\" }),\n ).toBe(\"anthropic/claude-sonnet-4-20250514\");\n });\n\n it(\"returns modelName from an object with extra properties\", () => {\n expect(\n extractModelName({\n modelName: \"openai/gpt-4o-mini\",\n apiKey: \"sk-test\",\n baseURL: \"https://custom.endpoint\",\n }),\n ).toBe(\"openai/gpt-4o-mini\");\n });\n});\n\ndescribe(\"resolveModel\", () => {\n it(\"extracts provider and modelName from a string\", () => {\n const result = resolveModel(\"openai/gpt-4o\");\n expect(result.provider).toBe(\"openai\");\n expect(result.modelName).toBe(\"gpt-4o\");\n expect(result.clientOptions).toEqual({});\n });\n\n it(\"extracts clientOptions from an object config\", () => {\n const result = resolveModel({\n modelName: \"openai/gpt-4o\" as never,\n apiKey: \"sk-test\",\n });\n expect(result.provider).toBe(\"openai\");\n expect(result.modelName).toBe(\"gpt-4o\");\n expect(result.clientOptions).toMatchObject({ apiKey: \"sk-test\" });\n // modelName should not leak into clientOptions\n expect(result.clientOptions).not.toHaveProperty(\"modelName\");\n });\n});\n"]}

package/dist/cjs/tests/unit/openai-cua-client.test.d.ts ADDED Viewed

	@@ -0,0 +1 @@
1	+ export {};

package/dist/cjs/tests/unit/openai-cua-client.test.js ADDED Viewed

@@ -0,0 +1,71 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const OpenAICUAClient_js_1 = require("../../lib/v3/agent/OpenAICUAClient.js");
+function createClient() {
+    return new OpenAICUAClient_js_1.OpenAICUAClient("openai", "computer-use-preview-2025-03-11", undefined, { apiKey: "test-key" });
+}
+(0, vitest_1.describe)("OpenAICUAClient", () => {
+    (0, vitest_1.it)("exposes captchaSolvedProceed tool after a captcha context note", () => {
+        const client = createClient();
+        // Before captcha note — tool should not be active
+        (0, vitest_1.expect)(client
+            .captchaSolvedToolActive).toBe(false);
+        // Simulate a captcha context note being added (as the CUA handler does)
+        client.addContextNote("A captcha was automatically detected and solved — no further interaction needed.");
+        (0, vitest_1.expect)(client
+            .captchaSolvedToolActive).toBe(true);
+    });
+    (0, vitest_1.it)("does NOT activate captcha tool for non-captcha context notes", () => {
+        const client = createClient();
+        client.addContextNote("The page has finished loading.");
+        (0, vitest_1.expect)(client
+            .captchaSolvedToolActive).toBe(false);
+    });
+    (0, vitest_1.it)("deactivates captcha tool after takeAction handles the function call", async () => {
+        const client = createClient();
+        client.addContextNote("A captcha was solved.");
+        (0, vitest_1.expect)(client
+            .captchaSolvedToolActive).toBe(true);
+        // Simulate the model calling the captchaSolvedProceed tool
+        const result = await client.takeAction([
+            {
+                type: "function_call",
+                name: "captchaSolvedProceed",
+                call_id: "call-1",
+                arguments: "{}",
+            },
+        ], vitest_1.vi.fn());
+        // Tool should be deactivated
+        (0, vitest_1.expect)(client
+            .captchaSolvedToolActive).toBe(false);
+        // Result should contain a function_call_output confirming proceed
+        (0, vitest_1.expect)(result).toEqual([
+            {
+                type: "function_call_output",
+                call_id: "call-1",
+                output: vitest_1.expect.stringContaining("Continue completing"),
+            },
+        ]);
+    });
+    (0, vitest_1.it)("does NOT auto-continue follow-up questions without a captcha context", async () => {
+        const client = createClient();
+        const executeStepSpy = vitest_1.vi.spyOn(client, "executeStep");
+        executeStepSpy.mockResolvedValueOnce({
+            actions: [],
+            message: "I've located the Submit button. Should I go ahead and submit it?",
+            completed: true,
+            nextInputItems: [],
+            responseId: "response-1",
+            usage: { input_tokens: 1, output_tokens: 1, inference_time_ms: 1 },
+        });
+        const result = await client.execute({
+            options: { instruction: "Submit the form.", maxSteps: 10 },
+            logger: vitest_1.vi.fn(),
+        });
+        // Should NOT have continued — the model's follow-up is treated as completion
+        (0, vitest_1.expect)(executeStepSpy).toHaveBeenCalledTimes(1);
+        (0, vitest_1.expect)(result.completed).toBe(true);
+    });
+});
+//# sourceMappingURL=openai-cua-client.test.js.map

package/dist/cjs/tests/unit/openai-cua-client.test.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"openai-cua-client.test.js","sourceRoot":"","sources":["../../../../tests/unit/openai-cua-client.test.ts"],"names":[],"mappings":";;AAAA,mCAAkD;AAClD,8EAAwE;AAExE,SAAS,YAAY;IACnB,OAAO,IAAI,oCAAe,CACxB,QAAQ,EACR,iCAAiC,EACjC,SAAS,EACT,EAAE,MAAM,EAAE,UAAU,EAAE,CACvB,CAAC;AACJ,CAAC;AAED,IAAA,iBAAQ,EAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAA,WAAE,EAAC,gEAAgE,EAAE,GAAG,EAAE;QACxE,MAAM,MAAM,GAAG,YAAY,EAAE,CAAC;QAE9B,kDAAkD;QAClD,IAAA,eAAM,EACH,MAA0D;aACxD,uBAAuB,CAC3B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEd,wEAAwE;QACxE,MAAM,CAAC,cAAc,CACnB,kFAAkF,CACnF,CAAC;QAEF,IAAA,eAAM,EACH,MAA0D;aACxD,uBAAuB,CAC3B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,MAAM,GAAG,YAAY,EAAE,CAAC;QAE9B,MAAM,CAAC,cAAc,CAAC,gCAAgC,CAAC,CAAC;QAExD,IAAA,eAAM,EACH,MAA0D;aACxD,uBAAuB,CAC3B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChB,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,MAAM,MAAM,GAAG,YAAY,EAAE,CAAC;QAC9B,MAAM,CAAC,cAAc,CAAC,uBAAuB,CAAC,CAAC;QAE/C,IAAA,eAAM,EACH,MAA0D;aACxD,uBAAuB,CAC3B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEb,2DAA2D;QAC3D,MAAM,MAAM,GAAG,MACb,MAMD,CAAC,UAAU,CACV;YACE;gBACE,IAAI,EAAE,eAAe;gBACrB,IAAI,EAAE,sBAAsB;gBAC5B,OAAO,EAAE,QAAQ;gBACjB,SAAS,EAAE,IAAI;aAChB;SACF,EACD,WAAE,CAAC,EAAE,EAAE,CACR,CAAC;QAEF,6BAA6B;QAC7B,IAAA,eAAM,EACH,MAA0D;aACxD,uBAAuB,CAC3B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEd,kEAAkE;QAClE,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB;gBACE,IAAI,EAAE,sBAAsB;gBAC5B,OAAO,EAAE,QAAQ;gBACjB,MAAM,EAAE,eAAM,CAAC,gBAAgB,CAAC,qBAAqB,CAAC;aACvD;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACpF,MAAM,MAAM,GAAG,YAAY,EAAE,CAAC;QAgB9B,MAAM,cAAc,GAAG,WAAE,CAAC,KAAK,CAC7B,MAMC,EACD,aAAa,CACd,CAAC;QAEF,cAAc,CAAC,qBAAqB,CAAC;YACnC,OAAO,EAAE,EAAE;YACX,OAAO,EACL,kEAAkE;YACpE,SAAS,EAAE,IAAI;YACf,cAAc,EAAE,EAAE;YAClB,UAAU,EAAE,YAAY;YACxB,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE;SACnE,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC;YAClC,OAAO,EAAE,EAAE,WAAW,EAAE,kBAAkB,EAAE,QAAQ,EAAE,EAAE,EAAW;YACnE,MAAM,EAAE,WAAE,CAAC,EAAE,EAAE;SAChB,CAAC,CAAC;QAEH,6EAA6E;QAC7E,IAAA,eAAM,EAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChD,IAAA,eAAM,EAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it, vi } from \"vitest\";\nimport { OpenAICUAClient } from \"../../lib/v3/agent/OpenAICUAClient.js\";\n\nfunction createClient() {\n return new OpenAICUAClient(\n \"openai\",\n \"computer-use-preview-2025-03-11\",\n undefined,\n { apiKey: \"test-key\" },\n );\n}\n\ndescribe(\"OpenAICUAClient\", () => {\n it(\"exposes captchaSolvedProceed tool after a captcha context note\", () => {\n const client = createClient();\n\n // Before captcha note — tool should not be active\n expect(\n (client as unknown as { captchaSolvedToolActive: boolean })\n .captchaSolvedToolActive,\n ).toBe(false);\n\n // Simulate a captcha context note being added (as the CUA handler does)\n client.addContextNote(\n \"A captcha was automatically detected and solved — no further interaction needed.\",\n );\n\n expect(\n (client as unknown as { captchaSolvedToolActive: boolean })\n .captchaSolvedToolActive,\n ).toBe(true);\n });\n\n it(\"does NOT activate captcha tool for non-captcha context notes\", () => {\n const client = createClient();\n\n client.addContextNote(\"The page has finished loading.\");\n\n expect(\n (client as unknown as { captchaSolvedToolActive: boolean })\n .captchaSolvedToolActive,\n ).toBe(false);\n });\n\n it(\"deactivates captcha tool after takeAction handles the function call\", async () => {\n const client = createClient();\n client.addContextNote(\"A captcha was solved.\");\n\n expect(\n (client as unknown as { captchaSolvedToolActive: boolean })\n .captchaSolvedToolActive,\n ).toBe(true);\n\n // Simulate the model calling the captchaSolvedProceed tool\n const result = await (\n client as unknown as {\n takeAction: (\n output: unknown[],\n logger: (msg: unknown) => void,\n ) => Promise<unknown[]>;\n }\n ).takeAction(\n [\n {\n type: \"function_call\",\n name: \"captchaSolvedProceed\",\n call_id: \"call-1\",\n arguments: \"{}\",\n },\n ],\n vi.fn(),\n );\n\n // Tool should be deactivated\n expect(\n (client as unknown as { captchaSolvedToolActive: boolean })\n .captchaSolvedToolActive,\n ).toBe(false);\n\n // Result should contain a function_call_output confirming proceed\n expect(result).toEqual([\n {\n type: \"function_call_output\",\n call_id: \"call-1\",\n output: expect.stringContaining(\"Continue completing\"),\n },\n ]);\n });\n\n it(\"does NOT auto-continue follow-up questions without a captcha context\", async () => {\n const client = createClient();\n // No captcha context note — no tool should be exposed\n\n type ExecuteStepResult = {\n actions: Array<{ type: string }>;\n message: string;\n completed: boolean;\n nextInputItems: unknown[];\n responseId: string;\n usage: {\n input_tokens: number;\n output_tokens: number;\n inference_time_ms: number;\n };\n };\n\n const executeStepSpy = vi.spyOn(\n client as unknown as {\n executeStep: (\n inputItems: unknown[],\n previousResponseId: string | undefined,\n logger: (message: { message: string }) => void,\n ) => Promise<ExecuteStepResult>;\n },\n \"executeStep\",\n );\n\n executeStepSpy.mockResolvedValueOnce({\n actions: [],\n message:\n \"I've located the Submit button. Should I go ahead and submit it?\",\n completed: true,\n nextInputItems: [],\n responseId: \"response-1\",\n usage: { input_tokens: 1, output_tokens: 1, inference_time_ms: 1 },\n });\n\n const result = await client.execute({\n options: { instruction: \"Submit the form.\", maxSteps: 10 } as never,\n logger: vi.fn(),\n });\n\n // Should NOT have continued — the model's follow-up is treated as completion\n expect(executeStepSpy).toHaveBeenCalledTimes(1);\n expect(result.completed).toBe(true);\n });\n});\n"]}

package/dist/cjs/tests/unit/page-extra-http-headers.test.d.ts ADDED Viewed

	@@ -0,0 +1 @@
1	+ export {};

package/dist/cjs/tests/unit/page-extra-http-headers.test.js ADDED Viewed

@@ -0,0 +1,92 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const page_js_1 = require("../../lib/v3/understudy/page.js");
+const mockCDPSession_js_1 = require("./helpers/mockCDPSession.js");
+const sdkErrors_js_1 = require("../../lib/v3/types/public/sdkErrors.js");
+const makePage = (sessions) => {
+    const mainSession = sessions[0] ?? new mockCDPSession_js_1.MockCDPSession({}, "main");
+    const stub = {
+        mainSession,
+        sessions: new Map(sessions.map((s) => [s.id, s])),
+        extraHTTPHeaders: {},
+        // Bind the private helper from Page.prototype so setExtraHTTPHeaders can call it
+        applyExtraHTTPHeadersToSession: page_js_1.Page.prototype
+            .applyExtraHTTPHeadersToSession,
+    };
+    return stub;
+};
+(0, vitest_1.describe)("Page.setExtraHTTPHeaders", () => {
+    const setExtraHTTPHeaders = page_js_1.Page.prototype.setExtraHTTPHeaders;
+    (0, vitest_1.it)("sends headers to all sessions owned by the page", async () => {
+        const sessionA = new mockCDPSession_js_1.MockCDPSession({}, "session-a");
+        const sessionB = new mockCDPSession_js_1.MockCDPSession({}, "session-b");
+        const page = makePage([sessionA, sessionB]);
+        await setExtraHTTPHeaders.call(page, {
+            "x-stagehand-test": "hello",
+        });
+        for (const session of [sessionA, sessionB]) {
+            (0, vitest_1.expect)(session.callsFor("Network.enable").length).toBe(1);
+            (0, vitest_1.expect)(session.callsFor("Network.setExtraHTTPHeaders")[0]?.params).toEqual({
+                headers: { "x-stagehand-test": "hello" },
+            });
+        }
+    });
+    (0, vitest_1.it)("applies headers to mainSession even when sessions map is empty", async () => {
+        const page = makePage([]);
+        await setExtraHTTPHeaders.call(page, { "x-test": "value" });
+        // mainSession should still receive headers even though it's not in the sessions map
+        (0, vitest_1.expect)(page.mainSession.callsFor("Network.enable").length).toBe(1);
+        (0, vitest_1.expect)(page.mainSession.callsFor("Network.setExtraHTTPHeaders")[0]?.params).toEqual({
+            headers: { "x-test": "value" },
+        });
+    });
+    (0, vitest_1.it)("throws StagehandSetExtraHTTPHeadersError with session failure details", async () => {
+        const sessionA = new mockCDPSession_js_1.MockCDPSession({
+            "Network.setExtraHTTPHeaders": () => {
+                throw new Error("connection closed");
+            },
+        }, "session-a");
+        const sessionB = new mockCDPSession_js_1.MockCDPSession({}, "session-b");
+        const page = makePage([sessionA, sessionB]);
+        let caughtError;
+        try {
+            await setExtraHTTPHeaders.call(page, {
+                "x-stagehand-test": "yes",
+            });
+        }
+        catch (error) {
+            caughtError = error;
+        }
+        (0, vitest_1.expect)(caughtError).toBeInstanceOf(sdkErrors_js_1.StagehandSetExtraHTTPHeadersError);
+        (0, vitest_1.expect)(caughtError?.failures).toHaveLength(1);
+        (0, vitest_1.expect)(caughtError?.failures[0]).toContain("session=session-a");
+        (0, vitest_1.expect)(caughtError?.failures[0]).toContain("connection closed");
+        // sessionB should still have been called successfully
+        (0, vitest_1.expect)(sessionB.callsFor("Network.setExtraHTTPHeaders").length).toBe(1);
+    });
+    (0, vitest_1.it)("applies headers to sessions adopted after the call", async () => {
+        const sessionA = new mockCDPSession_js_1.MockCDPSession({}, "session-a");
+        const page = makePage([sessionA]);
+        await setExtraHTTPHeaders.call(page, { "x-before": "yes" });
+        // A new OOPIF session is adopted after headers were set
+        const sessionB = new mockCDPSession_js_1.MockCDPSession({}, "session-b");
+        page.sessions.set(sessionB.id, sessionB);
+        // Simulate what adoptOopifSession does: replay headers onto the new session
+        await page.applyExtraHTTPHeadersToSession.call(page, sessionB, page.extraHTTPHeaders);
+        // The late-arriving session should have received the headers
+        (0, vitest_1.expect)(sessionB.callsFor("Network.enable").length).toBe(1);
+        (0, vitest_1.expect)(sessionB.callsFor("Network.setExtraHTTPHeaders")[0]?.params).toEqual({
+            headers: { "x-before": "yes" },
+        });
+    });
+    (0, vitest_1.it)("does not mutate the original headers object", async () => {
+        const session = new mockCDPSession_js_1.MockCDPSession({}, "session-a");
+        const page = makePage([session]);
+        const original = { "x-custom": "value" };
+        const frozen = { ...original };
+        await setExtraHTTPHeaders.call(page, original);
+        (0, vitest_1.expect)(original).toEqual(frozen);
+    });
+});
+//# sourceMappingURL=page-extra-http-headers.test.js.map

package/dist/cjs/tests/unit/page-extra-http-headers.test.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"page-extra-http-headers.test.js","sourceRoot":"","sources":["../../../../tests/unit/page-extra-http-headers.test.ts"],"names":[],"mappings":";;AAAA,mCAA8C;AAC9C,6DAAuD;AACvD,mEAA6D;AAC7D,yEAA2F;AAY3F,MAAM,QAAQ,GAAG,CAAC,QAA0B,EAAY,EAAE;IACxD,MAAM,WAAW,GAAG,QAAQ,CAAC,CAAC,CAAC,IAAI,IAAI,kCAAc,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;IAClE,MAAM,IAAI,GAAa;QACrB,WAAW;QACX,QAAQ,EAAE,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QACjD,gBAAgB,EAAE,EAAE;QACpB,iFAAiF;QACjF,8BAA8B,EAAG,cAAI,CAAC,SAAiC;aACpE,8BAA8B;KAClC,CAAC;IACF,OAAO,IAAI,CAAC;AACd,CAAC,CAAC;AAEF,IAAA,iBAAQ,EAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,MAAM,mBAAmB,GAAG,cAAI,CAAC,SAAS,CAAC,mBAGzB,CAAC;IAEnB,IAAA,WAAE,EAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,QAAQ,GAAG,IAAI,kCAAc,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QACrD,MAAM,QAAQ,GAAG,IAAI,kCAAc,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QACrD,MAAM,IAAI,GAAG,QAAQ,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;QAE5C,MAAM,mBAAmB,CAAC,IAAI,CAAC,IAAI,EAAE;YACnC,kBAAkB,EAAE,OAAO;SAC5B,CAAC,CAAC;QAEH,KAAK,MAAM,OAAO,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,EAAE,CAAC;YAC3C,IAAA,eAAM,EAAC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC1D,IAAA,eAAM,EACJ,OAAO,CAAC,QAAQ,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,CAC3D,CAAC,OAAO,CAAC;gBACR,OAAO,EAAE,EAAE,kBAAkB,EAAE,OAAO,EAAE;aACzC,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;QAC9E,MAAM,IAAI,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC;QAE1B,MAAM,mBAAmB,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;QAE5D,oFAAoF;QACpF,IAAA,eAAM,EAAC,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnE,IAAA,eAAM,EACJ,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,CACpE,CAAC,OAAO,CAAC;YACR,OAAO,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE;SAC/B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,QAAQ,GAAG,IAAI,kCAAc,CACjC;YACE,6BAA6B,EAAE,GAAG,EAAE;gBAClC,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;YACvC,CAAC;SACF,EACD,WAAW,CACZ,CAAC;QACF,MAAM,QAAQ,GAAG,IAAI,kCAAc,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QACrD,MAAM,IAAI,GAAG,QAAQ,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;QAE5C,IAAI,WAA0D,CAAC;QAC/D,IAAI,CAAC;YACH,MAAM,mBAAmB,CAAC,IAAI,CAAC,IAAI,EAAE;gBACnC,kBAAkB,EAAE,KAAK;aAC1B,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,WAAW,GAAG,KAA0C,CAAC;QAC3D,CAAC;QAED,IAAA,eAAM,EAAC,WAAW,CAAC,CAAC,cAAc,CAAC,gDAAiC,CAAC,CAAC;QACtE,IAAA,eAAM,EAAC,WAAW,EAAE,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9C,IAAA,eAAM,EAAC,WAAW,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAChE,IAAA,eAAM,EAAC,WAAW,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAEhE,sDAAsD;QACtD,IAAA,eAAM,EAAC,QAAQ,CAAC,QAAQ,CAAC,6BAA6B,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,QAAQ,GAAG,IAAI,kCAAc,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QACrD,MAAM,IAAI,GAAG,QAAQ,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;QAElC,MAAM,mBAAmB,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;QAE5D,wDAAwD;QACxD,MAAM,QAAQ,GAAG,IAAI,kCAAc,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QACrD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;QAEzC,4EAA4E;QAC5E,MAAM,IAAI,CAAC,8BAA8B,CAAC,IAAI,CAC5C,IAAI,EACJ,QAAQ,EACR,IAAI,CAAC,gBAAgB,CACtB,CAAC;QAEF,6DAA6D;QAC7D,IAAA,eAAM,EAAC,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC3D,IAAA,eAAM,EAAC,QAAQ,CAAC,QAAQ,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,OAAO,CACzE;YACE,OAAO,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE;SAC/B,CACF,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,OAAO,GAAG,IAAI,kCAAc,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QACpD,MAAM,IAAI,GAAG,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;QAEjC,MAAM,QAAQ,GAAG,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;QACzC,MAAM,MAAM,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;QAE/B,MAAM,mBAAmB,CAAC,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QAE/C,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it } from \"vitest\";\nimport { Page } from \"../../lib/v3/understudy/page.js\";\nimport { MockCDPSession } from \"./helpers/mockCDPSession.js\";\nimport { StagehandSetExtraHTTPHeadersError } from \"../../lib/v3/types/public/sdkErrors.js\";\n\ntype PageStub = {\n mainSession: MockCDPSession;\n sessions: Map<string, MockCDPSession>;\n extraHTTPHeaders: Record<string, string>;\n applyExtraHTTPHeadersToSession: (\n session: MockCDPSession,\n headers: Record<string, string>,\n ) => Promise<void>;\n};\n\nconst makePage = (sessions: MockCDPSession[]): PageStub => {\n const mainSession = sessions[0] ?? new MockCDPSession({}, \"main\");\n const stub: PageStub = {\n mainSession,\n sessions: new Map(sessions.map((s) => [s.id, s])),\n extraHTTPHeaders: {},\n // Bind the private helper from Page.prototype so setExtraHTTPHeaders can call it\n applyExtraHTTPHeadersToSession: (Page.prototype as unknown as PageStub)\n .applyExtraHTTPHeadersToSession,\n };\n return stub;\n};\n\ndescribe(\"Page.setExtraHTTPHeaders\", () => {\n const setExtraHTTPHeaders = Page.prototype.setExtraHTTPHeaders as (\n this: PageStub,\n headers: Record<string, string>,\n ) => Promise<void>;\n\n it(\"sends headers to all sessions owned by the page\", async () => {\n const sessionA = new MockCDPSession({}, \"session-a\");\n const sessionB = new MockCDPSession({}, \"session-b\");\n const page = makePage([sessionA, sessionB]);\n\n await setExtraHTTPHeaders.call(page, {\n \"x-stagehand-test\": \"hello\",\n });\n\n for (const session of [sessionA, sessionB]) {\n expect(session.callsFor(\"Network.enable\").length).toBe(1);\n expect(\n session.callsFor(\"Network.setExtraHTTPHeaders\")[0]?.params,\n ).toEqual({\n headers: { \"x-stagehand-test\": \"hello\" },\n });\n }\n });\n\n it(\"applies headers to mainSession even when sessions map is empty\", async () => {\n const page = makePage([]);\n\n await setExtraHTTPHeaders.call(page, { \"x-test\": \"value\" });\n\n // mainSession should still receive headers even though it's not in the sessions map\n expect(page.mainSession.callsFor(\"Network.enable\").length).toBe(1);\n expect(\n page.mainSession.callsFor(\"Network.setExtraHTTPHeaders\")[0]?.params,\n ).toEqual({\n headers: { \"x-test\": \"value\" },\n });\n });\n\n it(\"throws StagehandSetExtraHTTPHeadersError with session failure details\", async () => {\n const sessionA = new MockCDPSession(\n {\n \"Network.setExtraHTTPHeaders\": () => {\n throw new Error(\"connection closed\");\n },\n },\n \"session-a\",\n );\n const sessionB = new MockCDPSession({}, \"session-b\");\n const page = makePage([sessionA, sessionB]);\n\n let caughtError: StagehandSetExtraHTTPHeadersError | undefined;\n try {\n await setExtraHTTPHeaders.call(page, {\n \"x-stagehand-test\": \"yes\",\n });\n } catch (error) {\n caughtError = error as StagehandSetExtraHTTPHeadersError;\n }\n\n expect(caughtError).toBeInstanceOf(StagehandSetExtraHTTPHeadersError);\n expect(caughtError?.failures).toHaveLength(1);\n expect(caughtError?.failures[0]).toContain(\"session=session-a\");\n expect(caughtError?.failures[0]).toContain(\"connection closed\");\n\n // sessionB should still have been called successfully\n expect(sessionB.callsFor(\"Network.setExtraHTTPHeaders\").length).toBe(1);\n });\n\n it(\"applies headers to sessions adopted after the call\", async () => {\n const sessionA = new MockCDPSession({}, \"session-a\");\n const page = makePage([sessionA]);\n\n await setExtraHTTPHeaders.call(page, { \"x-before\": \"yes\" });\n\n // A new OOPIF session is adopted after headers were set\n const sessionB = new MockCDPSession({}, \"session-b\");\n page.sessions.set(sessionB.id, sessionB);\n\n // Simulate what adoptOopifSession does: replay headers onto the new session\n await page.applyExtraHTTPHeadersToSession.call(\n page,\n sessionB,\n page.extraHTTPHeaders,\n );\n\n // The late-arriving session should have received the headers\n expect(sessionB.callsFor(\"Network.enable\").length).toBe(1);\n expect(sessionB.callsFor(\"Network.setExtraHTTPHeaders\")[0]?.params).toEqual(\n {\n headers: { \"x-before\": \"yes\" },\n },\n );\n });\n\n it(\"does not mutate the original headers object\", async () => {\n const session = new MockCDPSession({}, \"session-a\");\n const page = makePage([session]);\n\n const original = { \"x-custom\": \"value\" };\n const frozen = { ...original };\n\n await setExtraHTTPHeaders.call(page, original);\n\n expect(original).toEqual(frozen);\n });\n});\n"]}