@google/gemini-cli-core 0.0.3-preview.4 → 0.0.3

package/dist/src/core/baseLlmClient.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"baseLlmClient.test.js","sourceRoot":"","sources":["../../../src/core/baseLlmClient.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,QAAQ,EACR,EAAE,EACF,MAAM,EACN,EAAE,EACF,UAAU,EACV,SAAS,GAEV,MAAM,QAAQ,CAAC;AAGhB,OAAO,EAAE,aAAa,EAA4B,MAAM,oBAAoB,CAAC;AAG7E,OAAO,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AACjD,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,EAAE,0BAA0B,EAAE,MAAM,uBAAuB,CAAC;AACnE,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErD,EAAE,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;AACtC,EAAE,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;AACnC,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IACrD,MAAM,MAAM,GAAG,MAAM,cAAc,EAAuC,CAAC;IAC3E,OAAO;QACL,GAAG,MAAM;QACT,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;KAC5E,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,gBAAgB,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC;CAClD,CAAC,CAAC,CAAC;AAEJ,MAAM,mBAAmB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AACpC,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AAEjC,MAAM,oBAAoB,GAAG;IAC3B,eAAe,EAAE,mBAAmB;IACpC,YAAY,EAAE,gBAAgB;CACQ,CAAC;AAEzC,MAAM,UAAU,GAAG;IACjB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,iBAAiB,CAAC;IACxD,yBAAyB,EAAE,EAAE;SAC1B,EAAE,EAAE;SACJ,eAAe,CAAC,EAAE,QAAQ,EAAE,QAAQ,CAAC,UAAU,EAAE,CAAC;IACrD,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,sBAAsB,CAAC;CACtC,CAAC;AAE/B,kDAAkD;AAClD,MAAM,kBAAkB,GAAG,CAAC,IAAY,EAA2B,EAAE,CACnE,CAAC;IACC,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;CAC1E,CAA4B,CAAC;AAEhC,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,IAAI,MAAqB,CAAC;IAC1B,IAAI,eAAgC,CAAC;IACrC,IAAI,cAAmC,CAAC;IAExC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,4FAA4F;QAC5F,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,EAAE,CAClD,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAC3C,CAAC;QACF,MAAM,GAAG,IAAI,aAAa,CAAC,oBAAoB,EAAE,UAAU,CAAC,CAAC;QAC7D,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QACxC,cAAc,GAAG;YACf,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,EAAE,CAAC;YACnE,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE;YACrE,KAAK,EAAE,YAAY;YACnB,WAAW,EAAE,eAAe,CAAC,MAAM;YACnC,QAAQ,EAAE,gBAAgB;SAC3B,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,eAAe,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAChD,EAAE,CAAC,4FAA4F,EAAE,KAAK,IAAI,EAAE;YAC1G,MAAM,YAAY,GAAG,kBAAkB,CAAC,mBAAmB,CAAC,CAAC;YAC7D,mBAAmB,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAEpD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC;YAEzD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;YAE1C,yCAAyC;YACzC,MAAM,CAAC,gBAAgB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAElD,6DAA6D;YAC7D,MAAM,CAAC,mBAAmB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACrD,MAAM,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAC9C;gBACE,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,cAAc,CAAC,QAAQ;gBACjC,MAAM,EAAE;oBACN,WAAW,EAAE,cAAc,CAAC,WAAW;oBACvC,WAAW,EAAE,CAAC;oBACd,IAAI,EAAE,CAAC;oBACP,kBAAkB,EAAE,cAAc,CAAC,MAAM;oBACzC,gBAAgB,EAAE,kBAAkB;oBACpC,gFAAgF;iBACjF;aACF,EACD,gBAAgB,CACjB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,YAAY,GAAG,kBAAkB,CAAC,kBAAkB,CAAC,CAAC;YAC5D,mBAAmB,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAEpD,MAAM,OAAO,GAAwB;gBACnC,GAAG,cAAc;gBACjB,MAAM,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE;aACvC,CAAC;YAEF,MAAM,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YAEnC,MAAM,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAC9C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC9B,WAAW,EAAE,GAAG;oBAChB,IAAI,EAAE,CAAC,EAAE,0CAA0C;oBACnD,IAAI,EAAE,EAAE;iBACT,CAAC;aACH,CAAC,EACF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,YAAY,GAAG,kBAAkB,CAAC,oBAAoB,CAAC,CAAC;YAC9D,mBAAmB,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YACpD,MAAM,iBAAiB,GAAG,8BAA8B,CAAC;YAEzD,MAAM,OAAO,GAAwB;gBACnC,GAAG,cAAc;gBACjB,iBAAiB;aAClB,CAAC;YAEF,MAAM,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YAEnC,MAAM,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAC9C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC9B,iBAAiB;iBAClB,CAAC;aACH,CAAC,EACF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,YAAY,GAAG,kBAAkB,CAAC,qBAAqB,CAAC,CAAC;YAC/D,mBAAmB,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YACpD,MAAM,cAAc,GAAG,eAAe,CAAC;YAEvC,MAAM,OAAO,GAAwB;gBACnC,GAAG,cAAc;gBACjB,QAAQ,EAAE,cAAc;aACzB,CAAC;YAEF,MAAM,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YAEnC,MAAM,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAC9C,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,cAAc,CACf,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAChD,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YACjF,MAAM,iBAAiB,GAAG,mCAAmC,CAAC;YAC9D,mBAAmB,CAAC,iBAAiB,CACnC,kBAAkB,CAAC,iBAAiB,CAAC,CACtC,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC;YAEzD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC5C,MAAM,CAAC,wBAAwB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC1D,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CACnD,UAAU,EACV,MAAM,CAAC,GAAG,CAAC,0BAA0B,CAAC,CACvC,CAAC;YACF,uCAAuC;YACvC,MAAM,KAAK,GAAG,EAAE,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,IAAI;iBACnD,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAA+B,CAAC;YAC7C,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;YAC5F,MAAM,sBAAsB,GAAG,+BAA+B,CAAC;YAC/D,mBAAmB,CAAC,iBAAiB,CACnC,kBAAkB,CAAC,sBAAsB,CAAC,CAC3C,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC;YAEzD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC5C,MAAM,CAAC,wBAAwB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,mBAAmB,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC,CAAC;YAE9D,uFAAuF;YACvF,MAAM,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC/D,mFAAmF,CACpF,CAAC;YAEF,iCAAiC;YACjC,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC7C,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CACtC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EACjB,wDAAwD,EACxD,cAAc,CAAC,QAAQ,EACvB,6BAA6B,CAC9B,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,WAAW,GAAG,kBAAkB,CAAC,CAAC,wBAAwB;YAChE,mBAAmB,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC,CAAC;YAEvE,MAAM,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC/D,yEAAyE,CAC1E,CAAC;YAEF,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC7C,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CACtC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EACjB,kDAAkD,EAClD,MAAM,CAAC,gBAAgB,CAAC,EAAE,yBAAyB,EAAE,WAAW,EAAE,CAAC,EACnE,oBAAoB,CACrB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,QAAQ,GAAG,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YACxD,iCAAiC;YACjC,mBAAmB,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAEhD,MAAM,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC/D,4DAA4D,CAC7D,CAAC;YAEF,iCAAiC;YACjC,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC7C,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CACtC,QAAQ,EACR,wCAAwC,EACxC,cAAc,CAAC,QAAQ,EACvB,kBAAkB,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,UAAU,GAAG,IAAI,YAAY,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;YAE7D,kDAAkD;YAClD,mBAAmB,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBAC1C,eAAe,CAAC,KAAK,EAAE,CAAC,CAAC,uDAAuD;gBAChF,MAAM,UAAU,CAAC;YACnB,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG;gBACd,GAAG,cAAc;gBACjB,WAAW,EAAE,eAAe,CAAC,MAAM;aACpC,CAAC;YAEF,MAAM,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAEvE,yEAAyE;YACzE,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,MAAM,KAAK,GAAG,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;QAC/C,MAAM,kBAAkB,GAAG,sBAAsB,CAAC;QAElD,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACtF,MAAM,cAAc,GAAG;gBACrB,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;gBACf,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;aAChB,CAAC;YACF,gBAAgB,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE;oBACV,EAAE,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;oBAC7B,EAAE,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;iBAC9B;aACF,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAErD,MAAM,CAAC,gBAAgB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAClD,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC;gBAC5C,KAAK,EAAE,kBAAkB;gBACzB,QAAQ,EAAE,KAAK;aAChB,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAEvC,MAAM,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC3D,sCAAsC,CACvC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACnF,gBAAgB,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,EAAE;aACf,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC3D,sCAAsC,CACvC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACtF,gBAAgB,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,yBAAyB;aAC/D,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC3D,oEAAoE,CACrE,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,gBAAgB,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,EAAE,oBAAoB;aACjF,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC3D,4EAA4E,CAC7E,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YAChF,gBAAgB,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,mBAAmB;aACzE,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC3D,0EAA0E,CAC3E,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,gBAAgB,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC;YAE7D,MAAM,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC3D,aAAa,CACd,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/core/client.d.ts

@@ -7,24 +7,26 @@ import type { GenerateContentConfig, PartListUnion, Content, GenerateContentResp
 import type { ServerGeminiStreamEvent, ChatCompressionInfo } from './turn.js';
 import { Turn } from './turn.js';
 import type { Config } from '../config/config.js';
-import type { UserTierId } from '../code_assist/types.js';
 import { GeminiChat } from './geminiChat.js';
-import type { ContentGenerator, ContentGeneratorConfig } from './contentGenerator.js';
+import type { ChatRecordingService } from '../services/chatRecordingService.js';
+import { LoopDetectionService } from '../services/loopDetectionService.js';
+export declare function isThinkingSupported(model: string): boolean;
+export declare function isThinkingDefault(model: string): boolean;
 /**
- * Returns the index of the content after the fraction of the total characters in the history.
+ * Returns the index of the oldest item to keep when compressing. May return
+ * contents.length which indicates that everything should be compressed.
  *
  * Exported for testing purposes.
  */
-export declare function findIndexAfterFraction(history: Content[], fraction: number): number;
+export declare function findCompressSplitPoint(contents: Content[], fraction: number): number;
 export declare class GeminiClient {
     private readonly config;
     private chat?;
-    private contentGenerator?;
-    private readonly embeddingModel;
     private readonly generateContentConfig;
     private sessionTurnCount;
     private readonly loopDetector;
     private lastPromptId;
+    private currentSequenceModel;
     private lastSentIdeContext;
     private forceFullIdeContext;
     /**
@@ -33,31 +35,24 @@ export declare class GeminiClient {
      */
     private hasFailedCompressionAttempt;
     constructor(config: Config);
-    initialize(contentGeneratorConfig: ContentGeneratorConfig): Promise<void>;
-    getContentGenerator(): ContentGenerator;
-    getUserTier(): UserTierId | undefined;
+    initialize(): Promise<void>;
+    private getContentGeneratorOrFail;
     addHistory(content: Content): Promise<void>;
     getChat(): GeminiChat;
     isInitialized(): boolean;
     getHistory(): Content[];
-    setHistory(history: Content[], { stripThoughts }?: {
-        stripThoughts?: boolean;
-    }): void;
+    stripThoughtsFromHistory(): void;
+    setHistory(history: Content[]): void;
     setTools(): Promise<void>;
     resetChat(): Promise<void>;
+    getChatRecordingService(): ChatRecordingService | undefined;
+    getLoopDetectionService(): LoopDetectionService;
     addDirectoryContext(): Promise<void>;
     startChat(extraHistory?: Content[]): Promise<GeminiChat>;
     private getIdeContextParts;
-    sendMessageStream(request: PartListUnion, signal: AbortSignal, prompt_id: string, turns?: number, originalModel?: string): AsyncGenerator<ServerGeminiStreamEvent, Turn>;
-    generateJson(contents: Content[], schema: Record<string, unknown>, abortSignal: AbortSignal, model?: string, config?: GenerateContentConfig): Promise<Record<string, unknown>>;
-    generateContent(contents: Content[], generationConfig: GenerateContentConfig, abortSignal: AbortSignal, model?: string): Promise<GenerateContentResponse>;
-    generateEmbedding(texts: string[]): Promise<number[][]>;
+    sendMessageStream(request: PartListUnion, signal: AbortSignal, prompt_id: string, turns?: number): AsyncGenerator<ServerGeminiStreamEvent, Turn>;
+    generateContent(contents: Content[], generationConfig: GenerateContentConfig, abortSignal: AbortSignal, model: string): Promise<GenerateContentResponse>;
     tryCompressChat(prompt_id: string, force?: boolean): Promise<ChatCompressionInfo>;
-    /**
-     * Handles falling back to Flash model when persistent 429 errors occur for OAuth users.
-     * Uses a fallback handler if provided by the config; otherwise, returns null.
-     */
-    private handleFlashFallback;
 }
 export declare const TEST_ONLY: {
     COMPRESSION_PRESERVE_THRESHOLD: number;

package/dist/src/core/client.js

@@ -13,40 +13,58 @@ import { reportError } from '../utils/errorReporting.js';
 import { GeminiChat } from './geminiChat.js';
 import { retryWithBackoff } from '../utils/retry.js';
 import { getErrorMessage } from '../utils/errors.js';
-import { isFunctionResponse } from '../utils/messageInspectors.js';
 import { tokenLimit } from './tokenLimits.js';
-import { AuthType, createContentGenerator } from './contentGenerator.js';
-import { ProxyAgent, setGlobalDispatcher } from 'undici';
-import { DEFAULT_GEMINI_FLASH_MODEL } from '../config/models.js';
+import { DEFAULT_GEMINI_FLASH_MODEL, DEFAULT_GEMINI_MODEL, DEFAULT_GEMINI_MODEL_AUTO, DEFAULT_THINKING_MODE, getEffectiveModel, } from '../config/models.js';
 import { LoopDetectionService } from '../services/loopDetectionService.js';
-import { ideContext } from '../ide/ideContext.js';
-import { logChatCompression, logNextSpeakerCheck, logMalformedJsonResponse, } from '../telemetry/loggers.js';
-import { makeChatCompressionEvent, MalformedJsonResponseEvent, NextSpeakerCheckEvent, } from '../telemetry/types.js';
-function isThinkingSupported(model) {
-    if (model.startsWith('gemini-2.5'))
-        return true;
-    return false;
+import { ideContextStore } from '../ide/ideContext.js';
+import { logChatCompression, logNextSpeakerCheck, } from '../telemetry/loggers.js';
+import { makeChatCompressionEvent, NextSpeakerCheckEvent, } from '../telemetry/types.js';
+import { handleFallback } from '../fallback/handler.js';
+import { uiTelemetryService } from '../telemetry/uiTelemetry.js';
+export function isThinkingSupported(model) {
+    return model.startsWith('gemini-2.5') || model === DEFAULT_GEMINI_MODEL_AUTO;
+}
+export function isThinkingDefault(model) {
+    if (model.startsWith('gemini-2.5-flash-lite')) {
+        return false;
+    }
+    return model.startsWith('gemini-2.5') || model === DEFAULT_GEMINI_MODEL_AUTO;
 }
 /**
- * Returns the index of the content after the fraction of the total characters in the history.
+ * Returns the index of the oldest item to keep when compressing. May return
+ * contents.length which indicates that everything should be compressed.
  *
  * Exported for testing purposes.
  */
-export function findIndexAfterFraction(history, fraction) {
+export function findCompressSplitPoint(contents, fraction) {
     if (fraction <= 0 || fraction >= 1) {
         throw new Error('Fraction must be between 0 and 1');
     }
-    const contentLengths = history.map((content) => JSON.stringify(content).length);
-    const totalCharacters = contentLengths.reduce((sum, length) => sum + length, 0);
-    const targetCharacters = totalCharacters * fraction;
-    let charactersSoFar = 0;
-    for (let i = 0; i < contentLengths.length; i++) {
-        charactersSoFar += contentLengths[i];
-        if (charactersSoFar >= targetCharacters) {
-            return i;
+    const charCounts = contents.map((content) => JSON.stringify(content).length);
+    const totalCharCount = charCounts.reduce((a, b) => a + b, 0);
+    const targetCharCount = totalCharCount * fraction;
+    let lastSplitPoint = 0; // 0 is always valid (compress nothing)
+    let cumulativeCharCount = 0;
+    for (let i = 0; i < contents.length; i++) {
+        cumulativeCharCount += charCounts[i];
+        const content = contents[i];
+        if (content.role === 'user' &&
+            !content.parts?.some((part) => !!part.functionResponse)) {
+            if (cumulativeCharCount >= targetCharCount) {
+                return i;
+            }
+            lastSplitPoint = i;
         }
     }
-    return contentLengths.length;
+    // We found no split points after targetCharCount.
+    // Check if it's safe to compress everything.
+    const lastContent = contents[contents.length - 1];
+    if (lastContent?.role === 'model' &&
+        !lastContent?.parts?.some((part) => part.functionCall)) {
+        return contents.length;
+    }
+    // Can't compress everything so just compress at last splitpoint.
+    return lastSplitPoint;
 }
 const MAX_TURNS = 100;
 /**
@@ -62,8 +80,6 @@ const COMPRESSION_PRESERVE_THRESHOLD = 0.3;
 export class GeminiClient {
     config;
     chat;
-    contentGenerator;
-    embeddingModel;
     generateContentConfig = {
         temperature: 0,
         topP: 1,
@@ -71,6 +87,7 @@ export class GeminiClient {
     sessionTurnCount = 0;
     loopDetector;
     lastPromptId;
+    currentSequenceModel = null;
     lastSentIdeContext;
     forceFullIdeContext = true;
     /**
@@ -80,25 +97,17 @@ export class GeminiClient {
     hasFailedCompressionAttempt = false;
     constructor(config) {
         this.config = config;
-        if (config.getProxy()) {
-            setGlobalDispatcher(new ProxyAgent(config.getProxy()));
-        }
-        this.embeddingModel = config.getEmbeddingModel();
         this.loopDetector = new LoopDetectionService(config);
         this.lastPromptId = this.config.getSessionId();
     }
-    async initialize(contentGeneratorConfig) {
-        this.contentGenerator = await createContentGenerator(contentGeneratorConfig, this.config, this.config.getSessionId());
+    async initialize() {
         this.chat = await this.startChat();
     }
-    getContentGenerator() {
-        if (!this.contentGenerator) {
+    getContentGeneratorOrFail() {
+        if (!this.config.getContentGenerator()) {
             throw new Error('Content generator not initialized');
         }
-        return this.contentGenerator;
-    }
-    getUserTier() {
-        return this.contentGenerator?.userTier;
+        return this.config.getContentGenerator();
     }
     async addHistory(content) {
         this.getChat().addHistory(content);
@@ -110,32 +119,16 @@ export class GeminiClient {
         return this.chat;
     }
     isInitialized() {
-        return this.chat !== undefined && this.contentGenerator !== undefined;
+        return this.chat !== undefined;
     }
     getHistory() {
         return this.getChat().getHistory();
     }
-    setHistory(history, { stripThoughts = false } = {}) {
-        const historyToSet = stripThoughts
-            ? history.map((content) => {
-                const newContent = { ...content };
-                if (newContent.parts) {
-                    newContent.parts = newContent.parts.map((part) => {
-                        if (part &&
-                            typeof part === 'object' &&
-                            'thoughtSignature' in part) {
-                            const newPart = { ...part };
-                            delete newPart
-                                .thoughtSignature;
-                            return newPart;
-                        }
-                        return part;
-                    });
-                }
-                return newContent;
-            })
-            : history;
-        this.getChat().setHistory(historyToSet);
+    stripThoughtsFromHistory() {
+        this.getChat().stripThoughtsFromHistory();
+    }
+    setHistory(history) {
+        this.getChat().setHistory(history);
         this.forceFullIdeContext = true;
     }
     async setTools() {
@@ -147,6 +140,12 @@ export class GeminiClient {
     async resetChat() {
         this.chat = await this.startChat();
     }
+    getChatRecordingService() {
+        return this.chat?.getChatRecordingService();
+    }
+    getLoopDetectionService() {
+        return this.loopDetector;
+    }
     async addDirectoryContext() {
         if (!this.chat) {
             return;
@@ -177,18 +176,17 @@ export class GeminiClient {
         try {
             const userMemory = this.config.getUserMemory();
             const systemInstruction = getCoreSystemPrompt(userMemory);
-            const generateContentConfigWithThinking = isThinkingSupported(this.config.getModel())
-                ? {
-                    ...this.generateContentConfig,
-                    thinkingConfig: {
-                        thinkingBudget: -1,
-                        includeThoughts: true,
-                    },
-                }
-                : this.generateContentConfig;
-            return new GeminiChat(this.config, this.getContentGenerator(), {
+            const model = this.config.getModel();
+            const config = { ...this.generateContentConfig };
+            if (isThinkingSupported(model)) {
+                config.thinkingConfig = {
+                    includeThoughts: true,
+                    thinkingBudget: DEFAULT_THINKING_MODE,
+                };
+            }
+            return new GeminiChat(this.config, {
                 systemInstruction,
-                ...generateContentConfigWithThinking,
+                ...config,
                 tools,
             }, history);
         }
@@ -198,7 +196,7 @@ export class GeminiClient {
         }
     }
     getIdeContextParts(forceFullContext) {
-        const currentIdeContext = ideContext.getIdeContext();
+        const currentIdeContext = ideContextStore.get();
         if (!currentIdeContext) {
             return { contextParts: [], newIdeContext: undefined };
         }
@@ -336,10 +334,11 @@ export class GeminiClient {
             };
         }
     }
-    async *sendMessageStream(request, signal, prompt_id, turns = MAX_TURNS, originalModel) {
+    async *sendMessageStream(request, signal, prompt_id, turns = MAX_TURNS) {
         if (this.lastPromptId !== prompt_id) {
             this.loopDetector.reset(prompt_id);
             this.lastPromptId = prompt_id;
+            this.currentSequenceModel = null;
         }
         this.sessionTurnCount++;
         if (this.config.getMaxSessionTurns() > 0 &&
@@ -352,9 +351,7 @@ export class GeminiClient {
         if (!boundedTurns) {
             return new Turn(this.getChat(), prompt_id);
         }
-        // Track the original model from the first call to detect model switching
-        const initialModel = originalModel || this.config.getModel();
-        const compressed = await this.tryCompressChat(prompt_id);
+        const compressed = await this.tryCompressChat(prompt_id, false);
         if (compressed.compressionStatus === CompressionStatus.COMPRESSED) {
             yield { type: GeminiEventType.ChatCompressed, value: compressed };
         }
@@ -380,15 +377,35 @@ export class GeminiClient {
             this.forceFullIdeContext = false;
         }
         const turn = new Turn(this.getChat(), prompt_id);
+        const controller = new AbortController();
+        const linkedSignal = AbortSignal.any([signal, controller.signal]);
         const loopDetected = await this.loopDetector.turnStarted(signal);
         if (loopDetected) {
             yield { type: GeminiEventType.LoopDetected };
             return turn;
         }
-        const resultStream = turn.run(request, signal);
+        const routingContext = {
+            history: this.getChat().getHistory(/*curated=*/ true),
+            request,
+            signal,
+        };
+        let modelToUse;
+        // Determine Model (Stickiness vs. Routing)
+        if (this.currentSequenceModel) {
+            modelToUse = this.currentSequenceModel;
+        }
+        else {
+            const router = await this.config.getModelRouterService();
+            const decision = await router.route(routingContext);
+            modelToUse = decision.model;
+            // Lock the model for the rest of the sequence
+            this.currentSequenceModel = modelToUse;
+        }
+        const resultStream = turn.run(modelToUse, request, linkedSignal);
         for await (const event of resultStream) {
             if (this.loopDetector.addAndCheck(event)) {
                 yield { type: GeminiEventType.LoopDetected };
+                controller.abort();
                 return turn;
             }
             yield event;
@@ -397,92 +414,26 @@ export class GeminiClient {
             }
         }
         if (!turn.pendingToolCalls.length && signal && !signal.aborted) {
-            // Check if model was switched during the call (likely due to quota error)
-            const currentModel = this.config.getModel();
-            if (currentModel !== initialModel) {
-                // Model was switched (likely due to quota error fallback)
-                // Don't continue with recursive call to prevent unwanted Flash execution
+            // Check if next speaker check is needed
+            if (this.config.getQuotaErrorOccurred()) {
                 return turn;
             }
             if (this.config.getSkipNextSpeakerCheck()) {
                 return turn;
             }
-            const nextSpeakerCheck = await checkNextSpeaker(this.getChat(), this, signal);
+            const nextSpeakerCheck = await checkNextSpeaker(this.getChat(), this.config.getBaseLlmClient(), signal, prompt_id);
             logNextSpeakerCheck(this.config, new NextSpeakerCheckEvent(prompt_id, turn.finishReason?.toString() || '', nextSpeakerCheck?.next_speaker || ''));
             if (nextSpeakerCheck?.next_speaker === 'model') {
                 const nextRequest = [{ text: 'Please continue.' }];
                 // This recursive call's events will be yielded out, but the final
                 // turn object will be from the top-level call.
-                yield* this.sendMessageStream(nextRequest, signal, prompt_id, boundedTurns - 1, initialModel);
+                yield* this.sendMessageStream(nextRequest, signal, prompt_id, boundedTurns - 1);
             }
         }
         return turn;
     }
-    async generateJson(contents, schema, abortSignal, model, config = {}) {
-        // Use current model from config instead of hardcoded Flash model
-        const modelToUse = model || this.config.getModel() || DEFAULT_GEMINI_FLASH_MODEL;
-        try {
-            const userMemory = this.config.getUserMemory();
-            const systemInstruction = getCoreSystemPrompt(userMemory);
-            const requestConfig = {
-                abortSignal,
-                ...this.generateContentConfig,
-                ...config,
-            };
-            const apiCall = () => this.getContentGenerator().generateContent({
-                model: modelToUse,
-                config: {
-                    ...requestConfig,
-                    systemInstruction,
-                    responseJsonSchema: schema,
-                    responseMimeType: 'application/json',
-                },
-                contents,
-            }, this.lastPromptId);
-            const result = await retryWithBackoff(apiCall, {
-                onPersistent429: async (authType, error) => await this.handleFlashFallback(authType, error),
-                authType: this.config.getContentGeneratorConfig()?.authType,
-            });
-            let text = getResponseText(result);
-            if (!text) {
-                const error = new Error('API returned an empty response for generateJson.');
-                await reportError(error, 'Error in generateJson: API returned an empty response.', contents, 'generateJson-empty-response');
-                throw error;
-            }
-            const prefix = '```json';
-            const suffix = '```';
-            if (text.startsWith(prefix) && text.endsWith(suffix)) {
-                logMalformedJsonResponse(this.config, new MalformedJsonResponseEvent(modelToUse));
-                text = text
-                    .substring(prefix.length, text.length - suffix.length)
-                    .trim();
-            }
-            try {
-                return JSON.parse(text);
-            }
-            catch (parseError) {
-                await reportError(parseError, 'Failed to parse JSON response from generateJson.', {
-                    responseTextFailedToParse: text,
-                    originalRequestContents: contents,
-                }, 'generateJson-parse');
-                throw new Error(`Failed to parse API response as JSON: ${getErrorMessage(parseError)}`);
-            }
-        }
-        catch (error) {
-            if (abortSignal.aborted) {
-                throw error;
-            }
-            // Avoid double reporting for the empty response case handled above
-            if (error instanceof Error &&
-                error.message === 'API returned an empty response for generateJson.') {
-                throw error;
-            }
-            await reportError(error, 'Error generating JSON content via API.', contents, 'generateJson-api');
-            throw new Error(`Failed to generate JSON content: ${getErrorMessage(error)}`);
-        }
-    }
     async generateContent(contents, generationConfig, abortSignal, model) {
-        const modelToUse = model ?? this.config.getModel();
+        let currentAttemptModel = model;
         const configToUse = {
             ...this.generateContentConfig,
             ...generationConfig,
@@ -495,13 +446,22 @@ export class GeminiClient {
                 ...configToUse,
                 systemInstruction,
             };
-            const apiCall = () => this.getContentGenerator().generateContent({
-                model: modelToUse,
-                config: requestConfig,
-                contents,
-            }, this.lastPromptId);
+            const apiCall = () => {
+                const modelToUse = this.config.isInFallbackMode()
+                    ? DEFAULT_GEMINI_FLASH_MODEL
+                    : model;
+                currentAttemptModel = modelToUse;
+                return this.getContentGeneratorOrFail().generateContent({
+                    model: modelToUse,
+                    config: requestConfig,
+                    contents,
+                }, this.lastPromptId);
+            };
+            const onPersistent429Callback = async (authType, error) => 
+            // Pass the captured model to the centralized handler.
+            await handleFallback(this.config, currentAttemptModel, authType, error);
             const result = await retryWithBackoff(apiCall, {
-                onPersistent429: async (authType, error) => await this.handleFlashFallback(authType, error),
+                onPersistent429: onPersistent429Callback,
                 authType: this.config.getContentGeneratorConfig()?.authType,
             });
             return result;
@@ -510,38 +470,23 @@ export class GeminiClient {
             if (abortSignal.aborted) {
                 throw error;
             }
-            await reportError(error, `Error generating content via API with model ${modelToUse}.`, {
+            await reportError(error, `Error generating content via API with model ${currentAttemptModel}.`, {
                 requestContents: contents,
                 requestConfig: configToUse,
             }, 'generateContent-api');
-            throw new Error(`Failed to generate content with model ${modelToUse}: ${getErrorMessage(error)}`);
+            throw new Error(`Failed to generate content with model ${currentAttemptModel}: ${getErrorMessage(error)}`);
         }
     }
-    async generateEmbedding(texts) {
-        if (!texts || texts.length === 0) {
-            return [];
-        }
-        const embedModelParams = {
-            model: this.embeddingModel,
-            contents: texts,
-        };
-        const embedContentResponse = await this.getContentGenerator().embedContent(embedModelParams);
-        if (!embedContentResponse.embeddings ||
-            embedContentResponse.embeddings.length === 0) {
-            throw new Error('No embeddings found in API response.');
-        }
-        if (embedContentResponse.embeddings.length !== texts.length) {
-            throw new Error(`API returned a mismatched number of embeddings. Expected ${texts.length}, got ${embedContentResponse.embeddings.length}.`);
-        }
-        return embedContentResponse.embeddings.map((embedding, index) => {
-            const values = embedding.values;
-            if (!values || values.length === 0) {
-                throw new Error(`API returned an empty embedding for input text at index ${index}: "${texts[index]}"`);
-            }
-            return values;
-        });
-    }
     async tryCompressChat(prompt_id, force = false) {
+        // If the model is 'auto', we will use a placeholder model to check.
+        // Compression occurs before we choose a model, so calling `count_tokens`
+        // before the model is chosen would result in an error.
+        const configModel = this.config.getModel();
+        let model = configModel === DEFAULT_GEMINI_MODEL_AUTO
+            ? DEFAULT_GEMINI_MODEL
+            : configModel;
+        // Check if the model needs to be a fallback
+        model = getEffectiveModel(this.config.isInFallbackMode(), model);
         const curatedHistory = this.getChat().getHistory(true);
         // Regardless of `force`, don't do anything if the history is empty.
         if (curatedHistory.length === 0 ||
@@ -552,8 +497,7 @@ export class GeminiClient {
                 compressionStatus: CompressionStatus.NOOP,
             };
         }
-        const model = this.config.getModel();
-        const { totalTokens: originalTokenCount } = await this.getContentGenerator().countTokens({
+        const { totalTokens: originalTokenCount } = await this.getContentGeneratorOrFail().countTokens({
             model,
             contents: curatedHistory,
         });
@@ -578,25 +522,29 @@ export class GeminiClient {
                 };
             }
         }
-        let compressBeforeIndex = findIndexAfterFraction(curatedHistory, 1 - COMPRESSION_PRESERVE_THRESHOLD);
-        // Find the first user message after the index. This is the start of the next turn.
-        while (compressBeforeIndex < curatedHistory.length &&
-            (curatedHistory[compressBeforeIndex]?.role === 'model' ||
-                isFunctionResponse(curatedHistory[compressBeforeIndex]))) {
-            compressBeforeIndex++;
-        }
-        const historyToCompress = curatedHistory.slice(0, compressBeforeIndex);
-        const historyToKeep = curatedHistory.slice(compressBeforeIndex);
-        this.getChat().setHistory(historyToCompress);
-        const { text: summary } = await this.getChat().sendMessage({
-            message: {
-                text: 'First, reason in your scratchpad. Then, generate the <state_snapshot>.',
-            },
+        const splitPoint = findCompressSplitPoint(curatedHistory, 1 - COMPRESSION_PRESERVE_THRESHOLD);
+        const historyToCompress = curatedHistory.slice(0, splitPoint);
+        const historyToKeep = curatedHistory.slice(splitPoint);
+        const summaryResponse = await this.config
+            .getContentGenerator()
+            .generateContent({
+            model,
+            contents: [
+                ...historyToCompress,
+                {
+                    role: 'user',
+                    parts: [
+                        {
+                            text: 'First, reason in your scratchpad. Then, generate the <state_snapshot>.',
+                        },
+                    ],
+                },
+            ],
             config: {
                 systemInstruction: { text: getCompressionPrompt() },
-                maxOutputTokens: originalTokenCount,
             },
         }, prompt_id);
+        const summary = getResponseText(summaryResponse) ?? '';
         const chat = await this.startChat([
             {
                 role: 'user',
@@ -609,7 +557,7 @@ export class GeminiClient {
             ...historyToKeep,
         ]);
         this.forceFullIdeContext = true;
-        const { totalTokens: newTokenCount } = await this.getContentGenerator().countTokens({
+        const { totalTokens: newTokenCount } = await this.getContentGeneratorOrFail().countTokens({
             // model might change after calling `sendMessage`, so we get the newest value from config
             model: this.config.getModel(),
             contents: chat.getHistory(),
@@ -623,6 +571,7 @@ export class GeminiClient {
                 compressionStatus: CompressionStatus.COMPRESSION_FAILED_TOKEN_COUNT_ERROR,
             };
         }
+        uiTelemetryService.setLastPromptTokenCount(newTokenCount);
         logChatCompression(this.config, makeChatCompressionEvent({
             tokens_before: originalTokenCount,
             tokens_after: newTokenCount,
@@ -645,42 +594,6 @@ export class GeminiClient {
             compressionStatus: CompressionStatus.COMPRESSED,
         };
     }
-    /**
-     * Handles falling back to Flash model when persistent 429 errors occur for OAuth users.
-     * Uses a fallback handler if provided by the config; otherwise, returns null.
-     */
-    async handleFlashFallback(authType, error) {
-        // Only handle fallback for OAuth users
-        if (authType !== AuthType.LOGIN_WITH_GOOGLE) {
-            return null;
-        }
-        const currentModel = this.config.getModel();
-        const fallbackModel = DEFAULT_GEMINI_FLASH_MODEL;
-        // Don't fallback if already using Flash model
-        if (currentModel === fallbackModel) {
-            return null;
-        }
-        // Check if config has a fallback handler (set by CLI package)
-        const fallbackHandler = this.config.flashFallbackHandler;
-        if (typeof fallbackHandler === 'function') {
-            try {
-                const accepted = await fallbackHandler(currentModel, fallbackModel, error);
-                if (accepted !== false && accepted !== null) {
-                    this.config.setModel(fallbackModel);
-                    this.config.setFallbackMode(true);
-                    return fallbackModel;
-                }
-                // Check if the model was switched manually in the handler
-                if (this.config.getModel() === fallbackModel) {
-                    return null; // Model was switched but don't continue with current prompt
-                }
-            }
-            catch (error) {
-                console.warn('Flash fallback handler failed:', error);
-            }
-        }
-        return null;
-    }
 }
 export const TEST_ONLY = {
     COMPRESSION_PRESERVE_THRESHOLD,