@langchain/langgraph 0.2.56 → 0.2.58
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/constants.cjs +99 -12
- package/dist/constants.d.ts +52 -6
- package/dist/constants.js +97 -11
- package/dist/constants.js.map +1 -1
- package/dist/graph/graph.d.ts +1 -1
- package/dist/graph/state.cjs +90 -11
- package/dist/graph/state.d.ts +23 -5
- package/dist/graph/state.js +90 -11
- package/dist/graph/state.js.map +1 -1
- package/dist/graph/zod/index.cjs +7 -0
- package/dist/graph/zod/index.d.ts +2 -0
- package/dist/graph/zod/index.js +3 -0
- package/dist/graph/zod/index.js.map +1 -0
- package/dist/graph/zod/plugin.cjs +41 -0
- package/dist/graph/zod/plugin.d.ts +15 -0
- package/dist/graph/zod/plugin.js +39 -0
- package/dist/graph/zod/plugin.js.map +1 -0
- package/dist/graph/zod/schema.cjs +79 -0
- package/dist/graph/zod/schema.d.ts +9 -0
- package/dist/graph/zod/schema.js +73 -0
- package/dist/graph/zod/schema.js.map +1 -0
- package/dist/graph/zod/state.cjs +73 -0
- package/dist/graph/zod/state.d.ts +30 -0
- package/dist/graph/zod/state.js +64 -0
- package/dist/graph/zod/state.js.map +1 -0
- package/dist/pregel/algo.cjs +26 -3
- package/dist/pregel/algo.js +27 -4
- package/dist/pregel/algo.js.map +1 -1
- package/dist/pregel/debug.cjs +2 -1
- package/dist/pregel/debug.d.ts +1 -1
- package/dist/pregel/debug.js +2 -1
- package/dist/pregel/debug.js.map +1 -1
- package/dist/pregel/index.cjs +133 -27
- package/dist/pregel/index.d.ts +19 -1
- package/dist/pregel/index.js +135 -29
- package/dist/pregel/index.js.map +1 -1
- package/dist/pregel/loop.cjs +48 -24
- package/dist/pregel/loop.d.ts +1 -0
- package/dist/pregel/loop.js +48 -24
- package/dist/pregel/loop.js.map +1 -1
- package/dist/pregel/retry.cjs +3 -1
- package/dist/pregel/retry.js +3 -1
- package/dist/pregel/retry.js.map +1 -1
- package/dist/pregel/runner.cjs +26 -17
- package/dist/pregel/runner.d.ts +4 -0
- package/dist/pregel/runner.js +26 -17
- package/dist/pregel/runner.js.map +1 -1
- package/dist/pregel/utils/index.d.ts +2 -0
- package/dist/pregel/utils/index.js.map +1 -1
- package/package.json +22 -3
- package/zod.cjs +1 -0
- package/zod.d.cts +1 -0
- package/zod.d.ts +1 -0
- package/zod.js +1 -0
package/dist/pregel/debug.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"debug.js","sourceRoot":"","sources":["../../src/pregel/debug.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,KAAK,EAAa,SAAS,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC1E,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AAMjD,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AAWzD,MAAM,UAAU,GAAoB;IAClC,IAAI,EAAE;QACJ,KAAK,EAAE,UAAU;QACjB,GAAG,EAAE,SAAS;KACf;IACD,KAAK,EAAE;QACL,KAAK,EAAE,UAAU;QACjB,GAAG,EAAE,SAAS;KACf;IACD,MAAM,EAAE;QACN,KAAK,EAAE,YAAY;QACnB,GAAG,EAAE,SAAS;KACf;CACF,CAAC;AAEF;;GAEG;AACH,MAAM,IAAI,GAAG,CAAC,KAAoB,EAAE,IAAY,EAAU,EAAE,CAC1D,GAAG,KAAK,CAAC,KAAK,GAAG,IAAI,GAAG,KAAK,CAAC,GAAG,EAAE,CAAC;AAEtC,MAAM,UAAU,eAAe,CAC7B,IAAY,EACZ,QAA4C;IAE5C,OAAO,CAAC,GAAG,CACT;QACE,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,wBAAwB,CAAC,EAAE;QACpD,kBAAkB,IAAI,qBAAqB;QAC3C,KAAK,IAAI,CAAC,SAAS,CACjB,MAAM,CAAC,WAAW,CAAC,aAAa,CAAQ,QAAQ,CAAC,CAAC,EAClD,IAAI,EACJ,CAAC,CACF,EAAE;KACJ,CAAC,IAAI,CAAC,EAAE,CAAC,CACX,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,CAAC,aAAa,CACrB,QAA4C;AAC5C,8DAA8D;;IAE9D,KAAK,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;QACtD,IAAI;YACF,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;YAC5B,8DAA8D;SAC/D;QAAC,OAAO,KAAU,EAAE;YACnB,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAiB,CAAC,iBAAiB,EAAE;gBACtD,iCAAiC;gBACjC,SAAS;aACV;iBAAM;gBACL,MAAM,KAAK,CAAC,CAAC,sDAAsD;aACpE;SACF;KACF;AACH,CAAC;AAED,MAAM,SAAS,CAAC,CAAC,aAAa,CAC5B,IAAY,EACZ,KAA4C;IAE5C,MAAM,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACpC,KAAK,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,KAAK,EAAE;QACjE,IAAI,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,UAAU,CAAC;YAAE,SAAS;QAEjD,MAAM,UAAU,GAAG,MAAM;aACtB,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE;YACvB,OAAO,OAAO,KAAK,EAAE,IAAI,CAAC,KAAK,SAAS,CAAC;QAC3C,CAAC,CAAC;aACD,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE;YACb,OAAO,CAAC,CAAC;QACX,CAAC,CAAC,CAAC;QACL,MAAM;YACJ,IAAI,EAAE,MAAM;YACZ,SAAS,EAAE,EAAE;YACb,IAAI;YACJ,OAAO,EAAE;gBACP,EAAE;gBACF,IAAI;gBACJ,KAAK;gBACL,QAAQ;gBACR,UAAU;aACX;SACF,CAAC;KACH;AACH,CAAC;AAED,MAAM,SAAS,CAAC,CAAC,mBAAmB,CAIlC,IAAY,EACZ,KAAiE,EACjE,cAAgD;IAEhD,MAAM,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACpC,KAAK,MAAM,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,MAAM,CAAC,IAAI,KAAK,EAAE;QAClD,IAAI,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,UAAU,CAAC;YAAE,SAAS;QACjD,MAAM;YACJ,IAAI,EAAE,aAAa;YACnB,SAAS,EAAE,EAAE;YACb,IAAI;YACJ,OAAO,EAAE;gBACP,EAAE;gBACF,IAAI;gBACJ,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE;oBAClC,OAAO,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC;wBAClC,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,OAAO,CAAC;wBAClC,CAAC,CAAC,OAAO,KAAK,cAAc,CAAC;gBACjC,CAAC,CAAC;gBACF,UAAU,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;aACtE;SACF,CAAC;KACH;AACH,CAAC;AAED,MAAM,SAAS,CAAC,CAAC,kBAAkB,CAIjC,IAAY,EACZ,MAAsB,EACtB,QAAqC,EACrC,cAAiC,EACjC,QAA4B,EAC5B,KAA4C,EAC5C,aAAuC,EACvC,YAAwC;IAExC,SAAS,YAAY,CAAC,MAAsB;QAY1C,iDAAiD;QACjD,MAAM,QAAQ,GAEV,EAAE,CAAC;QAEP,IAAI,MAAM,CAAC,SAAS,IAAI,IAAI;YAAE,QAAQ,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QACpE,IAAI,MAAM,CAAC,YAAY,IAAI,IAAI;YAC7B,QAAQ,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;QAC9C,IAAI,MAAM,CAAC,cAAc,IAAI,IAAI;YAC/B,QAAQ,CAAC,eAAe,GAAG,MAAM,CAAC,cAAc,CAAC;QAEnD,IAAI,MAAM,CAAC,QAAQ,IAAI,IAAI;YAAE,QAAQ,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QACjE,IAAI,MAAM,CAAC,cAAc,IAAI,IAAI;YAC/B,QAAQ,CAAC,eAAe,GAAG,MAAM,CAAC,cAAc,CAAC;QACnD,IAAI,MAAM,CAAC,KAAK,IAAI,IAAI;YAAE,QAAQ,CAAC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC;QACzD,IAAI,MAAM,CAAC,OAAO,IAAI,IAAI;YAAE,QAAQ,CAAC,QAAQ,GAAG,MAAM,CAAC,OAAO,CAAC;QAC/D,IAAI,MAAM,CAAC,IAAI,IAAI,IAAI;YAAE,QAAQ,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QAErD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,CAAC,YAAY,EAAE,aAAa,CAAC;IACpD,MAAM,UAAU,GAAmD,EAAE,CAAC;IAEtE,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;QACxB,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzE,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,kBAAkB,CAAC;YAAE,SAAS;QAEnD,IAAI,MAAM,GAAG,GAAG,IAAI,CAAC,IAAc,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;QACjD,IAAI,QAAQ;YAAE,MAAM,GAAG,GAAG,QAAQ,IAAI,MAAM,EAAE,CAAC;QAE/C,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG;YACpB,YAAY,EAAE;gBACZ,SAAS,EAAE,MAAM,CAAC,YAAY,EAAE,SAAS;gBACzC,aAAa,EAAE,MAAM;aACtB;SACF,CAAC;KACH;IAED,MAAM,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACpC,MAAM;QACJ,IAAI,EAAE,YAAY;QAClB,SAAS,EAAE,EAAE;QACb,IAAI;QACJ,OAAO,EAAE;YACP,MAAM,EAAE,YAAY,CAAC,MAAM,CAAC;YAC5B,MAAM,EAAE,YAAY,CAAC,QAAQ,EAAE,cAAc,CAAC;YAC9C,QAAQ;YACR,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;YACpC,KAAK,EAAE,eAAe,CAAC,KAAK,EAAE,aAAa,EAAE,UAAU,CAAC;YACxD,YAAY,EAAE,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS;SACpE;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,eAAe,CAC7B,KAAsE,EACtE,aAAuC,EACvC,MAAuD;IAEvD,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAyB,EAAE;QAC/C,MAAM,KAAK,GAAG,aAAa,CAAC,IAAI,CAC9B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,IAAI,CAAC,KAAK,KAAK,CAC3C,EAAE,CAAC,CAAC,CAAC,CAAC;QAEP,MAAM,UAAU,GAAG,aAAa;aAC7B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE;YAClB,OAAO,EAAE,KAAK,IAAI,CAAC,EAAE,IAAI,CAAC,KAAK,SAAS,CAAC;QAC3C,CAAC,CAAC;aACD,GAAG,CAAC,CAAC,CAAC,EAAE,AAAD,EAAG,CAAC,CAAC,EAAE,EAAE;YACf,OAAO,CAAC,CAAC;QACX,CAAC,CAAgB,CAAC;QACpB,IAAI,KAAK,EAAE;YACT,OAAO;gBACL,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,IAAI,EAAE,IAAI,CAAC,IAAc;gBACzB,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,KAAK;gBACL,UAAU;aACX,CAAC;SACH;QACD,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,IAAI,EAAE,IAAI,CAAC,IAAc;YACzB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,UAAU;YACV,KAAK,EAAE,
|
|
1
|
+
{"version":3,"file":"debug.js","sourceRoot":"","sources":["../../src/pregel/debug.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,KAAK,EAAa,SAAS,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC1E,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AAMjD,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AAWzD,MAAM,UAAU,GAAoB;IAClC,IAAI,EAAE;QACJ,KAAK,EAAE,UAAU;QACjB,GAAG,EAAE,SAAS;KACf;IACD,KAAK,EAAE;QACL,KAAK,EAAE,UAAU;QACjB,GAAG,EAAE,SAAS;KACf;IACD,MAAM,EAAE;QACN,KAAK,EAAE,YAAY;QACnB,GAAG,EAAE,SAAS;KACf;CACF,CAAC;AAEF;;GAEG;AACH,MAAM,IAAI,GAAG,CAAC,KAAoB,EAAE,IAAY,EAAU,EAAE,CAC1D,GAAG,KAAK,CAAC,KAAK,GAAG,IAAI,GAAG,KAAK,CAAC,GAAG,EAAE,CAAC;AAEtC,MAAM,UAAU,eAAe,CAC7B,IAAY,EACZ,QAA4C;IAE5C,OAAO,CAAC,GAAG,CACT;QACE,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,wBAAwB,CAAC,EAAE;QACpD,kBAAkB,IAAI,qBAAqB;QAC3C,KAAK,IAAI,CAAC,SAAS,CACjB,MAAM,CAAC,WAAW,CAAC,aAAa,CAAQ,QAAQ,CAAC,CAAC,EAClD,IAAI,EACJ,CAAC,CACF,EAAE;KACJ,CAAC,IAAI,CAAC,EAAE,CAAC,CACX,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,CAAC,aAAa,CACrB,QAA4C;AAC5C,8DAA8D;;IAE9D,KAAK,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;QACtD,IAAI;YACF,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;YAC5B,8DAA8D;SAC/D;QAAC,OAAO,KAAU,EAAE;YACnB,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAiB,CAAC,iBAAiB,EAAE;gBACtD,iCAAiC;gBACjC,SAAS;aACV;iBAAM;gBACL,MAAM,KAAK,CAAC,CAAC,sDAAsD;aACpE;SACF;KACF;AACH,CAAC;AAED,MAAM,SAAS,CAAC,CAAC,aAAa,CAC5B,IAAY,EACZ,KAA4C;IAE5C,MAAM,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACpC,KAAK,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,KAAK,EAAE;QACjE,IAAI,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,UAAU,CAAC;YAAE,SAAS;QAEjD,MAAM,UAAU,GAAG,MAAM;aACtB,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE;YACvB,OAAO,OAAO,KAAK,EAAE,IAAI,CAAC,KAAK,SAAS,CAAC;QAC3C,CAAC,CAAC;aACD,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE;YACb,OAAO,CAAC,CAAC;QACX,CAAC,CAAC,CAAC;QACL,MAAM;YACJ,IAAI,EAAE,MAAM;YACZ,SAAS,EAAE,EAAE;YACb,IAAI;YACJ,OAAO,EAAE;gBACP,EAAE;gBACF,IAAI;gBACJ,KAAK;gBACL,QAAQ;gBACR,UAAU;aACX;SACF,CAAC;KACH;AACH,CAAC;AAED,MAAM,SAAS,CAAC,CAAC,mBAAmB,CAIlC,IAAY,EACZ,KAAiE,EACjE,cAAgD;IAEhD,MAAM,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACpC,KAAK,MAAM,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,MAAM,CAAC,IAAI,KAAK,EAAE;QAClD,IAAI,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,UAAU,CAAC;YAAE,SAAS;QACjD,MAAM;YACJ,IAAI,EAAE,aAAa;YACnB,SAAS,EAAE,EAAE;YACb,IAAI;YACJ,OAAO,EAAE;gBACP,EAAE;gBACF,IAAI;gBACJ,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE;oBAClC,OAAO,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC;wBAClC,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,OAAO,CAAC;wBAClC,CAAC,CAAC,OAAO,KAAK,cAAc,CAAC;gBACjC,CAAC,CAAC;gBACF,UAAU,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;aACtE;SACF,CAAC;KACH;AACH,CAAC;AAED,MAAM,SAAS,CAAC,CAAC,kBAAkB,CAIjC,IAAY,EACZ,MAAsB,EACtB,QAAqC,EACrC,cAAiC,EACjC,QAA4B,EAC5B,KAA4C,EAC5C,aAAuC,EACvC,YAAwC;IAExC,SAAS,YAAY,CAAC,MAAsB;QAY1C,iDAAiD;QACjD,MAAM,QAAQ,GAEV,EAAE,CAAC;QAEP,IAAI,MAAM,CAAC,SAAS,IAAI,IAAI;YAAE,QAAQ,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QACpE,IAAI,MAAM,CAAC,YAAY,IAAI,IAAI;YAC7B,QAAQ,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;QAC9C,IAAI,MAAM,CAAC,cAAc,IAAI,IAAI;YAC/B,QAAQ,CAAC,eAAe,GAAG,MAAM,CAAC,cAAc,CAAC;QAEnD,IAAI,MAAM,CAAC,QAAQ,IAAI,IAAI;YAAE,QAAQ,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QACjE,IAAI,MAAM,CAAC,cAAc,IAAI,IAAI;YAC/B,QAAQ,CAAC,eAAe,GAAG,MAAM,CAAC,cAAc,CAAC;QACnD,IAAI,MAAM,CAAC,KAAK,IAAI,IAAI;YAAE,QAAQ,CAAC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC;QACzD,IAAI,MAAM,CAAC,OAAO,IAAI,IAAI;YAAE,QAAQ,CAAC,QAAQ,GAAG,MAAM,CAAC,OAAO,CAAC;QAC/D,IAAI,MAAM,CAAC,IAAI,IAAI,IAAI;YAAE,QAAQ,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QAErD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,CAAC,YAAY,EAAE,aAAa,CAAC;IACpD,MAAM,UAAU,GAAmD,EAAE,CAAC;IAEtE,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;QACxB,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzE,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,kBAAkB,CAAC;YAAE,SAAS;QAEnD,IAAI,MAAM,GAAG,GAAG,IAAI,CAAC,IAAc,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;QACjD,IAAI,QAAQ;YAAE,MAAM,GAAG,GAAG,QAAQ,IAAI,MAAM,EAAE,CAAC;QAE/C,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG;YACpB,YAAY,EAAE;gBACZ,SAAS,EAAE,MAAM,CAAC,YAAY,EAAE,SAAS;gBACzC,aAAa,EAAE,MAAM;aACtB;SACF,CAAC;KACH;IAED,MAAM,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACpC,MAAM;QACJ,IAAI,EAAE,YAAY;QAClB,SAAS,EAAE,EAAE;QACb,IAAI;QACJ,OAAO,EAAE;YACP,MAAM,EAAE,YAAY,CAAC,MAAM,CAAC;YAC5B,MAAM,EAAE,YAAY,CAAC,QAAQ,EAAE,cAAc,CAAC;YAC9C,QAAQ;YACR,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;YACpC,KAAK,EAAE,eAAe,CAAC,KAAK,EAAE,aAAa,EAAE,UAAU,CAAC;YACxD,YAAY,EAAE,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS;SACpE;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,eAAe,CAC7B,KAAsE,EACtE,aAAuC,EACvC,MAAuD;IAEvD,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAyB,EAAE;QAC/C,MAAM,KAAK,GAAG,aAAa,CAAC,IAAI,CAC9B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,IAAI,CAAC,KAAK,KAAK,CAC3C,EAAE,CAAC,CAAC,CAAC,CAAC;QAEP,MAAM,UAAU,GAAG,aAAa;aAC7B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE;YAClB,OAAO,EAAE,KAAK,IAAI,CAAC,EAAE,IAAI,CAAC,KAAK,SAAS,CAAC;QAC3C,CAAC,CAAC;aACD,GAAG,CAAC,CAAC,CAAC,EAAE,AAAD,EAAG,CAAC,CAAC,EAAE,EAAE;YACf,OAAO,CAAC,CAAC;QACX,CAAC,CAAgB,CAAC;QACpB,IAAI,KAAK,EAAE;YACT,OAAO;gBACL,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,IAAI,EAAE,IAAI,CAAC,IAAc;gBACzB,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,KAAK;gBACL,UAAU;aACX,CAAC;SACH;QACD,MAAM,SAAS,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACpC,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,IAAI,EAAE,IAAI,CAAC,IAAc;YACzB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,UAAU;YACV,GAAG,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACzD,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,IAAY,EACZ,QAA8C,EAC9C,SAAmB;IAEnB,OAAO,CAAC,GAAG,CACT;QACE,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,IAAI,IAAI,cAAc,CAAC,EAAE;QAClD,oCAAoC,IAAI,YAAY;QACpD,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,QAAQ,EAAE,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;KAC3D,CAAC,IAAI,CAAC,EAAE,CAAC,CACX,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,cAAc,CAC5B,IAAY,EACZ,SAAgD;IAEhD,MAAM,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC;IAChC,OAAO,CAAC,GAAG,CACT;QACE,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,IAAI,IAAI,SAAS,CAAC,EAAE;QAC7C,yBAAyB,IAAI,SAAS,MAAM,QAC1C,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GACtB,YAAY;QACZ,SAAS;aACN,GAAG,CACF,CAAC,IAAI,EAAE,EAAE,CACP,KAAK,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,OAAO,IAAI,CAAC,SAAS,CACjE,IAAI,CAAC,KAAK,EACV,IAAI,EACJ,CAAC,CACF,EAAE,CACN;aACA,IAAI,CAAC,IAAI,CAAC;KACd,CAAC,IAAI,CAAC,EAAE,CAAC,CACX,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,eAAe,CAC7B,IAAY,EACZ,MAAsB,EACtB,SAAmB;IAEnB,8DAA8D;IAC9D,MAAM,SAAS,GAA0B,EAAE,CAAC;IAE5C,KAAK,MAAM,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,MAAM,EAAE;QACrC,IAAI,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE;YAC/B,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE;gBACvB,SAAS,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;aACzB;YACD,SAAS,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;SAChC;KACF;IAED,OAAO,CAAC,GAAG,CACT;QACE,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,IAAI,IAAI,UAAU,CAAC,EAAE;QAC9C,yBAAyB,IAAI,mBAC3B,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MACzB,WAAW,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,YAAY;QACrE,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC;aACtB,GAAG,CACF,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CACf,KAAK,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,IAAI;aAC1C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;aAC7B,IAAI,CAAC,IAAI,CAAC,EAAE,CAClB;aACA,IAAI,CAAC,IAAI,CAAC;KACd,CAAC,IAAI,CAAC,EAAE,CAAC,CACX,CAAC;AACJ,CAAC"}
|
package/dist/pregel/index.cjs
CHANGED
|
@@ -519,10 +519,11 @@ class Pregel extends runnables_1.Runnable {
|
|
|
519
519
|
* @param config - Configuration for preparing the snapshot
|
|
520
520
|
* @param saved - Optional saved checkpoint data
|
|
521
521
|
* @param subgraphCheckpointer - Optional checkpointer for subgraphs
|
|
522
|
+
* @param applyPendingWrites - Whether to apply pending writes to tasks and then to channels
|
|
522
523
|
* @returns A snapshot of the graph state
|
|
523
524
|
* @internal
|
|
524
525
|
*/
|
|
525
|
-
async _prepareStateSnapshot({ config, saved, subgraphCheckpointer, }) {
|
|
526
|
+
async _prepareStateSnapshot({ config, saved, subgraphCheckpointer, applyPendingWrites = false, }) {
|
|
526
527
|
if (saved === undefined) {
|
|
527
528
|
return {
|
|
528
529
|
values: {},
|
|
@@ -531,20 +532,40 @@ class Pregel extends runnables_1.Runnable {
|
|
|
531
532
|
tasks: [],
|
|
532
533
|
};
|
|
533
534
|
}
|
|
534
|
-
//
|
|
535
|
-
const { managed } = await this.prepareSpecs(config, {
|
|
535
|
+
// Create all channels
|
|
536
|
+
const { managed } = await this.prepareSpecs(config, {
|
|
537
|
+
skipManaged: true,
|
|
538
|
+
});
|
|
536
539
|
const channels = (0, base_js_1.emptyChannels)(this.channels, saved.checkpoint);
|
|
537
|
-
|
|
540
|
+
// Apply null writes first (from NULL_TASK_ID)
|
|
541
|
+
if (saved.pendingWrites?.length) {
|
|
542
|
+
const nullWrites = saved.pendingWrites
|
|
543
|
+
.filter(([taskId, _]) => taskId === constants_js_1.NULL_TASK_ID)
|
|
544
|
+
.map(([_, channel, value]) => [String(channel), value]);
|
|
545
|
+
if (nullWrites.length > 0) {
|
|
546
|
+
(0, algo_js_1._applyWrites)(saved.checkpoint, channels, [
|
|
547
|
+
{
|
|
548
|
+
name: constants_js_1.INPUT,
|
|
549
|
+
writes: nullWrites,
|
|
550
|
+
triggers: [],
|
|
551
|
+
},
|
|
552
|
+
]);
|
|
553
|
+
}
|
|
554
|
+
}
|
|
555
|
+
// Prepare next tasks
|
|
556
|
+
const nextTasks = Object.values((0, algo_js_1._prepareNextTasks)(saved.checkpoint, saved.pendingWrites, this.nodes, channels, managed, saved.config, true, { step: (saved.metadata?.step ?? -1) + 1, store: this.store }));
|
|
557
|
+
// Find subgraphs
|
|
538
558
|
const subgraphs = await (0, utils_js_1.gatherIterator)(this.getSubgraphsAsync());
|
|
539
559
|
const parentNamespace = saved.config.configurable?.checkpoint_ns ?? "";
|
|
540
560
|
const taskStates = {};
|
|
561
|
+
// Prepare task states for subgraphs
|
|
541
562
|
for (const task of nextTasks) {
|
|
542
563
|
const matchingSubgraph = subgraphs.find(([name]) => name === task.name);
|
|
543
564
|
if (!matchingSubgraph) {
|
|
544
565
|
continue;
|
|
545
566
|
}
|
|
546
567
|
// assemble checkpoint_ns for this task
|
|
547
|
-
let taskNs = `${task.name}${constants_js_1.CHECKPOINT_NAMESPACE_END}${task.id}`;
|
|
568
|
+
let taskNs = `${String(task.name)}${constants_js_1.CHECKPOINT_NAMESPACE_END}${task.id}`;
|
|
548
569
|
if (parentNamespace) {
|
|
549
570
|
taskNs = `${parentNamespace}${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}${taskNs}`;
|
|
550
571
|
}
|
|
@@ -560,37 +581,59 @@ class Pregel extends runnables_1.Runnable {
|
|
|
560
581
|
}
|
|
561
582
|
else {
|
|
562
583
|
// get the state of the subgraph
|
|
563
|
-
const
|
|
584
|
+
const subgraphConfig = {
|
|
564
585
|
configurable: {
|
|
565
586
|
[constants_js_1.CONFIG_KEY_CHECKPOINTER]: subgraphCheckpointer,
|
|
566
587
|
thread_id: saved.config.configurable?.thread_id,
|
|
567
588
|
checkpoint_ns: taskNs,
|
|
568
589
|
},
|
|
569
590
|
};
|
|
570
|
-
|
|
591
|
+
const pregel = matchingSubgraph[1];
|
|
592
|
+
taskStates[task.id] = await pregel.getState(subgraphConfig, {
|
|
571
593
|
subgraphs: true,
|
|
572
594
|
});
|
|
573
595
|
}
|
|
574
596
|
}
|
|
575
|
-
//
|
|
576
|
-
|
|
577
|
-
|
|
578
|
-
.map((
|
|
579
|
-
|
|
580
|
-
(
|
|
581
|
-
|
|
582
|
-
|
|
583
|
-
|
|
584
|
-
|
|
585
|
-
|
|
586
|
-
|
|
597
|
+
// Apply pending writes to tasks and then to channels if applyPendingWrites is true
|
|
598
|
+
if (applyPendingWrites && saved.pendingWrites?.length) {
|
|
599
|
+
// Map task IDs to task objects for easy lookup
|
|
600
|
+
const nextTaskById = Object.fromEntries(nextTasks.map((task) => [task.id, task]));
|
|
601
|
+
// Apply pending writes to the appropriate tasks
|
|
602
|
+
for (const [taskId, channel, value] of saved.pendingWrites) {
|
|
603
|
+
// Skip special channels and tasks not in nextTasks
|
|
604
|
+
if ([constants_js_1.ERROR, constants_js_1.INTERRUPT, langgraph_checkpoint_1.SCHEDULED].includes(channel)) {
|
|
605
|
+
continue;
|
|
606
|
+
}
|
|
607
|
+
if (!(taskId in nextTaskById)) {
|
|
608
|
+
continue;
|
|
609
|
+
}
|
|
610
|
+
// Add the write to the task
|
|
611
|
+
nextTaskById[taskId].writes.push([String(channel), value]);
|
|
612
|
+
}
|
|
613
|
+
// Apply writes from tasks that have writes
|
|
614
|
+
const tasksWithWrites = nextTasks.filter((task) => task.writes.length > 0);
|
|
615
|
+
if (tasksWithWrites.length > 0) {
|
|
616
|
+
(0, algo_js_1._applyWrites)(saved.checkpoint, channels, tasksWithWrites);
|
|
617
|
+
}
|
|
587
618
|
}
|
|
619
|
+
// Preserve thread_id from the config in metadata
|
|
620
|
+
let metadata = saved?.metadata;
|
|
621
|
+
if (metadata && saved?.config?.configurable?.thread_id) {
|
|
622
|
+
metadata = {
|
|
623
|
+
...metadata,
|
|
624
|
+
thread_id: saved.config.configurable.thread_id,
|
|
625
|
+
};
|
|
626
|
+
}
|
|
627
|
+
// Filter next tasks - only include tasks without writes
|
|
628
|
+
const nextList = nextTasks
|
|
629
|
+
.filter((task) => task.writes.length === 0)
|
|
630
|
+
.map((task) => task.name);
|
|
588
631
|
// assemble the state snapshot
|
|
589
632
|
return {
|
|
590
633
|
values: (0, io_js_1.readChannels)(channels, this.streamChannelsAsIs),
|
|
591
|
-
next:
|
|
634
|
+
next: nextList,
|
|
592
635
|
tasks: (0, debug_js_1.tasksWithWrites)(nextTasks, saved?.pendingWrites ?? [], taskStates),
|
|
593
|
-
metadata
|
|
636
|
+
metadata,
|
|
594
637
|
config: (0, index_js_1.patchCheckpointMap)(saved.config, saved.metadata),
|
|
595
638
|
createdAt: saved.checkpoint.ts,
|
|
596
639
|
parentConfig: saved.parentConfig,
|
|
@@ -630,6 +673,7 @@ class Pregel extends runnables_1.Runnable {
|
|
|
630
673
|
config: mergedConfig,
|
|
631
674
|
saved,
|
|
632
675
|
subgraphCheckpointer: options?.subgraphs ? checkpointer : undefined,
|
|
676
|
+
applyPendingWrites: !config.configurable?.checkpoint_id,
|
|
633
677
|
});
|
|
634
678
|
return snapshot;
|
|
635
679
|
}
|
|
@@ -684,6 +728,8 @@ class Pregel extends runnables_1.Runnable {
|
|
|
684
728
|
* from a list of updates, especially if a checkpoint
|
|
685
729
|
* is created as a result of multiple tasks.
|
|
686
730
|
*
|
|
731
|
+
* @internal The API might change in the future.
|
|
732
|
+
*
|
|
687
733
|
* @param startConfig - Configuration for the update
|
|
688
734
|
* @param updates - The list of updates to apply to graph state
|
|
689
735
|
* @returns Updated configuration
|
|
@@ -761,7 +807,7 @@ class Pregel extends runnables_1.Runnable {
|
|
|
761
807
|
const { managed } = await this.prepareSpecs(config, {
|
|
762
808
|
skipManaged: true,
|
|
763
809
|
});
|
|
764
|
-
if (values === null && asNode ===
|
|
810
|
+
if (values === null && asNode === constants_js_1.END) {
|
|
765
811
|
if (updates.length > 1) {
|
|
766
812
|
throw new errors_js_1.InvalidUpdateError(`Cannot apply multiple updates when clearing state`);
|
|
767
813
|
}
|
|
@@ -808,7 +854,7 @@ class Pregel extends runnables_1.Runnable {
|
|
|
808
854
|
}, {});
|
|
809
855
|
return (0, index_js_1.patchCheckpointMap)(nextConfig, saved ? saved.metadata : undefined);
|
|
810
856
|
}
|
|
811
|
-
if (values == null && asNode ===
|
|
857
|
+
if (values == null && asNode === constants_js_1.COPY) {
|
|
812
858
|
if (updates.length > 1) {
|
|
813
859
|
throw new errors_js_1.InvalidUpdateError(`Cannot copy checkpoint with multiple updates`);
|
|
814
860
|
}
|
|
@@ -820,6 +866,34 @@ class Pregel extends runnables_1.Runnable {
|
|
|
820
866
|
}, {});
|
|
821
867
|
return (0, index_js_1.patchCheckpointMap)(nextConfig, saved ? saved.metadata : undefined);
|
|
822
868
|
}
|
|
869
|
+
if (asNode === constants_js_1.INPUT) {
|
|
870
|
+
if (updates.length > 1) {
|
|
871
|
+
throw new errors_js_1.InvalidUpdateError(`Cannot apply multiple updates when updating as input`);
|
|
872
|
+
}
|
|
873
|
+
const inputWrites = await (0, utils_js_1.gatherIterator)((0, io_js_1.mapInput)(this.inputChannels, values));
|
|
874
|
+
if (inputWrites.length === 0) {
|
|
875
|
+
throw new errors_js_1.InvalidUpdateError(`Received no input writes for ${JSON.stringify(this.inputChannels, null, 2)}`);
|
|
876
|
+
}
|
|
877
|
+
// apply to checkpoint
|
|
878
|
+
(0, algo_js_1._applyWrites)(checkpoint, channels, [
|
|
879
|
+
{
|
|
880
|
+
name: constants_js_1.INPUT,
|
|
881
|
+
writes: inputWrites,
|
|
882
|
+
triggers: [],
|
|
883
|
+
},
|
|
884
|
+
], checkpointer.getNextVersion.bind(this.checkpointer));
|
|
885
|
+
// apply input write to channels
|
|
886
|
+
const nextStep = saved?.metadata?.step != null ? saved.metadata.step + 1 : -1;
|
|
887
|
+
const nextConfig = await checkpointer.put(checkpointConfig, (0, base_js_1.createCheckpoint)(checkpoint, channels, nextStep), {
|
|
888
|
+
source: "input",
|
|
889
|
+
step: nextStep,
|
|
890
|
+
writes: Object.fromEntries(inputWrites),
|
|
891
|
+
parents: saved?.metadata?.parents ?? {},
|
|
892
|
+
}, (0, index_js_1.getNewChannelVersions)(checkpointPreviousVersions, checkpoint.channel_versions));
|
|
893
|
+
// Store the writes
|
|
894
|
+
await checkpointer.putWrites(nextConfig, inputWrites, (0, langgraph_checkpoint_1.uuid5)(constants_js_1.INPUT, checkpoint.id));
|
|
895
|
+
return (0, index_js_1.patchCheckpointMap)(nextConfig, saved ? saved.metadata : undefined);
|
|
896
|
+
}
|
|
823
897
|
// apply pending writes, if not on specific checkpoint
|
|
824
898
|
if (config.configurable?.checkpoint_id === undefined &&
|
|
825
899
|
saved?.pendingWrites !== undefined &&
|
|
@@ -1163,6 +1237,24 @@ class Pregel extends runnables_1.Runnable {
|
|
|
1163
1237
|
managed,
|
|
1164
1238
|
};
|
|
1165
1239
|
}
|
|
1240
|
+
/**
|
|
1241
|
+
* Validates the input for the graph.
|
|
1242
|
+
* @param input - The input to validate
|
|
1243
|
+
* @returns The validated input
|
|
1244
|
+
* @internal
|
|
1245
|
+
*/
|
|
1246
|
+
async _validateInput(input) {
|
|
1247
|
+
return input;
|
|
1248
|
+
}
|
|
1249
|
+
/**
|
|
1250
|
+
* Validates the configurable options for the graph.
|
|
1251
|
+
* @param config - The configurable options to validate
|
|
1252
|
+
* @returns The validated configurable options
|
|
1253
|
+
* @internal
|
|
1254
|
+
*/
|
|
1255
|
+
async _validateConfigurable(config) {
|
|
1256
|
+
return config;
|
|
1257
|
+
}
|
|
1166
1258
|
/**
|
|
1167
1259
|
* Internal iterator used by stream() to generate state updates.
|
|
1168
1260
|
* This method handles the core logic of graph execution and streaming.
|
|
@@ -1184,9 +1276,11 @@ class Pregel extends runnables_1.Runnable {
|
|
|
1184
1276
|
inputConfig.configurable === undefined) {
|
|
1185
1277
|
throw new Error(`Checkpointer requires one or more of the following "configurable" keys: "thread_id", "checkpoint_ns", "checkpoint_id"`);
|
|
1186
1278
|
}
|
|
1279
|
+
const validInput = await this._validateInput(input);
|
|
1187
1280
|
const { runId, ...restConfig } = inputConfig;
|
|
1188
1281
|
// assign defaults
|
|
1189
1282
|
const [debug, streamMode, , outputKeys, config, interruptBefore, interruptAfter, checkpointer, store, streamModeSingle,] = this._defaults(restConfig);
|
|
1283
|
+
config.configurable = await this._validateConfigurable(config.configurable);
|
|
1190
1284
|
const stream = new stream_js_1.IterableReadableWritableStream({
|
|
1191
1285
|
modes: new Set(streamMode),
|
|
1192
1286
|
});
|
|
@@ -1211,7 +1305,14 @@ class Pregel extends runnables_1.Runnable {
|
|
|
1211
1305
|
config.writer = (chunk) => stream.push([[], "custom", chunk]);
|
|
1212
1306
|
}
|
|
1213
1307
|
const callbackManager = await (0, runnables_1.getCallbackManagerForConfig)(config);
|
|
1214
|
-
const runManager = await callbackManager?.handleChainStart(this.toJSON(),
|
|
1308
|
+
const runManager = await callbackManager?.handleChainStart(this.toJSON(), // chain
|
|
1309
|
+
(0, index_js_1._coerceToDict)(input, "input"), // inputs
|
|
1310
|
+
runId, // run_id
|
|
1311
|
+
undefined, // run_type
|
|
1312
|
+
undefined, // tags
|
|
1313
|
+
undefined, // metadata
|
|
1314
|
+
config?.runName ?? this.getName() // run_name
|
|
1315
|
+
);
|
|
1215
1316
|
const { channelSpecs, managed } = await this.prepareSpecs(config);
|
|
1216
1317
|
let loop;
|
|
1217
1318
|
let loopError;
|
|
@@ -1226,7 +1327,7 @@ class Pregel extends runnables_1.Runnable {
|
|
|
1226
1327
|
const createAndRunLoop = async () => {
|
|
1227
1328
|
try {
|
|
1228
1329
|
loop = await loop_js_1.PregelLoop.initialize({
|
|
1229
|
-
input,
|
|
1330
|
+
input: validInput,
|
|
1230
1331
|
config,
|
|
1231
1332
|
checkpointer,
|
|
1232
1333
|
nodes: this.nodes,
|
|
@@ -1243,7 +1344,7 @@ class Pregel extends runnables_1.Runnable {
|
|
|
1243
1344
|
});
|
|
1244
1345
|
const runner = new runner_js_1.PregelRunner({
|
|
1245
1346
|
loop,
|
|
1246
|
-
nodeFinished: config.configurable?.
|
|
1347
|
+
nodeFinished: config.configurable?.[constants_js_1.CONFIG_KEY_NODE_FINISHED],
|
|
1247
1348
|
});
|
|
1248
1349
|
if (options?.subgraphs) {
|
|
1249
1350
|
loop.config.configurable = {
|
|
@@ -1315,7 +1416,11 @@ class Pregel extends runnables_1.Runnable {
|
|
|
1315
1416
|
finally {
|
|
1316
1417
|
await runLoopPromise;
|
|
1317
1418
|
}
|
|
1318
|
-
await runManager?.handleChainEnd(loop?.output ?? {}
|
|
1419
|
+
await runManager?.handleChainEnd(loop?.output ?? {}, runId, // run_id
|
|
1420
|
+
undefined, // run_type
|
|
1421
|
+
undefined, // tags
|
|
1422
|
+
undefined // metadata
|
|
1423
|
+
);
|
|
1319
1424
|
}
|
|
1320
1425
|
/**
|
|
1321
1426
|
* Run the graph with a single input and config.
|
|
@@ -1360,6 +1465,7 @@ class Pregel extends runnables_1.Runnable {
|
|
|
1360
1465
|
(0, debug_js_1.printStepWrites)(step, writes, this.streamChannelsList);
|
|
1361
1466
|
}
|
|
1362
1467
|
},
|
|
1468
|
+
maxConcurrency: config.maxConcurrency,
|
|
1363
1469
|
signal: config.signal,
|
|
1364
1470
|
});
|
|
1365
1471
|
}
|
package/dist/pregel/index.d.ts
CHANGED
|
@@ -310,13 +310,15 @@ export declare class Pregel<Nodes extends StrRecord<string, PregelNode>, Channel
|
|
|
310
310
|
* @param config - Configuration for preparing the snapshot
|
|
311
311
|
* @param saved - Optional saved checkpoint data
|
|
312
312
|
* @param subgraphCheckpointer - Optional checkpointer for subgraphs
|
|
313
|
+
* @param applyPendingWrites - Whether to apply pending writes to tasks and then to channels
|
|
313
314
|
* @returns A snapshot of the graph state
|
|
314
315
|
* @internal
|
|
315
316
|
*/
|
|
316
|
-
protected _prepareStateSnapshot({ config, saved, subgraphCheckpointer, }: {
|
|
317
|
+
protected _prepareStateSnapshot({ config, saved, subgraphCheckpointer, applyPendingWrites, }: {
|
|
317
318
|
config: RunnableConfig;
|
|
318
319
|
saved?: CheckpointTuple;
|
|
319
320
|
subgraphCheckpointer?: BaseCheckpointSaver;
|
|
321
|
+
applyPendingWrites?: boolean;
|
|
320
322
|
}): Promise<StateSnapshot>;
|
|
321
323
|
/**
|
|
322
324
|
* Gets the current state of the graph.
|
|
@@ -350,6 +352,8 @@ export declare class Pregel<Nodes extends StrRecord<string, PregelNode>, Channel
|
|
|
350
352
|
* from a list of updates, especially if a checkpoint
|
|
351
353
|
* is created as a result of multiple tasks.
|
|
352
354
|
*
|
|
355
|
+
* @internal The API might change in the future.
|
|
356
|
+
*
|
|
353
357
|
* @param startConfig - Configuration for the update
|
|
354
358
|
* @param updates - The list of updates to apply to graph state
|
|
355
359
|
* @returns Updated configuration
|
|
@@ -448,6 +452,20 @@ export declare class Pregel<Nodes extends StrRecord<string, PregelNode>, Channel
|
|
|
448
452
|
channelSpecs: Record<string, BaseChannel<unknown, unknown, unknown>>;
|
|
449
453
|
managed: ManagedValueMapping;
|
|
450
454
|
}>;
|
|
455
|
+
/**
|
|
456
|
+
* Validates the input for the graph.
|
|
457
|
+
* @param input - The input to validate
|
|
458
|
+
* @returns The validated input
|
|
459
|
+
* @internal
|
|
460
|
+
*/
|
|
461
|
+
protected _validateInput(input: PregelInputType): Promise<any>;
|
|
462
|
+
/**
|
|
463
|
+
* Validates the configurable options for the graph.
|
|
464
|
+
* @param config - The configurable options to validate
|
|
465
|
+
* @returns The validated configurable options
|
|
466
|
+
* @internal
|
|
467
|
+
*/
|
|
468
|
+
protected _validateConfigurable(config: Partial<LangGraphRunnableConfig["configurable"]>): Promise<LangGraphRunnableConfig["configurable"]>;
|
|
451
469
|
/**
|
|
452
470
|
* Internal iterator used by stream() to generate state updates.
|
|
453
471
|
* This method handles the core logic of graph execution and streaming.
|