@langchain/langgraph-sdk 1.6.4 → 1.7.0

package/dist/react/index.js

@@ -1,5 +1,15 @@
-import { SubagentManager, calculateDepthFromNamespace, extractParentIdFromNamespace, extractToolCallIdFromNamespace, isSubagentNamespace } from "../ui/subagents.js";
-import { FetchStreamTransport } from "./stream.custom.js";
-import { useStream } from "./stream.js";
+import { toMessageDict } from "../ui/messages.js";
+import { FetchStreamTransport, SubagentManager, calculateDepthFromNamespace, extractParentIdFromNamespace, extractToolCallIdFromNamespace, isSubagentNamespace, useStream as useStream$1 } from "@langchain/react";
 
-export { FetchStreamTransport, SubagentManager, calculateDepthFromNamespace, extractParentIdFromNamespace, extractToolCallIdFromNamespace, isSubagentNamespace, useStream };
+//#region src/react/index.ts
+function useStream(...args) {
+	const options = args[0];
+	return useStream$1({
+		...options,
+		toMessage: toMessageDict
+	});
+}
+
+//#endregion
+export { FetchStreamTransport, SubagentManager, calculateDepthFromNamespace, extractParentIdFromNamespace, extractToolCallIdFromNamespace, isSubagentNamespace, useStream };
+//# sourceMappingURL=index.js.map

package/dist/react/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","names":["_useStream"],"sources":["../../src/react/index.ts"],"sourcesContent":["import {\n  useStream as _useStream,\n  FetchStreamTransport,\n} from \"@langchain/react\";\nimport type { BagTemplate } from \"@langchain/langgraph-sdk\";\nimport type {\n  ResolveStreamInterface,\n  ResolveStreamOptions,\n  UseStreamCustomOptions,\n  InferBag,\n  InferStateType,\n} from \"@langchain/langgraph-sdk/ui\";\nimport { toMessageDict } from \"../ui/messages.js\";\n\nexport { FetchStreamTransport };\n\n/**\n * Re-export of useStream that forces plain Message[] output\n * for backward compatibility with @langchain/langgraph-sdk/react users.\n */\nexport function useStream<\n  T = Record<string, unknown>,\n  Bag extends BagTemplate = BagTemplate\n>(\n  options: ResolveStreamOptions<T, InferBag<T, Bag>>\n): ResolveStreamInterface<T, InferBag<T, Bag>>;\n\nexport function useStream<\n  T = Record<string, unknown>,\n  Bag extends BagTemplate = BagTemplate\n>(\n  options: UseStreamCustomOptions<InferStateType<T>, InferBag<T, Bag>>\n): ResolveStreamInterface<T, InferBag<T, Bag>>;\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport function useStream(...args: any[]): any {\n  const options = args[0];\n  return _useStream({\n    ...options,\n    toMessage: toMessageDict,\n  });\n}\n\nexport type {\n  UseStream,\n  UseStreamCustom,\n  SubagentStream,\n  SubagentStreamInterface,\n} from \"@langchain/react\";\nexport type {\n  BaseStream,\n  UseAgentStream,\n  UseAgentStreamOptions,\n  UseDeepAgentStream,\n  UseDeepAgentStreamOptions,\n  ResolveStreamInterface,\n  ResolveStreamOptions,\n  InferStateType,\n  InferToolCalls,\n  InferSubagentStates,\n  InferNodeNames,\n  InferBag,\n} from \"@langchain/react\";\nexport type {\n  MessageMetadata,\n  UseStreamOptions,\n  UseStreamCustomOptions,\n  UseStreamTransport,\n  UseStreamThread,\n  GetToolCallsType,\n  AgentTypeConfigLike,\n  IsAgentLike,\n  ExtractAgentConfig,\n  InferAgentToolCalls,\n  SubagentToolCall,\n  SubagentStatus,\n  SubAgentLike,\n  CompiledSubAgentLike,\n  DeepAgentTypeConfigLike,\n  IsDeepAgentLike,\n  ExtractDeepAgentConfig,\n  ExtractSubAgentMiddleware,\n  InferDeepAgentSubagents,\n  InferSubagentByName,\n  InferSubagentState,\n  InferSubagentNames,\n  SubagentStateMap,\n  DefaultSubagentStates,\n  BaseSubagentState,\n} from \"@langchain/react\";\nexport type {\n  ToolCallWithResult,\n  ToolCallState,\n  DefaultToolCall,\n  ToolCallFromTool,\n  ToolCallsFromTools,\n} from \"@langchain/langgraph-sdk\";\nexport {\n  SubagentManager,\n  extractToolCallIdFromNamespace,\n  calculateDepthFromNamespace,\n  extractParentIdFromNamespace,\n  isSubagentNamespace,\n} from \"@langchain/react\";\n"],"mappings":";;;;AAmCA,SAAgB,UAAU,GAAG,MAAkB;CAC7C,MAAM,UAAU,KAAK;AACrB,QAAOA,YAAW;EAChB,GAAG;EACH,WAAW;EACZ,CAAC"}

package/dist/react-ui/index.cjs

@@ -1,15 +1,40 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
-const require_client = require('./client.cjs');
-const require_types = require('./types.cjs');
+const require_runtime = require('../_virtual/_rolldown/runtime.cjs');
+let _langchain_react_react_ui = require("@langchain/react/react-ui");
 
-//#region src/react-ui/index.ts
-require_client.bootstrapUiContext();
-
-//#endregion
-exports.LoadExternalComponent = require_client.LoadExternalComponent;
-exports.experimental_loadShare = require_client.experimental_loadShare;
-exports.isRemoveUIMessage = require_types.isRemoveUIMessage;
-exports.isUIMessage = require_types.isUIMessage;
-exports.uiMessageReducer = require_types.uiMessageReducer;
-exports.useStreamContext = require_client.useStreamContext;
-//# sourceMappingURL=index.cjs.map
+Object.defineProperty(exports, 'LoadExternalComponent', {
+  enumerable: true,
+  get: function () {
+    return _langchain_react_react_ui.LoadExternalComponent;
+  }
+});
+Object.defineProperty(exports, 'experimental_loadShare', {
+  enumerable: true,
+  get: function () {
+    return _langchain_react_react_ui.experimental_loadShare;
+  }
+});
+Object.defineProperty(exports, 'isRemoveUIMessage', {
+  enumerable: true,
+  get: function () {
+    return _langchain_react_react_ui.isRemoveUIMessage;
+  }
+});
+Object.defineProperty(exports, 'isUIMessage', {
+  enumerable: true,
+  get: function () {
+    return _langchain_react_react_ui.isUIMessage;
+  }
+});
+Object.defineProperty(exports, 'uiMessageReducer', {
+  enumerable: true,
+  get: function () {
+    return _langchain_react_react_ui.uiMessageReducer;
+  }
+});
+Object.defineProperty(exports, 'useStreamContext', {
+  enumerable: true,
+  get: function () {
+    return _langchain_react_react_ui.useStreamContext;
+  }
+});

package/dist/react-ui/index.d.cts

@@ -1,3 +1,2 @@
-import { RemoveUIMessage, UIMessage, isRemoveUIMessage, isUIMessage, uiMessageReducer } from "./types.cjs";
-import { LoadExternalComponent, experimental_loadShare, useStreamContext } from "./client.cjs";
+import { LoadExternalComponent, RemoveUIMessage, UIMessage, experimental_loadShare, isRemoveUIMessage, isUIMessage, uiMessageReducer, useStreamContext } from "@langchain/react/react-ui";
 export { LoadExternalComponent, type RemoveUIMessage, type UIMessage, experimental_loadShare, isRemoveUIMessage, isUIMessage, uiMessageReducer, useStreamContext };

package/dist/react-ui/index.d.ts

@@ -1,3 +1,2 @@
-import { RemoveUIMessage, UIMessage, isRemoveUIMessage, isUIMessage, uiMessageReducer } from "./types.js";
-import { LoadExternalComponent, experimental_loadShare, useStreamContext } from "./client.js";
+import { LoadExternalComponent, RemoveUIMessage, UIMessage, experimental_loadShare, isRemoveUIMessage, isUIMessage, uiMessageReducer, useStreamContext } from "@langchain/react/react-ui";
 export { LoadExternalComponent, type RemoveUIMessage, type UIMessage, experimental_loadShare, isRemoveUIMessage, isUIMessage, uiMessageReducer, useStreamContext };

package/dist/react-ui/index.js

@@ -1,9 +1,3 @@
-import { LoadExternalComponent, bootstrapUiContext, experimental_loadShare, useStreamContext } from "./client.js";
-import { isRemoveUIMessage, isUIMessage, uiMessageReducer } from "./types.js";
+import { LoadExternalComponent, experimental_loadShare, isRemoveUIMessage, isUIMessage, uiMessageReducer, useStreamContext } from "@langchain/react/react-ui";
 
-//#region src/react-ui/index.ts
-bootstrapUiContext();
-
-//#endregion
-export { LoadExternalComponent, experimental_loadShare, isRemoveUIMessage, isUIMessage, uiMessageReducer, useStreamContext };
-//# sourceMappingURL=index.js.map
+export { LoadExternalComponent, experimental_loadShare, isRemoveUIMessage, isUIMessage, uiMessageReducer, useStreamContext };

package/dist/react-ui/server/index.cjs

@@ -1,6 +1,16 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
-const require_types = require('../types.cjs');
-const require_server = require('./server.cjs');
+const require_runtime = require('../../_virtual/_rolldown/runtime.cjs');
+let _langchain_react_react_ui_server = require("@langchain/react/react-ui/server");
 
-exports.typedUi = require_server.typedUi;
-exports.uiMessageReducer = require_types.uiMessageReducer;
+Object.defineProperty(exports, 'typedUi', {
+  enumerable: true,
+  get: function () {
+    return _langchain_react_react_ui_server.typedUi;
+  }
+});
+Object.defineProperty(exports, 'uiMessageReducer', {
+  enumerable: true,
+  get: function () {
+    return _langchain_react_react_ui_server.uiMessageReducer;
+  }
+});

package/dist/react-ui/server/index.d.cts

@@ -1,3 +1,2 @@
-import { RemoveUIMessage, UIMessage, uiMessageReducer } from "../types.cjs";
-import { typedUi } from "./server.cjs";
+import { RemoveUIMessage, UIMessage, typedUi, uiMessageReducer } from "@langchain/react/react-ui/server";
 export { type RemoveUIMessage, type UIMessage, typedUi, uiMessageReducer };

package/dist/react-ui/server/index.d.ts

@@ -1,3 +1,2 @@
-import { RemoveUIMessage, UIMessage, uiMessageReducer } from "../types.js";
-import { typedUi } from "./server.js";
+import { RemoveUIMessage, UIMessage, typedUi, uiMessageReducer } from "@langchain/react/react-ui/server";
 export { type RemoveUIMessage, type UIMessage, typedUi, uiMessageReducer };

package/dist/react-ui/server/index.js

@@ -1,4 +1,3 @@
-import { uiMessageReducer } from "../types.js";
-import { typedUi } from "./server.js";
+import { typedUi, uiMessageReducer } from "@langchain/react/react-ui/server";
 
 export { typedUi, uiMessageReducer };

package/dist/schema.d.cts

@@ -165,6 +165,8 @@ interface Thread<ValuesType = DefaultValues, TInterruptValue = unknown> {
   config?: Config;
   /** The error for the thread (if status == "error") */
   error?: Optional<string | Record<string, unknown>>;
+  /** Extracted values from thread data. Only present when `extract` is used in search. */
+  extracted?: Record<string, unknown>;
 }
 interface Cron {
   /** The ID of the cron */

package/dist/schema.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"schema.d.cts","names":[],"sources":["../src/schema.ts"],"mappings":";;;KAEK,QAAA,MAAc,CAAA;AAAA,KAEP,SAAA;AAAA,KAQA,YAAA;AAAA,KAEP,iBAAA;AAAA,KAEO,YAAA;AAAA,KAEA,MAAA;EAhBQ;AAEpB;;;EAmBE,IAAA;EAnBmB;AAQrB;;;EAiBE,eAAA;EAjBsB;AAA6C;;EAsBnE,YAAA;IApBoB;;AAEtB;IAsBI,SAAA,GAAY,QAAA;IAtBQ;;;IA2BpB,aAAA,GAAgB,QAAA;IAAA,CAEf,GAAA;EAAA;AAAA;AAAA,UAIY,WAAA;EApBf;;;EAwBA,QAAA;EAVE;;;;EAgBF,YAAA,GAAe,WAAA;EAVA;;;;EAgBf,aAAA,GAAgB,WAAA;EAMD;;;;EAAf,YAAA,GAAe,WAAA;EAlBf;;;;EAwBA,aAAA,GAAgB,WAAA;EANhB;;;;EAYA,cAAA,GAAiB,WAAA;AAAA;AAAA,KAGP,SAAA,GAAY,MAAA,SAAe,WAAA;AAAA,KAE3B,QAAA,GAAW,QAAA;EACrB,MAAA;EAEA,IAAA;EAEA,MAAA,GAAS,MAAA;EAET,OAAA,GAAU,MAAA;EAAA,CAET,GAAA;AAAA;AAAA,UAGc,aAAA;;EAEf,YAAA;EAPU;EAUV,QAAA;EAjB6B;EAoB7B,MAAA,EAAQ,MAAA;EApBa;EAuBrB,OAAA;EApBA;EAuBA,UAAA;EArBS;EAwBT,QAAA,EAAU,QAAA;EAtBA;EAyBV,OAAA;EAvBY;EA0BZ,IAAA;EAvBe;EA0Bf,WAAA;AAAA;AAAA,UAGe,gBAAA,SAAyB,aAAA;AAAA,UAEzB,SAAA,SAAkB,aAAA;EAvBjC;EAyBA,UAAA;AAAA;AAAA,UAGe,wBAAA;EAnBf;EAqBA,UAAA,EAAY,SAAA;EAlBZ;EAoBA,IAAA;AAAA;AAAA,UAGe,cAAA;EACf,KAAA,EAAO,KAAA;IACL,EAAA;IACA,IAAA;IAEA,IAAA,GAAO,MAAA;IACP,QAAA;EAAA;EAEF,KAAA,EAAO,KAAA;IACL,MAAA;IACA,MAAA;IACA,IAAA;IACA,WAAA;EAAA;AAAA;;;;UAOa,SAAA;EAtBf;;;EA0BA,EAAA;EAvB6B;;;EA4B7B,KAAA,GAAQ,MAAA;EApBD;;;;EA0BP,IAAA;EAhCE;;;;EAsCF,SAAA;EAhCA;;;;EAsCA,EAAA;AAAA;AAAA,UAGe,MAAA,cAAoB,aAAA;EArCtB;EAuCb,SAAA;EAhCwB;EAmCxB,UAAA;EA1Bc;EA6Bd,UAAA;EAlCA;EAqCA,gBAAA;EAhCQ;EAmCR,QAAA,EAAU,QAAA;EAvBV;EA0BA,MAAA,EAAQ,YAAA;EApBN;EAuBF,MAAA,EAAQ,UAAA;EApBO;EAuBf,UAAA,EAAY,MAAA,SAAe,KAAA,CAAM,SAAA,CAAU,eAAA;EAvBtB;EA0BrB,MAAA,GAAS,MAAA;EAZC;EAeV,KAAA,GAAQ,QAAA,UAAkB,MAAA;AAAA;AAAA,UAGX,IAAA;EATkB;EAWjC,OAAA;EAXY;EAcZ,YAAA;EAR0B;EAW1B,SAAA,EAAW,QAAA;EAXK;EAchB,gBAAA;EA3CsB;EA8CtB,QAAA,EAAU,QAAA;EA9CwC;EAiDlD,QAAA;EA5CA;EA+CA,UAAA;EAzCA;EA4CA,UAAA;EAzCU;EA4CV,OAAA,EAAS,MAAA;EAzCD;EA4CR,OAAA,EAAS,QAAA;EAzCD;EA4CR,aAAA,EAAe,QAAA;EAzCH;EA4CZ,QAAA,EAAU,MAAA;EA5CuB;EA+CjC,OAAA;AAAA;AAAA,KAGU,aAAA,GAAgB,MAAA,sBAA4B,MAAA;AAAA,KAE5C,kBAAA,GAAqB,MAAA;AAAA,UAEhB,WAAA,cAAyB,aAAA;EAhDd;EAkD1B,MAAA,EAAQ,UAAA;EAlDwB;EAqDhC,IAAA;EAlDmB;EAqDnB,UAAA,EAAY,UAAA;EA7CD;EAgDX,QAAA,EAAU,QAAA;EA9BD;EAiCT,UAAA,EAAY,QAAA;EA3BG;EA8Bf,iBAAA,EAAmB,QAAA,CAAS,UAAA;EA3BZ;EA8BhB,KAAA,EAAO,KAAA,CAAM,UAAA;AAAA;AAAA,UAGE,UAAA,cACF,aAAA;EAGb,EAAA;EACA,IAAA;EACA,MAAA;EACA,KAAA,EAAO,QAAA;EACP,UAAA,EAAY,KAAA,CAAM,SAAA,CAAU,eAAA;EAC5B,UAAA,EAAY,QAAA,CAAS,UAAA;EACrB,KAAA,EAAO,QAAA,CAAS,WAAA,CAAY,UAAA;AAAA;AAAA,UAGb,GAAA;EAvDN;EAyDT,MAAA;EAtDS;EAyDT,SAAA;EAtDe;EAyDf,YAAA;EAtDU;EAyDV,UAAA;EAtDO;EAyDP,UAAA;EAtDU;EAyDV,MAAA,EAAQ,SAAA;;EAGR,QAAA,EAAU,QAAA;EA5DkD;EA+D5D,kBAAA,EAAoB,QAAA,CAAS,iBAAA;AAAA;AAAA,KAGnB,UAAA;EACV,SAAA;EACA,aAAA;EACA,aAAA,EAAe,QAAA;EACf,cAAA,EAAgB,QAAA,CAAS,MAAA;AAAA;AAAA,UAGV,qBAAA;EACf,UAAA;AAAA;AAAA,UAEe,IAAA;EACf,SAAA;EACA,GAAA;EAEA,KAAA,EAAO,MAAA;EACP,SAAA;EACA,SAAA;AAAA;AAAA,UAGe,UAAA,SAAmB,IAAA;EAClC,KAAA;AAAA;AAAA,UAEe,mBAAA;EACf,KAAA,EAAO,UAAA;AAAA;AAAA,UAGQ,kBAAA;EACf,OAAA;EACA,YAAA;EACA,SAAA;EACA,OAAA;EACA,OAAA,EAAS,MAAA;EACT,QAAA;EACA,aAAA;EACA,QAAA;EACA,UAAA;EACA,UAAA;EACA,QAAA,EAAU,QAAA;AAAA;AAAA,UAGK,2BAAA,SACP,IAAA,CAAK,kBAAA;EACb,SAAA;AAAA;AAAA,KAGU,eAAA;AAAA,KAOA,YAAA;AAAA,KAOA,UAAA;AAAA,KAQA,SAAA;AAAA,KAKA,oBAAA;AAAA,KAYA,iBAAA;AAAA,KAYA,cAAA;AAAA,KAWA,eAAA"}
+{"version":3,"file":"schema.d.cts","names":[],"sources":["../src/schema.ts"],"mappings":";;;KAEK,QAAA,MAAc,CAAA;AAAA,KAEP,SAAA;AAAA,KAQA,YAAA;AAAA,KAEP,iBAAA;AAAA,KAEO,YAAA;AAAA,KAEA,MAAA;EAhBQ;AAEpB;;;EAmBE,IAAA;EAnBmB;AAQrB;;;EAiBE,eAAA;EAjBsB;AAA6C;;EAsBnE,YAAA;IApBoB;;AAEtB;IAsBI,SAAA,GAAY,QAAA;IAtBQ;;;IA2BpB,aAAA,GAAgB,QAAA;IAAA,CAEf,GAAA;EAAA;AAAA;AAAA,UAIY,WAAA;EApBf;;;EAwBA,QAAA;EAVE;;;;EAgBF,YAAA,GAAe,WAAA;EAVA;;;;EAgBf,aAAA,GAAgB,WAAA;EAMD;;;;EAAf,YAAA,GAAe,WAAA;EAlBf;;;;EAwBA,aAAA,GAAgB,WAAA;EANhB;;;;EAYA,cAAA,GAAiB,WAAA;AAAA;AAAA,KAGP,SAAA,GAAY,MAAA,SAAe,WAAA;AAAA,KAE3B,QAAA,GAAW,QAAA;EACrB,MAAA;EAEA,IAAA;EAEA,MAAA,GAAS,MAAA;EAET,OAAA,GAAU,MAAA;EAAA,CAET,GAAA;AAAA;AAAA,UAGc,aAAA;;EAEf,YAAA;EAPU;EAUV,QAAA;EAjB6B;EAoB7B,MAAA,EAAQ,MAAA;EApBa;EAuBrB,OAAA;EApBA;EAuBA,UAAA;EArBS;EAwBT,QAAA,EAAU,QAAA;EAtBA;EAyBV,OAAA;EAvBY;EA0BZ,IAAA;EAvBe;EA0Bf,WAAA;AAAA;AAAA,UAGe,gBAAA,SAAyB,aAAA;AAAA,UAEzB,SAAA,SAAkB,aAAA;EAvBjC;EAyBA,UAAA;AAAA;AAAA,UAGe,wBAAA;EAnBf;EAqBA,UAAA,EAAY,SAAA;EAlBZ;EAoBA,IAAA;AAAA;AAAA,UAGe,cAAA;EACf,KAAA,EAAO,KAAA;IACL,EAAA;IACA,IAAA;IAEA,IAAA,GAAO,MAAA;IACP,QAAA;EAAA;EAEF,KAAA,EAAO,KAAA;IACL,MAAA;IACA,MAAA;IACA,IAAA;IACA,WAAA;EAAA;AAAA;;;;UAOa,SAAA;EAtBf;;;EA0BA,EAAA;EAvB6B;;;EA4B7B,KAAA,GAAQ,MAAA;EApBD;;;;EA0BP,IAAA;EAhCE;;;;EAsCF,SAAA;EAhCA;;;;EAsCA,EAAA;AAAA;AAAA,UAGe,MAAA,cAAoB,aAAA;EArCtB;EAuCb,SAAA;EAhCwB;EAmCxB,UAAA;EA1Bc;EA6Bd,UAAA;EAlCA;EAqCA,gBAAA;EAhCQ;EAmCR,QAAA,EAAU,QAAA;EAvBV;EA0BA,MAAA,EAAQ,YAAA;EApBN;EAuBF,MAAA,EAAQ,UAAA;EApBO;EAuBf,UAAA,EAAY,MAAA,SAAe,KAAA,CAAM,SAAA,CAAU,eAAA;EAvBtB;EA0BrB,MAAA,GAAS,MAAA;EAZC;EAeV,KAAA,GAAQ,QAAA,UAAkB,MAAA;EATlB;EAYR,SAAA,GAAY,MAAA;AAAA;AAAA,UAGG,IAAA;EAZH;EAcZ,OAAA;EAR0B;EAW1B,YAAA;EARY;EAWZ,SAAA,EAAW,QAAA;EAXO;EAclB,gBAAA;EA9CmC;EAiDnC,QAAA,EAAU,QAAA;EA/CV;EAkDA,QAAA;EA5CA;EA+CA,UAAA;EAzCA;EA4CA,UAAA;EAzCA;EA4CA,OAAA,EAAS,MAAA;EAzCT;EA4CA,OAAA,EAAS,QAAA;EAzCT;EA4CA,aAAA,EAAe,QAAA;EA5CY;EA+C3B,QAAA,EAAU,MAAA;EA/CiC;EAkD3C,OAAA;AAAA;AAAA,KAGU,aAAA,GAAgB,MAAA,sBAA4B,MAAA;AAAA,KAE5C,kBAAA,GAAqB,MAAA;AAAA,UAEhB,WAAA,cAAyB,aAAA;EAhDxC;EAkDA,MAAA,EAAQ,UAAA;EAlDU;EAqDlB,IAAA;EAlDe;EAqDf,UAAA,EAAY,UAAA;;EAGZ,QAAA,EAAU,QAAA;EA1CA;EA6CV,UAAA,EAAY,QAAA;EA9BH;EAiCT,iBAAA,EAAmB,QAAA,CAAS,UAAA;EA3BlB;EA8BV,KAAA,EAAO,KAAA,CAAM,UAAA;AAAA;AAAA,UAGE,UAAA,cACF,aAAA;EAGb,EAAA;EACA,IAAA;EACA,MAAA;EACA,KAAA,EAAO,QAAA;EACP,UAAA,EAAY,KAAA,CAAM,SAAA,CAAU,eAAA;EAC5B,UAAA,EAAY,QAAA,CAAS,UAAA;EACrB,KAAA,EAAO,QAAA,CAAS,WAAA,CAAY,UAAA;AAAA;AAAA,UAGb,GAAA;EAvDf;EAyDA,MAAA;EAtDA;EAyDA,SAAA;EAtDA;EAyDA,YAAA;EAtDA;EAyDA,UAAA;EAtDA;EAyDA,UAAA;EAzDO;EA4DP,MAAA,EAAQ,SAAA;EAzDe;EA4DvB,QAAA,EAAU,QAAA;EA5DgB;EA+D1B,kBAAA,EAAoB,QAAA,CAAS,iBAAA;AAAA;AAAA,KAGnB,UAAA;EACV,SAAA;EACA,aAAA;EACA,aAAA,EAAe,QAAA;EACf,cAAA,EAAgB,QAAA,CAAS,MAAA;AAAA;AAAA,UAGV,qBAAA;EACf,UAAA;AAAA;AAAA,UAEe,IAAA;EACf,SAAA;EACA,GAAA;EAEA,KAAA,EAAO,MAAA;EACP,SAAA;EACA,SAAA;AAAA;AAAA,UAGe,UAAA,SAAmB,IAAA;EAClC,KAAA;AAAA;AAAA,UAEe,mBAAA;EACf,KAAA,EAAO,UAAA;AAAA;AAAA,UAGQ,kBAAA;EACf,OAAA;EACA,YAAA;EACA,SAAA;EACA,OAAA;EACA,OAAA,EAAS,MAAA;EACT,QAAA;EACA,aAAA;EACA,QAAA;EACA,UAAA;EACA,UAAA;EACA,QAAA,EAAU,QAAA;AAAA;AAAA,UAGK,2BAAA,SACP,IAAA,CAAK,kBAAA;EACb,SAAA;AAAA;AAAA,KAGU,eAAA;AAAA,KAOA,YAAA;AAAA,KAOA,UAAA;AAAA,KAQA,SAAA;AAAA,KAKA,oBAAA;AAAA,KAYA,iBAAA;AAAA,KAYA,cAAA;AAAA,KAWA,eAAA"}

package/dist/schema.d.ts

@@ -165,6 +165,8 @@ interface Thread<ValuesType = DefaultValues, TInterruptValue = unknown> {
   config?: Config;
   /** The error for the thread (if status == "error") */
   error?: Optional<string | Record<string, unknown>>;
+  /** Extracted values from thread data. Only present when `extract` is used in search. */
+  extracted?: Record<string, unknown>;
 }
 interface Cron {
   /** The ID of the cron */

package/dist/schema.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"schema.d.ts","names":[],"sources":["../src/schema.ts"],"mappings":";;;KAEK,QAAA,MAAc,CAAA;AAAA,KAEP,SAAA;AAAA,KAQA,YAAA;AAAA,KAEP,iBAAA;AAAA,KAEO,YAAA;AAAA,KAEA,MAAA;EAhBQ;AAEpB;;;EAmBE,IAAA;EAnBmB;AAQrB;;;EAiBE,eAAA;EAjBsB;AAA6C;;EAsBnE,YAAA;IApBoB;;AAEtB;IAsBI,SAAA,GAAY,QAAA;IAtBQ;;;IA2BpB,aAAA,GAAgB,QAAA;IAAA,CAEf,GAAA;EAAA;AAAA;AAAA,UAIY,WAAA;EApBf;;;EAwBA,QAAA;EAVE;;;;EAgBF,YAAA,GAAe,WAAA;EAVA;;;;EAgBf,aAAA,GAAgB,WAAA;EAMD;;;;EAAf,YAAA,GAAe,WAAA;EAlBf;;;;EAwBA,aAAA,GAAgB,WAAA;EANhB;;;;EAYA,cAAA,GAAiB,WAAA;AAAA;AAAA,KAGP,SAAA,GAAY,MAAA,SAAe,WAAA;AAAA,KAE3B,QAAA,GAAW,QAAA;EACrB,MAAA;EAEA,IAAA;EAEA,MAAA,GAAS,MAAA;EAET,OAAA,GAAU,MAAA;EAAA,CAET,GAAA;AAAA;AAAA,UAGc,aAAA;;EAEf,YAAA;EAPU;EAUV,QAAA;EAjB6B;EAoB7B,MAAA,EAAQ,MAAA;EApBa;EAuBrB,OAAA;EApBA;EAuBA,UAAA;EArBS;EAwBT,QAAA,EAAU,QAAA;EAtBA;EAyBV,OAAA;EAvBY;EA0BZ,IAAA;EAvBe;EA0Bf,WAAA;AAAA;AAAA,UAGe,gBAAA,SAAyB,aAAA;AAAA,UAEzB,SAAA,SAAkB,aAAA;EAvBjC;EAyBA,UAAA;AAAA;AAAA,UAGe,wBAAA;EAnBf;EAqBA,UAAA,EAAY,SAAA;EAlBZ;EAoBA,IAAA;AAAA;AAAA,UAGe,cAAA;EACf,KAAA,EAAO,KAAA;IACL,EAAA;IACA,IAAA;IAEA,IAAA,GAAO,MAAA;IACP,QAAA;EAAA;EAEF,KAAA,EAAO,KAAA;IACL,MAAA;IACA,MAAA;IACA,IAAA;IACA,WAAA;EAAA;AAAA;;;;UAOa,SAAA;EAtBf;;;EA0BA,EAAA;EAvB6B;;;EA4B7B,KAAA,GAAQ,MAAA;EApBD;;;;EA0BP,IAAA;EAhCE;;;;EAsCF,SAAA;EAhCA;;;;EAsCA,EAAA;AAAA;AAAA,UAGe,MAAA,cAAoB,aAAA;EArCtB;EAuCb,SAAA;EAhCwB;EAmCxB,UAAA;EA1Bc;EA6Bd,UAAA;EAlCA;EAqCA,gBAAA;EAhCQ;EAmCR,QAAA,EAAU,QAAA;EAvBV;EA0BA,MAAA,EAAQ,YAAA;EApBN;EAuBF,MAAA,EAAQ,UAAA;EApBO;EAuBf,UAAA,EAAY,MAAA,SAAe,KAAA,CAAM,SAAA,CAAU,eAAA;EAvBtB;EA0BrB,MAAA,GAAS,MAAA;EAZC;EAeV,KAAA,GAAQ,QAAA,UAAkB,MAAA;AAAA;AAAA,UAGX,IAAA;EATkB;EAWjC,OAAA;EAXY;EAcZ,YAAA;EAR0B;EAW1B,SAAA,EAAW,QAAA;EAXK;EAchB,gBAAA;EA3CsB;EA8CtB,QAAA,EAAU,QAAA;EA9CwC;EAiDlD,QAAA;EA5CA;EA+CA,UAAA;EAzCA;EA4CA,UAAA;EAzCU;EA4CV,OAAA,EAAS,MAAA;EAzCD;EA4CR,OAAA,EAAS,QAAA;EAzCD;EA4CR,aAAA,EAAe,QAAA;EAzCH;EA4CZ,QAAA,EAAU,MAAA;EA5CuB;EA+CjC,OAAA;AAAA;AAAA,KAGU,aAAA,GAAgB,MAAA,sBAA4B,MAAA;AAAA,KAE5C,kBAAA,GAAqB,MAAA;AAAA,UAEhB,WAAA,cAAyB,aAAA;EAhDd;EAkD1B,MAAA,EAAQ,UAAA;EAlDwB;EAqDhC,IAAA;EAlDmB;EAqDnB,UAAA,EAAY,UAAA;EA7CD;EAgDX,QAAA,EAAU,QAAA;EA9BD;EAiCT,UAAA,EAAY,QAAA;EA3BG;EA8Bf,iBAAA,EAAmB,QAAA,CAAS,UAAA;EA3BZ;EA8BhB,KAAA,EAAO,KAAA,CAAM,UAAA;AAAA;AAAA,UAGE,UAAA,cACF,aAAA;EAGb,EAAA;EACA,IAAA;EACA,MAAA;EACA,KAAA,EAAO,QAAA;EACP,UAAA,EAAY,KAAA,CAAM,SAAA,CAAU,eAAA;EAC5B,UAAA,EAAY,QAAA,CAAS,UAAA;EACrB,KAAA,EAAO,QAAA,CAAS,WAAA,CAAY,UAAA;AAAA;AAAA,UAGb,GAAA;EAvDN;EAyDT,MAAA;EAtDS;EAyDT,SAAA;EAtDe;EAyDf,YAAA;EAtDU;EAyDV,UAAA;EAtDO;EAyDP,UAAA;EAtDU;EAyDV,MAAA,EAAQ,SAAA;;EAGR,QAAA,EAAU,QAAA;EA5DkD;EA+D5D,kBAAA,EAAoB,QAAA,CAAS,iBAAA;AAAA;AAAA,KAGnB,UAAA;EACV,SAAA;EACA,aAAA;EACA,aAAA,EAAe,QAAA;EACf,cAAA,EAAgB,QAAA,CAAS,MAAA;AAAA;AAAA,UAGV,qBAAA;EACf,UAAA;AAAA;AAAA,UAEe,IAAA;EACf,SAAA;EACA,GAAA;EAEA,KAAA,EAAO,MAAA;EACP,SAAA;EACA,SAAA;AAAA;AAAA,UAGe,UAAA,SAAmB,IAAA;EAClC,KAAA;AAAA;AAAA,UAEe,mBAAA;EACf,KAAA,EAAO,UAAA;AAAA;AAAA,UAGQ,kBAAA;EACf,OAAA;EACA,YAAA;EACA,SAAA;EACA,OAAA;EACA,OAAA,EAAS,MAAA;EACT,QAAA;EACA,aAAA;EACA,QAAA;EACA,UAAA;EACA,UAAA;EACA,QAAA,EAAU,QAAA;AAAA;AAAA,UAGK,2BAAA,SACP,IAAA,CAAK,kBAAA;EACb,SAAA;AAAA;AAAA,KAGU,eAAA;AAAA,KAOA,YAAA;AAAA,KAOA,UAAA;AAAA,KAQA,SAAA;AAAA,KAKA,oBAAA;AAAA,KAYA,iBAAA;AAAA,KAYA,cAAA;AAAA,KAWA,eAAA"}
+{"version":3,"file":"schema.d.ts","names":[],"sources":["../src/schema.ts"],"mappings":";;;KAEK,QAAA,MAAc,CAAA;AAAA,KAEP,SAAA;AAAA,KAQA,YAAA;AAAA,KAEP,iBAAA;AAAA,KAEO,YAAA;AAAA,KAEA,MAAA;EAhBQ;AAEpB;;;EAmBE,IAAA;EAnBmB;AAQrB;;;EAiBE,eAAA;EAjBsB;AAA6C;;EAsBnE,YAAA;IApBoB;;AAEtB;IAsBI,SAAA,GAAY,QAAA;IAtBQ;;;IA2BpB,aAAA,GAAgB,QAAA;IAAA,CAEf,GAAA;EAAA;AAAA;AAAA,UAIY,WAAA;EApBf;;;EAwBA,QAAA;EAVE;;;;EAgBF,YAAA,GAAe,WAAA;EAVA;;;;EAgBf,aAAA,GAAgB,WAAA;EAMD;;;;EAAf,YAAA,GAAe,WAAA;EAlBf;;;;EAwBA,aAAA,GAAgB,WAAA;EANhB;;;;EAYA,cAAA,GAAiB,WAAA;AAAA;AAAA,KAGP,SAAA,GAAY,MAAA,SAAe,WAAA;AAAA,KAE3B,QAAA,GAAW,QAAA;EACrB,MAAA;EAEA,IAAA;EAEA,MAAA,GAAS,MAAA;EAET,OAAA,GAAU,MAAA;EAAA,CAET,GAAA;AAAA;AAAA,UAGc,aAAA;;EAEf,YAAA;EAPU;EAUV,QAAA;EAjB6B;EAoB7B,MAAA,EAAQ,MAAA;EApBa;EAuBrB,OAAA;EApBA;EAuBA,UAAA;EArBS;EAwBT,QAAA,EAAU,QAAA;EAtBA;EAyBV,OAAA;EAvBY;EA0BZ,IAAA;EAvBe;EA0Bf,WAAA;AAAA;AAAA,UAGe,gBAAA,SAAyB,aAAA;AAAA,UAEzB,SAAA,SAAkB,aAAA;EAvBjC;EAyBA,UAAA;AAAA;AAAA,UAGe,wBAAA;EAnBf;EAqBA,UAAA,EAAY,SAAA;EAlBZ;EAoBA,IAAA;AAAA;AAAA,UAGe,cAAA;EACf,KAAA,EAAO,KAAA;IACL,EAAA;IACA,IAAA;IAEA,IAAA,GAAO,MAAA;IACP,QAAA;EAAA;EAEF,KAAA,EAAO,KAAA;IACL,MAAA;IACA,MAAA;IACA,IAAA;IACA,WAAA;EAAA;AAAA;;;;UAOa,SAAA;EAtBf;;;EA0BA,EAAA;EAvB6B;;;EA4B7B,KAAA,GAAQ,MAAA;EApBD;;;;EA0BP,IAAA;EAhCE;;;;EAsCF,SAAA;EAhCA;;;;EAsCA,EAAA;AAAA;AAAA,UAGe,MAAA,cAAoB,aAAA;EArCtB;EAuCb,SAAA;EAhCwB;EAmCxB,UAAA;EA1Bc;EA6Bd,UAAA;EAlCA;EAqCA,gBAAA;EAhCQ;EAmCR,QAAA,EAAU,QAAA;EAvBV;EA0BA,MAAA,EAAQ,YAAA;EApBN;EAuBF,MAAA,EAAQ,UAAA;EApBO;EAuBf,UAAA,EAAY,MAAA,SAAe,KAAA,CAAM,SAAA,CAAU,eAAA;EAvBtB;EA0BrB,MAAA,GAAS,MAAA;EAZC;EAeV,KAAA,GAAQ,QAAA,UAAkB,MAAA;EATlB;EAYR,SAAA,GAAY,MAAA;AAAA;AAAA,UAGG,IAAA;EAZH;EAcZ,OAAA;EAR0B;EAW1B,YAAA;EARY;EAWZ,SAAA,EAAW,QAAA;EAXO;EAclB,gBAAA;EA9CmC;EAiDnC,QAAA,EAAU,QAAA;EA/CV;EAkDA,QAAA;EA5CA;EA+CA,UAAA;EAzCA;EA4CA,UAAA;EAzCA;EA4CA,OAAA,EAAS,MAAA;EAzCT;EA4CA,OAAA,EAAS,QAAA;EAzCT;EA4CA,aAAA,EAAe,QAAA;EA5CY;EA+C3B,QAAA,EAAU,MAAA;EA/CiC;EAkD3C,OAAA;AAAA;AAAA,KAGU,aAAA,GAAgB,MAAA,sBAA4B,MAAA;AAAA,KAE5C,kBAAA,GAAqB,MAAA;AAAA,UAEhB,WAAA,cAAyB,aAAA;EAhDxC;EAkDA,MAAA,EAAQ,UAAA;EAlDU;EAqDlB,IAAA;EAlDe;EAqDf,UAAA,EAAY,UAAA;;EAGZ,QAAA,EAAU,QAAA;EA1CA;EA6CV,UAAA,EAAY,QAAA;EA9BH;EAiCT,iBAAA,EAAmB,QAAA,CAAS,UAAA;EA3BlB;EA8BV,KAAA,EAAO,KAAA,CAAM,UAAA;AAAA;AAAA,UAGE,UAAA,cACF,aAAA;EAGb,EAAA;EACA,IAAA;EACA,MAAA;EACA,KAAA,EAAO,QAAA;EACP,UAAA,EAAY,KAAA,CAAM,SAAA,CAAU,eAAA;EAC5B,UAAA,EAAY,QAAA,CAAS,UAAA;EACrB,KAAA,EAAO,QAAA,CAAS,WAAA,CAAY,UAAA;AAAA;AAAA,UAGb,GAAA;EAvDf;EAyDA,MAAA;EAtDA;EAyDA,SAAA;EAtDA;EAyDA,YAAA;EAtDA;EAyDA,UAAA;EAtDA;EAyDA,UAAA;EAzDO;EA4DP,MAAA,EAAQ,SAAA;EAzDe;EA4DvB,QAAA,EAAU,QAAA;EA5DgB;EA+D1B,kBAAA,EAAoB,QAAA,CAAS,iBAAA;AAAA;AAAA,KAGnB,UAAA;EACV,SAAA;EACA,aAAA;EACA,aAAA,EAAe,QAAA;EACf,cAAA,EAAgB,QAAA,CAAS,MAAA;AAAA;AAAA,UAGV,qBAAA;EACf,UAAA;AAAA;AAAA,UAEe,IAAA;EACf,SAAA;EACA,GAAA;EAEA,KAAA,EAAO,MAAA;EACP,SAAA;EACA,SAAA;AAAA;AAAA,UAGe,UAAA,SAAmB,IAAA;EAClC,KAAA;AAAA;AAAA,UAEe,mBAAA;EACf,KAAA,EAAO,UAAA;AAAA;AAAA,UAGQ,kBAAA;EACf,OAAA;EACA,YAAA;EACA,SAAA;EACA,OAAA;EACA,OAAA,EAAS,MAAA;EACT,QAAA;EACA,aAAA;EACA,QAAA;EACA,UAAA;EACA,UAAA;EACA,QAAA,EAAU,QAAA;AAAA;AAAA,UAGK,2BAAA,SACP,IAAA,CAAK,kBAAA;EACb,SAAA;AAAA;AAAA,KAGU,eAAA;AAAA,KAOA,YAAA;AAAA,KAOA,UAAA;AAAA,KAQA,SAAA;AAAA,KAKA,oBAAA;AAAA,KAYA,iBAAA;AAAA,KAYA,cAAA;AAAA,KAWA,eAAA"}

package/dist/types.messages.d.cts

@@ -226,8 +226,10 @@ type ToolCallState = "pending" | "completed" | "error";
  * Useful for rendering tool invocations and their outputs together.
  *
  * @template ToolCall The type of the tool call.
+ * @template TToolMessage The type of the tool result message. Defaults to the SDK's plain ToolMessage interface.
+ * @template TAIMessage The type of the AI message. Defaults to the SDK's plain AIMessage interface.
  */
-type ToolCallWithResult<ToolCall = DefaultToolCall> = {
+type ToolCallWithResult<ToolCall = DefaultToolCall, TToolMessage = ToolMessage, TAIMessage = AIMessage<ToolCall>> = {
   /**
    * Unique identifier for this tool call.
    * Uses the tool call's id if available, otherwise generates one from aiMessage.id and index.
@@ -241,11 +243,11 @@ type ToolCallWithResult<ToolCall = DefaultToolCall> = {
    * The result message from tool execution.
    * `undefined` if the tool is still being executed or no result was received.
    */
-  result: ToolMessage | undefined;
+  result: TToolMessage | undefined;
   /**
    * The AI message that initiated this tool call.
    */
-  aiMessage: AIMessage<ToolCall>;
+  aiMessage: TAIMessage;
   /**
    * Index of this tool call within the AI message's tool_calls array.
    */

package/dist/types.messages.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"types.messages.d.cts","names":[],"sources":["../src/types.messages.ts"],"mappings":";KAAK,WAAA;AAAA,KACA,sBAAA;EACH,IAAA;EACA,SAAA;IAAsB,GAAA;IAAa,MAAA,GAAS,WAAA;EAAA;AAAA;AAAA,KAGzC,kBAAA;EAAuB,IAAA;EAAc,IAAA;AAAA;AAAA,KACrC,qBAAA,GAAwB,kBAAA,GAAqB,sBAAA;AAAA,KAC7C,cAAA,YAA0B,qBAAA;;;;KAK1B,uBAAA,GAA0B,MAAA;AAAA,KAE1B,WAAA;EACH,iBAAA,GAAoB,uBAAA;EACpB,OAAA,EAAS,cAAA;EACT,EAAA;EACA,IAAA;EACA,iBAAA,GAAoB,MAAA;AAAA;AAAA,KAGV,YAAA,GAAe,WAAA;EACzB,IAAA;EACA,OAAA;AAAA;;;;KAMU,eAAA;EACV,IAAA;EAEA,IAAA;IAAA,CAAS,CAAA;EAAA;EACT,EAAA;EACA,IAAA;AAAA;;;;KAMU,eAAA;EACV,IAAA;EACA,IAAA;EACA,EAAA;EACA,KAAA;EACA,IAAA;AAAA;;;;KAMU,aAAA;EACV,YAAA;EACA,aAAA;EACA,YAAA;EACA,mBAAA;IAEM,KAAA;IACA,UAAA;IACA,cAAA;EAAA;EAGN,oBAAA;IACM,KAAA;IAA4B,SAAA;EAAA;AAAA;;;;;;;;;AAvBpC;;;;;;;;;;;AAWA;;;;;;;;KA2CY,SAAA,YAAqB,eAAA,IAAmB,WAAA;EAClD,IAAA;EACA,OAAA;EACA,UAAA,GAAa,QAAA;EACb,kBAAA,GAAqB,eAAA;EACrB,cAAA,GAAiB,aAAA;AAAA;AAAA,KAGP,WAAA,GAAc,WAAA;EACxB,IAAA;EACA,MAAA;EACA,YAAA;EAX+B;;;;;;;EAoB/B,QAAA;AAAA;AAAA,KAGU,aAAA,GAAgB,WAAA;EAC1B,IAAA;AAAA;AAAA,KAGU,eAAA,GAAkB,WAAA;EAC5B,IAAA;AAAA;AAAA,KAGU,aAAA,GAAgB,WAAA;EAC1B,IAAA;AAAA;;;AAxBF;;;KAgCY,OAAA,YAAmB,eAAA,IAC3B,YAAA,GACA,SAAA,CAAU,QAAA,IACV,WAAA,GACA,aAAA,GACA,eAAA,GACA,aAAA;;;;;;;;KAUC,gBAAA,MAAsB,CAAA;EAAY,IAAA;IAAQ,KAAA;EAAA;AAAA,IAC3C,IAAA,GACA,CAAA;EAAY,MAAA;AAAA,IACZ,IAAA;;;;AA5BJ;;KAoCK,cAAA,MAAoB,CAAA;EAEvB,KAAA,GAAQ,GAAA,iBAAoB,IAAA;AAAA,IAE1B,IAAA,GACA,CAAA;EAAY,MAAA;AAAA,IACZ,gBAAA,CAAiB,CAAA;;;;;;;;;;;;;;;;;;;;;;;AA3BH;KAsDN,gBAAA,MAAsB,CAAA;EAAY,IAAA;AAAA,IAC1C,cAAA,CAAe,CAAA,uBACb,IAAA,yBAGA,IAAA,SAAa,MAAA;EACT,IAAA,EAAM,CAAA;EAAG,IAAA,EAAM,IAAA;EAAM,EAAA;EAAa,IAAA;AAAA;;;;;;;;AA/CpC;;;;;;;;;;;;;;;;;;;;;;KAiFI,kBAAA,iCACV,CAAA,iDACI,gBAAA,CAAiB,KAAA,IAAS,kBAAA,CAAmB,IAAA;AA1CnD;;;;;;;AAAA,KAoDY,aAAA;;;;;;;KAQA,kBAAA,YAA8B,eAAA;EA3DtC;;;;EAgEF,EAAA;EA5DiB;;;EAiEjB,IAAA,EAAM,QAAA;EAhEiB;;;;EAsEvB,MAAA,EAAQ,WAAA;EApCE;;;EAyCV,SAAA,EAAW,SAAA,CAAU,QAAA;EAvCA;;;EA4CrB,KAAA;EA5CgD;;;;;;;EAqDhD,KAAA,EAAO,aAAA;AAAA"}
+{"version":3,"file":"types.messages.d.cts","names":[],"sources":["../src/types.messages.ts"],"mappings":";KAAK,WAAA;AAAA,KACA,sBAAA;EACH,IAAA;EACA,SAAA;IAAsB,GAAA;IAAa,MAAA,GAAS,WAAA;EAAA;AAAA;AAAA,KAGzC,kBAAA;EAAuB,IAAA;EAAc,IAAA;AAAA;AAAA,KACrC,qBAAA,GAAwB,kBAAA,GAAqB,sBAAA;AAAA,KAC7C,cAAA,YAA0B,qBAAA;;;;KAK1B,uBAAA,GAA0B,MAAA;AAAA,KAE1B,WAAA;EACH,iBAAA,GAAoB,uBAAA;EACpB,OAAA,EAAS,cAAA;EACT,EAAA;EACA,IAAA;EACA,iBAAA,GAAoB,MAAA;AAAA;AAAA,KAGV,YAAA,GAAe,WAAA;EACzB,IAAA;EACA,OAAA;AAAA;;;;KAMU,eAAA;EACV,IAAA;EAEA,IAAA;IAAA,CAAS,CAAA;EAAA;EACT,EAAA;EACA,IAAA;AAAA;;;;KAMU,eAAA;EACV,IAAA;EACA,IAAA;EACA,EAAA;EACA,KAAA;EACA,IAAA;AAAA;;;;KAMU,aAAA;EACV,YAAA;EACA,aAAA;EACA,YAAA;EACA,mBAAA;IAEM,KAAA;IACA,UAAA;IACA,cAAA;EAAA;EAGN,oBAAA;IACM,KAAA;IAA4B,SAAA;EAAA;AAAA;;;;;;;;;AAvBpC;;;;;;;;;;;AAWA;;;;;;;;KA2CY,SAAA,YAAqB,eAAA,IAAmB,WAAA;EAClD,IAAA;EACA,OAAA;EACA,UAAA,GAAa,QAAA;EACb,kBAAA,GAAqB,eAAA;EACrB,cAAA,GAAiB,aAAA;AAAA;AAAA,KAGP,WAAA,GAAc,WAAA;EACxB,IAAA;EACA,MAAA;EACA,YAAA;EAX+B;;;;;;;EAoB/B,QAAA;AAAA;AAAA,KAGU,aAAA,GAAgB,WAAA;EAC1B,IAAA;AAAA;AAAA,KAGU,eAAA,GAAkB,WAAA;EAC5B,IAAA;AAAA;AAAA,KAGU,aAAA,GAAgB,WAAA;EAC1B,IAAA;AAAA;;;AAxBF;;;KAgCY,OAAA,YAAmB,eAAA,IAC3B,YAAA,GACA,SAAA,CAAU,QAAA,IACV,WAAA,GACA,aAAA,GACA,eAAA,GACA,aAAA;;;;;;;;KAUC,gBAAA,MAAsB,CAAA;EAAY,IAAA;IAAQ,KAAA;EAAA;AAAA,IAC3C,IAAA,GACA,CAAA;EAAY,MAAA;AAAA,IACZ,IAAA;;;;AA5BJ;;KAoCK,cAAA,MAAoB,CAAA;EAEvB,KAAA,GAAQ,GAAA,iBAAoB,IAAA;AAAA,IAE1B,IAAA,GACA,CAAA;EAAY,MAAA;AAAA,IACZ,gBAAA,CAAiB,CAAA;;;;;;;;;;;;;;;;;;;;;;;AA3BH;KAsDN,gBAAA,MAAsB,CAAA;EAAY,IAAA;AAAA,IAC1C,cAAA,CAAe,CAAA,uBACb,IAAA,yBAGA,IAAA,SAAa,MAAA;EACT,IAAA,EAAM,CAAA;EAAG,IAAA,EAAM,IAAA;EAAM,EAAA;EAAa,IAAA;AAAA;;;;;;;;AA/CpC;;;;;;;;;;;;;;;;;;;;;;KAiFI,kBAAA,iCACV,CAAA,iDACI,gBAAA,CAAiB,KAAA,IAAS,kBAAA,CAAmB,IAAA;AA1CnD;;;;;;;AAAA,KAoDY,aAAA;;;;;;;;;KAUA,kBAAA,YACC,eAAA,iBACI,WAAA,eACF,SAAA,CAAU,QAAA;EAhEW;;;;EAsElC,EAAA;EAjEc;;;EAsEd,IAAA,EAAM,QAAA;EAtEoC;;;AAkC5C;EA0CE,MAAA,EAAQ,YAAA;EA1CoB;;;EA+C5B,SAAA,EAAW,UAAA;EA7CsC;;;EAkDjD,KAAA;EApD6B;;;;;;;EA6D7B,KAAA,EAAO,aAAA;AAAA"}

package/dist/types.messages.d.ts

@@ -226,8 +226,10 @@ type ToolCallState = "pending" | "completed" | "error";
  * Useful for rendering tool invocations and their outputs together.
  *
  * @template ToolCall The type of the tool call.
+ * @template TToolMessage The type of the tool result message. Defaults to the SDK's plain ToolMessage interface.
+ * @template TAIMessage The type of the AI message. Defaults to the SDK's plain AIMessage interface.
  */
-type ToolCallWithResult<ToolCall = DefaultToolCall> = {
+type ToolCallWithResult<ToolCall = DefaultToolCall, TToolMessage = ToolMessage, TAIMessage = AIMessage<ToolCall>> = {
   /**
    * Unique identifier for this tool call.
    * Uses the tool call's id if available, otherwise generates one from aiMessage.id and index.
@@ -241,11 +243,11 @@ type ToolCallWithResult<ToolCall = DefaultToolCall> = {
    * The result message from tool execution.
    * `undefined` if the tool is still being executed or no result was received.
    */
-  result: ToolMessage | undefined;
+  result: TToolMessage | undefined;
   /**
    * The AI message that initiated this tool call.
    */
-  aiMessage: AIMessage<ToolCall>;
+  aiMessage: TAIMessage;
   /**
    * Index of this tool call within the AI message's tool_calls array.
    */

package/dist/types.messages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.messages.d.ts","names":[],"sources":["../src/types.messages.ts"],"mappings":";KAAK,WAAA;AAAA,KACA,sBAAA;EACH,IAAA;EACA,SAAA;IAAsB,GAAA;IAAa,MAAA,GAAS,WAAA;EAAA;AAAA;AAAA,KAGzC,kBAAA;EAAuB,IAAA;EAAc,IAAA;AAAA;AAAA,KACrC,qBAAA,GAAwB,kBAAA,GAAqB,sBAAA;AAAA,KAC7C,cAAA,YAA0B,qBAAA;;;;KAK1B,uBAAA,GAA0B,MAAA;AAAA,KAE1B,WAAA;EACH,iBAAA,GAAoB,uBAAA;EACpB,OAAA,EAAS,cAAA;EACT,EAAA;EACA,IAAA;EACA,iBAAA,GAAoB,MAAA;AAAA;AAAA,KAGV,YAAA,GAAe,WAAA;EACzB,IAAA;EACA,OAAA;AAAA;;;;KAMU,eAAA;EACV,IAAA;EAEA,IAAA;IAAA,CAAS,CAAA;EAAA;EACT,EAAA;EACA,IAAA;AAAA;;;;KAMU,eAAA;EACV,IAAA;EACA,IAAA;EACA,EAAA;EACA,KAAA;EACA,IAAA;AAAA;;;;KAMU,aAAA;EACV,YAAA;EACA,aAAA;EACA,YAAA;EACA,mBAAA;IAEM,KAAA;IACA,UAAA;IACA,cAAA;EAAA;EAGN,oBAAA;IACM,KAAA;IAA4B,SAAA;EAAA;AAAA;;;;;;;;;AAvBpC;;;;;;;;;;;AAWA;;;;;;;;KA2CY,SAAA,YAAqB,eAAA,IAAmB,WAAA;EAClD,IAAA;EACA,OAAA;EACA,UAAA,GAAa,QAAA;EACb,kBAAA,GAAqB,eAAA;EACrB,cAAA,GAAiB,aAAA;AAAA;AAAA,KAGP,WAAA,GAAc,WAAA;EACxB,IAAA;EACA,MAAA;EACA,YAAA;EAX+B;;;;;;;EAoB/B,QAAA;AAAA;AAAA,KAGU,aAAA,GAAgB,WAAA;EAC1B,IAAA;AAAA;AAAA,KAGU,eAAA,GAAkB,WAAA;EAC5B,IAAA;AAAA;AAAA,KAGU,aAAA,GAAgB,WAAA;EAC1B,IAAA;AAAA;;;AAxBF;;;KAgCY,OAAA,YAAmB,eAAA,IAC3B,YAAA,GACA,SAAA,CAAU,QAAA,IACV,WAAA,GACA,aAAA,GACA,eAAA,GACA,aAAA;;;;;;;;KAUC,gBAAA,MAAsB,CAAA;EAAY,IAAA;IAAQ,KAAA;EAAA;AAAA,IAC3C,IAAA,GACA,CAAA;EAAY,MAAA;AAAA,IACZ,IAAA;;;;AA5BJ;;KAoCK,cAAA,MAAoB,CAAA;EAEvB,KAAA,GAAQ,GAAA,iBAAoB,IAAA;AAAA,IAE1B,IAAA,GACA,CAAA;EAAY,MAAA;AAAA,IACZ,gBAAA,CAAiB,CAAA;;;;;;;;;;;;;;;;;;;;;;;AA3BH;KAsDN,gBAAA,MAAsB,CAAA;EAAY,IAAA;AAAA,IAC1C,cAAA,CAAe,CAAA,uBACb,IAAA,yBAGA,IAAA,SAAa,MAAA;EACT,IAAA,EAAM,CAAA;EAAG,IAAA,EAAM,IAAA;EAAM,EAAA;EAAa,IAAA;AAAA;;;;;;;;AA/CpC;;;;;;;;;;;;;;;;;;;;;;KAiFI,kBAAA,iCACV,CAAA,iDACI,gBAAA,CAAiB,KAAA,IAAS,kBAAA,CAAmB,IAAA;AA1CnD;;;;;;;AAAA,KAoDY,aAAA;;;;;;;KAQA,kBAAA,YAA8B,eAAA;EA3DtC;;;;EAgEF,EAAA;EA5DiB;;;EAiEjB,IAAA,EAAM,QAAA;EAhEiB;;;;EAsEvB,MAAA,EAAQ,WAAA;EApCE;;;EAyCV,SAAA,EAAW,SAAA,CAAU,QAAA;EAvCA;;;EA4CrB,KAAA;EA5CgD;;;;;;;EAqDhD,KAAA,EAAO,aAAA;AAAA"}
+{"version":3,"file":"types.messages.d.ts","names":[],"sources":["../src/types.messages.ts"],"mappings":";KAAK,WAAA;AAAA,KACA,sBAAA;EACH,IAAA;EACA,SAAA;IAAsB,GAAA;IAAa,MAAA,GAAS,WAAA;EAAA;AAAA;AAAA,KAGzC,kBAAA;EAAuB,IAAA;EAAc,IAAA;AAAA;AAAA,KACrC,qBAAA,GAAwB,kBAAA,GAAqB,sBAAA;AAAA,KAC7C,cAAA,YAA0B,qBAAA;;;;KAK1B,uBAAA,GAA0B,MAAA;AAAA,KAE1B,WAAA;EACH,iBAAA,GAAoB,uBAAA;EACpB,OAAA,EAAS,cAAA;EACT,EAAA;EACA,IAAA;EACA,iBAAA,GAAoB,MAAA;AAAA;AAAA,KAGV,YAAA,GAAe,WAAA;EACzB,IAAA;EACA,OAAA;AAAA;;;;KAMU,eAAA;EACV,IAAA;EAEA,IAAA;IAAA,CAAS,CAAA;EAAA;EACT,EAAA;EACA,IAAA;AAAA;;;;KAMU,eAAA;EACV,IAAA;EACA,IAAA;EACA,EAAA;EACA,KAAA;EACA,IAAA;AAAA;;;;KAMU,aAAA;EACV,YAAA;EACA,aAAA;EACA,YAAA;EACA,mBAAA;IAEM,KAAA;IACA,UAAA;IACA,cAAA;EAAA;EAGN,oBAAA;IACM,KAAA;IAA4B,SAAA;EAAA;AAAA;;;;;;;;;AAvBpC;;;;;;;;;;;AAWA;;;;;;;;KA2CY,SAAA,YAAqB,eAAA,IAAmB,WAAA;EAClD,IAAA;EACA,OAAA;EACA,UAAA,GAAa,QAAA;EACb,kBAAA,GAAqB,eAAA;EACrB,cAAA,GAAiB,aAAA;AAAA;AAAA,KAGP,WAAA,GAAc,WAAA;EACxB,IAAA;EACA,MAAA;EACA,YAAA;EAX+B;;;;;;;EAoB/B,QAAA;AAAA;AAAA,KAGU,aAAA,GAAgB,WAAA;EAC1B,IAAA;AAAA;AAAA,KAGU,eAAA,GAAkB,WAAA;EAC5B,IAAA;AAAA;AAAA,KAGU,aAAA,GAAgB,WAAA;EAC1B,IAAA;AAAA;;;AAxBF;;;KAgCY,OAAA,YAAmB,eAAA,IAC3B,YAAA,GACA,SAAA,CAAU,QAAA,IACV,WAAA,GACA,aAAA,GACA,eAAA,GACA,aAAA;;;;;;;;KAUC,gBAAA,MAAsB,CAAA;EAAY,IAAA;IAAQ,KAAA;EAAA;AAAA,IAC3C,IAAA,GACA,CAAA;EAAY,MAAA;AAAA,IACZ,IAAA;;;;AA5BJ;;KAoCK,cAAA,MAAoB,CAAA;EAEvB,KAAA,GAAQ,GAAA,iBAAoB,IAAA;AAAA,IAE1B,IAAA,GACA,CAAA;EAAY,MAAA;AAAA,IACZ,gBAAA,CAAiB,CAAA;;;;;;;;;;;;;;;;;;;;;;;AA3BH;KAsDN,gBAAA,MAAsB,CAAA;EAAY,IAAA;AAAA,IAC1C,cAAA,CAAe,CAAA,uBACb,IAAA,yBAGA,IAAA,SAAa,MAAA;EACT,IAAA,EAAM,CAAA;EAAG,IAAA,EAAM,IAAA;EAAM,EAAA;EAAa,IAAA;AAAA;;;;;;;;AA/CpC;;;;;;;;;;;;;;;;;;;;;;KAiFI,kBAAA,iCACV,CAAA,iDACI,gBAAA,CAAiB,KAAA,IAAS,kBAAA,CAAmB,IAAA;AA1CnD;;;;;;;AAAA,KAoDY,aAAA;;;;;;;;;KAUA,kBAAA,YACC,eAAA,iBACI,WAAA,eACF,SAAA,CAAU,QAAA;EAhEW;;;;EAsElC,EAAA;EAjEc;;;EAsEd,IAAA,EAAM,QAAA;EAtEoC;;;AAkC5C;EA0CE,MAAA,EAAQ,YAAA;EA1CoB;;;EA+C5B,SAAA,EAAW,UAAA;EA7CsC;;;EAkDjD,KAAA;EApD6B;;;;;;;EA6D7B,KAAA,EAAO,aAAA;AAAA"}

package/dist/types.stream.d.cts

@@ -13,7 +13,7 @@ import type { SubgraphCheckpointsStreamEvent } from "./types.stream.subgraph.js"
  * - "debug": Stream detailed debug information.
  * - "custom": Stream custom events.
  */
-type StreamMode = "values" | "messages" | "updates" | "events" | "debug" | "tasks" | "checkpoints" | "custom" | "messages-tuple";
+type StreamMode = "values" | "messages" | "updates" | "events" | "debug" | "tasks" | "checkpoints" | "custom" | "messages-tuple" | "tools";
 type ThreadStreamMode = "run_modes" | "lifecycle" | "state_update";
 type MessageTupleMetadata = {
   tags: string[];
@@ -204,6 +204,42 @@ type FeedbackStreamEvent = {
     [feedbackKey: string]: string;
   };
 };
+type ToolStreamEventData = {
+  event: "on_tool_start";
+  toolCallId?: string;
+  name: string;
+  input: unknown;
+} | {
+  event: "on_tool_event";
+  toolCallId?: string;
+  name: string;
+  data: unknown;
+} | {
+  event: "on_tool_end";
+  toolCallId?: string;
+  name: string;
+  output: unknown;
+} | {
+  event: "on_tool_error";
+  toolCallId?: string;
+  name: string;
+  error: unknown;
+};
+type ToolsStreamEvent = {
+  event: "tools";
+  data: ToolStreamEventData;
+};
+type ToolProgress = {
+  toolCallId?: string;
+  name: string;
+  state: "starting" | "running" | "completed" | "error";
+  input?: unknown;
+  data?: unknown;
+  result?: unknown;
+  error?: unknown;
+};
+/** @internal */
+type SubgraphToolsStreamEvent = AsSubgraph<ToolsStreamEvent>;
 type GetStreamModeMap<TStreamMode extends StreamMode | StreamMode[], TStateType = unknown, TUpdateType = TStateType, TCustomType = unknown> = {
   values: ValuesStreamEvent<TStateType>;
   updates: UpdatesStreamEvent<TUpdateType>;
@@ -214,6 +250,7 @@ type GetStreamModeMap<TStreamMode extends StreamMode | StreamMode[], TStateType
   tasks: TasksStreamEvent<TStateType, TUpdateType>;
   checkpoints: CheckpointsStreamEvent<TStateType>;
   events: EventsStreamEvent;
+  tools: ToolsStreamEvent;
 }[TStreamMode extends StreamMode[] ? TStreamMode[number] : TStreamMode] | ErrorStreamEvent | MetadataStreamEvent | FeedbackStreamEvent;
 type GetSubgraphsStreamModeMap<TStreamMode extends StreamMode | StreamMode[], TStateType = unknown, TUpdateType = TStateType, TCustomType = unknown> = {
   values: SubgraphValuesStreamEvent<TStateType>;
@@ -225,8 +262,9 @@ type GetSubgraphsStreamModeMap<TStreamMode extends StreamMode | StreamMode[], TS
   events: SubgraphEventsStreamEvent;
   tasks: SubgraphTasksStreamEvent<TStateType, TUpdateType>;
   checkpoints: SubgraphCheckpointsStreamEvent<TStateType>;
+  tools: SubgraphToolsStreamEvent;
 }[TStreamMode extends StreamMode[] ? TStreamMode[number] : TStreamMode] | SubgraphErrorStreamEvent | MetadataStreamEvent | FeedbackStreamEvent;
 type TypedAsyncGenerator<TStreamMode extends StreamMode | StreamMode[] = [], TSubgraphs extends boolean = false, TStateType = unknown, TUpdateType = TStateType, TCustomType = unknown> = AsyncGenerator<TSubgraphs extends true ? GetSubgraphsStreamModeMap<TStreamMode, TStateType, TUpdateType, TCustomType> : GetStreamModeMap<TStreamMode, TStateType, TUpdateType, TCustomType>>;
 //#endregion
-export { CheckpointsStreamEvent, CustomStreamEvent, DebugStreamEvent, ErrorStreamEvent, EventsStreamEvent, FeedbackStreamEvent, MessagesStreamEvent, MessagesTupleStreamEvent, MetadataStreamEvent, StreamMode, TasksStreamEvent, ThreadStreamMode, TypedAsyncGenerator, UpdatesStreamEvent, ValuesStreamEvent };
+export { CheckpointsStreamEvent, CustomStreamEvent, DebugStreamEvent, ErrorStreamEvent, EventsStreamEvent, FeedbackStreamEvent, MessagesStreamEvent, MessagesTupleStreamEvent, MetadataStreamEvent, StreamMode, TasksStreamEvent, ThreadStreamMode, ToolProgress, ToolsStreamEvent, TypedAsyncGenerator, UpdatesStreamEvent, ValuesStreamEvent };
 //# sourceMappingURL=types.stream.d.cts.map

package/dist/types.stream.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"types.stream.d.cts","names":[],"sources":["../src/types.stream.ts"],"mappings":";;;;;;AAaA;;;;;AAWA;;;;KAXY,UAAA;AAAA,KAWA,gBAAA;AAAA,KAEP,oBAAA;EACH,IAAA;EAAA,CACC,GAAA;AAAA;AAAA,KAGE,UAAA;EAA4B,EAAA;EAAa,KAAA;EAAe,IAAA;AAAA;EAEzD,EAAA,GAAK,MAAA;EACL,KAAA,EAAO,MAAA,eAAqB,MAAA;EAC5B,IAAA,EAAM,MAAA;AAAA;;;;KAME,iBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA,EAAM,SAAA;AAAA;;KAII,yBAAA,cAAuC,UAAA,CACjD,iBAAA,CAAkB,SAAA;;;;KAMR,wBAAA;EACV,KAAA;EAGA,IAAA,GAAO,OAAA,EAAS,OAAA,EAAS,MAAA,EAAQ,oBAAA;AAAA;;KAIvB,gCAAA,GACV,UAAA,CAAW,wBAAA;;;;KAKD,mBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IAAQ,MAAA;IAAgB,SAAA;EAAA;AAAA;;;;KAMd,gBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IAAQ,KAAA;IAAe,OAAA;EAAA;AAAA;;KAIb,wBAAA,GAA2B,UAAA,CAAW,gBAAA;;;;;;KAOtC,kBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IAAA,CAAS,IAAA,WAAe,UAAA;EAAA;AAAA;AA1B1B;AAAA,KA8BY,0BAAA,eAAyC,UAAA,CACnD,kBAAA,CAAmB,UAAA;;;;KAMT,iBAAA;EAAyB,KAAA;EAAiB,IAAA,EAAM,CAAA;AAAA;;KAGhD,yBAAA,MAA+B,UAAA,CAAW,iBAAA,CAAkB,CAAA;AAAA,KAEnE,2BAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA;IAAA,CAAS,SAAA;MAAsB,QAAA;IAAA;EAAA;AAAA;AAAA,KAE5B,2BAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA,EAAM,OAAA;AAAA;AAAA,KAEH,0BAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA,EAAM,OAAA;AAAA;AAAA,KAGH,sBAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA;IACE,EAAA;IACA,IAAA;IACA,UAAA,EAAY,SAAA;IACZ,KAAA,EAAO,SAAA;IACP,QAAA;EAAA;AAAA;AAAA,KAIC,sBAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA;IACE,EAAA;IACA,IAAA;IACA,UAAA,EAAY,SAAA;IACZ,MAAA,WAAiB,UAAA;EAAA;AAAA;AAAA,KAIhB,qBAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA;IACE,EAAA;IACA,IAAA;IACA,UAAA,EAAY,SAAA;IACZ,KAAA;EAAA;AAAA;AAAA,KAIQ,gBAAA,0BACR,sBAAA,CAAuB,SAAA,IACvB,sBAAA,CAAuB,UAAA,IACvB,qBAAA;AAAA,KAEC,wBAAA,0BACD,UAAA,CAAW,sBAAA,CAAuB,SAAA,KAClC,UAAA,CAAW,sBAAA,CAAuB,UAAA,KAClC,UAAA,CAAW,qBAAA;AAAA,KAEH,sBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IACE,MAAA,EAAQ,SAAA;IACR,IAAA;IACA,MAAA,EAAQ,MAAA;IACR,QAAA,EAAU,QAAA;IACV,KAAA,EAAO,UAAA;EAAA;AAAA;AAAA,KAIN,8BAAA,cAA4C,UAAA,CAC/C,sBAAA,CAAuB,SAAA;;;AA3EmD;;KAkFhE,mBAAA,GACR,2BAAA,GACA,2BAAA,GACA,0BAAA;;KAGQ,2BAAA,GACR,UAAA,CAAW,2BAAA,IACX,UAAA,CAAW,2BAAA,IACX,UAAA,CAAW,0BAAA;;;;KAKH,gBAAA;EAAqB,EAAA;EAAa,KAAA;EAAgB,IAAA;AAAA;;KAGlD,wBAAA,GAA2B,UAAA,CAAW,gBAAA;;;;KAKtC,iBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IACE,KAAA;IASA,IAAA;IACA,IAAA;IACA,MAAA;IACA,QAAA,EAAU,MAAA;IACV,UAAA;IACA,IAAA;EAAA;AAAA;AA3GW;AAAA,KAgHH,yBAAA,GAA4B,UAAA,CAAW,iBAAA;;;;;KAMvC,mBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IAAA,CAAS,WAAA;EAAA;AAAA;AAAA,KAGN,gBAAA,qBACiB,UAAA,GAAa,UAAA,wCAEnB,UAAA;EAIV,MAAA,EAAQ,iBAAA,CAAkB,UAAA;EAC1B,OAAA,EAAS,kBAAA,CAAmB,WAAA;EAC5B,MAAA,EAAQ,iBAAA,CAAkB,WAAA;EAC1B,KAAA,EAAO,gBAAA;EACP,QAAA,EAAU,mBAAA;EACV,gBAAA,EAAkB,wBAAA;EAClB,KAAA,EAAO,gBAAA,CAAiB,UAAA,EAAY,WAAA;EACpC,WAAA,EAAa,sBAAA,CAAuB,UAAA;EACpC,MAAA,EAAQ,iBAAA;AAAA,EACR,WAAA,SAAoB,UAAA,KAAe,WAAA,WAAsB,WAAA,IAC3D,gBAAA,GACA,mBAAA,GACA,mBAAA;AAAA,KAEC,yBAAA,qBACiB,UAAA,GAAa,UAAA,wCAEnB,UAAA;EAIV,MAAA,EAAQ,yBAAA,CAA0B,UAAA;EAClC,OAAA,EAAS,0BAAA,CAA2B,WAAA;EACpC,MAAA,EAAQ,yBAAA,CAA0B,WAAA;EAClC,KAAA,EAAO,wBAAA;EACP,QAAA,EAAU,2BAAA;EACV,gBAAA,EAAkB,gCAAA;EAClB,MAAA,EAAQ,yBAAA;EACR,KAAA,EAAO,wBAAA,CAAyB,UAAA,EAAY,WAAA;EAC5C,WAAA,EAAa,8BAAA,CAA+B,UAAA;AAAA,EAC5C,WAAA,SAAoB,UAAA,KAAe,WAAA,WAAsB,WAAA,IAC3D,wBAAA,GACA,mBAAA,GACA,mBAAA;AAAA,KAEQ,mBAAA,qBACU,UAAA,GAAa,UAAA,iFAGnB,UAAA,2BAEZ,cAAA,CACF,UAAA,gBACI,yBAAA,CACE,WAAA,EACA,UAAA,EACA,WAAA,EACA,WAAA,IAEF,gBAAA,CAAiB,WAAA,EAAa,UAAA,EAAY,WAAA,EAAa,WAAA"}
+{"version":3,"file":"types.stream.d.cts","names":[],"sources":["../src/types.stream.ts"],"mappings":";;;;;;AAaA;;;;;AAYA;;;;KAZY,UAAA;AAAA,KAYA,gBAAA;AAAA,KAEP,oBAAA;EACH,IAAA;EAAA,CACC,GAAA;AAAA;AAAA,KAGE,UAAA;EAA4B,EAAA;EAAa,KAAA;EAAe,IAAA;AAAA;EAEzD,EAAA,GAAK,MAAA;EACL,KAAA,EAAO,MAAA,eAAqB,MAAA;EAC5B,IAAA,EAAM,MAAA;AAAA;;;;KAME,iBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA,EAAM,SAAA;AAAA;;KAII,yBAAA,cAAuC,UAAA,CACjD,iBAAA,CAAkB,SAAA;;;;KAMR,wBAAA;EACV,KAAA;EAGA,IAAA,GAAO,OAAA,EAAS,OAAA,EAAS,MAAA,EAAQ,oBAAA;AAAA;;KAIvB,gCAAA,GACV,UAAA,CAAW,wBAAA;;;;KAKD,mBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IAAQ,MAAA;IAAgB,SAAA;EAAA;AAAA;;;;KAMd,gBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IAAQ,KAAA;IAAe,OAAA;EAAA;AAAA;;KAIb,wBAAA,GAA2B,UAAA,CAAW,gBAAA;;;;;;KAOtC,kBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IAAA,CAAS,IAAA,WAAe,UAAA;EAAA;AAAA;AA1B1B;AAAA,KA8BY,0BAAA,eAAyC,UAAA,CACnD,kBAAA,CAAmB,UAAA;;;;KAMT,iBAAA;EAAyB,KAAA;EAAiB,IAAA,EAAM,CAAA;AAAA;;KAGhD,yBAAA,MAA+B,UAAA,CAAW,iBAAA,CAAkB,CAAA;AAAA,KAEnE,2BAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA;IAAA,CAAS,SAAA;MAAsB,QAAA;IAAA;EAAA;AAAA;AAAA,KAE5B,2BAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA,EAAM,OAAA;AAAA;AAAA,KAEH,0BAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA,EAAM,OAAA;AAAA;AAAA,KAGH,sBAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA;IACE,EAAA;IACA,IAAA;IACA,UAAA,EAAY,SAAA;IACZ,KAAA,EAAO,SAAA;IACP,QAAA;EAAA;AAAA;AAAA,KAIC,sBAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA;IACE,EAAA;IACA,IAAA;IACA,UAAA,EAAY,SAAA;IACZ,MAAA,WAAiB,UAAA;EAAA;AAAA;AAAA,KAIhB,qBAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA;IACE,EAAA;IACA,IAAA;IACA,UAAA,EAAY,SAAA;IACZ,KAAA;EAAA;AAAA;AAAA,KAIQ,gBAAA,0BACR,sBAAA,CAAuB,SAAA,IACvB,sBAAA,CAAuB,UAAA,IACvB,qBAAA;AAAA,KAEC,wBAAA,0BACD,UAAA,CAAW,sBAAA,CAAuB,SAAA,KAClC,UAAA,CAAW,sBAAA,CAAuB,UAAA,KAClC,UAAA,CAAW,qBAAA;AAAA,KAEH,sBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IACE,MAAA,EAAQ,SAAA;IACR,IAAA;IACA,MAAA,EAAQ,MAAA;IACR,QAAA,EAAU,QAAA;IACV,KAAA,EAAO,UAAA;EAAA;AAAA;AAAA,KAIN,8BAAA,cAA4C,UAAA,CAC/C,sBAAA,CAAuB,SAAA;;;AA3EmD;;KAkFhE,mBAAA,GACR,2BAAA,GACA,2BAAA,GACA,0BAAA;;KAGQ,2BAAA,GACR,UAAA,CAAW,2BAAA,IACX,UAAA,CAAW,2BAAA,IACX,UAAA,CAAW,0BAAA;;;;KAKH,gBAAA;EAAqB,EAAA;EAAa,KAAA;EAAgB,IAAA;AAAA;;KAGlD,wBAAA,GAA2B,UAAA,CAAW,gBAAA;;;;KAKtC,iBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IACE,KAAA;IASA,IAAA;IACA,IAAA;IACA,MAAA;IACA,QAAA,EAAU,MAAA;IACV,UAAA;IACA,IAAA;EAAA;AAAA;AA3GW;AAAA,KAgHH,yBAAA,GAA4B,UAAA,CAAW,iBAAA;;;;;KAMvC,mBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IAAA,CAAS,WAAA;EAAA;AAAA;AAAA,KAGC,mBAAA;EAEN,KAAA;EACA,UAAA;EACA,IAAA;EACA,KAAA;AAAA;EAGA,KAAA;EACA,UAAA;EACA,IAAA;EACA,IAAA;AAAA;EAGA,KAAA;EACA,UAAA;EACA,IAAA;EACA,MAAA;AAAA;EAGA,KAAA;EACA,UAAA;EACA,IAAA;EACA,KAAA;AAAA;AAAA,KAGM,gBAAA;EACV,KAAA;EACA,IAAA,EAAM,mBAAA;AAAA;AAAA,KAGI,YAAA;EACV,UAAA;EACA,IAAA;EACA,KAAA;EACA,KAAA;EACA,IAAA;EACA,MAAA;EACA,KAAA;AAAA;AA7HF;AAAA,KAiIY,wBAAA,GAA2B,UAAA,CAAW,gBAAA;AAAA,KAE7C,gBAAA,qBACiB,UAAA,GAAa,UAAA,wCAEnB,UAAA;EAIV,MAAA,EAAQ,iBAAA,CAAkB,UAAA;EAC1B,OAAA,EAAS,kBAAA,CAAmB,WAAA;EAC5B,MAAA,EAAQ,iBAAA,CAAkB,WAAA;EAC1B,KAAA,EAAO,gBAAA;EACP,QAAA,EAAU,mBAAA;EACV,gBAAA,EAAkB,wBAAA;EAClB,KAAA,EAAO,gBAAA,CAAiB,UAAA,EAAY,WAAA;EACpC,WAAA,EAAa,sBAAA,CAAuB,UAAA;EACpC,MAAA,EAAQ,iBAAA;EACR,KAAA,EAAO,gBAAA;AAAA,EACP,WAAA,SAAoB,UAAA,KAAe,WAAA,WAAsB,WAAA,IAC3D,gBAAA,GACA,mBAAA,GACA,mBAAA;AAAA,KAEC,yBAAA,qBACiB,UAAA,GAAa,UAAA,wCAEnB,UAAA;EAIV,MAAA,EAAQ,yBAAA,CAA0B,UAAA;EAClC,OAAA,EAAS,0BAAA,CAA2B,WAAA;EACpC,MAAA,EAAQ,yBAAA,CAA0B,WAAA;EAClC,KAAA,EAAO,wBAAA;EACP,QAAA,EAAU,2BAAA;EACV,gBAAA,EAAkB,gCAAA;EAClB,MAAA,EAAQ,yBAAA;EACR,KAAA,EAAO,wBAAA,CAAyB,UAAA,EAAY,WAAA;EAC5C,WAAA,EAAa,8BAAA,CAA+B,UAAA;EAC5C,KAAA,EAAO,wBAAA;AAAA,EACP,WAAA,SAAoB,UAAA,KAAe,WAAA,WAAsB,WAAA,IAC3D,wBAAA,GACA,mBAAA,GACA,mBAAA;AAAA,KAEQ,mBAAA,qBACU,UAAA,GAAa,UAAA,iFAGnB,UAAA,2BAEZ,cAAA,CACF,UAAA,gBACI,yBAAA,CACE,WAAA,EACA,UAAA,EACA,WAAA,EACA,WAAA,IAEF,gBAAA,CAAiB,WAAA,EAAa,UAAA,EAAY,WAAA,EAAa,WAAA"}

package/dist/types.stream.d.ts

@@ -13,7 +13,7 @@ import type { SubgraphCheckpointsStreamEvent } from "./types.stream.subgraph.js"
  * - "debug": Stream detailed debug information.
  * - "custom": Stream custom events.
  */
-type StreamMode = "values" | "messages" | "updates" | "events" | "debug" | "tasks" | "checkpoints" | "custom" | "messages-tuple";
+type StreamMode = "values" | "messages" | "updates" | "events" | "debug" | "tasks" | "checkpoints" | "custom" | "messages-tuple" | "tools";
 type ThreadStreamMode = "run_modes" | "lifecycle" | "state_update";
 type MessageTupleMetadata = {
   tags: string[];
@@ -204,6 +204,42 @@ type FeedbackStreamEvent = {
     [feedbackKey: string]: string;
   };
 };
+type ToolStreamEventData = {
+  event: "on_tool_start";
+  toolCallId?: string;
+  name: string;
+  input: unknown;
+} | {
+  event: "on_tool_event";
+  toolCallId?: string;
+  name: string;
+  data: unknown;
+} | {
+  event: "on_tool_end";
+  toolCallId?: string;
+  name: string;
+  output: unknown;
+} | {
+  event: "on_tool_error";
+  toolCallId?: string;
+  name: string;
+  error: unknown;
+};
+type ToolsStreamEvent = {
+  event: "tools";
+  data: ToolStreamEventData;
+};
+type ToolProgress = {
+  toolCallId?: string;
+  name: string;
+  state: "starting" | "running" | "completed" | "error";
+  input?: unknown;
+  data?: unknown;
+  result?: unknown;
+  error?: unknown;
+};
+/** @internal */
+type SubgraphToolsStreamEvent = AsSubgraph<ToolsStreamEvent>;
 type GetStreamModeMap<TStreamMode extends StreamMode | StreamMode[], TStateType = unknown, TUpdateType = TStateType, TCustomType = unknown> = {
   values: ValuesStreamEvent<TStateType>;
   updates: UpdatesStreamEvent<TUpdateType>;
@@ -214,6 +250,7 @@ type GetStreamModeMap<TStreamMode extends StreamMode | StreamMode[], TStateType
   tasks: TasksStreamEvent<TStateType, TUpdateType>;
   checkpoints: CheckpointsStreamEvent<TStateType>;
   events: EventsStreamEvent;
+  tools: ToolsStreamEvent;
 }[TStreamMode extends StreamMode[] ? TStreamMode[number] : TStreamMode] | ErrorStreamEvent | MetadataStreamEvent | FeedbackStreamEvent;
 type GetSubgraphsStreamModeMap<TStreamMode extends StreamMode | StreamMode[], TStateType = unknown, TUpdateType = TStateType, TCustomType = unknown> = {
   values: SubgraphValuesStreamEvent<TStateType>;
@@ -225,8 +262,9 @@ type GetSubgraphsStreamModeMap<TStreamMode extends StreamMode | StreamMode[], TS
   events: SubgraphEventsStreamEvent;
   tasks: SubgraphTasksStreamEvent<TStateType, TUpdateType>;
   checkpoints: SubgraphCheckpointsStreamEvent<TStateType>;
+  tools: SubgraphToolsStreamEvent;
 }[TStreamMode extends StreamMode[] ? TStreamMode[number] : TStreamMode] | SubgraphErrorStreamEvent | MetadataStreamEvent | FeedbackStreamEvent;
 type TypedAsyncGenerator<TStreamMode extends StreamMode | StreamMode[] = [], TSubgraphs extends boolean = false, TStateType = unknown, TUpdateType = TStateType, TCustomType = unknown> = AsyncGenerator<TSubgraphs extends true ? GetSubgraphsStreamModeMap<TStreamMode, TStateType, TUpdateType, TCustomType> : GetStreamModeMap<TStreamMode, TStateType, TUpdateType, TCustomType>>;
 //#endregion
-export { CheckpointsStreamEvent, CustomStreamEvent, DebugStreamEvent, ErrorStreamEvent, EventsStreamEvent, FeedbackStreamEvent, MessagesStreamEvent, MessagesTupleStreamEvent, MetadataStreamEvent, StreamMode, TasksStreamEvent, ThreadStreamMode, TypedAsyncGenerator, UpdatesStreamEvent, ValuesStreamEvent };
+export { CheckpointsStreamEvent, CustomStreamEvent, DebugStreamEvent, ErrorStreamEvent, EventsStreamEvent, FeedbackStreamEvent, MessagesStreamEvent, MessagesTupleStreamEvent, MetadataStreamEvent, StreamMode, TasksStreamEvent, ThreadStreamMode, ToolProgress, ToolsStreamEvent, TypedAsyncGenerator, UpdatesStreamEvent, ValuesStreamEvent };
 //# sourceMappingURL=types.stream.d.ts.map

package/dist/types.stream.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.stream.d.ts","names":[],"sources":["../src/types.stream.ts"],"mappings":";;;;;;AAaA;;;;;AAWA;;;;KAXY,UAAA;AAAA,KAWA,gBAAA;AAAA,KAEP,oBAAA;EACH,IAAA;EAAA,CACC,GAAA;AAAA;AAAA,KAGE,UAAA;EAA4B,EAAA;EAAa,KAAA;EAAe,IAAA;AAAA;EAEzD,EAAA,GAAK,MAAA;EACL,KAAA,EAAO,MAAA,eAAqB,MAAA;EAC5B,IAAA,EAAM,MAAA;AAAA;;;;KAME,iBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA,EAAM,SAAA;AAAA;;KAII,yBAAA,cAAuC,UAAA,CACjD,iBAAA,CAAkB,SAAA;;;;KAMR,wBAAA;EACV,KAAA;EAGA,IAAA,GAAO,OAAA,EAAS,OAAA,EAAS,MAAA,EAAQ,oBAAA;AAAA;;KAIvB,gCAAA,GACV,UAAA,CAAW,wBAAA;;;;KAKD,mBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IAAQ,MAAA;IAAgB,SAAA;EAAA;AAAA;;;;KAMd,gBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IAAQ,KAAA;IAAe,OAAA;EAAA;AAAA;;KAIb,wBAAA,GAA2B,UAAA,CAAW,gBAAA;;;;;;KAOtC,kBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IAAA,CAAS,IAAA,WAAe,UAAA;EAAA;AAAA;AA1B1B;AAAA,KA8BY,0BAAA,eAAyC,UAAA,CACnD,kBAAA,CAAmB,UAAA;;;;KAMT,iBAAA;EAAyB,KAAA;EAAiB,IAAA,EAAM,CAAA;AAAA;;KAGhD,yBAAA,MAA+B,UAAA,CAAW,iBAAA,CAAkB,CAAA;AAAA,KAEnE,2BAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA;IAAA,CAAS,SAAA;MAAsB,QAAA;IAAA;EAAA;AAAA;AAAA,KAE5B,2BAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA,EAAM,OAAA;AAAA;AAAA,KAEH,0BAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA,EAAM,OAAA;AAAA;AAAA,KAGH,sBAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA;IACE,EAAA;IACA,IAAA;IACA,UAAA,EAAY,SAAA;IACZ,KAAA,EAAO,SAAA;IACP,QAAA;EAAA;AAAA;AAAA,KAIC,sBAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA;IACE,EAAA;IACA,IAAA;IACA,UAAA,EAAY,SAAA;IACZ,MAAA,WAAiB,UAAA;EAAA;AAAA;AAAA,KAIhB,qBAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA;IACE,EAAA;IACA,IAAA;IACA,UAAA,EAAY,SAAA;IACZ,KAAA;EAAA;AAAA;AAAA,KAIQ,gBAAA,0BACR,sBAAA,CAAuB,SAAA,IACvB,sBAAA,CAAuB,UAAA,IACvB,qBAAA;AAAA,KAEC,wBAAA,0BACD,UAAA,CAAW,sBAAA,CAAuB,SAAA,KAClC,UAAA,CAAW,sBAAA,CAAuB,UAAA,KAClC,UAAA,CAAW,qBAAA;AAAA,KAEH,sBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IACE,MAAA,EAAQ,SAAA;IACR,IAAA;IACA,MAAA,EAAQ,MAAA;IACR,QAAA,EAAU,QAAA;IACV,KAAA,EAAO,UAAA;EAAA;AAAA;AAAA,KAIN,8BAAA,cAA4C,UAAA,CAC/C,sBAAA,CAAuB,SAAA;;;AA3EmD;;KAkFhE,mBAAA,GACR,2BAAA,GACA,2BAAA,GACA,0BAAA;;KAGQ,2BAAA,GACR,UAAA,CAAW,2BAAA,IACX,UAAA,CAAW,2BAAA,IACX,UAAA,CAAW,0BAAA;;;;KAKH,gBAAA;EAAqB,EAAA;EAAa,KAAA;EAAgB,IAAA;AAAA;;KAGlD,wBAAA,GAA2B,UAAA,CAAW,gBAAA;;;;KAKtC,iBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IACE,KAAA;IASA,IAAA;IACA,IAAA;IACA,MAAA;IACA,QAAA,EAAU,MAAA;IACV,UAAA;IACA,IAAA;EAAA;AAAA;AA3GW;AAAA,KAgHH,yBAAA,GAA4B,UAAA,CAAW,iBAAA;;;;;KAMvC,mBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IAAA,CAAS,WAAA;EAAA;AAAA;AAAA,KAGN,gBAAA,qBACiB,UAAA,GAAa,UAAA,wCAEnB,UAAA;EAIV,MAAA,EAAQ,iBAAA,CAAkB,UAAA;EAC1B,OAAA,EAAS,kBAAA,CAAmB,WAAA;EAC5B,MAAA,EAAQ,iBAAA,CAAkB,WAAA;EAC1B,KAAA,EAAO,gBAAA;EACP,QAAA,EAAU,mBAAA;EACV,gBAAA,EAAkB,wBAAA;EAClB,KAAA,EAAO,gBAAA,CAAiB,UAAA,EAAY,WAAA;EACpC,WAAA,EAAa,sBAAA,CAAuB,UAAA;EACpC,MAAA,EAAQ,iBAAA;AAAA,EACR,WAAA,SAAoB,UAAA,KAAe,WAAA,WAAsB,WAAA,IAC3D,gBAAA,GACA,mBAAA,GACA,mBAAA;AAAA,KAEC,yBAAA,qBACiB,UAAA,GAAa,UAAA,wCAEnB,UAAA;EAIV,MAAA,EAAQ,yBAAA,CAA0B,UAAA;EAClC,OAAA,EAAS,0BAAA,CAA2B,WAAA;EACpC,MAAA,EAAQ,yBAAA,CAA0B,WAAA;EAClC,KAAA,EAAO,wBAAA;EACP,QAAA,EAAU,2BAAA;EACV,gBAAA,EAAkB,gCAAA;EAClB,MAAA,EAAQ,yBAAA;EACR,KAAA,EAAO,wBAAA,CAAyB,UAAA,EAAY,WAAA;EAC5C,WAAA,EAAa,8BAAA,CAA+B,UAAA;AAAA,EAC5C,WAAA,SAAoB,UAAA,KAAe,WAAA,WAAsB,WAAA,IAC3D,wBAAA,GACA,mBAAA,GACA,mBAAA;AAAA,KAEQ,mBAAA,qBACU,UAAA,GAAa,UAAA,iFAGnB,UAAA,2BAEZ,cAAA,CACF,UAAA,gBACI,yBAAA,CACE,WAAA,EACA,UAAA,EACA,WAAA,EACA,WAAA,IAEF,gBAAA,CAAiB,WAAA,EAAa,UAAA,EAAY,WAAA,EAAa,WAAA"}
+{"version":3,"file":"types.stream.d.ts","names":[],"sources":["../src/types.stream.ts"],"mappings":";;;;;;AAaA;;;;;AAYA;;;;KAZY,UAAA;AAAA,KAYA,gBAAA;AAAA,KAEP,oBAAA;EACH,IAAA;EAAA,CACC,GAAA;AAAA;AAAA,KAGE,UAAA;EAA4B,EAAA;EAAa,KAAA;EAAe,IAAA;AAAA;EAEzD,EAAA,GAAK,MAAA;EACL,KAAA,EAAO,MAAA,eAAqB,MAAA;EAC5B,IAAA,EAAM,MAAA;AAAA;;;;KAME,iBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA,EAAM,SAAA;AAAA;;KAII,yBAAA,cAAuC,UAAA,CACjD,iBAAA,CAAkB,SAAA;;;;KAMR,wBAAA;EACV,KAAA;EAGA,IAAA,GAAO,OAAA,EAAS,OAAA,EAAS,MAAA,EAAQ,oBAAA;AAAA;;KAIvB,gCAAA,GACV,UAAA,CAAW,wBAAA;;;;KAKD,mBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IAAQ,MAAA;IAAgB,SAAA;EAAA;AAAA;;;;KAMd,gBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IAAQ,KAAA;IAAe,OAAA;EAAA;AAAA;;KAIb,wBAAA,GAA2B,UAAA,CAAW,gBAAA;;;;;;KAOtC,kBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IAAA,CAAS,IAAA,WAAe,UAAA;EAAA;AAAA;AA1B1B;AAAA,KA8BY,0BAAA,eAAyC,UAAA,CACnD,kBAAA,CAAmB,UAAA;;;;KAMT,iBAAA;EAAyB,KAAA;EAAiB,IAAA,EAAM,CAAA;AAAA;;KAGhD,yBAAA,MAA+B,UAAA,CAAW,iBAAA,CAAkB,CAAA;AAAA,KAEnE,2BAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA;IAAA,CAAS,SAAA;MAAsB,QAAA;IAAA;EAAA;AAAA;AAAA,KAE5B,2BAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA,EAAM,OAAA;AAAA;AAAA,KAEH,0BAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA,EAAM,OAAA;AAAA;AAAA,KAGH,sBAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA;IACE,EAAA;IACA,IAAA;IACA,UAAA,EAAY,SAAA;IACZ,KAAA,EAAO,SAAA;IACP,QAAA;EAAA;AAAA;AAAA,KAIC,sBAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA;IACE,EAAA;IACA,IAAA;IACA,UAAA,EAAY,SAAA;IACZ,MAAA,WAAiB,UAAA;EAAA;AAAA;AAAA,KAIhB,qBAAA;EACH,EAAA;EACA,KAAA;EACA,IAAA;IACE,EAAA;IACA,IAAA;IACA,UAAA,EAAY,SAAA;IACZ,KAAA;EAAA;AAAA;AAAA,KAIQ,gBAAA,0BACR,sBAAA,CAAuB,SAAA,IACvB,sBAAA,CAAuB,UAAA,IACvB,qBAAA;AAAA,KAEC,wBAAA,0BACD,UAAA,CAAW,sBAAA,CAAuB,SAAA,KAClC,UAAA,CAAW,sBAAA,CAAuB,UAAA,KAClC,UAAA,CAAW,qBAAA;AAAA,KAEH,sBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IACE,MAAA,EAAQ,SAAA;IACR,IAAA;IACA,MAAA,EAAQ,MAAA;IACR,QAAA,EAAU,QAAA;IACV,KAAA,EAAO,UAAA;EAAA;AAAA;AAAA,KAIN,8BAAA,cAA4C,UAAA,CAC/C,sBAAA,CAAuB,SAAA;;;AA3EmD;;KAkFhE,mBAAA,GACR,2BAAA,GACA,2BAAA,GACA,0BAAA;;KAGQ,2BAAA,GACR,UAAA,CAAW,2BAAA,IACX,UAAA,CAAW,2BAAA,IACX,UAAA,CAAW,0BAAA;;;;KAKH,gBAAA;EAAqB,EAAA;EAAa,KAAA;EAAgB,IAAA;AAAA;;KAGlD,wBAAA,GAA2B,UAAA,CAAW,gBAAA;;;;KAKtC,iBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IACE,KAAA;IASA,IAAA;IACA,IAAA;IACA,MAAA;IACA,QAAA,EAAU,MAAA;IACV,UAAA;IACA,IAAA;EAAA;AAAA;AA3GW;AAAA,KAgHH,yBAAA,GAA4B,UAAA,CAAW,iBAAA;;;;;KAMvC,mBAAA;EACV,EAAA;EACA,KAAA;EACA,IAAA;IAAA,CAAS,WAAA;EAAA;AAAA;AAAA,KAGC,mBAAA;EAEN,KAAA;EACA,UAAA;EACA,IAAA;EACA,KAAA;AAAA;EAGA,KAAA;EACA,UAAA;EACA,IAAA;EACA,IAAA;AAAA;EAGA,KAAA;EACA,UAAA;EACA,IAAA;EACA,MAAA;AAAA;EAGA,KAAA;EACA,UAAA;EACA,IAAA;EACA,KAAA;AAAA;AAAA,KAGM,gBAAA;EACV,KAAA;EACA,IAAA,EAAM,mBAAA;AAAA;AAAA,KAGI,YAAA;EACV,UAAA;EACA,IAAA;EACA,KAAA;EACA,KAAA;EACA,IAAA;EACA,MAAA;EACA,KAAA;AAAA;AA7HF;AAAA,KAiIY,wBAAA,GAA2B,UAAA,CAAW,gBAAA;AAAA,KAE7C,gBAAA,qBACiB,UAAA,GAAa,UAAA,wCAEnB,UAAA;EAIV,MAAA,EAAQ,iBAAA,CAAkB,UAAA;EAC1B,OAAA,EAAS,kBAAA,CAAmB,WAAA;EAC5B,MAAA,EAAQ,iBAAA,CAAkB,WAAA;EAC1B,KAAA,EAAO,gBAAA;EACP,QAAA,EAAU,mBAAA;EACV,gBAAA,EAAkB,wBAAA;EAClB,KAAA,EAAO,gBAAA,CAAiB,UAAA,EAAY,WAAA;EACpC,WAAA,EAAa,sBAAA,CAAuB,UAAA;EACpC,MAAA,EAAQ,iBAAA;EACR,KAAA,EAAO,gBAAA;AAAA,EACP,WAAA,SAAoB,UAAA,KAAe,WAAA,WAAsB,WAAA,IAC3D,gBAAA,GACA,mBAAA,GACA,mBAAA;AAAA,KAEC,yBAAA,qBACiB,UAAA,GAAa,UAAA,wCAEnB,UAAA;EAIV,MAAA,EAAQ,yBAAA,CAA0B,UAAA;EAClC,OAAA,EAAS,0BAAA,CAA2B,WAAA;EACpC,MAAA,EAAQ,yBAAA,CAA0B,WAAA;EAClC,KAAA,EAAO,wBAAA;EACP,QAAA,EAAU,2BAAA;EACV,gBAAA,EAAkB,gCAAA;EAClB,MAAA,EAAQ,yBAAA;EACR,KAAA,EAAO,wBAAA,CAAyB,UAAA,EAAY,WAAA;EAC5C,WAAA,EAAa,8BAAA,CAA+B,UAAA;EAC5C,KAAA,EAAO,wBAAA;AAAA,EACP,WAAA,SAAoB,UAAA,KAAe,WAAA,WAAsB,WAAA,IAC3D,wBAAA,GACA,mBAAA,GACA,mBAAA;AAAA,KAEQ,mBAAA,qBACU,UAAA,GAAa,UAAA,iFAGnB,UAAA,2BAEZ,cAAA,CACF,UAAA,gBACI,yBAAA,CACE,WAAA,EACA,UAAA,EACA,WAAA,EACA,WAAA,IAEF,gBAAA,CAAiB,WAAA,EAAa,UAAA,EAAY,WAAA,EAAa,WAAA"}

package/dist/ui/branching.cjs

@@ -1,3 +1,4 @@
+const require_utils = require('./utils.cjs');
 
 //#region src/ui/branching.ts
 function getBranchSequence(history) {
@@ -147,7 +148,30 @@ function getBranchContext(branch, history) {
 		threadHead: flatHistory.at(-1)
 	};
 }
+function getMessagesMetadataMap(options) {
+	const currentValues = options.branchContext.threadHead?.values ?? options.initialValues ?? {};
+	const alreadyShown = /* @__PURE__ */ new Set();
+	return options.getMessages(currentValues).map((message, idx) => {
+		const messageId = message.id ?? idx;
+		const firstSeenState = require_utils.findLast(options.history ?? [], (state) => options.getMessages(state.values).map((m, idx) => m.id ?? idx).includes(messageId));
+		const checkpointId = firstSeenState?.checkpoint?.checkpoint_id;
+		let branch = checkpointId != null ? options.branchContext.branchByCheckpoint[checkpointId] : void 0;
+		if (!branch?.branch?.length) branch = void 0;
+		const optionsShown = branch?.branchOptions?.flat(2).join(",");
+		if (optionsShown) {
+			if (alreadyShown.has(optionsShown)) branch = void 0;
+			alreadyShown.add(optionsShown);
+		}
+		return {
+			messageId: messageId.toString(),
+			firstSeenState,
+			branch: branch?.branch,
+			branchOptions: branch?.branchOptions
+		};
+	});
+}
 
 //#endregion
 exports.getBranchContext = getBranchContext;
+exports.getMessagesMetadataMap = getMessagesMetadataMap;
 //# sourceMappingURL=branching.cjs.map

package/dist/ui/branching.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"branching.cjs","names":[],"sources":["../../src/ui/branching.ts"],"sourcesContent":["import type { ThreadState } from \"../schema.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ninterface Node<StateType = any> {\n  type: \"node\";\n  value: ThreadState<StateType>;\n  path: string[];\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ninterface Fork<StateType = any> {\n  type: \"fork\";\n  items: Array<Sequence<StateType>>;\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport interface Sequence<StateType = any> {\n  type: \"sequence\";\n  items: Array<Node<StateType> | Fork<StateType>>;\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ninterface ValidFork<StateType = any> {\n  type: \"fork\";\n  items: Array<ValidSequence<StateType>>;\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ninterface ValidSequence<StateType = any> {\n  type: \"sequence\";\n  items: [Node<StateType>, ...(Node<StateType> | ValidFork<StateType>)[]];\n}\n\nexport function getBranchSequence<StateType extends Record<string, unknown>>(\n  history: ThreadState<StateType>[]\n) {\n  const nodeIds = new Set<string>();\n  const childrenMap: Record<string, ThreadState<StateType>[]> = {};\n\n  // Short circuit if there's only a singular one state\n  if (history.length <= 1) {\n    return {\n      rootSequence: {\n        type: \"sequence\",\n        items: history.map((value) => ({ type: \"node\", value, path: [] })),\n      } satisfies Sequence<StateType>,\n      paths: [],\n    };\n  }\n\n  // First pass - collect nodes for each checkpoint\n  history.forEach((state) => {\n    const checkpointId = state.parent_checkpoint?.checkpoint_id ?? \"$\";\n    childrenMap[checkpointId] ??= [];\n    childrenMap[checkpointId].push(state);\n\n    if (state.checkpoint?.checkpoint_id != null) {\n      nodeIds.add(state.checkpoint.checkpoint_id);\n    }\n  });\n\n  // If dealing with partial history, take the branch\n  // with the latest checkpoint and mark it as the root.\n  const maxId = (...ids: (string | null)[]) =>\n    ids\n      .filter((i): i is string => i != null)\n      .sort((a, b) => a.localeCompare(b))\n      .at(-1)!;\n\n  const lastOrphanedNode =\n    childrenMap.$ == null\n      ? Object.keys(childrenMap)\n          .filter((parentId) => !nodeIds.has(parentId))\n          .map((parentId) => {\n            const queue: string[] = [parentId];\n            const seen = new Set<string>();\n\n            let lastId = parentId;\n\n            while (queue.length > 0) {\n              const current = queue.shift()!;\n\n              if (seen.has(current)) continue;\n              seen.add(current);\n\n              const children = (childrenMap[current] ?? []).flatMap(\n                (i) => i.checkpoint?.checkpoint_id ?? []\n              );\n\n              lastId = maxId(lastId, ...children);\n              queue.push(...children);\n            }\n\n            return { parentId, lastId };\n          })\n          .sort((a, b) => a.lastId.localeCompare(b.lastId))\n          .at(-1)?.parentId\n      : undefined;\n\n  if (lastOrphanedNode != null) childrenMap.$ = childrenMap[lastOrphanedNode];\n\n  // Second pass - create a tree of sequences\n  type Task = { id: string; sequence: Sequence; path: string[] };\n  const rootSequence: Sequence = { type: \"sequence\", items: [] };\n  const queue: Task[] = [{ id: \"$\", sequence: rootSequence, path: [] }];\n\n  const paths: string[][] = [];\n\n  const visited = new Set<string>();\n  while (queue.length > 0) {\n    const task = queue.shift()!;\n    if (visited.has(task.id)) continue;\n    visited.add(task.id);\n\n    const children = childrenMap[task.id];\n    if (children == null || children.length === 0) continue;\n\n    // If we've encountered a fork (2+ children), push the fork\n    // to the sequence and add a new sequence for each child\n    let fork: Fork | undefined;\n    if (children.length > 1) {\n      fork = { type: \"fork\", items: [] };\n      task.sequence.items.push(fork);\n    }\n\n    for (const value of children) {\n      const id = value.checkpoint?.checkpoint_id;\n      if (id == null) continue;\n\n      let { sequence } = task;\n      let { path } = task;\n      if (fork != null) {\n        sequence = { type: \"sequence\", items: [] };\n        fork.items.unshift(sequence);\n\n        path = path.slice();\n        path.push(id);\n        paths.push(path);\n      }\n\n      sequence.items.push({ type: \"node\", value, path });\n      queue.push({ id, sequence, path });\n    }\n  }\n\n  return { rootSequence, paths };\n}\n\nconst PATH_SEP = \">\";\nconst ROOT_ID = \"$\";\n\n// Get flat view\nexport function getBranchView<StateType extends Record<string, unknown>>(\n  sequence: Sequence<StateType>,\n  paths: string[][],\n  branch: string\n) {\n  const path = branch.split(PATH_SEP);\n  const pathMap: Record<string, string[][]> = {};\n\n  for (const path of paths) {\n    const parent = path.at(-2) ?? ROOT_ID;\n    pathMap[parent] ??= [];\n    pathMap[parent].unshift(path);\n  }\n\n  const history: ThreadState<StateType>[] = [];\n  const branchByCheckpoint: Record<\n    string,\n    { branch: string | undefined; branchOptions: string[] | undefined }\n  > = {};\n\n  const forkStack = path.slice();\n  const queue: (Node<StateType> | Fork<StateType>)[] = [...sequence.items];\n\n  while (queue.length > 0) {\n    const item = queue.shift()!;\n\n    if (item.type === \"node\") {\n      history.push(item.value);\n      const checkpointId = item.value.checkpoint?.checkpoint_id;\n      if (checkpointId == null) continue;\n\n      branchByCheckpoint[checkpointId] = {\n        branch: item.path.join(PATH_SEP),\n        branchOptions: (item.path.length > 0\n          ? pathMap[item.path.at(-2) ?? ROOT_ID] ?? []\n          : []\n        ).map((p) => p.join(PATH_SEP)),\n      };\n    }\n    if (item.type === \"fork\") {\n      const forkId = forkStack.shift();\n      const index =\n        forkId != null\n          ? item.items.findIndex((value) => {\n              const firstItem = value.items.at(0);\n              if (!firstItem || firstItem.type !== \"node\") return false;\n              return firstItem.value.checkpoint?.checkpoint_id === forkId;\n            })\n          : -1;\n\n      const nextItems = item.items.at(index)?.items ?? [];\n      queue.push(...nextItems);\n    }\n  }\n\n  return { history, branchByCheckpoint };\n}\n\nexport function getBranchContext<StateType extends Record<string, unknown>>(\n  branch: string,\n  history: ThreadState<StateType>[] | undefined\n) {\n  const { rootSequence: branchTree, paths } = getBranchSequence(history ?? []);\n  const { history: flatHistory, branchByCheckpoint } = getBranchView(\n    branchTree,\n    paths,\n    branch\n  );\n\n  return {\n    branchTree,\n    flatHistory,\n    branchByCheckpoint,\n    threadHead: flatHistory.at(-1),\n  };\n}\n"],"mappings":";;AAiCA,SAAgB,kBACd,SACA;CACA,MAAM,0BAAU,IAAI,KAAa;CACjC,MAAM,cAAwD,EAAE;AAGhE,KAAI,QAAQ,UAAU,EACpB,QAAO;EACL,cAAc;GACZ,MAAM;GACN,OAAO,QAAQ,KAAK,WAAW;IAAE,MAAM;IAAQ;IAAO,MAAM,EAAE;IAAE,EAAE;GACnE;EACD,OAAO,EAAE;EACV;AAIH,SAAQ,SAAS,UAAU;EACzB,MAAM,eAAe,MAAM,mBAAmB,iBAAiB;AAC/D,cAAY,kBAAkB,EAAE;AAChC,cAAY,cAAc,KAAK,MAAM;AAErC,MAAI,MAAM,YAAY,iBAAiB,KACrC,SAAQ,IAAI,MAAM,WAAW,cAAc;GAE7C;CAIF,MAAM,SAAS,GAAG,QAChB,IACG,QAAQ,MAAmB,KAAK,KAAK,CACrC,MAAM,GAAG,MAAM,EAAE,cAAc,EAAE,CAAC,CAClC,GAAG,GAAG;CAEX,MAAM,mBACJ,YAAY,KAAK,OACb,OAAO,KAAK,YAAY,CACrB,QAAQ,aAAa,CAAC,QAAQ,IAAI,SAAS,CAAC,CAC5C,KAAK,aAAa;EACjB,MAAM,QAAkB,CAAC,SAAS;EAClC,MAAM,uBAAO,IAAI,KAAa;EAE9B,IAAI,SAAS;AAEb,SAAO,MAAM,SAAS,GAAG;GACvB,MAAM,UAAU,MAAM,OAAO;AAE7B,OAAI,KAAK,IAAI,QAAQ,CAAE;AACvB,QAAK,IAAI,QAAQ;GAEjB,MAAM,YAAY,YAAY,YAAY,EAAE,EAAE,SAC3C,MAAM,EAAE,YAAY,iBAAiB,EAAE,CACzC;AAED,YAAS,MAAM,QAAQ,GAAG,SAAS;AACnC,SAAM,KAAK,GAAG,SAAS;;AAGzB,SAAO;GAAE;GAAU;GAAQ;GAC3B,CACD,MAAM,GAAG,MAAM,EAAE,OAAO,cAAc,EAAE,OAAO,CAAC,CAChD,GAAG,GAAG,EAAE,WACX;AAEN,KAAI,oBAAoB,KAAM,aAAY,IAAI,YAAY;CAI1D,MAAM,eAAyB;EAAE,MAAM;EAAY,OAAO,EAAE;EAAE;CAC9D,MAAM,QAAgB,CAAC;EAAE,IAAI;EAAK,UAAU;EAAc,MAAM,EAAE;EAAE,CAAC;CAErE,MAAM,QAAoB,EAAE;CAE5B,MAAM,0BAAU,IAAI,KAAa;AACjC,QAAO,MAAM,SAAS,GAAG;EACvB,MAAM,OAAO,MAAM,OAAO;AAC1B,MAAI,QAAQ,IAAI,KAAK,GAAG,CAAE;AAC1B,UAAQ,IAAI,KAAK,GAAG;EAEpB,MAAM,WAAW,YAAY,KAAK;AAClC,MAAI,YAAY,QAAQ,SAAS,WAAW,EAAG;EAI/C,IAAI;AACJ,MAAI,SAAS,SAAS,GAAG;AACvB,UAAO;IAAE,MAAM;IAAQ,OAAO,EAAE;IAAE;AAClC,QAAK,SAAS,MAAM,KAAK,KAAK;;AAGhC,OAAK,MAAM,SAAS,UAAU;GAC5B,MAAM,KAAK,MAAM,YAAY;AAC7B,OAAI,MAAM,KAAM;GAEhB,IAAI,EAAE,aAAa;GACnB,IAAI,EAAE,SAAS;AACf,OAAI,QAAQ,MAAM;AAChB,eAAW;KAAE,MAAM;KAAY,OAAO,EAAE;KAAE;AAC1C,SAAK,MAAM,QAAQ,SAAS;AAE5B,WAAO,KAAK,OAAO;AACnB,SAAK,KAAK,GAAG;AACb,UAAM,KAAK,KAAK;;AAGlB,YAAS,MAAM,KAAK;IAAE,MAAM;IAAQ;IAAO;IAAM,CAAC;AAClD,SAAM,KAAK;IAAE;IAAI;IAAU;IAAM,CAAC;;;AAItC,QAAO;EAAE;EAAc;EAAO;;AAGhC,MAAM,WAAW;AACjB,MAAM,UAAU;AAGhB,SAAgB,cACd,UACA,OACA,QACA;CACA,MAAM,OAAO,OAAO,MAAM,SAAS;CACnC,MAAM,UAAsC,EAAE;AAE9C,MAAK,MAAM,QAAQ,OAAO;EACxB,MAAM,SAAS,KAAK,GAAG,GAAG,IAAI;AAC9B,UAAQ,YAAY,EAAE;AACtB,UAAQ,QAAQ,QAAQ,KAAK;;CAG/B,MAAM,UAAoC,EAAE;CAC5C,MAAM,qBAGF,EAAE;CAEN,MAAM,YAAY,KAAK,OAAO;CAC9B,MAAM,QAA+C,CAAC,GAAG,SAAS,MAAM;AAExE,QAAO,MAAM,SAAS,GAAG;EACvB,MAAM,OAAO,MAAM,OAAO;AAE1B,MAAI,KAAK,SAAS,QAAQ;AACxB,WAAQ,KAAK,KAAK,MAAM;GACxB,MAAM,eAAe,KAAK,MAAM,YAAY;AAC5C,OAAI,gBAAgB,KAAM;AAE1B,sBAAmB,gBAAgB;IACjC,QAAQ,KAAK,KAAK,KAAK,SAAS;IAChC,gBAAgB,KAAK,KAAK,SAAS,IAC/B,QAAQ,KAAK,KAAK,GAAG,GAAG,IAAI,YAAY,EAAE,GAC1C,EAAE,EACJ,KAAK,MAAM,EAAE,KAAK,SAAS,CAAC;IAC/B;;AAEH,MAAI,KAAK,SAAS,QAAQ;GACxB,MAAM,SAAS,UAAU,OAAO;GAChC,MAAM,QACJ,UAAU,OACN,KAAK,MAAM,WAAW,UAAU;IAC9B,MAAM,YAAY,MAAM,MAAM,GAAG,EAAE;AACnC,QAAI,CAAC,aAAa,UAAU,SAAS,OAAQ,QAAO;AACpD,WAAO,UAAU,MAAM,YAAY,kBAAkB;KACrD,GACF;GAEN,MAAM,YAAY,KAAK,MAAM,GAAG,MAAM,EAAE,SAAS,EAAE;AACnD,SAAM,KAAK,GAAG,UAAU;;;AAI5B,QAAO;EAAE;EAAS;EAAoB;;AAGxC,SAAgB,iBACd,QACA,SACA;CACA,MAAM,EAAE,cAAc,YAAY,UAAU,kBAAkB,WAAW,EAAE,CAAC;CAC5E,MAAM,EAAE,SAAS,aAAa,uBAAuB,cACnD,YACA,OACA,OACD;AAED,QAAO;EACL;EACA;EACA;EACA,YAAY,YAAY,GAAG,GAAG;EAC/B"}
+{"version":3,"file":"branching.cjs","names":["findLast"],"sources":["../../src/ui/branching.ts"],"sourcesContent":["import type { ThreadState } from \"../schema.js\";\nimport { Message } from \"../types.messages.js\";\nimport { findLast } from \"./utils.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ninterface Node<StateType = any> {\n  type: \"node\";\n  value: ThreadState<StateType>;\n  path: string[];\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ninterface Fork<StateType = any> {\n  type: \"fork\";\n  items: Array<Sequence<StateType>>;\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport interface Sequence<StateType = any> {\n  type: \"sequence\";\n  items: Array<Node<StateType> | Fork<StateType>>;\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ninterface ValidFork<StateType = any> {\n  type: \"fork\";\n  items: Array<ValidSequence<StateType>>;\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ninterface ValidSequence<StateType = any> {\n  type: \"sequence\";\n  items: [Node<StateType>, ...(Node<StateType> | ValidFork<StateType>)[]];\n}\n\nexport function getBranchSequence<StateType extends Record<string, unknown>>(\n  history: ThreadState<StateType>[]\n) {\n  const nodeIds = new Set<string>();\n  const childrenMap: Record<string, ThreadState<StateType>[]> = {};\n\n  // Short circuit if there's only a singular one state\n  if (history.length <= 1) {\n    return {\n      rootSequence: {\n        type: \"sequence\",\n        items: history.map((value) => ({ type: \"node\", value, path: [] })),\n      } satisfies Sequence<StateType>,\n      paths: [],\n    };\n  }\n\n  // First pass - collect nodes for each checkpoint\n  history.forEach((state) => {\n    const checkpointId = state.parent_checkpoint?.checkpoint_id ?? \"$\";\n    childrenMap[checkpointId] ??= [];\n    childrenMap[checkpointId].push(state);\n\n    if (state.checkpoint?.checkpoint_id != null) {\n      nodeIds.add(state.checkpoint.checkpoint_id);\n    }\n  });\n\n  // If dealing with partial history, take the branch\n  // with the latest checkpoint and mark it as the root.\n  const maxId = (...ids: (string | null)[]) =>\n    ids\n      .filter((i): i is string => i != null)\n      .sort((a, b) => a.localeCompare(b))\n      .at(-1)!;\n\n  const lastOrphanedNode =\n    childrenMap.$ == null\n      ? Object.keys(childrenMap)\n          .filter((parentId) => !nodeIds.has(parentId))\n          .map((parentId) => {\n            const queue: string[] = [parentId];\n            const seen = new Set<string>();\n\n            let lastId = parentId;\n\n            while (queue.length > 0) {\n              const current = queue.shift()!;\n\n              if (seen.has(current)) continue;\n              seen.add(current);\n\n              const children = (childrenMap[current] ?? []).flatMap(\n                (i) => i.checkpoint?.checkpoint_id ?? []\n              );\n\n              lastId = maxId(lastId, ...children);\n              queue.push(...children);\n            }\n\n            return { parentId, lastId };\n          })\n          .sort((a, b) => a.lastId.localeCompare(b.lastId))\n          .at(-1)?.parentId\n      : undefined;\n\n  if (lastOrphanedNode != null) childrenMap.$ = childrenMap[lastOrphanedNode];\n\n  // Second pass - create a tree of sequences\n  type Task = { id: string; sequence: Sequence; path: string[] };\n  const rootSequence: Sequence = { type: \"sequence\", items: [] };\n  const queue: Task[] = [{ id: \"$\", sequence: rootSequence, path: [] }];\n\n  const paths: string[][] = [];\n\n  const visited = new Set<string>();\n  while (queue.length > 0) {\n    const task = queue.shift()!;\n    if (visited.has(task.id)) continue;\n    visited.add(task.id);\n\n    const children = childrenMap[task.id];\n    if (children == null || children.length === 0) continue;\n\n    // If we've encountered a fork (2+ children), push the fork\n    // to the sequence and add a new sequence for each child\n    let fork: Fork | undefined;\n    if (children.length > 1) {\n      fork = { type: \"fork\", items: [] };\n      task.sequence.items.push(fork);\n    }\n\n    for (const value of children) {\n      const id = value.checkpoint?.checkpoint_id;\n      if (id == null) continue;\n\n      let { sequence } = task;\n      let { path } = task;\n      if (fork != null) {\n        sequence = { type: \"sequence\", items: [] };\n        fork.items.unshift(sequence);\n\n        path = path.slice();\n        path.push(id);\n        paths.push(path);\n      }\n\n      sequence.items.push({ type: \"node\", value, path });\n      queue.push({ id, sequence, path });\n    }\n  }\n\n  return { rootSequence, paths };\n}\n\nconst PATH_SEP = \">\";\nconst ROOT_ID = \"$\";\n\ntype BranchByCheckpoint = Record<\n  string,\n  { branch: string | undefined; branchOptions: string[] | undefined }\n>;\n\n// Get flat view\nexport function getBranchView<StateType extends Record<string, unknown>>(\n  sequence: Sequence<StateType>,\n  paths: string[][],\n  branch: string\n) {\n  const path = branch.split(PATH_SEP);\n  const pathMap: Record<string, string[][]> = {};\n\n  for (const path of paths) {\n    const parent = path.at(-2) ?? ROOT_ID;\n    pathMap[parent] ??= [];\n    pathMap[parent].unshift(path);\n  }\n\n  const history: ThreadState<StateType>[] = [];\n  const branchByCheckpoint: BranchByCheckpoint = {};\n\n  const forkStack = path.slice();\n  const queue: (Node<StateType> | Fork<StateType>)[] = [...sequence.items];\n\n  while (queue.length > 0) {\n    const item = queue.shift()!;\n\n    if (item.type === \"node\") {\n      history.push(item.value);\n      const checkpointId = item.value.checkpoint?.checkpoint_id;\n      if (checkpointId == null) continue;\n\n      branchByCheckpoint[checkpointId] = {\n        branch: item.path.join(PATH_SEP),\n        branchOptions: (item.path.length > 0\n          ? pathMap[item.path.at(-2) ?? ROOT_ID] ?? []\n          : []\n        ).map((p) => p.join(PATH_SEP)),\n      };\n    }\n    if (item.type === \"fork\") {\n      const forkId = forkStack.shift();\n      const index =\n        forkId != null\n          ? item.items.findIndex((value) => {\n              const firstItem = value.items.at(0);\n              if (!firstItem || firstItem.type !== \"node\") return false;\n              return firstItem.value.checkpoint?.checkpoint_id === forkId;\n            })\n          : -1;\n\n      const nextItems = item.items.at(index)?.items ?? [];\n      queue.push(...nextItems);\n    }\n  }\n\n  return { history, branchByCheckpoint };\n}\n\nexport function getBranchContext<StateType extends Record<string, unknown>>(\n  branch: string,\n  history: ThreadState<StateType>[] | undefined\n) {\n  const { rootSequence: branchTree, paths } = getBranchSequence(history ?? []);\n  const { history: flatHistory, branchByCheckpoint } = getBranchView(\n    branchTree,\n    paths,\n    branch\n  );\n\n  return {\n    branchTree,\n    flatHistory,\n    branchByCheckpoint,\n    threadHead: flatHistory.at(-1),\n  };\n}\n\nexport function getMessagesMetadataMap<\n  StateType extends Record<string, unknown>\n>(options: {\n  initialValues: StateType | null | undefined;\n  history: ThreadState<StateType>[] | null | undefined;\n  getMessages: (values: StateType) => Message[];\n\n  branchContext: {\n    threadHead: ThreadState<StateType> | undefined;\n    branchByCheckpoint: BranchByCheckpoint;\n  };\n}) {\n  const currentValues =\n    options.branchContext.threadHead?.values ??\n    options.initialValues ??\n    ({} as StateType);\n\n  const alreadyShown = new Set<string>();\n  return options.getMessages(currentValues).map((message, idx) => {\n    const messageId = message.id ?? idx;\n\n    // Find the first checkpoint where the message was seen\n    const firstSeenState = findLast(options.history ?? [], (state) =>\n      options\n        .getMessages(state.values)\n        .map((m, idx) => m.id ?? idx)\n        .includes(messageId)\n    );\n\n    const checkpointId = firstSeenState?.checkpoint?.checkpoint_id;\n    let branch =\n      checkpointId != null\n        ? options.branchContext.branchByCheckpoint[checkpointId]\n        : undefined;\n    if (!branch?.branch?.length) branch = undefined;\n\n    // serialize branches\n    const optionsShown = branch?.branchOptions?.flat(2).join(\",\");\n    if (optionsShown) {\n      if (alreadyShown.has(optionsShown)) branch = undefined;\n      alreadyShown.add(optionsShown);\n    }\n\n    return {\n      messageId: messageId.toString(),\n      firstSeenState,\n\n      branch: branch?.branch,\n      branchOptions: branch?.branchOptions,\n    };\n  });\n}\n"],"mappings":";;;AAmCA,SAAgB,kBACd,SACA;CACA,MAAM,0BAAU,IAAI,KAAa;CACjC,MAAM,cAAwD,EAAE;AAGhE,KAAI,QAAQ,UAAU,EACpB,QAAO;EACL,cAAc;GACZ,MAAM;GACN,OAAO,QAAQ,KAAK,WAAW;IAAE,MAAM;IAAQ;IAAO,MAAM,EAAE;IAAE,EAAE;GACnE;EACD,OAAO,EAAE;EACV;AAIH,SAAQ,SAAS,UAAU;EACzB,MAAM,eAAe,MAAM,mBAAmB,iBAAiB;AAC/D,cAAY,kBAAkB,EAAE;AAChC,cAAY,cAAc,KAAK,MAAM;AAErC,MAAI,MAAM,YAAY,iBAAiB,KACrC,SAAQ,IAAI,MAAM,WAAW,cAAc;GAE7C;CAIF,MAAM,SAAS,GAAG,QAChB,IACG,QAAQ,MAAmB,KAAK,KAAK,CACrC,MAAM,GAAG,MAAM,EAAE,cAAc,EAAE,CAAC,CAClC,GAAG,GAAG;CAEX,MAAM,mBACJ,YAAY,KAAK,OACb,OAAO,KAAK,YAAY,CACrB,QAAQ,aAAa,CAAC,QAAQ,IAAI,SAAS,CAAC,CAC5C,KAAK,aAAa;EACjB,MAAM,QAAkB,CAAC,SAAS;EAClC,MAAM,uBAAO,IAAI,KAAa;EAE9B,IAAI,SAAS;AAEb,SAAO,MAAM,SAAS,GAAG;GACvB,MAAM,UAAU,MAAM,OAAO;AAE7B,OAAI,KAAK,IAAI,QAAQ,CAAE;AACvB,QAAK,IAAI,QAAQ;GAEjB,MAAM,YAAY,YAAY,YAAY,EAAE,EAAE,SAC3C,MAAM,EAAE,YAAY,iBAAiB,EAAE,CACzC;AAED,YAAS,MAAM,QAAQ,GAAG,SAAS;AACnC,SAAM,KAAK,GAAG,SAAS;;AAGzB,SAAO;GAAE;GAAU;GAAQ;GAC3B,CACD,MAAM,GAAG,MAAM,EAAE,OAAO,cAAc,EAAE,OAAO,CAAC,CAChD,GAAG,GAAG,EAAE,WACX;AAEN,KAAI,oBAAoB,KAAM,aAAY,IAAI,YAAY;CAI1D,MAAM,eAAyB;EAAE,MAAM;EAAY,OAAO,EAAE;EAAE;CAC9D,MAAM,QAAgB,CAAC;EAAE,IAAI;EAAK,UAAU;EAAc,MAAM,EAAE;EAAE,CAAC;CAErE,MAAM,QAAoB,EAAE;CAE5B,MAAM,0BAAU,IAAI,KAAa;AACjC,QAAO,MAAM,SAAS,GAAG;EACvB,MAAM,OAAO,MAAM,OAAO;AAC1B,MAAI,QAAQ,IAAI,KAAK,GAAG,CAAE;AAC1B,UAAQ,IAAI,KAAK,GAAG;EAEpB,MAAM,WAAW,YAAY,KAAK;AAClC,MAAI,YAAY,QAAQ,SAAS,WAAW,EAAG;EAI/C,IAAI;AACJ,MAAI,SAAS,SAAS,GAAG;AACvB,UAAO;IAAE,MAAM;IAAQ,OAAO,EAAE;IAAE;AAClC,QAAK,SAAS,MAAM,KAAK,KAAK;;AAGhC,OAAK,MAAM,SAAS,UAAU;GAC5B,MAAM,KAAK,MAAM,YAAY;AAC7B,OAAI,MAAM,KAAM;GAEhB,IAAI,EAAE,aAAa;GACnB,IAAI,EAAE,SAAS;AACf,OAAI,QAAQ,MAAM;AAChB,eAAW;KAAE,MAAM;KAAY,OAAO,EAAE;KAAE;AAC1C,SAAK,MAAM,QAAQ,SAAS;AAE5B,WAAO,KAAK,OAAO;AACnB,SAAK,KAAK,GAAG;AACb,UAAM,KAAK,KAAK;;AAGlB,YAAS,MAAM,KAAK;IAAE,MAAM;IAAQ;IAAO;IAAM,CAAC;AAClD,SAAM,KAAK;IAAE;IAAI;IAAU;IAAM,CAAC;;;AAItC,QAAO;EAAE;EAAc;EAAO;;AAGhC,MAAM,WAAW;AACjB,MAAM,UAAU;AAQhB,SAAgB,cACd,UACA,OACA,QACA;CACA,MAAM,OAAO,OAAO,MAAM,SAAS;CACnC,MAAM,UAAsC,EAAE;AAE9C,MAAK,MAAM,QAAQ,OAAO;EACxB,MAAM,SAAS,KAAK,GAAG,GAAG,IAAI;AAC9B,UAAQ,YAAY,EAAE;AACtB,UAAQ,QAAQ,QAAQ,KAAK;;CAG/B,MAAM,UAAoC,EAAE;CAC5C,MAAM,qBAAyC,EAAE;CAEjD,MAAM,YAAY,KAAK,OAAO;CAC9B,MAAM,QAA+C,CAAC,GAAG,SAAS,MAAM;AAExE,QAAO,MAAM,SAAS,GAAG;EACvB,MAAM,OAAO,MAAM,OAAO;AAE1B,MAAI,KAAK,SAAS,QAAQ;AACxB,WAAQ,KAAK,KAAK,MAAM;GACxB,MAAM,eAAe,KAAK,MAAM,YAAY;AAC5C,OAAI,gBAAgB,KAAM;AAE1B,sBAAmB,gBAAgB;IACjC,QAAQ,KAAK,KAAK,KAAK,SAAS;IAChC,gBAAgB,KAAK,KAAK,SAAS,IAC/B,QAAQ,KAAK,KAAK,GAAG,GAAG,IAAI,YAAY,EAAE,GAC1C,EAAE,EACJ,KAAK,MAAM,EAAE,KAAK,SAAS,CAAC;IAC/B;;AAEH,MAAI,KAAK,SAAS,QAAQ;GACxB,MAAM,SAAS,UAAU,OAAO;GAChC,MAAM,QACJ,UAAU,OACN,KAAK,MAAM,WAAW,UAAU;IAC9B,MAAM,YAAY,MAAM,MAAM,GAAG,EAAE;AACnC,QAAI,CAAC,aAAa,UAAU,SAAS,OAAQ,QAAO;AACpD,WAAO,UAAU,MAAM,YAAY,kBAAkB;KACrD,GACF;GAEN,MAAM,YAAY,KAAK,MAAM,GAAG,MAAM,EAAE,SAAS,EAAE;AACnD,SAAM,KAAK,GAAG,UAAU;;;AAI5B,QAAO;EAAE;EAAS;EAAoB;;AAGxC,SAAgB,iBACd,QACA,SACA;CACA,MAAM,EAAE,cAAc,YAAY,UAAU,kBAAkB,WAAW,EAAE,CAAC;CAC5E,MAAM,EAAE,SAAS,aAAa,uBAAuB,cACnD,YACA,OACA,OACD;AAED,QAAO;EACL;EACA;EACA;EACA,YAAY,YAAY,GAAG,GAAG;EAC/B;;AAGH,SAAgB,uBAEd,SASC;CACD,MAAM,gBACJ,QAAQ,cAAc,YAAY,UAClC,QAAQ,iBACP,EAAE;CAEL,MAAM,+BAAe,IAAI,KAAa;AACtC,QAAO,QAAQ,YAAY,cAAc,CAAC,KAAK,SAAS,QAAQ;EAC9D,MAAM,YAAY,QAAQ,MAAM;EAGhC,MAAM,iBAAiBA,uBAAS,QAAQ,WAAW,EAAE,GAAG,UACtD,QACG,YAAY,MAAM,OAAO,CACzB,KAAK,GAAG,QAAQ,EAAE,MAAM,IAAI,CAC5B,SAAS,UAAU,CACvB;EAED,MAAM,eAAe,gBAAgB,YAAY;EACjD,IAAI,SACF,gBAAgB,OACZ,QAAQ,cAAc,mBAAmB,gBACzC;AACN,MAAI,CAAC,QAAQ,QAAQ,OAAQ,UAAS;EAGtC,MAAM,eAAe,QAAQ,eAAe,KAAK,EAAE,CAAC,KAAK,IAAI;AAC7D,MAAI,cAAc;AAChB,OAAI,aAAa,IAAI,aAAa,CAAE,UAAS;AAC7C,gBAAa,IAAI,aAAa;;AAGhC,SAAO;GACL,WAAW,UAAU,UAAU;GAC/B;GAEA,QAAQ,QAAQ;GAChB,eAAe,QAAQ;GACxB;GACD"}

package/dist/ui/branching.d.cts

@@ -1,4 +1,5 @@
 import { ThreadState } from "../schema.cjs";
+import { Message } from "../types.messages.cjs";
 
 //#region src/ui/branching.d.ts
 interface Node<StateType = any> {
@@ -14,6 +15,30 @@ interface Sequence<StateType = any> {
   type: "sequence";
   items: Array<Node<StateType> | Fork<StateType>>;
 }
+type BranchByCheckpoint = Record<string, {
+  branch: string | undefined;
+  branchOptions: string[] | undefined;
+}>;
+declare function getBranchContext<StateType extends Record<string, unknown>>(branch: string, history: ThreadState<StateType>[] | undefined): {
+  branchTree: Sequence<any>;
+  flatHistory: ThreadState<any>[];
+  branchByCheckpoint: BranchByCheckpoint;
+  threadHead: ThreadState<any> | undefined;
+};
+declare function getMessagesMetadataMap<StateType extends Record<string, unknown>>(options: {
+  initialValues: StateType | null | undefined;
+  history: ThreadState<StateType>[] | null | undefined;
+  getMessages: (values: StateType) => Message[];
+  branchContext: {
+    threadHead: ThreadState<StateType> | undefined;
+    branchByCheckpoint: BranchByCheckpoint;
+  };
+}): {
+  messageId: string;
+  firstSeenState: ThreadState<StateType> | undefined;
+  branch: string | undefined;
+  branchOptions: string[] | undefined;
+}[];
 //#endregion
-export { Sequence };
+export { Sequence, getBranchContext, getMessagesMetadataMap };
 //# sourceMappingURL=branching.d.cts.map

package/dist/ui/branching.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"branching.d.cts","names":[],"sources":["../../src/ui/branching.ts"],"mappings":";;;UAGU,IAAA;EACR,IAAA;EACA,KAAA,EAAO,WAAA,CAAY,SAAA;EACnB,IAAA;AAAA;AAAA,UAIQ,IAAA;EACR,IAAA;EACA,KAAA,EAAO,KAAA,CAAM,QAAA,CAAS,SAAA;AAAA;AAAA,UAIP,QAAA;EACf,IAAA;EACA,KAAA,EAAO,KAAA,CAAM,IAAA,CAAK,SAAA,IAAa,IAAA,CAAK,SAAA;AAAA"}
+{"version":3,"file":"branching.d.cts","names":[],"sources":["../../src/ui/branching.ts"],"mappings":";;;;UAKU,IAAA;EACR,IAAA;EACA,KAAA,EAAO,WAAA,CAAY,SAAA;EACnB,IAAA;AAAA;AAAA,UAIQ,IAAA;EACR,IAAA;EACA,KAAA,EAAO,KAAA,CAAM,QAAA,CAAS,SAAA;AAAA;AAAA,UAIP,QAAA;EACf,IAAA;EACA,KAAA,EAAO,KAAA,CAAM,IAAA,CAAK,SAAA,IAAa,IAAA,CAAK,SAAA;AAAA;AAAA,KAqIjC,kBAAA,GAAqB,MAAA;EAEtB,MAAA;EAA4B,aAAA;AAAA;AAAA,iBA2DhB,gBAAA,mBAAmC,MAAA,kBAAA,CACjD,MAAA,UACA,OAAA,EAAS,WAAA,CAAY,SAAA;;;;;;iBAiBP,sBAAA,mBACI,MAAA,kBAAA,CAClB,OAAA;EACA,aAAA,EAAe,SAAA;EACf,OAAA,EAAS,WAAA,CAAY,SAAA;EACrB,WAAA,GAAc,MAAA,EAAQ,SAAA,KAAc,OAAA;EAEpC,aAAA;IACE,UAAA,EAAY,WAAA,CAAY,SAAA;IACxB,kBAAA,EAAoB,kBAAA;EAAA;AAAA"}