@dremio/js-sdk 0.59.0 → 0.60.0

package/dist/enterprise/ai/conversations/createConversationMachine.js.map

@@ -1 +1 @@
-{"version":3,"file":"createConversationMachine.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/conversations/createConversationMachine.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AAE1F,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAG3D,OAAO,EAAE,kBAAkB,EAAE,uBAAuB,EAAE,MAAM,iCAAiC,CAAC;AAC9F,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EAAE,oBAAoB,EAAE,MAAM,mCAAmC,CAAC;AACzE,OAAO,EAAE,2BAA2B,EAAE,MAAM,0CAA0C,CAAC;AACvF,OAAO,EAAE,iBAAiB,EAAE,MAAM,gCAAgC,CAAC;AACnE,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,mBAAmB,EAAE,MAAM,wCAAwC,CAAC;AAC7E,OAAO,EAAE,GAAG,EAAE,MAAM,MAAM,CAAC;AAC3B,OAAO,EAAE,cAAc,EAAkB,MAAM,4BAA4B,CAAC;AAkC5E,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,MAAqB,EAAE,EAAE,CACjE,KAAK,CAAC;IACJ,MAAM,EAAE;QACN,kBAAkB,EAAE,WAAW,CAC7B,CAAC,EAAE,KAAK,EAAsD,EAAE,EAAE,CAChE,kBAAkB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YAChD,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CAAC,CACL;QACD,oBAAoB,EAAE,WAAW,CAC/B,CAAC,EAAE,KAAK,EAAE,MAAM,EAA8D,EAAE,EAAE,CAChF,oBAAoB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YAC7E,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CAAC,CACL;QACD,eAAe,EAAE,WAAW,CAC1B,CAAC,EAAE,KAAK,EAAE,MAAM,EAA8D,EAAE,EAAE,CAChF,2BAA2B,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YACpF,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CAAC,CACL;QACD,aAAa,EAAE,WAAW,CACxB,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAA8D,EAAE,EAAE;YACtF,MAAM,CAAC,kBAAkB,EAAE,aAAa,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC5D,oBAAoB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,CAAC;gBAC9D,2BAA2B,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,CAAC;aACtE,CAAC,CAAC;YAEH,IAAI,kBAAkB,CAAC,KAAK,EAAE,EAAE,CAAC;gBAC/B,MAAM,kBAAkB,CAAC,KAAK,CAAC;YACjC,CAAC;YAED,IAAI,aAAa,CAAC,KAAK,EAAE,EAAE,CAAC;gBAC1B,MAAM,aAAa,CAAC,KAAK,CAAC;YAC5B,CAAC;YAED,OAAO;gBACL,YAAY,EAAE,kBAAkB,CAAC,KAAK;gBACtC,OAAO,EAAE,aAAa,CAAC,KAAK;aAC7B,CAAC;QACJ,CAAC,CACF;QACD,QAAQ,EAAE,WAAW,CACnB,CAAC,EAAE,KAAK,EAAmE,EAAE,EAAE,CAC7E,iBAAiB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAC9E,CAAC,MAAM,EAAE,EAAE;YACT,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CACF,CACJ;QACD,OAAO,EAAE,WAAW,CAAC,CAAC,EAAE,KAAK,EAAwD,EAAE,EAAE,CACvF,eAAe,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YACzE,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CAAC,CACH;QACD,SAAS,EAAE,cAAc,CAAC,CAAC,EAAE,KAAK,EAAwD,EAAE,EAAE,CAC5F,mBAAmB,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CACjC,eAAe,CAAC,MAAM,CAAC,CACrB,EAAE,cAAc,EAAE,KAAK,CAAC,cAAc,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,EAC5D,EAAE,MAAM,EAAE,CACX,CACF,CAAC,IAAI,CACJ,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,CAAC,CAAC,MAAM,CAAC,cAAc,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAmC,CAAC,CACnF,CACF,CACF;KACF;IACD,KAAK,EAAE;QACL,OAAO,EAAE,EAAgC;QACzC,OAAO,EAAE,EAAgC;QACzC,MAAM,EAAE,EAA+B;QACvC,KAAK,EAAE,EAA8B;KACtC;CACF,CAAC,CAAC,aAAa,CAAC;IACf,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QACvB,cAAc,EAAE,KAAK,CAAC,cAAc;QACpC,oBAAoB,EAAE,SAAS;QAC/B,YAAY,EAAE,SAAS;QACvB,cAAc,EAAE,SAAS;KAC1B,CAAC;IACF,EAAE,EAAE,mBAAmB;IACvB,OAAO,EAAE,eAAe;IACxB,MAAM,EAAE;QACN,qBAAqB,EAAE;YACrB,WAAW,EAAE,oCAAoC;YACjD,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;oBACvB,OAAO,EAAE,OAAO,CAAC,cAAe;iBACjC,CAAC;gBACF,MAAM,EAAE;oBACN;wBACE,OAAO,EAAE;4BACP,MAAM,CAAC;gCACL,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE;gCAC9C,qDAAqD;gCACrD,oBAAoB,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;oCAC5C,GAAG,CAAC,OAAO,CAAC,oBAAoB,IAAI,EAAE,CAAC;oCACvC;wCACE,EAAE,EAAE,KAAK,CAAC,MAAM,CAAC,YAAa;wCAC9B,QAAQ,EAAE,IAAI,GAAG,EAAE;wCACnB,oBAAoB,EAAE,OAAO,CAAC,cAAc;wCAC5C,SAAS,EAAE,IAAI,GAAG,EAAE;qCACU;iCACjC;gCACD,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY;gCACtD,cAAc,EAAE,SAAS;6BAC1B,CAAC;4BACF,IAAI,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;gCACnB,EAAE,EAAE,KAAK,CAAC,MAAM,CAAC,EAAE;gCACnB,IAAI,EAAE,qBAAqB;6BAC5B,CAAC,CAAC;yBACJ;wBACD,sFAAsF;wBACtF,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY;wBACjD,MAAM,EAAE,WAAW;qBACpB;oBACD;;uBAEG;oBACH;wBACE,OAAO,EAAE,MAAM,CAAC;4BACd,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE;4BAC9C,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY;4BACtD,cAAc,EAAE,SAAS;yBAC1B,CAAC;wBACF,MAAM,EAAE,oBAAoB;qBAC7B;iBACF;gBACD,OAAO,EAAE;oBACP,OAAO,EAAE,IAAI,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;wBAC5B,KAAK,EAAE,KAAK,CAAC,KAAkB;wBAC/B,IAAI,EAAE,UAAU;qBACjB,CAAC,CAAC;oBACH,MAAM,EAAE,eAAe;iBACxB;gBACD,GAAG,EAAE,oBAAoB;aAC1B;SACF;QACD,IAAI,EAAE;YACJ,WAAW,EAAE,uDAAuD;YACpE,EAAE,EAAE;gBACF,eAAe,EAAE;oBACf,MAAM,EAAE,oBAAoB;iBAC7B;gBACD,mBAAmB,EAAE;oBACnB,OAAO,EAAE,MAAM,CAAC;wBACd,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO;qBAC7C,CAAC;oBACF,MAAM,EAAE,oBAAoB;iBAC7B;aACF;SACF;QACD,uBAAuB,EAAE;YACvB,EAAE,EAAE;gBACF,eAAe,EAAE;oBACf,OAAO,EAAE,MAAM,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;oBACzC,MAAM,EAAE,oBAAoB;iBAC7B;aACF;SACF;QACD,kBAAkB,EAAE;YAClB,WAAW,EAAE,gDAAgD;YAC7D,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,cAAc,EAAE,OAAO,CAAC,cAAe,EAAE,CAAC;gBACrE,MAAM,EAAE;oBACN;wBACE,OAAO,EAAE,MAAM,CAAC;4BACd,oBAAoB,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAC3C,iBAAiB,CAAC,gBAAgB,CAChC,OAAO,CAAC,oBAAoB,EAC5B,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAC1B;4BACH,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,YAAY;yBACpE,CAAC;wBACF,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,YAAY;wBAC9D,MAAM,EAAE,WAAW;qBACpB;oBACD;wBACE,OAAO,EAAE,MAAM,CAAC;4BACd,oBAAoB,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAC3C,iBAAiB,CAAC,gBAAgB,CAChC,OAAO,CAAC,oBAAoB,EAC5B,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAC1B;4BACH,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,YAAY;yBACpE,CAAC;wBACF,MAAM,EAAE,MAAM;qBACf;iBACF;gBACD,OAAO,EAAE;oBACP;wBACE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE;4BACnB,MAAM,CAAC,GAAG,KAAK,CAAC,KAAkC,CAAC;4BACnD,OAAO,CACL,CAAC,EAAE,IAAI,KAAK,YAAY;gCACxB,CAAC,EAAE,OAAO,EAAE,QAAQ,CAAC,SAAS,CAAC;gCAC/B,CAAC,EAAE,IAAI,KAAK,WAAW,CACxB,CAAC;wBACJ,CAAC;wBACD,MAAM,EAAE,eAAe;qBACxB;oBACD;wBACE,OAAO,EAAE;4BACP,MAAM,CAAC;gCACL,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK;6BACtC,CAAC;4BACF,IAAI,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;gCACnB,KAAK,EAAE,KAAK,CAAC,KAAkB;gCAC/B,IAAI,EAAE,UAAU;6BACjB,CAAC,CAAC;yBACJ;wBACD,MAAM,EAAE,yBAAyB;qBAClC;iBACF;gBACD,GAAG,EAAE,eAAe;aACrB;SACF;QACD,SAAS,EAAE;YACT,WAAW,EAAE,kDAAkD;YAC/D,OAAO,EAAE,oBAAoB;YAC7B,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE;oBACrB,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;wBAC1B,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;oBACzD,CAAC;oBACD,OAAO;wBACL,cAAc,EAAE,OAAO,CAAC,cAAe;wBACvC,KAAK,EAAE,OAAO,CAAC,YAAY;qBAC5B,CAAC;gBACJ,CAAC;gBACD,MAAM,EAAE;oBACN,OAAO,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;oBACvC,MAAM,EAAE,MAAM;iBACf;gBACD,OAAO,EAAE;oBACP,OAAO,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;oBACvC,MAAM,EAAE,MAAM;iBACf;gBACD,UAAU,EAAE;oBACV,OAAO,EAAE,cAAc,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE;wBACtD,MAAM,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;wBACzC,IAAI,CAAC,SAAS,EAAE,CAAC;4BACf,OAAO;wBACT,CAAC;wBAED,MAAM,YAAY,GAAG,iBAAiB,CAAC,gBAAgB,CACrD,OAAO,CAAC,oBAAoB,EAC5B,CAAC,SAAS,CAAC,CACZ,CAAC;wBAEF,OAAO,CAAC,MAAM,CAAC;4BACb,oBAAoB,EAAE,YAAY;yBACnC,CAAC,CAAC;wBACH,OAAO,CAAC,IAAI,CAAC;4BACX,SAAS;4BACT,IAAI,EAAE,mBAAmB;yBAC1B,CAAC,CAAC;oBACL,CAAC,CAAC;iBACH;gBACD,GAAG,EAAE,WAAW;aACjB;YACD,MAAM,EAAE;gBACN,kBAAkB,EAAE;oBAClB,WAAW,EAAE,kCAAkC;oBAC/C,EAAE,EAAE;wBACF,QAAQ,EAAE;4BACR,MAAM,EAAE,UAAU;yBACnB;qBACF;iBACF;gBACD,QAAQ,EAAE;oBACR,WAAW,EAAE,0CAA0C;oBACvD,MAAM,EAAE;wBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;4BACvB,cAAc,EAAE,OAAO,CAAC,cAAe;4BACvC,KAAK,EAAE,OAAO,CAAC,YAAa;yBAC7B,CAAC;wBACF,MAAM,EAAE;4BACN,OAAO,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC;4BAC5C,MAAM,EAAE,yBAAyB;yBAClC;wBACD,OAAO,EAAE;4BACP,MAAM,EAAE,uCAAuC;yBAChD;wBACD,GAAG,EAAE,SAAS;qBACf;iBACF;aACF;SACF;QACD,kBAAkB,EAAE;YAClB,WAAW,EAAE,gDAAgD;YAC7D,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE;oBAC5B,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;wBACnB,KAAK,eAAe;4BAClB,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;gCAC5B,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;4BAC3D,CAAC;4BACD,OAAO;gCACL,cAAc,EAAE,OAAO,CAAC,cAAe;gCACvC,OAAO,EAAE,OAAO,CAAC,cAAc;6BAChC,CAAC;wBACJ,KAAK,qBAAqB;4BACxB,OAAO;gCACL,cAAc,EAAE,OAAO,CAAC,cAAe;gCACvC,OAAO,EAAE,KAAK,CAAC,OAAO;6BACvB,CAAC;oBACN,CAAC;oBACD,MAAM,IAAI,KAAK,CAAC,yDAAyD,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;gBACzF,CAAC;gBACD,MAAM,EAAE;oBACN,OAAO,EAAE,MAAM,CAAC;wBACd,qDAAqD;wBACrD,oBAAoB,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;4BAC5C,GAAG,CAAC,OAAO,CAAC,oBAAoB,IAAI,EAAE,CAAC;4BACvC;gCACE,EAAE,EAAE,KAAK,CAAC,MAAM,CAAC,YAAY;gCAC7B,QAAQ,EAAE,IAAI,GAAG,EAAE;gCACnB,oBAAoB,EAAE,OAAO,CAAC,cAAc,IAAI,SAAS;gCACzD,SAAS,EAAE,IAAI,GAAG,EAAE;6BACU;yBACjC;wBACD,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY;wBACtD,cAAc,EAAE,SAAS;qBAC1B,CAAC;oBACF,MAAM,EAAE,WAAW;iBACpB;gBACD,OAAO,EAAE;oBACP,OAAO,EAAE;wBACP,MAAM,CAAC;4BACL,YAAY,EAAE,GAAG,EAAE,CAAC,IAAI;4BACxB,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK;yBACtC,CAAC;wBACF,IAAI,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;4BACnB,KAAK,EAAE,KAAK,CAAC,KAAkB;4BAC/B,IAAI,EAAE,UAAU;yBACjB,CAAC,CAAC;qBACJ;oBACD,MAAM,EAAE,2BAA2B;iBACpC;gBACD,GAAG,EAAE,UAAU;aAChB;SACF;QACD,yBAAyB,EAAE;YACzB,WAAW,EAAE,0BAA0B;YACvC,EAAE,EAAE;gBACF,eAAe,EAAE;oBACf,OAAO,EAAE,MAAM,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;oBACzC,MAAM,EAAE,oBAAoB;iBAC7B;gBACD,aAAa,EAAE;oBACb,OAAO,EAAE,MAAM,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;oBACzC,MAAM,EAAE,oBAAoB;iBAC7B;gBACD,mBAAmB,EAAE;oBACnB,OAAO,EAAE,MAAM,CAAC;wBACd,SAAS,EAAE,SAAS;wBACpB,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO;qBAC7C,CAAC;oBACF,MAAM,EAAE,oBAAoB;iBAC7B;aACF;SACF;QACD,aAAa,EAAE;YACb,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc;gBAChD,MAAM,EAAE,oBAAoB;aAC7B;YACD,WAAW,EACT,gFAAgF;YAClF,EAAE,EAAE;gBACF,mBAAmB,EAAE;oBACnB,OAAO,EAAE,MAAM,CAAC;wBACd,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO;qBAC7C,CAAC;oBACF,MAAM,EAAE,uBAAuB;iBAChC;aACF;SACF;KACF;CACF,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { setup, fromObservable, fromPromise, assign, enqueueActions, emit } from \"xstate\";\nimport type { UserChatMessage } from \"../chat/UserChatMessage.ts\";\nimport { AgentConversation } from \"./AgentConversation.ts\";\nimport type { ConversationExchange } from \"./reduceChatEvents.ts\";\nimport type { SonarV4Config } from \"../../../common/Config.ts\";\nimport { createConversation, createConversationCodec } from \"./methods/createConversation.ts\";\nimport * as z from \"zod/mini\";\nimport { retrieveConversation } from \"./methods/retrieveConversation.ts\";\nimport { retrieveConversationHistory } from \"./methods/retrieveConversationHistory.ts\";\nimport { createExchangeRun } from \"./methods/createExchangeRun.ts\";\nimport { stopExchangeRun } from \"./methods/stopExchangeRun.ts\";\nimport { streamRunEvents } from \"./methods/streamRunEvents.ts\";\nimport { fromTextEventStream } from \"../../../common/fromTextEventStream.ts\";\nimport { map } from \"rxjs\";\nimport { chatEventCodec, type ChatEvent } from \"../chat/chatEventSchema.ts\";\nimport type { HttpError } from \"../../../common/HttpError.ts\";\n\nexport type ConversationMachineInput = {\n  conversationId: string | null;\n};\n\nexport type ConversationMachineContext = {\n  conversationId: string | null;\n  conversationSnapshot?: ConversationExchange[];\n  currentRunId?: string | null;\n  lastError?: unknown;\n  messageAttempt?: UserChatMessage;\n};\nexport type ConversationMachineEvents =\n  | { type: \"STOP_RUN\" }\n  | { type: \"SUBMIT_USER_MESSAGE\"; message: UserChatMessage }\n  | { type: \"RETRY_MESSAGE\" }\n  | { type: \"REFRESH_HISTORY\" };\n\nexport type ConversationMachineEmitted =\n  | {\n      type: \"apiError\";\n      error: HttpError;\n    }\n  | {\n      type: \"chatEventReceived\";\n      chatEvent: ChatEvent;\n    }\n  | {\n      type: \"conversationCreated\";\n      id: string;\n    };\n\nexport const createConversationMachine = (config: SonarV4Config) =>\n  setup({\n    actors: {\n      createConversation: fromPromise(\n        ({ input }: { input: z.input<typeof createConversationCodec> }) =>\n          createConversation(config)(input).then((result) => {\n            if (result.isErr()) {\n              throw result.error;\n            }\n            return result.value;\n          }),\n      ),\n      retrieveConversation: fromPromise(\n        ({ input, signal }: { input: { conversationId: string }; signal: AbortSignal }) =>\n          retrieveConversation(config)(input.conversationId, { signal }).then((result) => {\n            if (result.isErr()) {\n              throw result.error;\n            }\n            return result.value;\n          }),\n      ),\n      retrieveHistory: fromPromise(\n        ({ input, signal }: { input: { conversationId: string }; signal: AbortSignal }) =>\n          retrieveConversationHistory(config)(input.conversationId, { signal }).then((result) => {\n            if (result.isErr()) {\n              throw result.error;\n            }\n            return result.value;\n          }),\n      ),\n      retrieveState: fromPromise(\n        async ({ input, signal }: { input: { conversationId: string }; signal: AbortSignal }) => {\n          const [conversationResult, historyResult] = await Promise.all([\n            retrieveConversation(config)(input.conversationId, { signal }),\n            retrieveConversationHistory(config)(input.conversationId, { signal }),\n          ]);\n\n          if (conversationResult.isErr()) {\n            throw conversationResult.error;\n          }\n\n          if (historyResult.isErr()) {\n            throw historyResult.error;\n          }\n\n          return {\n            conversation: conversationResult.value,\n            history: historyResult.value,\n          };\n        },\n      ),\n      startRun: fromPromise(\n        ({ input }: { input: { conversationId: string; message: UserChatMessage } }) =>\n          createExchangeRun(config)(input.conversationId, { message: input.message }).then(\n            (result) => {\n              if (result.isErr()) {\n                throw result.error;\n              }\n              return result.value;\n            },\n          ),\n      ),\n      stopRun: fromPromise(({ input }: { input: { conversationId: string; runId: string } }) =>\n        stopExchangeRun(config)(input.conversationId, input.runId).then((result) => {\n          if (result.isErr()) {\n            throw result.error;\n          }\n          return result.value;\n        }),\n      ),\n      streamRun: fromObservable(({ input }: { input: { conversationId: string; runId: string } }) =>\n        fromTextEventStream(({ signal }) =>\n          streamRunEvents(config)(\n            { conversationId: input.conversationId, runId: input.runId },\n            { signal },\n          ),\n        ).pipe(\n          map((event) =>\n            z.decode(chatEventCodec, JSON.parse(event.data) as z.input<typeof chatEventCodec>),\n          ),\n        ),\n      ),\n    },\n    types: {\n      context: {} as ConversationMachineContext,\n      emitted: {} as ConversationMachineEmitted,\n      events: {} as ConversationMachineEvents,\n      input: {} as ConversationMachineInput,\n    },\n  }).createMachine({\n    context: ({ input }) => ({\n      conversationId: input.conversationId,\n      conversationSnapshot: undefined,\n      currentRunId: undefined,\n      messageAttempt: undefined,\n    }),\n    id: \"agentConversation\",\n    initial: \"uninitialized\",\n    states: {\n      creating_conversation: {\n        description: \"Creating a new conversation object\",\n        invoke: {\n          input: ({ context }) => ({\n            message: context.messageAttempt!,\n          }),\n          onDone: [\n            {\n              actions: [\n                assign({\n                  conversationId: ({ event }) => event.output.id,\n                  // Add the message attempt into the snapshot manually\n                  conversationSnapshot: ({ context, event }) => [\n                    ...(context.conversationSnapshot ?? []),\n                    {\n                      id: event.output.currentRunId!,\n                      messages: new Map(),\n                      submittedUserMessage: context.messageAttempt,\n                      toolCalls: new Map(),\n                    } satisfies ConversationExchange,\n                  ],\n                  currentRunId: ({ event }) => event.output.currentRunId,\n                  messageAttempt: undefined,\n                }),\n                emit(({ event }) => ({\n                  id: event.output.id,\n                  type: \"conversationCreated\",\n                })),\n              ],\n              // Don't transition to streaming in a situation where we didn't receive a currentRunId\n              guard: ({ event }) => !!event.output.currentRunId,\n              target: \"streaming\",\n            },\n            /**\n             * Fall back to the retrieving_history state if we didn't receive a currentRunId\n             */\n            {\n              actions: assign({\n                conversationId: ({ event }) => event.output.id,\n                currentRunId: ({ event }) => event.output.currentRunId,\n                messageAttempt: undefined,\n              }),\n              target: \"retrieving_history\",\n            },\n          ],\n          onError: {\n            actions: emit(({ event }) => ({\n              error: event.error as HttpError,\n              type: \"apiError\",\n            })),\n            target: \"uninitialized\",\n          },\n          src: \"createConversation\",\n        },\n      },\n      idle: {\n        description: \"Waiting for user input to start the next reply stream\",\n        on: {\n          REFRESH_HISTORY: {\n            target: \"retrieving_history\",\n          },\n          SUBMIT_USER_MESSAGE: {\n            actions: assign({\n              messageAttempt: ({ event }) => event.message,\n            }),\n            target: \"submitting_message\",\n          },\n        },\n      },\n      retrieve_history_failed: {\n        on: {\n          REFRESH_HISTORY: {\n            actions: assign({ lastError: undefined }),\n            target: \"retrieving_history\",\n          },\n        },\n      },\n      retrieving_history: {\n        description: \"Retrieving latest copy of conversation history\",\n        invoke: {\n          input: ({ context }) => ({ conversationId: context.conversationId! }),\n          onDone: [\n            {\n              actions: assign({\n                conversationSnapshot: ({ context, event }) =>\n                  AgentConversation.reduceChatEvents(\n                    context.conversationSnapshot,\n                    event.output.history.data,\n                  ),\n                currentRunId: ({ event }) => event.output.conversation.currentRunId,\n              }),\n              guard: ({ event }) => !!event.output.conversation.currentRunId,\n              target: \"streaming\",\n            },\n            {\n              actions: assign({\n                conversationSnapshot: ({ context, event }) =>\n                  AgentConversation.reduceChatEvents(\n                    context.conversationSnapshot,\n                    event.output.history.data,\n                  ),\n                currentRunId: ({ event }) => event.output.conversation.currentRunId,\n              }),\n              target: \"idle\",\n            },\n          ],\n          onError: [\n            {\n              guard: ({ event }) => {\n                const e = event.error as Error & { code?: string };\n                return (\n                  e?.name === \"AbortError\" ||\n                  e?.message?.includes(\"aborted\") ||\n                  e?.code === \"ABORT_ERR\"\n                );\n              },\n              target: \"uninitialized\",\n            },\n            {\n              actions: [\n                assign({\n                  lastError: ({ event }) => event.error,\n                }),\n                emit(({ event }) => ({\n                  error: event.error as HttpError,\n                  type: \"apiError\",\n                })),\n              ],\n              target: \"retrieve_history_failed\",\n            },\n          ],\n          src: \"retrieveState\",\n        },\n      },\n      streaming: {\n        description: \"Manages monitoring and canceling the current run\",\n        initial: \"monitoring_replies\",\n        invoke: {\n          input: ({ context }) => {\n            if (!context.currentRunId) {\n              throw new Error(\"Attempted to stream without a runId\");\n            }\n            return {\n              conversationId: context.conversationId!,\n              runId: context.currentRunId,\n            };\n          },\n          onDone: {\n            actions: assign({ currentRunId: null }),\n            target: \"idle\",\n          },\n          onError: {\n            actions: assign({ currentRunId: null }),\n            target: \"idle\",\n          },\n          onSnapshot: {\n            actions: enqueueActions(({ context, enqueue, event }) => {\n              const chatEvent = event.snapshot.context;\n              if (!chatEvent) {\n                return;\n              }\n\n              const nextSnapshot = AgentConversation.reduceChatEvents(\n                context.conversationSnapshot,\n                [chatEvent],\n              );\n\n              enqueue.assign({\n                conversationSnapshot: nextSnapshot,\n              });\n              enqueue.emit({\n                chatEvent,\n                type: \"chatEventReceived\",\n              });\n            }),\n          },\n          src: \"streamRun\",\n        },\n        states: {\n          monitoring_replies: {\n            description: \"Attached to the run event stream\",\n            on: {\n              STOP_RUN: {\n                target: \"stopping\",\n              },\n            },\n          },\n          stopping: {\n            description: \"Background cancelation of the run stream\",\n            invoke: {\n              input: ({ context }) => ({\n                conversationId: context.conversationId!,\n                runId: context.currentRunId!,\n              }),\n              onDone: {\n                actions: assign({ currentRunId: undefined }),\n                target: \"#agentConversation.idle\",\n              },\n              onError: {\n                target: \"#agentConversation.retrieving_history\",\n              },\n              src: \"stopRun\",\n            },\n          },\n        },\n      },\n      submitting_message: {\n        description: \"Submitting message to start a new reply stream\",\n        invoke: {\n          input: ({ context, event }) => {\n            switch (event.type) {\n              case \"RETRY_MESSAGE\":\n                if (!context.messageAttempt) {\n                  throw new Error(\"Tried to resubmit nonexistent message\");\n                }\n                return {\n                  conversationId: context.conversationId!,\n                  message: context.messageAttempt,\n                };\n              case \"SUBMIT_USER_MESSAGE\":\n                return {\n                  conversationId: context.conversationId!,\n                  message: event.message,\n                };\n            }\n            throw new Error(`Unexpected event type received in submitting_message: ${event.type}`);\n          },\n          onDone: {\n            actions: assign({\n              // Add the message attempt into the snapshot manually\n              conversationSnapshot: ({ context, event }) => [\n                ...(context.conversationSnapshot ?? []),\n                {\n                  id: event.output.currentRunId,\n                  messages: new Map(),\n                  submittedUserMessage: context.messageAttempt || undefined,\n                  toolCalls: new Map(),\n                } satisfies ConversationExchange,\n              ],\n              currentRunId: ({ event }) => event.output.currentRunId,\n              messageAttempt: undefined,\n            }),\n            target: \"streaming\",\n          },\n          onError: {\n            actions: [\n              assign({\n                currentRunId: () => null,\n                lastError: ({ event }) => event.error,\n              }),\n              emit(({ event }) => ({\n                error: event.error as HttpError,\n                type: \"apiError\",\n              })),\n            ],\n            target: \"submitting_message_failed\",\n          },\n          src: \"startRun\",\n        },\n      },\n      submitting_message_failed: {\n        description: \"Failed to submit message\",\n        on: {\n          REFRESH_HISTORY: {\n            actions: assign({ lastError: undefined }),\n            target: \"retrieving_history\",\n          },\n          RETRY_MESSAGE: {\n            actions: assign({ lastError: undefined }),\n            target: \"submitting_message\",\n          },\n          SUBMIT_USER_MESSAGE: {\n            actions: assign({\n              lastError: undefined,\n              messageAttempt: ({ event }) => event.message,\n            }),\n            target: \"submitting_message\",\n          },\n        },\n      },\n      uninitialized: {\n        always: {\n          guard: ({ context }) => !!context.conversationId,\n          target: \"retrieving_history\",\n        },\n        description:\n          \"Entry state that the machine remains in when no conversation has been provided\",\n        on: {\n          SUBMIT_USER_MESSAGE: {\n            actions: assign({\n              messageAttempt: ({ event }) => event.message,\n            }),\n            target: \"creating_conversation\",\n          },\n        },\n      },\n    },\n  });\n\nexport type AgentConversationMachine = ReturnType<typeof createConversationMachine>;\n"]}
+{"version":3,"file":"createConversationMachine.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/conversations/createConversationMachine.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AAE1F,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAG3D,OAAO,EAAE,kBAAkB,EAAE,uBAAuB,EAAE,MAAM,iCAAiC,CAAC;AAC9F,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EAAE,oBAAoB,EAAE,MAAM,mCAAmC,CAAC;AACzE,OAAO,EAAE,2BAA2B,EAAE,MAAM,0CAA0C,CAAC;AACvF,OAAO,EAAE,iBAAiB,EAAE,MAAM,gCAAgC,CAAC;AACnE,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,mBAAmB,EAAE,MAAM,wCAAwC,CAAC;AAC7E,OAAO,EAAE,GAAG,EAAE,MAAM,MAAM,CAAC;AAC3B,OAAO,EAAE,cAAc,EAAkB,MAAM,4BAA4B,CAAC;AAqC5E,MAAM,yBAAyB,GAAG,MAAM,CAAC;AACzC,MAAM,qBAAqB,GAAG,IAAI,CAAC;AACnC,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC;AACpC,MAAM,kBAAkB,GAAG;IACzB,mBAAmB,EAAE,KAAK;IAC1B,gBAAgB,EAAE,CAAC;IACnB,iBAAiB,EAAE,KAAK;CACzB,CAAC;AAEF,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,MAAqB,EAAE,EAAE,CACjE,KAAK,CAAC;IACJ,MAAM,EAAE;QACN,kBAAkB,EAAE,WAAW,CAC7B,CAAC,EAAE,KAAK,EAAsD,EAAE,EAAE,CAChE,kBAAkB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YAChD,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CAAC,CACL;QACD,oBAAoB,EAAE,WAAW,CAC/B,CAAC,EAAE,KAAK,EAAE,MAAM,EAA8D,EAAE,EAAE,CAChF,oBAAoB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YAC7E,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CAAC,CACL;QACD,eAAe,EAAE,WAAW,CAC1B,CAAC,EAAE,KAAK,EAAE,MAAM,EAA8D,EAAE,EAAE,CAChF,2BAA2B,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YACpF,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CAAC,CACL;QACD,aAAa,EAAE,WAAW,CACxB,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAA8D,EAAE,EAAE;YACtF,MAAM,aAAa,GAAG,WAAW,CAAC,GAAG,CAAC;gBACpC,MAAM;gBACN,WAAW,CAAC,OAAO,CAAC,yBAAyB,CAAC;aAC/C,CAAC,CAAC;YACH,MAAM,CAAC,kBAAkB,EAAE,aAAa,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC5D,oBAAoB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;gBAC7E,2BAA2B,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;aACrF,CAAC,CAAC;YAEH,IAAI,kBAAkB,CAAC,KAAK,EAAE,EAAE,CAAC;gBAC/B,MAAM,kBAAkB,CAAC,KAAK,CAAC;YACjC,CAAC;YAED,IAAI,aAAa,CAAC,KAAK,EAAE,EAAE,CAAC;gBAC1B,MAAM,aAAa,CAAC,KAAK,CAAC;YAC5B,CAAC;YAED,OAAO;gBACL,YAAY,EAAE,kBAAkB,CAAC,KAAK;gBACtC,OAAO,EAAE,aAAa,CAAC,KAAK;aAC7B,CAAC;QACJ,CAAC,CACF;QACD,QAAQ,EAAE,WAAW,CACnB,CAAC,EAAE,KAAK,EAAmE,EAAE,EAAE,CAC7E,iBAAiB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAC9E,CAAC,MAAM,EAAE,EAAE;YACT,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CACF,CACJ;QACD,OAAO,EAAE,WAAW,CAAC,CAAC,EAAE,KAAK,EAAwD,EAAE,EAAE,CACvF,eAAe,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YACzE,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CAAC,CACH;QACD,SAAS,EAAE,cAAc,CAAC,CAAC,EAAE,KAAK,EAAwD,EAAE,EAAE,CAC5F,mBAAmB,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CACjC,eAAe,CAAC,MAAM,CAAC,CACrB,EAAE,cAAc,EAAE,KAAK,CAAC,cAAc,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,EAC5D,EAAE,MAAM,EAAE,CACX,CACF,CAAC,IAAI,CACJ,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,CAAC,CAAC,MAAM,CAAC,cAAc,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAmC,CAAC,CACnF,CACF,CACF;KACF;IACD,MAAM,EAAE;QACN,kBAAkB,EAAE,qBAAqB;KAC1C;IACD,KAAK,EAAE;QACL,OAAO,EAAE,EAAgC;QACzC,OAAO,EAAE,EAAgC;QACzC,MAAM,EAAE,EAA+B;QACvC,KAAK,EAAE,EAA8B;KACtC;CACF,CAAC,CAAC,aAAa,CAAC;IACf,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QACvB,cAAc,EAAE,KAAK,CAAC,cAAc;QACpC,oBAAoB,EAAE,SAAS;QAC/B,YAAY,EAAE,SAAS;QACvB,cAAc,EAAE,SAAS;QACzB,GAAG,kBAAkB;KACtB,CAAC;IACF,EAAE,EAAE,mBAAmB;IACvB,OAAO,EAAE,eAAe;IACxB,MAAM,EAAE;QACN,qBAAqB,EAAE;YACrB,WAAW,EAAE,oCAAoC;YACjD,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;oBACvB,OAAO,EAAE,OAAO,CAAC,cAAe;iBACjC,CAAC;gBACF,MAAM,EAAE;oBACN;wBACE,OAAO,EAAE;4BACP,MAAM,CAAC;gCACL,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE;gCAC9C,qDAAqD;gCACrD,oBAAoB,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;oCAC5C,GAAG,CAAC,OAAO,CAAC,oBAAoB,IAAI,EAAE,CAAC;oCACvC;wCACE,EAAE,EAAE,KAAK,CAAC,MAAM,CAAC,YAAa;wCAC9B,QAAQ,EAAE,IAAI,GAAG,EAAE;wCACnB,oBAAoB,EAAE,OAAO,CAAC,cAAc;wCAC5C,SAAS,EAAE,IAAI,GAAG,EAAE;qCACU;iCACjC;gCACD,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY;gCACtD,cAAc,EAAE,SAAS;gCACzB,GAAG,kBAAkB;6BACtB,CAAC;4BACF,IAAI,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;gCACnB,EAAE,EAAE,KAAK,CAAC,MAAM,CAAC,EAAE;gCACnB,IAAI,EAAE,qBAAqB;6BAC5B,CAAC,CAAC;yBACJ;wBACD,sFAAsF;wBACtF,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY;wBACjD,MAAM,EAAE,WAAW;qBACpB;oBACD;;uBAEG;oBACH;wBACE,OAAO,EAAE,MAAM,CAAC;4BACd,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE;4BAC9C,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY;4BACtD,cAAc,EAAE,SAAS;yBAC1B,CAAC;wBACF,MAAM,EAAE,oBAAoB;qBAC7B;iBACF;gBACD,OAAO,EAAE;oBACP,OAAO,EAAE,IAAI,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;wBAC5B,KAAK,EAAE,KAAK,CAAC,KAAkB;wBAC/B,IAAI,EAAE,UAAU;qBACjB,CAAC,CAAC;oBACH,MAAM,EAAE,eAAe;iBACxB;gBACD,GAAG,EAAE,oBAAoB;aAC1B;SACF;QACD,IAAI,EAAE;YACJ,WAAW,EAAE,uDAAuD;YACpE,EAAE,EAAE;gBACF,eAAe,EAAE;oBACf,MAAM,EAAE,oBAAoB;iBAC7B;gBACD,mBAAmB,EAAE;oBACnB,OAAO,EAAE,MAAM,CAAC;wBACd,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO;qBAC7C,CAAC;oBACF,MAAM,EAAE,oBAAoB;iBAC7B;aACF;SACF;QACD,uBAAuB,EAAE;YACvB,EAAE,EAAE;gBACF,eAAe,EAAE;oBACf,OAAO,EAAE,MAAM,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;oBACzC,MAAM,EAAE,oBAAoB;iBAC7B;aACF;SACF;QACD,kBAAkB,EAAE;YAClB,WAAW,EAAE,gDAAgD;YAC7D,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,cAAc,EAAE,OAAO,CAAC,cAAe,EAAE,CAAC;gBACrE,MAAM,EAAE;oBACN;wBACE,OAAO,EAAE,MAAM,CAAC;4BACd,oBAAoB,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAC3C,iBAAiB,CAAC,gBAAgB,CAChC,OAAO,CAAC,oBAAoB,EAC5B,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAC1B;4BACH,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,YAAY;4BACnE,mBAAmB,EAAE,KAAK;yBAC3B,CAAC;wBACF,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,YAAY;wBAC9D,MAAM,EAAE,WAAW;qBACpB;oBACD;wBACE,OAAO,EAAE,MAAM,CAAC;4BACd,oBAAoB,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAC3C,iBAAiB,CAAC,gBAAgB,CAChC,OAAO,CAAC,oBAAoB,EAC5B,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAC1B;4BACH,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,YAAY;4BACnE,GAAG,kBAAkB;yBACtB,CAAC;wBACF,MAAM,EAAE,MAAM;qBACf;iBACF;gBACD,OAAO,EAAE;oBACP;wBACE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE;4BACnB,MAAM,CAAC,GAAG,KAAK,CAAC,KAAkC,CAAC;4BACnD,IAAI,CAAC,EAAE,IAAI,KAAK,cAAc;gCAAE,OAAO,KAAK,CAAC;4BAC7C,OAAO,CACL,CAAC,EAAE,IAAI,KAAK,YAAY;gCACxB,CAAC,EAAE,OAAO,EAAE,QAAQ,CAAC,SAAS,CAAC;gCAC/B,CAAC,EAAE,IAAI,KAAK,WAAW,CACxB,CAAC;wBACJ,CAAC;wBACD,MAAM,EAAE,eAAe;qBACxB;oBACD;wBACE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,gBAAgB,GAAG,CAAC;wBACpD,MAAM,EAAE,oBAAoB;qBAC7B;oBACD;wBACE,OAAO,EAAE;4BACP,MAAM,CAAC;gCACL,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK;6BACtC,CAAC;4BACF,IAAI,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;gCACnB,KAAK,EAAE,KAAK,CAAC,KAAkB;gCAC/B,IAAI,EAAE,UAAU;6BACjB,CAAC,CAAC;yBACJ;wBACD,MAAM,EAAE,yBAAyB;qBAClC;iBACF;gBACD,GAAG,EAAE,eAAe;aACrB;SACF;QACD,kBAAkB,EAAE;YAClB,KAAK,EAAE;gBACL,kBAAkB,EAAE;oBAClB;wBACE,OAAO,EAAE,MAAM,CAAC;4BACd,gBAAgB,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,gBAAgB,GAAG,CAAC;yBAChE,CAAC;wBACF,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,gBAAgB,GAAG,kBAAkB;wBACrE,MAAM,EAAE,oBAAoB;qBAC7B;oBACD;wBACE,MAAM,EAAE,wBAAwB;qBACjC;iBACF;aACF;SACF;QACD,sBAAsB,EAAE;YACtB,EAAE,EAAE;gBACF,eAAe,EAAE;oBACf,OAAO,EAAE,MAAM,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,gBAAgB,EAAE,CAAC,EAAE,CAAC;oBAC9D,MAAM,EAAE,oBAAoB;iBAC7B;aACF;SACF;QACD,SAAS,EAAE;YACT,WAAW,EAAE,kDAAkD;YAC/D,OAAO,EAAE,oBAAoB;YAC7B,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE;oBACrB,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;wBAC1B,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;oBACzD,CAAC;oBACD,OAAO;wBACL,cAAc,EAAE,OAAO,CAAC,cAAe;wBACvC,KAAK,EAAE,OAAO,CAAC,YAAY;qBAC5B,CAAC;gBACJ,CAAC;gBACD,MAAM,EAAE;oBACN;wBACE,OAAO,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;wBACjE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,mBAAmB,IAAI,CAAC,CAAC,OAAO,CAAC,iBAAiB;wBACpF,MAAM,EAAE,MAAM;qBACf;oBACD;wBACE,OAAO,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;wBACvC,MAAM,EAAE,oBAAoB;qBAC7B;iBACF;gBACD,OAAO,EAAE;oBACP;wBACE,OAAO,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;wBACjE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,mBAAmB,IAAI,CAAC,CAAC,OAAO,CAAC,iBAAiB;wBACpF,MAAM,EAAE,MAAM;qBACf;oBACD;wBACE,OAAO,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;wBACvC,MAAM,EAAE,oBAAoB;qBAC7B;iBACF;gBACD,UAAU,EAAE;oBACV,OAAO,EAAE,cAAc,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE;wBACtD,MAAM,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;wBACzC,IAAI,CAAC,SAAS,EAAE,CAAC;4BACf,OAAO;wBACT,CAAC;wBAED,IAAI,SAAS,CAAC,OAAO,CAAC,SAAS,KAAK,aAAa,IAAI,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC;4BAClF,OAAO,CAAC,MAAM,CAAC,EAAE,mBAAmB,EAAE,IAAI,EAAE,CAAC,CAAC;wBAChD,CAAC;wBAED,MAAM,YAAY,GAAG,iBAAiB,CAAC,gBAAgB,CACrD,OAAO,CAAC,oBAAoB,EAC5B,CAAC,SAAS,CAAC,CACZ,CAAC;wBAEF,OAAO,CAAC,MAAM,CAAC;4BACb,oBAAoB,EAAE,YAAY;yBACnC,CAAC,CAAC;wBACH,OAAO,CAAC,IAAI,CAAC;4BACX,SAAS;4BACT,IAAI,EAAE,mBAAmB;yBAC1B,CAAC,CAAC;oBACL,CAAC,CAAC;iBACH;gBACD,GAAG,EAAE,WAAW;aACjB;YACD,MAAM,EAAE;gBACN,kBAAkB,EAAE;oBAClB,WAAW,EAAE,kCAAkC;oBAC/C,EAAE,EAAE;wBACF,QAAQ,EAAE;4BACR,OAAO,EAAE,MAAM,CAAC,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC;4BAC5C,MAAM,EAAE,UAAU;yBACnB;qBACF;iBACF;gBACD,QAAQ,EAAE;oBACR,WAAW,EAAE,0CAA0C;oBACvD,MAAM,EAAE;wBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;4BACvB,cAAc,EAAE,OAAO,CAAC,cAAe;4BACvC,KAAK,EAAE,OAAO,CAAC,YAAa;yBAC7B,CAAC;wBACF,MAAM,EAAE;4BACN,OAAO,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,SAAS,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;4BACtE,MAAM,EAAE,yBAAyB;yBAClC;wBACD,OAAO,EAAE;4BACP,OAAO,EAAE,MAAM,CAAC,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;4BAC7C,MAAM,EAAE,uCAAuC;yBAChD;wBACD,GAAG,EAAE,SAAS;qBACf;iBACF;aACF;SACF;QACD,kBAAkB,EAAE;YAClB,WAAW,EAAE,gDAAgD;YAC7D,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE;oBAC5B,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;wBACnB,KAAK,eAAe;4BAClB,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;gCAC5B,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;4BAC3D,CAAC;4BACD,OAAO;gCACL,cAAc,EAAE,OAAO,CAAC,cAAe;gCACvC,OAAO,EAAE,OAAO,CAAC,cAAc;6BAChC,CAAC;wBACJ,KAAK,qBAAqB;4BACxB,OAAO;gCACL,cAAc,EAAE,OAAO,CAAC,cAAe;gCACvC,OAAO,EAAE,KAAK,CAAC,OAAO;6BACvB,CAAC;oBACN,CAAC;oBACD,MAAM,IAAI,KAAK,CAAC,yDAAyD,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;gBACzF,CAAC;gBACD,MAAM,EAAE;oBACN,OAAO,EAAE,MAAM,CAAC;wBACd,qDAAqD;wBACrD,oBAAoB,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;4BAC5C,GAAG,CAAC,OAAO,CAAC,oBAAoB,IAAI,EAAE,CAAC;4BACvC;gCACE,EAAE,EAAE,KAAK,CAAC,MAAM,CAAC,YAAY;gCAC7B,QAAQ,EAAE,IAAI,GAAG,EAAE;gCACnB,oBAAoB,EAAE,OAAO,CAAC,cAAc,IAAI,SAAS;gCACzD,SAAS,EAAE,IAAI,GAAG,EAAE;6BACU;yBACjC;wBACD,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY;wBACtD,cAAc,EAAE,SAAS;wBACzB,GAAG,kBAAkB;qBACtB,CAAC;oBACF,MAAM,EAAE,WAAW;iBACpB;gBACD,OAAO,EAAE;oBACP,OAAO,EAAE;wBACP,MAAM,CAAC;4BACL,YAAY,EAAE,GAAG,EAAE,CAAC,IAAI;4BACxB,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK;yBACtC,CAAC;wBACF,IAAI,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;4BACnB,KAAK,EAAE,KAAK,CAAC,KAAkB;4BAC/B,IAAI,EAAE,UAAU;yBACjB,CAAC,CAAC;qBACJ;oBACD,MAAM,EAAE,2BAA2B;iBACpC;gBACD,GAAG,EAAE,UAAU;aAChB;SACF;QACD,yBAAyB,EAAE;YACzB,WAAW,EAAE,0BAA0B;YACvC,EAAE,EAAE;gBACF,eAAe,EAAE;oBACf,OAAO,EAAE,MAAM,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;oBACzC,MAAM,EAAE,oBAAoB;iBAC7B;gBACD,aAAa,EAAE;oBACb,OAAO,EAAE,MAAM,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;oBACzC,MAAM,EAAE,oBAAoB;iBAC7B;gBACD,mBAAmB,EAAE;oBACnB,OAAO,EAAE,MAAM,CAAC;wBACd,SAAS,EAAE,SAAS;wBACpB,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO;qBAC7C,CAAC;oBACF,MAAM,EAAE,oBAAoB;iBAC7B;aACF;SACF;QACD,aAAa,EAAE;YACb,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc;gBAChD,MAAM,EAAE,oBAAoB;aAC7B;YACD,WAAW,EACT,gFAAgF;YAClF,EAAE,EAAE;gBACF,mBAAmB,EAAE;oBACnB,OAAO,EAAE,MAAM,CAAC;wBACd,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO;qBAC7C,CAAC;oBACF,MAAM,EAAE,uBAAuB;iBAChC;aACF;SACF;KACF;CACF,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { setup, fromObservable, fromPromise, assign, enqueueActions, emit } from \"xstate\";\nimport type { UserChatMessage } from \"../chat/UserChatMessage.ts\";\nimport { AgentConversation } from \"./AgentConversation.ts\";\nimport type { ConversationExchange } from \"./reduceChatEvents.ts\";\nimport type { SonarV4Config } from \"../../../common/Config.ts\";\nimport { createConversation, createConversationCodec } from \"./methods/createConversation.ts\";\nimport * as z from \"zod/mini\";\nimport { retrieveConversation } from \"./methods/retrieveConversation.ts\";\nimport { retrieveConversationHistory } from \"./methods/retrieveConversationHistory.ts\";\nimport { createExchangeRun } from \"./methods/createExchangeRun.ts\";\nimport { stopExchangeRun } from \"./methods/stopExchangeRun.ts\";\nimport { streamRunEvents } from \"./methods/streamRunEvents.ts\";\nimport { fromTextEventStream } from \"../../../common/fromTextEventStream.ts\";\nimport { map } from \"rxjs\";\nimport { chatEventCodec, type ChatEvent } from \"../chat/chatEventSchema.ts\";\nimport type { HttpError } from \"../../../common/HttpError.ts\";\n\nexport type ConversationMachineInput = {\n  conversationId: string | null;\n};\n\nexport type ConversationMachineContext = {\n  conversationId: string | null;\n  conversationSnapshot?: ConversationExchange[];\n  currentRunId?: string | null;\n  lastError?: unknown;\n  messageAttempt?: UserChatMessage;\n  receivedEndOfStream: boolean;\n  streamRetryCount: number;\n  userRequestedStop: boolean;\n};\nexport type ConversationMachineEvents =\n  | { type: \"STOP_RUN\" }\n  | { type: \"SUBMIT_USER_MESSAGE\"; message: UserChatMessage }\n  | { type: \"RETRY_MESSAGE\" }\n  | { type: \"REFRESH_HISTORY\" };\n\nexport type ConversationMachineEmitted =\n  | {\n      type: \"apiError\";\n      error: HttpError;\n    }\n  | {\n      type: \"chatEventReceived\";\n      chatEvent: ChatEvent;\n    }\n  | {\n      type: \"conversationCreated\";\n      id: string;\n    };\n\nconst RETRIEVE_STATE_TIMEOUT_MS = 10_000;\nconst STREAM_RETRY_DELAY_MS = 2500;\nexport const MAX_STREAM_RETRIES = 5;\nconst STREAM_STATE_RESET = {\n  receivedEndOfStream: false,\n  streamRetryCount: 0,\n  userRequestedStop: false,\n};\n\nexport const createConversationMachine = (config: SonarV4Config) =>\n  setup({\n    actors: {\n      createConversation: fromPromise(\n        ({ input }: { input: z.input<typeof createConversationCodec> }) =>\n          createConversation(config)(input).then((result) => {\n            if (result.isErr()) {\n              throw result.error;\n            }\n            return result.value;\n          }),\n      ),\n      retrieveConversation: fromPromise(\n        ({ input, signal }: { input: { conversationId: string }; signal: AbortSignal }) =>\n          retrieveConversation(config)(input.conversationId, { signal }).then((result) => {\n            if (result.isErr()) {\n              throw result.error;\n            }\n            return result.value;\n          }),\n      ),\n      retrieveHistory: fromPromise(\n        ({ input, signal }: { input: { conversationId: string }; signal: AbortSignal }) =>\n          retrieveConversationHistory(config)(input.conversationId, { signal }).then((result) => {\n            if (result.isErr()) {\n              throw result.error;\n            }\n            return result.value;\n          }),\n      ),\n      retrieveState: fromPromise(\n        async ({ input, signal }: { input: { conversationId: string }; signal: AbortSignal }) => {\n          const timeoutSignal = AbortSignal.any([\n            signal,\n            AbortSignal.timeout(RETRIEVE_STATE_TIMEOUT_MS),\n          ]);\n          const [conversationResult, historyResult] = await Promise.all([\n            retrieveConversation(config)(input.conversationId, { signal: timeoutSignal }),\n            retrieveConversationHistory(config)(input.conversationId, { signal: timeoutSignal }),\n          ]);\n\n          if (conversationResult.isErr()) {\n            throw conversationResult.error;\n          }\n\n          if (historyResult.isErr()) {\n            throw historyResult.error;\n          }\n\n          return {\n            conversation: conversationResult.value,\n            history: historyResult.value,\n          };\n        },\n      ),\n      startRun: fromPromise(\n        ({ input }: { input: { conversationId: string; message: UserChatMessage } }) =>\n          createExchangeRun(config)(input.conversationId, { message: input.message }).then(\n            (result) => {\n              if (result.isErr()) {\n                throw result.error;\n              }\n              return result.value;\n            },\n          ),\n      ),\n      stopRun: fromPromise(({ input }: { input: { conversationId: string; runId: string } }) =>\n        stopExchangeRun(config)(input.conversationId, input.runId).then((result) => {\n          if (result.isErr()) {\n            throw result.error;\n          }\n          return result.value;\n        }),\n      ),\n      streamRun: fromObservable(({ input }: { input: { conversationId: string; runId: string } }) =>\n        fromTextEventStream(({ signal }) =>\n          streamRunEvents(config)(\n            { conversationId: input.conversationId, runId: input.runId },\n            { signal },\n          ),\n        ).pipe(\n          map((event) =>\n            z.decode(chatEventCodec, JSON.parse(event.data) as z.input<typeof chatEventCodec>),\n          ),\n        ),\n      ),\n    },\n    delays: {\n      STREAM_RETRY_DELAY: STREAM_RETRY_DELAY_MS,\n    },\n    types: {\n      context: {} as ConversationMachineContext,\n      emitted: {} as ConversationMachineEmitted,\n      events: {} as ConversationMachineEvents,\n      input: {} as ConversationMachineInput,\n    },\n  }).createMachine({\n    context: ({ input }) => ({\n      conversationId: input.conversationId,\n      conversationSnapshot: undefined,\n      currentRunId: undefined,\n      messageAttempt: undefined,\n      ...STREAM_STATE_RESET,\n    }),\n    id: \"agentConversation\",\n    initial: \"uninitialized\",\n    states: {\n      creating_conversation: {\n        description: \"Creating a new conversation object\",\n        invoke: {\n          input: ({ context }) => ({\n            message: context.messageAttempt!,\n          }),\n          onDone: [\n            {\n              actions: [\n                assign({\n                  conversationId: ({ event }) => event.output.id,\n                  // Add the message attempt into the snapshot manually\n                  conversationSnapshot: ({ context, event }) => [\n                    ...(context.conversationSnapshot ?? []),\n                    {\n                      id: event.output.currentRunId!,\n                      messages: new Map(),\n                      submittedUserMessage: context.messageAttempt,\n                      toolCalls: new Map(),\n                    } satisfies ConversationExchange,\n                  ],\n                  currentRunId: ({ event }) => event.output.currentRunId,\n                  messageAttempt: undefined,\n                  ...STREAM_STATE_RESET,\n                }),\n                emit(({ event }) => ({\n                  id: event.output.id,\n                  type: \"conversationCreated\",\n                })),\n              ],\n              // Don't transition to streaming in a situation where we didn't receive a currentRunId\n              guard: ({ event }) => !!event.output.currentRunId,\n              target: \"streaming\",\n            },\n            /**\n             * Fall back to the retrieving_history state if we didn't receive a currentRunId\n             */\n            {\n              actions: assign({\n                conversationId: ({ event }) => event.output.id,\n                currentRunId: ({ event }) => event.output.currentRunId,\n                messageAttempt: undefined,\n              }),\n              target: \"retrieving_history\",\n            },\n          ],\n          onError: {\n            actions: emit(({ event }) => ({\n              error: event.error as HttpError,\n              type: \"apiError\",\n            })),\n            target: \"uninitialized\",\n          },\n          src: \"createConversation\",\n        },\n      },\n      idle: {\n        description: \"Waiting for user input to start the next reply stream\",\n        on: {\n          REFRESH_HISTORY: {\n            target: \"retrieving_history\",\n          },\n          SUBMIT_USER_MESSAGE: {\n            actions: assign({\n              messageAttempt: ({ event }) => event.message,\n            }),\n            target: \"submitting_message\",\n          },\n        },\n      },\n      retrieve_history_failed: {\n        on: {\n          REFRESH_HISTORY: {\n            actions: assign({ lastError: undefined }),\n            target: \"retrieving_history\",\n          },\n        },\n      },\n      retrieving_history: {\n        description: \"Retrieving latest copy of conversation history\",\n        invoke: {\n          input: ({ context }) => ({ conversationId: context.conversationId! }),\n          onDone: [\n            {\n              actions: assign({\n                conversationSnapshot: ({ context, event }) =>\n                  AgentConversation.reduceChatEvents(\n                    context.conversationSnapshot,\n                    event.output.history.data,\n                  ),\n                currentRunId: ({ event }) => event.output.conversation.currentRunId,\n                receivedEndOfStream: false,\n              }),\n              guard: ({ event }) => !!event.output.conversation.currentRunId,\n              target: \"streaming\",\n            },\n            {\n              actions: assign({\n                conversationSnapshot: ({ context, event }) =>\n                  AgentConversation.reduceChatEvents(\n                    context.conversationSnapshot,\n                    event.output.history.data,\n                  ),\n                currentRunId: ({ event }) => event.output.conversation.currentRunId,\n                ...STREAM_STATE_RESET,\n              }),\n              target: \"idle\",\n            },\n          ],\n          onError: [\n            {\n              guard: ({ event }) => {\n                const e = event.error as Error & { code?: string };\n                if (e?.name === \"TimeoutError\") return false;\n                return (\n                  e?.name === \"AbortError\" ||\n                  e?.message?.includes(\"aborted\") ||\n                  e?.code === \"ABORT_ERR\"\n                );\n              },\n              target: \"uninitialized\",\n            },\n            {\n              guard: ({ context }) => context.streamRetryCount > 0,\n              target: \"stream_interrupted\",\n            },\n            {\n              actions: [\n                assign({\n                  lastError: ({ event }) => event.error,\n                }),\n                emit(({ event }) => ({\n                  error: event.error as HttpError,\n                  type: \"apiError\",\n                })),\n              ],\n              target: \"retrieve_history_failed\",\n            },\n          ],\n          src: \"retrieveState\",\n        },\n      },\n      stream_interrupted: {\n        after: {\n          STREAM_RETRY_DELAY: [\n            {\n              actions: assign({\n                streamRetryCount: ({ context }) => context.streamRetryCount + 1,\n              }),\n              guard: ({ context }) => context.streamRetryCount < MAX_STREAM_RETRIES,\n              target: \"retrieving_history\",\n            },\n            {\n              target: \"stream_recovery_failed\",\n            },\n          ],\n        },\n      },\n      stream_recovery_failed: {\n        on: {\n          REFRESH_HISTORY: {\n            actions: assign({ lastError: undefined, streamRetryCount: 0 }),\n            target: \"retrieving_history\",\n          },\n        },\n      },\n      streaming: {\n        description: \"Manages monitoring and canceling the current run\",\n        initial: \"monitoring_replies\",\n        invoke: {\n          input: ({ context }) => {\n            if (!context.currentRunId) {\n              throw new Error(\"Attempted to stream without a runId\");\n            }\n            return {\n              conversationId: context.conversationId!,\n              runId: context.currentRunId,\n            };\n          },\n          onDone: [\n            {\n              actions: assign({ currentRunId: null, userRequestedStop: false }),\n              guard: ({ context }) => !!context.receivedEndOfStream || !!context.userRequestedStop,\n              target: \"idle\",\n            },\n            {\n              actions: assign({ currentRunId: null }),\n              target: \"stream_interrupted\",\n            },\n          ],\n          onError: [\n            {\n              actions: assign({ currentRunId: null, userRequestedStop: false }),\n              guard: ({ context }) => !!context.receivedEndOfStream || !!context.userRequestedStop,\n              target: \"idle\",\n            },\n            {\n              actions: assign({ currentRunId: null }),\n              target: \"stream_interrupted\",\n            },\n          ],\n          onSnapshot: {\n            actions: enqueueActions(({ context, enqueue, event }) => {\n              const chatEvent = event.snapshot.context;\n              if (!chatEvent) {\n                return;\n              }\n\n              if (chatEvent.content.chunkType === \"endOfStream\" && !context.receivedEndOfStream) {\n                enqueue.assign({ receivedEndOfStream: true });\n              }\n\n              const nextSnapshot = AgentConversation.reduceChatEvents(\n                context.conversationSnapshot,\n                [chatEvent],\n              );\n\n              enqueue.assign({\n                conversationSnapshot: nextSnapshot,\n              });\n              enqueue.emit({\n                chatEvent,\n                type: \"chatEventReceived\",\n              });\n            }),\n          },\n          src: \"streamRun\",\n        },\n        states: {\n          monitoring_replies: {\n            description: \"Attached to the run event stream\",\n            on: {\n              STOP_RUN: {\n                actions: assign({ userRequestedStop: true }),\n                target: \"stopping\",\n              },\n            },\n          },\n          stopping: {\n            description: \"Background cancelation of the run stream\",\n            invoke: {\n              input: ({ context }) => ({\n                conversationId: context.conversationId!,\n                runId: context.currentRunId!,\n              }),\n              onDone: {\n                actions: assign({ currentRunId: undefined, userRequestedStop: false }),\n                target: \"#agentConversation.idle\",\n              },\n              onError: {\n                actions: assign({ userRequestedStop: false }),\n                target: \"#agentConversation.retrieving_history\",\n              },\n              src: \"stopRun\",\n            },\n          },\n        },\n      },\n      submitting_message: {\n        description: \"Submitting message to start a new reply stream\",\n        invoke: {\n          input: ({ context, event }) => {\n            switch (event.type) {\n              case \"RETRY_MESSAGE\":\n                if (!context.messageAttempt) {\n                  throw new Error(\"Tried to resubmit nonexistent message\");\n                }\n                return {\n                  conversationId: context.conversationId!,\n                  message: context.messageAttempt,\n                };\n              case \"SUBMIT_USER_MESSAGE\":\n                return {\n                  conversationId: context.conversationId!,\n                  message: event.message,\n                };\n            }\n            throw new Error(`Unexpected event type received in submitting_message: ${event.type}`);\n          },\n          onDone: {\n            actions: assign({\n              // Add the message attempt into the snapshot manually\n              conversationSnapshot: ({ context, event }) => [\n                ...(context.conversationSnapshot ?? []),\n                {\n                  id: event.output.currentRunId,\n                  messages: new Map(),\n                  submittedUserMessage: context.messageAttempt || undefined,\n                  toolCalls: new Map(),\n                } satisfies ConversationExchange,\n              ],\n              currentRunId: ({ event }) => event.output.currentRunId,\n              messageAttempt: undefined,\n              ...STREAM_STATE_RESET,\n            }),\n            target: \"streaming\",\n          },\n          onError: {\n            actions: [\n              assign({\n                currentRunId: () => null,\n                lastError: ({ event }) => event.error,\n              }),\n              emit(({ event }) => ({\n                error: event.error as HttpError,\n                type: \"apiError\",\n              })),\n            ],\n            target: \"submitting_message_failed\",\n          },\n          src: \"startRun\",\n        },\n      },\n      submitting_message_failed: {\n        description: \"Failed to submit message\",\n        on: {\n          REFRESH_HISTORY: {\n            actions: assign({ lastError: undefined }),\n            target: \"retrieving_history\",\n          },\n          RETRY_MESSAGE: {\n            actions: assign({ lastError: undefined }),\n            target: \"submitting_message\",\n          },\n          SUBMIT_USER_MESSAGE: {\n            actions: assign({\n              lastError: undefined,\n              messageAttempt: ({ event }) => event.message,\n            }),\n            target: \"submitting_message\",\n          },\n        },\n      },\n      uninitialized: {\n        always: {\n          guard: ({ context }) => !!context.conversationId,\n          target: \"retrieving_history\",\n        },\n        description:\n          \"Entry state that the machine remains in when no conversation has been provided\",\n        on: {\n          SUBMIT_USER_MESSAGE: {\n            actions: assign({\n              messageAttempt: ({ event }) => event.message,\n            }),\n            target: \"creating_conversation\",\n          },\n        },\n      },\n    },\n  });\n\nexport type AgentConversationMachine = ReturnType<typeof createConversationMachine>;\n"]}

package/dist/enterprise/ai/conversations/methods/retrieveConversationHistory.d.ts

@@ -32,12 +32,11 @@ export declare const retrieveConversationHistory: (config: SonarV4Config) => (id
         };
         role: "agent";
     } | {
+        createdAt: import("temporal-polyfill").Temporal.Instant;
         conversationId: string;
         modelName: string;
         modelProviderId: string;
         runId: string;
-        createdAt: import("temporal-polyfill").Temporal.Instant;
-        id: string;
         content: {
             chunkType: "endOfStream";
         };
@@ -94,6 +93,20 @@ export declare const retrieveConversationHistory: (config: SonarV4Config) => (id
             text: string;
         };
         role: "agent";
+    } | {
+        conversationId: string;
+        modelName: string;
+        modelProviderId: string;
+        runId: string;
+        createdAt: import("temporal-polyfill").Temporal.Instant;
+        id: string;
+        content: {
+            callId: string;
+            chunkType: "sandboxStdout";
+            isFinal: boolean;
+            text: string;
+        };
+        role: "agent";
     } | {
         conversationId: string;
         modelName: string;

package/dist/enterprise/ai/conversations/reduceChatEvents.d.ts

@@ -6,7 +6,7 @@ export type AgentToolCall = {
     request: ChatEventWithChunkType<"toolRequest"> | undefined;
     result: ChatEventWithChunkType<"toolResponse"> | undefined;
 };
-export type ConversationExchangeMessage = ChatEventWithChunkType<"error"> | ChatEventWithChunkType<"model"> | ChatEventWithChunkType<"sandboxProgress"> | ChatEventWithChunkType<"userMessage">;
+export type ConversationExchangeMessage = ChatEventWithChunkType<"error"> | ChatEventWithChunkType<"model"> | ChatEventWithChunkType<"sandboxProgress"> | ChatEventWithChunkType<"sandboxStdout"> | ChatEventWithChunkType<"userMessage">;
 export type ConversationExchange = {
     id: string;
     messages: Map<string, ConversationExchangeMessage>;

package/dist/enterprise/ai/conversations/reduceChatEvents.js

@@ -40,8 +40,11 @@ function applyChatEventToConversationExchange(conversationExchange, chatEvent) {
         case "error":
         case "model":
         case "sandboxProgress":
+        case "sandboxStdout":
         case "userMessage": {
-            conversationExchange.messages.set(chatEvent.id, chatEvent);
+            // TypeScript doesn't narrow outer union via nested content.chunkType; cast is sound here
+            const event = chatEvent;
+            conversationExchange.messages.set(event.id, event);
             break;
         }
         case "toolRequest":

package/dist/enterprise/ai/conversations/reduceChatEvents.js.map

@@ -1 +1 @@
-{"version":3,"file":"reduceChatEvents.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/conversations/reduceChatEvents.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAc,MAAM,UAAU,CAAC;AA0B9C,SAAS,4BAA4B,CACnC,KAAmC,EACnC,SAAoB;IAEpB,IAAI,oBAAoB,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,KAAK,SAAS,CAAC,KAAK,CAAC,CAAC;IAEzF,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC1B,oBAAoB,GAAG;YACrB,EAAE,EAAE,SAAS,CAAC,KAAK;YACnB,QAAQ,EAAE,IAAI,GAAG,EAAE;YACnB,SAAS,EAAE,IAAI,GAAG,EAAE;SACrB,CAAC;QACF,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACjC,oBAAoB,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACjD,CAAC;IAED,oCAAoC,CAAC,oBAAqB,EAAE,SAAS,CAAC,CAAC;AACzE,CAAC;AAED,SAAS,oCAAoC,CAC3C,oBAAiD,EACjD,SAAoB;IAEpB,QAAQ,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;QACpC,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,KAAK,MAAM,CAAC,EAAE,QAAQ,CAAC,IAAI,oBAAoB,CAAC,SAAS,EAAE,CAAC;gBAC1D,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;oBACrB,QAAQ,CAAC,KAAK,GAAG,UAAU,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,MAAM;QACR,CAAC;QACD,KAAK,OAAO,CAAC;QACb,KAAK,OAAO,CAAC;QACb,KAAK,iBAAiB,CAAC;QACvB,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,oBAAoB,CAAC,QAAQ,CAAC,GAAG,CAC/B,SAAS,CAAC,EAAE,EACZ,SAA+C,CAChD,CAAC;YACF,MAAM;QACR,CAAC;QACD,KAAK,aAAa,CAAC;QACnB,KAAK,cAAc,CAAC,CAAC,CAAC;YACpB,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC;YAExC,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;gBAChD,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,EAAE;oBACzC,EAAE,EAAE,MAAM;oBACV,OAAO,EAAE,SAAS;oBAClB,MAAM,EAAE,SAAS;oBACjB,KAAK,EAAE,SAAS;iBACjB,CAAC,CAAC;YACL,CAAC;YAED,MAAM,QAAQ,GAAG,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAE,CAAC;YAE7D,QAAQ,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;gBACpC,KAAK,aAAa;oBAChB,QAAQ,CAAC,OAAO,GAAG,SAAoE,CAAC;oBACxF,MAAM;gBACR,KAAK,cAAc;oBACjB,QAAQ,CAAC,MAAM,GAAG,SAAqE,CAAC;oBACxF,IAAI,cAAc,IAAI,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;wBACrD,QAAQ,CAAC,KAAK,GAAG,OAAO,CAAC;oBAC3B,CAAC;yBAAM,CAAC;wBACN,QAAQ,CAAC,KAAK,GAAG,SAAS,CAAC;oBAC7B,CAAC;oBACD,MAAM;YACV,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,QAA+B,EAAE,EAAE,UAAuB;IACzF,OAAO,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;QAC7B,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,4BAA4B,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;QAChE,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { create, type Draft } from \"mutative\";\nimport type { ChatEvent, ChatEventWithChunkType } from \"../chat/chatEventSchema.ts\";\nimport type { UserChatMessage } from \"../chat/UserChatMessage.ts\";\n\nexport type AgentToolCall = {\n  id: string;\n  state: \"canceled\" | \"error\" | \"pending\" | \"success\";\n  request: ChatEventWithChunkType<\"toolRequest\"> | undefined;\n  result: ChatEventWithChunkType<\"toolResponse\"> | undefined;\n};\n\nexport type ConversationExchangeMessage =\n  | ChatEventWithChunkType<\"error\">\n  | ChatEventWithChunkType<\"model\">\n  | ChatEventWithChunkType<\"sandboxProgress\">\n  | ChatEventWithChunkType<\"userMessage\">;\n\nexport type ConversationExchange = {\n  id: string;\n  messages: Map<string, ConversationExchangeMessage>;\n  submittedUserMessage?: UserChatMessage;\n  toolCalls: Map<string, AgentToolCall>;\n};\n\nexport type ChatEventReducerState = ConversationExchange[];\n\nfunction applyChatEventToConversation(\n  draft: Draft<ChatEventReducerState>,\n  chatEvent: ChatEvent,\n): void {\n  let conversationExchange = draft.findLast((exchange) => exchange.id === chatEvent.runId);\n\n  if (!conversationExchange) {\n    conversationExchange = {\n      id: chatEvent.runId,\n      messages: new Map(),\n      toolCalls: new Map(),\n    };\n    draft.push(conversationExchange);\n    conversationExchange = draft[draft.length - 1];\n  }\n\n  applyChatEventToConversationExchange(conversationExchange!, chatEvent);\n}\n\nfunction applyChatEventToConversationExchange(\n  conversationExchange: Draft<ConversationExchange>,\n  chatEvent: ChatEvent,\n): void {\n  switch (chatEvent.content.chunkType) {\n    case \"endOfStream\": {\n      for (const [, toolCall] of conversationExchange.toolCalls) {\n        if (!toolCall.result) {\n          toolCall.state = \"canceled\";\n        }\n      }\n      break;\n    }\n    case \"error\":\n    case \"model\":\n    case \"sandboxProgress\":\n    case \"userMessage\": {\n      conversationExchange.messages.set(\n        chatEvent.id,\n        chatEvent as Draft<ConversationExchangeMessage>,\n      );\n      break;\n    }\n    case \"toolRequest\":\n    case \"toolResponse\": {\n      const callId = chatEvent.content.callId;\n\n      if (!conversationExchange.toolCalls.has(callId)) {\n        conversationExchange.toolCalls.set(callId, {\n          id: callId,\n          request: undefined,\n          result: undefined,\n          state: \"pending\",\n        });\n      }\n\n      const toolCall = conversationExchange.toolCalls.get(callId)!;\n\n      switch (chatEvent.content.chunkType) {\n        case \"toolRequest\":\n          toolCall.request = chatEvent as unknown as Draft<ChatEventWithChunkType<\"toolRequest\">>;\n          break;\n        case \"toolResponse\":\n          toolCall.result = chatEvent as unknown as Draft<ChatEventWithChunkType<\"toolResponse\">>;\n          if (\"errorMessage\" in toolCall.result.content.result) {\n            toolCall.state = \"error\";\n          } else {\n            toolCall.state = \"success\";\n          }\n          break;\n      }\n    }\n  }\n}\n\nexport function reduceChatEvents(state: ChatEventReducerState = [], chatEvents: ChatEvent[]) {\n  return create(state, (draft) => {\n    for (const chatEvent of chatEvents) {\n      applyChatEventToConversation(draft, Object.freeze(chatEvent));\n    }\n  });\n}\n"]}
+{"version":3,"file":"reduceChatEvents.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/conversations/reduceChatEvents.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAc,MAAM,UAAU,CAAC;AA2B9C,SAAS,4BAA4B,CACnC,KAAmC,EACnC,SAAoB;IAEpB,IAAI,oBAAoB,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,KAAK,SAAS,CAAC,KAAK,CAAC,CAAC;IAEzF,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC1B,oBAAoB,GAAG;YACrB,EAAE,EAAE,SAAS,CAAC,KAAK;YACnB,QAAQ,EAAE,IAAI,GAAG,EAAE;YACnB,SAAS,EAAE,IAAI,GAAG,EAAE;SACrB,CAAC;QACF,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACjC,oBAAoB,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACjD,CAAC;IAED,oCAAoC,CAAC,oBAAqB,EAAE,SAAS,CAAC,CAAC;AACzE,CAAC;AAED,SAAS,oCAAoC,CAC3C,oBAAiD,EACjD,SAAoB;IAEpB,QAAQ,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;QACpC,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,KAAK,MAAM,CAAC,EAAE,QAAQ,CAAC,IAAI,oBAAoB,CAAC,SAAS,EAAE,CAAC;gBAC1D,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;oBACrB,QAAQ,CAAC,KAAK,GAAG,UAAU,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,MAAM;QACR,CAAC;QACD,KAAK,OAAO,CAAC;QACb,KAAK,OAAO,CAAC;QACb,KAAK,iBAAiB,CAAC;QACvB,KAAK,eAAe,CAAC;QACrB,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,yFAAyF;YACzF,MAAM,KAAK,GAAG,SAA+C,CAAC;YAC9D,oBAAoB,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAA2C,CAAC,CAAC;YACzF,MAAM;QACR,CAAC;QACD,KAAK,aAAa,CAAC;QACnB,KAAK,cAAc,CAAC,CAAC,CAAC;YACpB,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC;YAExC,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;gBAChD,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,EAAE;oBACzC,EAAE,EAAE,MAAM;oBACV,OAAO,EAAE,SAAS;oBAClB,MAAM,EAAE,SAAS;oBACjB,KAAK,EAAE,SAAS;iBACjB,CAAC,CAAC;YACL,CAAC;YAED,MAAM,QAAQ,GAAG,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAE,CAAC;YAE7D,QAAQ,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;gBACpC,KAAK,aAAa;oBAChB,QAAQ,CAAC,OAAO,GAAG,SAAoE,CAAC;oBACxF,MAAM;gBACR,KAAK,cAAc;oBACjB,QAAQ,CAAC,MAAM,GAAG,SAAqE,CAAC;oBACxF,IAAI,cAAc,IAAI,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;wBACrD,QAAQ,CAAC,KAAK,GAAG,OAAO,CAAC;oBAC3B,CAAC;yBAAM,CAAC;wBACN,QAAQ,CAAC,KAAK,GAAG,SAAS,CAAC;oBAC7B,CAAC;oBACD,MAAM;YACV,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,QAA+B,EAAE,EAAE,UAAuB;IACzF,OAAO,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;QAC7B,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,4BAA4B,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;QAChE,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { create, type Draft } from \"mutative\";\nimport type { ChatEvent, ChatEventWithChunkType } from \"../chat/chatEventSchema.ts\";\nimport type { UserChatMessage } from \"../chat/UserChatMessage.ts\";\n\nexport type AgentToolCall = {\n  id: string;\n  state: \"canceled\" | \"error\" | \"pending\" | \"success\";\n  request: ChatEventWithChunkType<\"toolRequest\"> | undefined;\n  result: ChatEventWithChunkType<\"toolResponse\"> | undefined;\n};\n\nexport type ConversationExchangeMessage =\n  | ChatEventWithChunkType<\"error\">\n  | ChatEventWithChunkType<\"model\">\n  | ChatEventWithChunkType<\"sandboxProgress\">\n  | ChatEventWithChunkType<\"sandboxStdout\">\n  | ChatEventWithChunkType<\"userMessage\">;\n\nexport type ConversationExchange = {\n  id: string;\n  messages: Map<string, ConversationExchangeMessage>;\n  submittedUserMessage?: UserChatMessage;\n  toolCalls: Map<string, AgentToolCall>;\n};\n\nexport type ChatEventReducerState = ConversationExchange[];\n\nfunction applyChatEventToConversation(\n  draft: Draft<ChatEventReducerState>,\n  chatEvent: ChatEvent,\n): void {\n  let conversationExchange = draft.findLast((exchange) => exchange.id === chatEvent.runId);\n\n  if (!conversationExchange) {\n    conversationExchange = {\n      id: chatEvent.runId,\n      messages: new Map(),\n      toolCalls: new Map(),\n    };\n    draft.push(conversationExchange);\n    conversationExchange = draft[draft.length - 1];\n  }\n\n  applyChatEventToConversationExchange(conversationExchange!, chatEvent);\n}\n\nfunction applyChatEventToConversationExchange(\n  conversationExchange: Draft<ConversationExchange>,\n  chatEvent: ChatEvent,\n): void {\n  switch (chatEvent.content.chunkType) {\n    case \"endOfStream\": {\n      for (const [, toolCall] of conversationExchange.toolCalls) {\n        if (!toolCall.result) {\n          toolCall.state = \"canceled\";\n        }\n      }\n      break;\n    }\n    case \"error\":\n    case \"model\":\n    case \"sandboxProgress\":\n    case \"sandboxStdout\":\n    case \"userMessage\": {\n      // TypeScript doesn't narrow outer union via nested content.chunkType; cast is sound here\n      const event = chatEvent as Extract<ChatEvent, { id: string }>;\n      conversationExchange.messages.set(event.id, event as Draft<ConversationExchangeMessage>);\n      break;\n    }\n    case \"toolRequest\":\n    case \"toolResponse\": {\n      const callId = chatEvent.content.callId;\n\n      if (!conversationExchange.toolCalls.has(callId)) {\n        conversationExchange.toolCalls.set(callId, {\n          id: callId,\n          request: undefined,\n          result: undefined,\n          state: \"pending\",\n        });\n      }\n\n      const toolCall = conversationExchange.toolCalls.get(callId)!;\n\n      switch (chatEvent.content.chunkType) {\n        case \"toolRequest\":\n          toolCall.request = chatEvent as unknown as Draft<ChatEventWithChunkType<\"toolRequest\">>;\n          break;\n        case \"toolResponse\":\n          toolCall.result = chatEvent as unknown as Draft<ChatEventWithChunkType<\"toolResponse\">>;\n          if (\"errorMessage\" in toolCall.result.content.result) {\n            toolCall.state = \"error\";\n          } else {\n            toolCall.state = \"success\";\n          }\n          break;\n      }\n    }\n  }\n}\n\nexport function reduceChatEvents(state: ChatEventReducerState = [], chatEvents: ChatEvent[]) {\n  return create(state, (draft) => {\n    for (const chatEvent of chatEvents) {\n      applyChatEventToConversation(draft, Object.freeze(chatEvent));\n    }\n  });\n}\n"]}

package/dist-iife/cloud.js

@@ -17131,6 +17131,14 @@ var DremioCloud = (() => {
   var endOfStreamChunkSchema = object({
     chunkType: literal("endOfStream")
   });
+  var endOfStreamInputSchema = object({
+    chunkType: literal("endOfStream"),
+    conversationId: string2(),
+    createdAt: string2().check(iso_exports.datetime()),
+    modelName: string2(),
+    modelProviderId: string2(),
+    runId: string2()
+  });
   var interruptChunkSchema = object({
     chunkType: literal("interrupt")
   });
@@ -17150,6 +17158,14 @@ var DremioCloud = (() => {
     stepName: string2(),
     text: string2()
   });
+  var sandboxStdoutChunkSchema = object({
+    callId: string2().check(_trim(), _minLength(1)),
+    chunkType: literal("sandboxStdout"),
+    isFinal: boolean2(),
+    text: string2()
+  });
+  var sandboxStdoutContentSchema = sandboxStdoutChunkSchema;
+  var sandboxStdoutChatEventSchema = extend2(sandboxStdoutChunkSchema, chatEventSharedSchema);
   var toolPlanSchema = record(string2(), unknown());
   var toolRequestChunkSchema = object({
     arguments: record(string2(), unknown()),
@@ -17187,12 +17203,13 @@ var DremioCloud = (() => {
   ]);
   var chatEventInputSchema = discriminatedUnion("chunkType", [
     extend2(errorChunkV2Schema, chatEventSharedSchema),
-    extend2(endOfStreamChunkSchema, chatEventSharedSchema),
+    endOfStreamInputSchema,
     extend2(interruptChunkSchema, chatEventSharedSchema),
     extend2(conversationUpdateChunkSchema, chatEventSharedSchema),
     extend2(jobUpdateChunkSchema, chatEventSharedSchema),
     extend2(modelChunkSchema, chatEventSharedSchema),
     extend2(sandboxProgressChunkSchema, chatEventSharedSchema),
+    sandboxStdoutChatEventSchema,
     extend2(toolRequestChunkSchema, chatEventSharedSchema),
     extend2(toolResponseChunkSchema, chatEventSharedSchema),
     extend2(userMessageChunkSchema, chatEventSharedSchema)
@@ -17201,13 +17218,14 @@ var DremioCloud = (() => {
     createdAt: _instanceof(Xn.Instant),
     id: string2()
   });
+  var endOfStreamOutputSchema = omit2(chatEventOutputSharedSchema, { id: true });
   var chatEventOutputSchema = union([
     extend2(chatEventOutputSharedSchema, {
       content: conversationUpdateChunkSchema,
       role: literal("agent")
     }),
     extend2(chatEventOutputSharedSchema, { content: errorChunkV2Schema, role: literal("agent") }),
-    extend2(chatEventOutputSharedSchema, {
+    extend2(endOfStreamOutputSchema, {
       content: endOfStreamChunkSchema,
       role: literal("agent")
     }),
@@ -17224,6 +17242,10 @@ var DremioCloud = (() => {
       content: sandboxProgressChunkSchema,
       role: literal("agent")
     }),
+    extend2(chatEventOutputSharedSchema, {
+      content: sandboxStdoutContentSchema,
+      role: literal("agent")
+    }),
     extend2(chatEventOutputSharedSchema, {
       content: toolRequestChunkSchema,
       role: literal("agent")
@@ -17239,6 +17261,17 @@ var DremioCloud = (() => {
   ]);
   var chatEventCodec = codec(chatEventInputSchema, chatEventOutputSchema, {
     decode(v2) {
+      if (v2.chunkType === "endOfStream") {
+        return {
+          content: { chunkType: "endOfStream" },
+          conversationId: v2.conversationId,
+          createdAt: Xn.Instant.from(v2.createdAt),
+          modelName: v2.modelName,
+          modelProviderId: v2.modelProviderId,
+          role: "agent",
+          runId: v2.runId
+        };
+      }
       const sharedProperties = {
         conversationId: v2.conversationId,
         createdAt: Xn.Instant.from(v2.createdAt),
@@ -17268,14 +17301,6 @@ var DremioCloud = (() => {
             },
             role: "agent"
           };
-        case "endOfStream":
-          return {
-            ...sharedProperties,
-            content: {
-              chunkType: "endOfStream"
-            },
-            role: "agent"
-          };
         case "interrupt":
           return {
             ...sharedProperties,
@@ -17316,6 +17341,17 @@ var DremioCloud = (() => {
             },
             role: "agent"
           };
+        case "sandboxStdout":
+          return {
+            ...sharedProperties,
+            content: {
+              callId: v2.callId,
+              chunkType: "sandboxStdout",
+              isFinal: v2.isFinal,
+              text: v2.text
+            },
+            role: "agent"
+          };
         case "toolRequest":
           return {
             ...sharedProperties,
@@ -17352,10 +17388,21 @@ var DremioCloud = (() => {
       }
     },
     encode(v2) {
+      if (v2.content.chunkType === "endOfStream") {
+        return {
+          chunkType: "endOfStream",
+          conversationId: v2.conversationId,
+          createdAt: v2.createdAt.toString(),
+          modelName: v2.modelName,
+          modelProviderId: v2.modelProviderId,
+          runId: v2.runId
+        };
+      }
+      const id = v2.id;
       return {
         conversationId: v2.conversationId,
         createdAt: v2.createdAt.toString(),
-        messageId: v2.id,
+        messageId: id,
         modelName: v2.modelName,
         modelProviderId: v2.modelProviderId,
         runId: v2.runId,
@@ -19376,8 +19423,10 @@ var DremioCloud = (() => {
       case "error":
       case "model":
       case "sandboxProgress":
+      case "sandboxStdout":
       case "userMessage": {
-        conversationExchange.messages.set(chatEvent.id, chatEvent);
+        const event = chatEvent;
+        conversationExchange.messages.set(event.id, event);
         break;
       }
       case "toolRequest":
@@ -22876,6 +22925,14 @@ ${err.message}`);
   }
 
   // dist/enterprise/ai/conversations/createConversationMachine.js
+  var RETRIEVE_STATE_TIMEOUT_MS = 1e4;
+  var STREAM_RETRY_DELAY_MS = 2500;
+  var MAX_STREAM_RETRIES = 5;
+  var STREAM_STATE_RESET = {
+    receivedEndOfStream: false,
+    streamRetryCount: 0,
+    userRequestedStop: false
+  };
   var createConversationMachine = (config3) => setup({
     actors: {
       createConversation: fromPromise2(({ input }) => createConversation(config3)(input).then((result) => {
@@ -22897,9 +22954,13 @@ ${err.message}`);
         return result.value;
       })),
       retrieveState: fromPromise2(async ({ input, signal }) => {
+        const timeoutSignal = AbortSignal.any([
+          signal,
+          AbortSignal.timeout(RETRIEVE_STATE_TIMEOUT_MS)
+        ]);
         const [conversationResult, historyResult] = await Promise.all([
-          retrieveConversation(config3)(input.conversationId, { signal }),
-          retrieveConversationHistory(config3)(input.conversationId, { signal })
+          retrieveConversation(config3)(input.conversationId, { signal: timeoutSignal }),
+          retrieveConversationHistory(config3)(input.conversationId, { signal: timeoutSignal })
         ]);
         if (conversationResult.isErr()) {
           throw conversationResult.error;
@@ -22926,6 +22987,9 @@ ${err.message}`);
       })),
       streamRun: fromObservable(({ input }) => fromTextEventStream(({ signal }) => streamRunEvents(config3)({ conversationId: input.conversationId, runId: input.runId }, { signal })).pipe(map((event) => decode(chatEventCodec, JSON.parse(event.data)))))
     },
+    delays: {
+      STREAM_RETRY_DELAY: STREAM_RETRY_DELAY_MS
+    },
     types: {
       context: {},
       emitted: {},
@@ -22937,7 +23001,8 @@ ${err.message}`);
       conversationId: input.conversationId,
       conversationSnapshot: void 0,
       currentRunId: void 0,
-      messageAttempt: void 0
+      messageAttempt: void 0,
+      ...STREAM_STATE_RESET
     }),
     id: "agentConversation",
     initial: "uninitialized",
@@ -22964,7 +23029,8 @@ ${err.message}`);
                     }
                   ],
                   currentRunId: ({ event }) => event.output.currentRunId,
-                  messageAttempt: void 0
+                  messageAttempt: void 0,
+                  ...STREAM_STATE_RESET
                 }),
                 emit(({ event }) => ({
                   id: event.output.id,
@@ -23027,7 +23093,8 @@ ${err.message}`);
             {
               actions: assign({
                 conversationSnapshot: ({ context: context2, event }) => AgentConversation.reduceChatEvents(context2.conversationSnapshot, event.output.history.data),
-                currentRunId: ({ event }) => event.output.conversation.currentRunId
+                currentRunId: ({ event }) => event.output.conversation.currentRunId,
+                receivedEndOfStream: false
               }),
               guard: ({ event }) => !!event.output.conversation.currentRunId,
               target: "streaming"
@@ -23035,7 +23102,8 @@ ${err.message}`);
             {
               actions: assign({
                 conversationSnapshot: ({ context: context2, event }) => AgentConversation.reduceChatEvents(context2.conversationSnapshot, event.output.history.data),
-                currentRunId: ({ event }) => event.output.conversation.currentRunId
+                currentRunId: ({ event }) => event.output.conversation.currentRunId,
+                ...STREAM_STATE_RESET
               }),
               target: "idle"
             }
@@ -23044,10 +23112,16 @@ ${err.message}`);
             {
               guard: ({ event }) => {
                 const e2 = event.error;
+                if (e2?.name === "TimeoutError")
+                  return false;
                 return e2?.name === "AbortError" || e2?.message?.includes("aborted") || e2?.code === "ABORT_ERR";
               },
               target: "uninitialized"
             },
+            {
+              guard: ({ context: context2 }) => context2.streamRetryCount > 0,
+              target: "stream_interrupted"
+            },
             {
               actions: [
                 assign({
@@ -23064,6 +23138,30 @@ ${err.message}`);
           src: "retrieveState"
         }
       },
+      stream_interrupted: {
+        after: {
+          STREAM_RETRY_DELAY: [
+            {
+              actions: assign({
+                streamRetryCount: ({ context: context2 }) => context2.streamRetryCount + 1
+              }),
+              guard: ({ context: context2 }) => context2.streamRetryCount < MAX_STREAM_RETRIES,
+              target: "retrieving_history"
+            },
+            {
+              target: "stream_recovery_failed"
+            }
+          ]
+        }
+      },
+      stream_recovery_failed: {
+        on: {
+          REFRESH_HISTORY: {
+            actions: assign({ lastError: void 0, streamRetryCount: 0 }),
+            target: "retrieving_history"
+          }
+        }
+      },
       streaming: {
         description: "Manages monitoring and canceling the current run",
         initial: "monitoring_replies",
@@ -23077,20 +23175,37 @@ ${err.message}`);
               runId: context2.currentRunId
             };
           },
-          onDone: {
-            actions: assign({ currentRunId: null }),
-            target: "idle"
-          },
-          onError: {
-            actions: assign({ currentRunId: null }),
-            target: "idle"
-          },
+          onDone: [
+            {
+              actions: assign({ currentRunId: null, userRequestedStop: false }),
+              guard: ({ context: context2 }) => !!context2.receivedEndOfStream || !!context2.userRequestedStop,
+              target: "idle"
+            },
+            {
+              actions: assign({ currentRunId: null }),
+              target: "stream_interrupted"
+            }
+          ],
+          onError: [
+            {
+              actions: assign({ currentRunId: null, userRequestedStop: false }),
+              guard: ({ context: context2 }) => !!context2.receivedEndOfStream || !!context2.userRequestedStop,
+              target: "idle"
+            },
+            {
+              actions: assign({ currentRunId: null }),
+              target: "stream_interrupted"
+            }
+          ],
           onSnapshot: {
             actions: enqueueActions(({ context: context2, enqueue, event }) => {
               const chatEvent = event.snapshot.context;
               if (!chatEvent) {
                 return;
               }
+              if (chatEvent.content.chunkType === "endOfStream" && !context2.receivedEndOfStream) {
+                enqueue.assign({ receivedEndOfStream: true });
+              }
               const nextSnapshot = AgentConversation.reduceChatEvents(context2.conversationSnapshot, [chatEvent]);
               enqueue.assign({
                 conversationSnapshot: nextSnapshot
@@ -23108,6 +23223,7 @@ ${err.message}`);
             description: "Attached to the run event stream",
             on: {
               STOP_RUN: {
+                actions: assign({ userRequestedStop: true }),
                 target: "stopping"
               }
             }
@@ -23120,10 +23236,11 @@ ${err.message}`);
                 runId: context2.currentRunId
               }),
               onDone: {
-                actions: assign({ currentRunId: void 0 }),
+                actions: assign({ currentRunId: void 0, userRequestedStop: false }),
                 target: "#agentConversation.idle"
               },
               onError: {
+                actions: assign({ userRequestedStop: false }),
                 target: "#agentConversation.retrieving_history"
               },
               src: "stopRun"
@@ -23165,7 +23282,8 @@ ${err.message}`);
                 }
               ],
               currentRunId: ({ event }) => event.output.currentRunId,
-              messageAttempt: void 0
+              messageAttempt: void 0,
+              ...STREAM_STATE_RESET
             }),
             target: "streaming"
           },