@langchain/angular 0.0.1 → 0.1.0

package/dist/stream.custom.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"stream.custom.cjs","names":["StreamManager","MessageTupleManager","toMessageClass"],"sources":["../src/stream.custom.ts"],"sourcesContent":["import { signal, computed, effect } from \"@angular/core\";\nimport {\n  StreamManager,\n  MessageTupleManager,\n  extractInterrupts,\n  toMessageClass,\n  type EventStreamEvent,\n  type GetUpdateType,\n  type GetCustomEventType,\n  type GetInterruptType,\n  type GetConfigurableType,\n  type GetToolCallsType,\n  type AnyStreamCustomOptions,\n  type CustomSubmitOptions,\n} from \"@langchain/langgraph-sdk/ui\";\nimport { getToolCallsWithResults } from \"@langchain/langgraph-sdk/utils\";\nimport type { BagTemplate, Message, Interrupt } from \"@langchain/langgraph-sdk\";\n\nexport function useStreamCustom<\n  StateType extends Record<string, unknown> = Record<string, unknown>,\n  Bag extends BagTemplate = BagTemplate,\n>(options: AnyStreamCustomOptions<StateType, Bag>) {\n  type UpdateType = GetUpdateType<Bag, StateType>;\n  type CustomType = GetCustomEventType<Bag>;\n  type InterruptType = GetInterruptType<Bag>;\n  type ConfigurableType = GetConfigurableType<Bag>;\n  type ToolCallType = GetToolCallsType<StateType>;\n\n  const messageManager = new MessageTupleManager();\n  const stream = new StreamManager<StateType, Bag>(messageManager, {\n    throttle: options.throttle ?? false,\n    subagentToolNames: options.subagentToolNames,\n    filterSubagentMessages: options.filterSubagentMessages,\n    toMessage: toMessageClass,\n  });\n\n  const streamValues = signal<StateType | null>(stream.values);\n  const streamError = signal<unknown>(stream.error);\n  const isLoading = signal(stream.isLoading);\n\n  effect((onCleanup) => {\n    const unsubscribe = stream.subscribe(() => {\n      streamValues.set(stream.values);\n      streamError.set(stream.error);\n      isLoading.set(stream.isLoading);\n    });\n\n    onCleanup(() => unsubscribe());\n  });\n\n  let threadId: string | null = options.threadId ?? null;\n\n  const getMessages = (value: StateType): Message[] => {\n    const messagesKey = options.messagesKey ?? \"messages\";\n    return Array.isArray(value[messagesKey])\n      ? (value[messagesKey] as Message[])\n      : [];\n  };\n\n  const setMessages = (current: StateType, messages: Message[]): StateType => {\n    const messagesKey = options.messagesKey ?? \"messages\";\n    return { ...current, [messagesKey]: messages };\n  };\n\n  const historyValues = options.initialValues ?? ({} as StateType);\n\n  const historyMessages = getMessages(historyValues);\n  const shouldReconstructSubagents =\n    options.filterSubagentMessages &&\n    !stream.isLoading &&\n    historyMessages.length > 0;\n\n  effect(() => {\n    const loading = isLoading();\n    const hvMessages = getMessages(historyValues);\n    const should =\n      options.filterSubagentMessages && !loading && hvMessages.length > 0;\n    if (should) {\n      stream.reconstructSubagents(hvMessages, { skipIfPopulated: true });\n    }\n  });\n\n  if (shouldReconstructSubagents) {\n    stream.reconstructSubagents(historyMessages, { skipIfPopulated: true });\n  }\n\n  function switchThread(newThreadId: string | null) {\n    if (newThreadId !== threadId) {\n      threadId = newThreadId;\n      stream.clear();\n    }\n  }\n\n  function stop() {\n    return stream.stop(historyValues, { onStop: options.onStop });\n  }\n\n  async function submitDirect(\n    values: UpdateType | null | undefined,\n    submitOptions?: CustomSubmitOptions<StateType, ConfigurableType>,\n  ) {\n    const currentThreadId = options.threadId ?? null;\n    if (currentThreadId !== threadId) {\n      threadId = currentThreadId;\n      stream.clear();\n    }\n\n    let usableThreadId = threadId;\n\n    stream.setStreamValues(() => {\n      if (submitOptions?.optimisticValues != null) {\n        return {\n          ...historyValues,\n          ...(typeof submitOptions.optimisticValues === \"function\"\n            ? submitOptions.optimisticValues(historyValues)\n            : submitOptions.optimisticValues),\n        };\n      }\n\n      return { ...historyValues };\n    });\n\n    await stream.start(\n      async (signal: AbortSignal) => {\n        if (!usableThreadId) {\n          usableThreadId = crypto.randomUUID();\n          threadId = usableThreadId;\n          options.onThreadId?.(usableThreadId);\n        }\n\n        if (!usableThreadId) {\n          throw new Error(\"Failed to obtain valid thread ID.\");\n        }\n\n        return options.transport.stream({\n          input: values,\n          context: submitOptions?.context,\n          command: submitOptions?.command,\n          signal,\n          config: {\n            ...submitOptions?.config,\n            configurable: {\n              thread_id: usableThreadId,\n              ...submitOptions?.config?.configurable,\n            } as unknown as GetConfigurableType<Bag>,\n          },\n        }) as Promise<\n          AsyncGenerator<EventStreamEvent<StateType, UpdateType, CustomType>>\n        >;\n      },\n      {\n        getMessages,\n        setMessages,\n\n        initialValues: {} as StateType,\n        callbacks: options,\n\n        onSuccess: () => undefined,\n        onError(error) {\n          options.onError?.(error, undefined);\n        },\n      },\n    );\n  }\n\n  async function submit(\n    values: UpdateType | null | undefined,\n    submitOptions?: CustomSubmitOptions<StateType, ConfigurableType>,\n  ) {\n    await submitDirect(values, submitOptions);\n  }\n\n  const values = computed(() => streamValues() ?? ({} as StateType));\n\n  return {\n    values,\n    error: streamError,\n    isLoading,\n\n    stop,\n    submit,\n    switchThread,\n\n    queue: {\n      entries: signal([]),\n      size: signal(0),\n      async cancel() {\n        return false;\n      },\n      async clear() {},\n    },\n\n    get interrupts(): Interrupt<InterruptType>[] {\n      const vals = streamValues();\n      if (\n        vals != null &&\n        \"__interrupt__\" in vals &&\n        Array.isArray(vals.__interrupt__)\n      ) {\n        const valueInterrupts = vals.__interrupt__;\n        if (valueInterrupts.length === 0) return [{ when: \"breakpoint\" }];\n        return valueInterrupts;\n      }\n\n      return [];\n    },\n\n    get interrupt(): Interrupt<InterruptType> | undefined {\n      return extractInterrupts<InterruptType>(streamValues());\n    },\n\n    get messages(): Message<ToolCallType>[] {\n      const vals = streamValues();\n      if (!vals) return [];\n      return getMessages(vals);\n    },\n\n    get toolCalls() {\n      const vals = streamValues();\n      if (!vals) return [];\n      const msgs = getMessages(vals);\n      return getToolCallsWithResults<ToolCallType>(msgs);\n    },\n\n    getToolCalls(message: Message<ToolCallType>) {\n      const vals = streamValues();\n      if (!vals) return [];\n      const msgs = getMessages(vals);\n      const allToolCalls = getToolCallsWithResults<ToolCallType>(msgs);\n      return allToolCalls.filter((tc) => tc.aiMessage.id === message.id);\n    },\n\n    get subagents() {\n      return stream.getSubagents();\n    },\n\n    get activeSubagents() {\n      return stream.getActiveSubagents();\n    },\n\n    getSubagent(toolCallId: string) {\n      return stream.getSubagent(toolCallId);\n    },\n\n    getSubagentsByType(type: string) {\n      return stream.getSubagentsByType(type);\n    },\n\n    getSubagentsByMessage(messageId: string) {\n      return stream.getSubagentsByMessage(messageId);\n    },\n  };\n}\n"],"mappings":";;;;;AAkBA,SAAgB,gBAGd,SAAiD;CAQjD,MAAM,SAAS,IAAIA,0CADI,IAAIC,iDAAqB,EACiB;EAC/D,UAAU,QAAQ,YAAY;EAC9B,mBAAmB,QAAQ;EAC3B,wBAAwB,QAAQ;EAChC,WAAWC;EACZ,CAAC;CAEF,MAAM,yCAAwC,OAAO,OAAO;CAC5D,MAAM,wCAA8B,OAAO,MAAM;CACjD,MAAM,sCAAmB,OAAO,UAAU;AAE1C,4BAAQ,cAAc;EACpB,MAAM,cAAc,OAAO,gBAAgB;AACzC,gBAAa,IAAI,OAAO,OAAO;AAC/B,eAAY,IAAI,OAAO,MAAM;AAC7B,aAAU,IAAI,OAAO,UAAU;IAC/B;AAEF,kBAAgB,aAAa,CAAC;GAC9B;CAEF,IAAI,WAA0B,QAAQ,YAAY;CAElD,MAAM,eAAe,UAAgC;EACnD,MAAM,cAAc,QAAQ,eAAe;AAC3C,SAAO,MAAM,QAAQ,MAAM,aAAa,GACnC,MAAM,eACP,EAAE;;CAGR,MAAM,eAAe,SAAoB,aAAmC;EAC1E,MAAM,cAAc,QAAQ,eAAe;AAC3C,SAAO;GAAE,GAAG;IAAU,cAAc;GAAU;;CAGhD,MAAM,gBAAgB,QAAQ,iBAAkB,EAAE;CAElD,MAAM,kBAAkB,YAAY,cAAc;CAClD,MAAM,6BACJ,QAAQ,0BACR,CAAC,OAAO,aACR,gBAAgB,SAAS;AAE3B,iCAAa;EACX,MAAM,UAAU,WAAW;EAC3B,MAAM,aAAa,YAAY,cAAc;AAG7C,MADE,QAAQ,0BAA0B,CAAC,WAAW,WAAW,SAAS,EAElE,QAAO,qBAAqB,YAAY,EAAE,iBAAiB,MAAM,CAAC;GAEpE;AAEF,KAAI,2BACF,QAAO,qBAAqB,iBAAiB,EAAE,iBAAiB,MAAM,CAAC;CAGzE,SAAS,aAAa,aAA4B;AAChD,MAAI,gBAAgB,UAAU;AAC5B,cAAW;AACX,UAAO,OAAO;;;CAIlB,SAAS,OAAO;AACd,SAAO,OAAO,KAAK,eAAe,EAAE,QAAQ,QAAQ,QAAQ,CAAC;;CAG/D,eAAe,aACb,QACA,eACA;EACA,MAAM,kBAAkB,QAAQ,YAAY;AAC5C,MAAI,oBAAoB,UAAU;AAChC,cAAW;AACX,UAAO,OAAO;;EAGhB,IAAI,iBAAiB;AAErB,SAAO,sBAAsB;AAC3B,OAAI,eAAe,oBAAoB,KACrC,QAAO;IACL,GAAG;IACH,GAAI,OAAO,cAAc,qBAAqB,aAC1C,cAAc,iBAAiB,cAAc,GAC7C,cAAc;IACnB;AAGH,UAAO,EAAE,GAAG,eAAe;IAC3B;AAEF,QAAM,OAAO,MACX,OAAO,WAAwB;AAC7B,OAAI,CAAC,gBAAgB;AACnB,qBAAiB,OAAO,YAAY;AACpC,eAAW;AACX,YAAQ,aAAa,eAAe;;AAGtC,OAAI,CAAC,eACH,OAAM,IAAI,MAAM,oCAAoC;AAGtD,UAAO,QAAQ,UAAU,OAAO;IAC9B,OAAO;IACP,SAAS,eAAe;IACxB,SAAS,eAAe;IACxB;IACA,QAAQ;KACN,GAAG,eAAe;KAClB,cAAc;MACZ,WAAW;MACX,GAAG,eAAe,QAAQ;MAC3B;KACF;IACF,CAAC;KAIJ;GACE;GACA;GAEA,eAAe,EAAE;GACjB,WAAW;GAEX,iBAAiB;GACjB,QAAQ,OAAO;AACb,YAAQ,UAAU,OAAO,OAAU;;GAEtC,CACF;;CAGH,eAAe,OACb,QACA,eACA;AACA,QAAM,aAAa,QAAQ,cAAc;;AAK3C,QAAO;EACL,0CAH4B,cAAc,IAAK,EAAE,CAAe;EAIhE,OAAO;EACP;EAEA;EACA;EACA;EAEA,OAAO;GACL,mCAAgB,EAAE,CAAC;GACnB,gCAAa,EAAE;GACf,MAAM,SAAS;AACb,WAAO;;GAET,MAAM,QAAQ;GACf;EAED,IAAI,aAAyC;GAC3C,MAAM,OAAO,cAAc;AAC3B,OACE,QAAQ,QACR,mBAAmB,QACnB,MAAM,QAAQ,KAAK,cAAc,EACjC;IACA,MAAM,kBAAkB,KAAK;AAC7B,QAAI,gBAAgB,WAAW,EAAG,QAAO,CAAC,EAAE,MAAM,cAAc,CAAC;AACjE,WAAO;;AAGT,UAAO,EAAE;;EAGX,IAAI,YAAkD;AACpD,6DAAwC,cAAc,CAAC;;EAGzD,IAAI,WAAoC;GACtC,MAAM,OAAO,cAAc;AAC3B,OAAI,CAAC,KAAM,QAAO,EAAE;AACpB,UAAO,YAAY,KAAK;;EAG1B,IAAI,YAAY;GACd,MAAM,OAAO,cAAc;AAC3B,OAAI,CAAC,KAAM,QAAO,EAAE;AAEpB,sEADa,YAAY,KAAK,CACoB;;EAGpD,aAAa,SAAgC;GAC3C,MAAM,OAAO,cAAc;AAC3B,OAAI,CAAC,KAAM,QAAO,EAAE;AAGpB,sEAFa,YAAY,KAAK,CACkC,CAC5C,QAAQ,OAAO,GAAG,UAAU,OAAO,QAAQ,GAAG;;EAGpE,IAAI,YAAY;AACd,UAAO,OAAO,cAAc;;EAG9B,IAAI,kBAAkB;AACpB,UAAO,OAAO,oBAAoB;;EAGpC,YAAY,YAAoB;AAC9B,UAAO,OAAO,YAAY,WAAW;;EAGvC,mBAAmB,MAAc;AAC/B,UAAO,OAAO,mBAAmB,KAAK;;EAGxC,sBAAsB,WAAmB;AACvC,UAAO,OAAO,sBAAsB,UAAU;;EAEjD"}
+{"version":3,"file":"stream.custom.cjs","names":["MessageTupleManager","StreamManager","toMessageClass"],"sources":["../src/stream.custom.ts"],"sourcesContent":["import { signal, computed, effect } from \"@angular/core\";\nimport {\n  StreamManager,\n  MessageTupleManager,\n  extractInterrupts,\n  toMessageClass,\n  ensureMessageInstances,\n  type EventStreamEvent,\n  type GetUpdateType,\n  type GetCustomEventType,\n  type GetInterruptType,\n  type GetConfigurableType,\n  type GetToolCallsType,\n  type AnyStreamCustomOptions,\n  type CustomSubmitOptions,\n  type MessageMetadata,\n} from \"@langchain/langgraph-sdk/ui\";\nimport { getToolCallsWithResults } from \"@langchain/langgraph-sdk/utils\";\nimport type { BagTemplate, Message, Interrupt } from \"@langchain/langgraph-sdk\";\n\nexport function useStreamCustom<\n  StateType extends Record<string, unknown> = Record<string, unknown>,\n  Bag extends BagTemplate = BagTemplate,\n>(options: AnyStreamCustomOptions<StateType, Bag>) {\n  type UpdateType = GetUpdateType<Bag, StateType>;\n  type CustomType = GetCustomEventType<Bag>;\n  type InterruptType = GetInterruptType<Bag>;\n  type ConfigurableType = GetConfigurableType<Bag>;\n  type ToolCallType = GetToolCallsType<StateType>;\n\n  const messageManager = new MessageTupleManager();\n  const stream = new StreamManager<StateType, Bag>(messageManager, {\n    throttle: options.throttle ?? false,\n    subagentToolNames: options.subagentToolNames,\n    filterSubagentMessages: options.filterSubagentMessages,\n    toMessage: toMessageClass,\n  });\n\n  const streamValues = signal<StateType | null>(stream.values);\n  const streamError = signal<unknown>(stream.error);\n  const isLoading = signal(stream.isLoading);\n\n  const subagentVersion = signal(0);\n\n  effect((onCleanup) => {\n    const unsubscribe = stream.subscribe(() => {\n      streamValues.set(stream.values);\n      streamError.set(stream.error);\n      isLoading.set(stream.isLoading);\n      subagentVersion.update((v) => v + 1);\n    });\n\n    onCleanup(() => unsubscribe());\n  });\n\n  let threadId: string | null = options.threadId ?? null;\n\n  const branch = signal<string>(\"\");\n\n  const getMessages = (value: StateType): Message[] => {\n    const messagesKey = options.messagesKey ?? \"messages\";\n    return Array.isArray(value[messagesKey])\n      ? (value[messagesKey] as Message[])\n      : [];\n  };\n\n  const setMessages = (current: StateType, messages: Message[]): StateType => {\n    const messagesKey = options.messagesKey ?? \"messages\";\n    return { ...current, [messagesKey]: messages };\n  };\n\n  const historyValues = options.initialValues ?? ({} as StateType);\n\n  const historyMessages = getMessages(historyValues);\n  const shouldReconstructSubagents =\n    options.filterSubagentMessages &&\n    !stream.isLoading &&\n    historyMessages.length > 0;\n\n  effect(() => {\n    const loading = isLoading();\n    const hvMessages = getMessages(historyValues);\n    const should =\n      options.filterSubagentMessages && !loading && hvMessages.length > 0;\n    if (should) {\n      stream.reconstructSubagents(hvMessages, { skipIfPopulated: true });\n    }\n  });\n\n  if (shouldReconstructSubagents) {\n    stream.reconstructSubagents(historyMessages, { skipIfPopulated: true });\n  }\n\n  function switchThread(newThreadId: string | null) {\n    if (newThreadId !== threadId) {\n      threadId = newThreadId;\n      stream.clear();\n    }\n  }\n\n  function stop() {\n    return stream.stop(historyValues, { onStop: options.onStop });\n  }\n\n  async function submitDirect(\n    values: UpdateType | null | undefined,\n    submitOptions?: CustomSubmitOptions<StateType, ConfigurableType>,\n  ) {\n    const currentThreadId = options.threadId ?? null;\n    if (currentThreadId !== threadId) {\n      threadId = currentThreadId;\n      stream.clear();\n    }\n\n    let usableThreadId = threadId;\n\n    stream.setStreamValues(() => {\n      if (submitOptions?.optimisticValues != null) {\n        return {\n          ...historyValues,\n          ...(typeof submitOptions.optimisticValues === \"function\"\n            ? submitOptions.optimisticValues(historyValues)\n            : submitOptions.optimisticValues),\n        };\n      }\n\n      return { ...historyValues };\n    });\n\n    await stream.start(\n      async (signal: AbortSignal) => {\n        if (!usableThreadId) {\n          usableThreadId = crypto.randomUUID();\n          threadId = usableThreadId;\n          options.onThreadId?.(usableThreadId);\n        }\n\n        if (!usableThreadId) {\n          throw new Error(\"Failed to obtain valid thread ID.\");\n        }\n\n        return options.transport.stream({\n          input: values,\n          context: submitOptions?.context,\n          command: submitOptions?.command,\n          signal,\n          config: {\n            ...submitOptions?.config,\n            configurable: {\n              thread_id: usableThreadId,\n              ...submitOptions?.config?.configurable,\n            } as unknown as GetConfigurableType<Bag>,\n          },\n        }) as Promise<\n          AsyncGenerator<EventStreamEvent<StateType, UpdateType, CustomType>>\n        >;\n      },\n      {\n        getMessages,\n        setMessages,\n\n        initialValues: {} as StateType,\n        callbacks: options,\n\n        onSuccess: () => undefined,\n        onError(error) {\n          options.onError?.(error, undefined);\n          submitOptions?.onError?.(error, undefined);\n        },\n      },\n    );\n  }\n\n  async function submit(\n    values: UpdateType | null | undefined,\n    submitOptions?: CustomSubmitOptions<StateType, ConfigurableType>,\n  ) {\n    await submitDirect(values, submitOptions);\n  }\n\n  const values = computed(() => streamValues() ?? ({} as StateType));\n\n  function setBranch(value: string) {\n    branch.set(value);\n  }\n\n  function getMessagesMetadata(\n    message: Message<ToolCallType>,\n    index?: number,\n  ): MessageMetadata<StateType> | undefined {\n    const streamMetadata = messageManager.get(message.id)?.metadata;\n    if (streamMetadata != null) {\n      return {\n        messageId: message.id ?? String(index),\n        firstSeenState: undefined,\n        branch: undefined,\n        branchOptions: undefined,\n        streamMetadata,\n      } as MessageMetadata<StateType>;\n    }\n    return undefined;\n  }\n\n  return {\n    values,\n    error: streamError,\n    isLoading,\n\n    stop,\n    submit,\n    switchThread,\n\n    branch,\n    setBranch,\n    getMessagesMetadata,\n\n    queue: {\n      entries: signal([]),\n      size: signal(0),\n      async cancel() {\n        return false;\n      },\n      async clear() {},\n    },\n\n    interrupts: computed((): Interrupt<InterruptType>[] => {\n      const vals = streamValues();\n      if (\n        vals != null &&\n        \"__interrupt__\" in vals &&\n        Array.isArray(vals.__interrupt__)\n      ) {\n        const valueInterrupts = vals.__interrupt__;\n        if (valueInterrupts.length === 0) return [{ when: \"breakpoint\" }];\n        return valueInterrupts;\n      }\n\n      return [];\n    }),\n\n    interrupt: computed((): Interrupt<InterruptType> | undefined => {\n      return extractInterrupts<InterruptType>(streamValues());\n    }),\n\n    messages: computed(() => {\n      const vals = streamValues();\n      if (!vals) return [];\n      return ensureMessageInstances(getMessages(vals));\n    }),\n\n    toolCalls: computed(() => {\n      const vals = streamValues();\n      if (!vals) return [];\n      const msgs = getMessages(vals);\n      return getToolCallsWithResults<ToolCallType>(msgs);\n    }),\n\n    getToolCalls(message: Message<ToolCallType>) {\n      const vals = streamValues();\n      if (!vals) return [];\n      const msgs = getMessages(vals);\n      const allToolCalls = getToolCallsWithResults<ToolCallType>(msgs);\n      return allToolCalls.filter((tc) => tc.aiMessage.id === message.id);\n    },\n\n    get subagents() {\n      void subagentVersion();\n      return stream.getSubagents();\n    },\n\n    get activeSubagents() {\n      void subagentVersion();\n      return stream.getActiveSubagents();\n    },\n\n    getSubagent(toolCallId: string) {\n      return stream.getSubagent(toolCallId);\n    },\n\n    getSubagentsByType(type: string) {\n      return stream.getSubagentsByType(type);\n    },\n\n    getSubagentsByMessage(messageId: string) {\n      return stream.getSubagentsByMessage(messageId);\n    },\n  };\n}\n"],"mappings":";;;;;AAoBA,SAAgB,gBAGd,SAAiD;CAOjD,MAAM,iBAAiB,IAAIA,iDAAqB;CAChD,MAAM,SAAS,IAAIC,0CAA8B,gBAAgB;EAC/D,UAAU,QAAQ,YAAY;EAC9B,mBAAmB,QAAQ;EAC3B,wBAAwB,QAAQ;EAChC,WAAWC;EACZ,CAAC;CAEF,MAAM,yCAAwC,OAAO,OAAO;CAC5D,MAAM,wCAA8B,OAAO,MAAM;CACjD,MAAM,sCAAmB,OAAO,UAAU;CAE1C,MAAM,4CAAyB,EAAE;AAEjC,4BAAQ,cAAc;EACpB,MAAM,cAAc,OAAO,gBAAgB;AACzC,gBAAa,IAAI,OAAO,OAAO;AAC/B,eAAY,IAAI,OAAO,MAAM;AAC7B,aAAU,IAAI,OAAO,UAAU;AAC/B,mBAAgB,QAAQ,MAAM,IAAI,EAAE;IACpC;AAEF,kBAAgB,aAAa,CAAC;GAC9B;CAEF,IAAI,WAA0B,QAAQ,YAAY;CAElD,MAAM,mCAAwB,GAAG;CAEjC,MAAM,eAAe,UAAgC;EACnD,MAAM,cAAc,QAAQ,eAAe;AAC3C,SAAO,MAAM,QAAQ,MAAM,aAAa,GACnC,MAAM,eACP,EAAE;;CAGR,MAAM,eAAe,SAAoB,aAAmC;EAC1E,MAAM,cAAc,QAAQ,eAAe;AAC3C,SAAO;GAAE,GAAG;IAAU,cAAc;GAAU;;CAGhD,MAAM,gBAAgB,QAAQ,iBAAkB,EAAE;CAElD,MAAM,kBAAkB,YAAY,cAAc;CAClD,MAAM,6BACJ,QAAQ,0BACR,CAAC,OAAO,aACR,gBAAgB,SAAS;AAE3B,iCAAa;EACX,MAAM,UAAU,WAAW;EAC3B,MAAM,aAAa,YAAY,cAAc;AAG7C,MADE,QAAQ,0BAA0B,CAAC,WAAW,WAAW,SAAS,EAElE,QAAO,qBAAqB,YAAY,EAAE,iBAAiB,MAAM,CAAC;GAEpE;AAEF,KAAI,2BACF,QAAO,qBAAqB,iBAAiB,EAAE,iBAAiB,MAAM,CAAC;CAGzE,SAAS,aAAa,aAA4B;AAChD,MAAI,gBAAgB,UAAU;AAC5B,cAAW;AACX,UAAO,OAAO;;;CAIlB,SAAS,OAAO;AACd,SAAO,OAAO,KAAK,eAAe,EAAE,QAAQ,QAAQ,QAAQ,CAAC;;CAG/D,eAAe,aACb,QACA,eACA;EACA,MAAM,kBAAkB,QAAQ,YAAY;AAC5C,MAAI,oBAAoB,UAAU;AAChC,cAAW;AACX,UAAO,OAAO;;EAGhB,IAAI,iBAAiB;AAErB,SAAO,sBAAsB;AAC3B,OAAI,eAAe,oBAAoB,KACrC,QAAO;IACL,GAAG;IACH,GAAI,OAAO,cAAc,qBAAqB,aAC1C,cAAc,iBAAiB,cAAc,GAC7C,cAAc;IACnB;AAGH,UAAO,EAAE,GAAG,eAAe;IAC3B;AAEF,QAAM,OAAO,MACX,OAAO,WAAwB;AAC7B,OAAI,CAAC,gBAAgB;AACnB,qBAAiB,OAAO,YAAY;AACpC,eAAW;AACX,YAAQ,aAAa,eAAe;;AAGtC,OAAI,CAAC,eACH,OAAM,IAAI,MAAM,oCAAoC;AAGtD,UAAO,QAAQ,UAAU,OAAO;IAC9B,OAAO;IACP,SAAS,eAAe;IACxB,SAAS,eAAe;IACxB;IACA,QAAQ;KACN,GAAG,eAAe;KAClB,cAAc;MACZ,WAAW;MACX,GAAG,eAAe,QAAQ;MAC3B;KACF;IACF,CAAC;KAIJ;GACE;GACA;GAEA,eAAe,EAAE;GACjB,WAAW;GAEX,iBAAiB;GACjB,QAAQ,OAAO;AACb,YAAQ,UAAU,OAAO,OAAU;AACnC,mBAAe,UAAU,OAAO,OAAU;;GAE7C,CACF;;CAGH,eAAe,OACb,QACA,eACA;AACA,QAAM,aAAa,QAAQ,cAAc;;CAG3C,MAAM,2CAAwB,cAAc,IAAK,EAAE,CAAe;CAElE,SAAS,UAAU,OAAe;AAChC,SAAO,IAAI,MAAM;;CAGnB,SAAS,oBACP,SACA,OACwC;EACxC,MAAM,iBAAiB,eAAe,IAAI,QAAQ,GAAG,EAAE;AACvD,MAAI,kBAAkB,KACpB,QAAO;GACL,WAAW,QAAQ,MAAM,OAAO,MAAM;GACtC,gBAAgB;GAChB,QAAQ;GACR,eAAe;GACf;GACD;;AAKL,QAAO;EACL;EACA,OAAO;EACP;EAEA;EACA;EACA;EAEA;EACA;EACA;EAEA,OAAO;GACL,mCAAgB,EAAE,CAAC;GACnB,gCAAa,EAAE;GACf,MAAM,SAAS;AACb,WAAO;;GAET,MAAM,QAAQ;GACf;EAED,8CAAuD;GACrD,MAAM,OAAO,cAAc;AAC3B,OACE,QAAQ,QACR,mBAAmB,QACnB,MAAM,QAAQ,KAAK,cAAc,EACjC;IACA,MAAM,kBAAkB,KAAK;AAC7B,QAAI,gBAAgB,WAAW,EAAG,QAAO,CAAC,EAAE,MAAM,cAAc,CAAC;AACjE,WAAO;;AAGT,UAAO,EAAE;IACT;EAEF,6CAAgE;AAC9D,6DAAwC,cAAc,CAAC;IACvD;EAEF,4CAAyB;GACvB,MAAM,OAAO,cAAc;AAC3B,OAAI,CAAC,KAAM,QAAO,EAAE;AACpB,kEAA8B,YAAY,KAAK,CAAC;IAChD;EAEF,6CAA0B;GACxB,MAAM,OAAO,cAAc;AAC3B,OAAI,CAAC,KAAM,QAAO,EAAE;AAEpB,sEADa,YAAY,KAAK,CACoB;IAClD;EAEF,aAAa,SAAgC;GAC3C,MAAM,OAAO,cAAc;AAC3B,OAAI,CAAC,KAAM,QAAO,EAAE;AAGpB,sEAFa,YAAY,KAAK,CACkC,CAC5C,QAAQ,OAAO,GAAG,UAAU,OAAO,QAAQ,GAAG;;EAGpE,IAAI,YAAY;AACd,GAAK,iBAAiB;AACtB,UAAO,OAAO,cAAc;;EAG9B,IAAI,kBAAkB;AACpB,GAAK,iBAAiB;AACtB,UAAO,OAAO,oBAAoB;;EAGpC,YAAY,YAAoB;AAC9B,UAAO,OAAO,YAAY,WAAW;;EAGvC,mBAAmB,MAAc;AAC/B,UAAO,OAAO,mBAAmB,KAAK;;EAGxC,sBAAsB,WAAmB;AACvC,UAAO,OAAO,sBAAsB,UAAU;;EAEjD"}

package/dist/stream.custom.js

@@ -1,10 +1,11 @@
 import { computed, effect, signal } from "@angular/core";
-import { MessageTupleManager, StreamManager, extractInterrupts, toMessageClass } from "@langchain/langgraph-sdk/ui";
+import { MessageTupleManager, StreamManager, ensureMessageInstances, extractInterrupts, toMessageClass } from "@langchain/langgraph-sdk/ui";
 import { getToolCallsWithResults } from "@langchain/langgraph-sdk/utils";
 
 //#region src/stream.custom.ts
 function useStreamCustom(options) {
-	const stream = new StreamManager(new MessageTupleManager(), {
+	const messageManager = new MessageTupleManager();
+	const stream = new StreamManager(messageManager, {
 		throttle: options.throttle ?? false,
 		subagentToolNames: options.subagentToolNames,
 		filterSubagentMessages: options.filterSubagentMessages,
@@ -13,15 +14,18 @@ function useStreamCustom(options) {
 	const streamValues = signal(stream.values);
 	const streamError = signal(stream.error);
 	const isLoading = signal(stream.isLoading);
+	const subagentVersion = signal(0);
 	effect((onCleanup) => {
 		const unsubscribe = stream.subscribe(() => {
 			streamValues.set(stream.values);
 			streamError.set(stream.error);
 			isLoading.set(stream.isLoading);
+			subagentVersion.update((v) => v + 1);
 		});
 		onCleanup(() => unsubscribe());
 	});
 	let threadId = options.threadId ?? null;
+	const branch = signal("");
 	const getMessages = (value) => {
 		const messagesKey = options.messagesKey ?? "messages";
 		return Array.isArray(value[messagesKey]) ? value[messagesKey] : [];
@@ -93,19 +97,37 @@ function useStreamCustom(options) {
 			onSuccess: () => void 0,
 			onError(error) {
 				options.onError?.(error, void 0);
+				submitOptions?.onError?.(error, void 0);
 			}
 		});
 	}
 	async function submit(values, submitOptions) {
 		await submitDirect(values, submitOptions);
 	}
+	const values = computed(() => streamValues() ?? {});
+	function setBranch(value) {
+		branch.set(value);
+	}
+	function getMessagesMetadata(message, index) {
+		const streamMetadata = messageManager.get(message.id)?.metadata;
+		if (streamMetadata != null) return {
+			messageId: message.id ?? String(index),
+			firstSeenState: void 0,
+			branch: void 0,
+			branchOptions: void 0,
+			streamMetadata
+		};
+	}
 	return {
-		values: computed(() => streamValues() ?? {}),
+		values,
 		error: streamError,
 		isLoading,
 		stop,
 		submit,
 		switchThread,
+		branch,
+		setBranch,
+		getMessagesMetadata,
 		queue: {
 			entries: signal([]),
 			size: signal(0),
@@ -114,7 +136,7 @@ function useStreamCustom(options) {
 			},
 			async clear() {}
 		},
-		get interrupts() {
+		interrupts: computed(() => {
 			const vals = streamValues();
 			if (vals != null && "__interrupt__" in vals && Array.isArray(vals.__interrupt__)) {
 				const valueInterrupts = vals.__interrupt__;
@@ -122,29 +144,31 @@ function useStreamCustom(options) {
 				return valueInterrupts;
 			}
 			return [];
-		},
-		get interrupt() {
+		}),
+		interrupt: computed(() => {
 			return extractInterrupts(streamValues());
-		},
-		get messages() {
+		}),
+		messages: computed(() => {
 			const vals = streamValues();
 			if (!vals) return [];
-			return getMessages(vals);
-		},
-		get toolCalls() {
+			return ensureMessageInstances(getMessages(vals));
+		}),
+		toolCalls: computed(() => {
 			const vals = streamValues();
 			if (!vals) return [];
 			return getToolCallsWithResults(getMessages(vals));
-		},
+		}),
 		getToolCalls(message) {
 			const vals = streamValues();
 			if (!vals) return [];
 			return getToolCallsWithResults(getMessages(vals)).filter((tc) => tc.aiMessage.id === message.id);
 		},
 		get subagents() {
+			subagentVersion();
 			return stream.getSubagents();
 		},
 		get activeSubagents() {
+			subagentVersion();
 			return stream.getActiveSubagents();
 		},
 		getSubagent(toolCallId) {

package/dist/stream.custom.js.map

@@ -1 +1 @@
-{"version":3,"file":"stream.custom.js","names":[],"sources":["../src/stream.custom.ts"],"sourcesContent":["import { signal, computed, effect } from \"@angular/core\";\nimport {\n  StreamManager,\n  MessageTupleManager,\n  extractInterrupts,\n  toMessageClass,\n  type EventStreamEvent,\n  type GetUpdateType,\n  type GetCustomEventType,\n  type GetInterruptType,\n  type GetConfigurableType,\n  type GetToolCallsType,\n  type AnyStreamCustomOptions,\n  type CustomSubmitOptions,\n} from \"@langchain/langgraph-sdk/ui\";\nimport { getToolCallsWithResults } from \"@langchain/langgraph-sdk/utils\";\nimport type { BagTemplate, Message, Interrupt } from \"@langchain/langgraph-sdk\";\n\nexport function useStreamCustom<\n  StateType extends Record<string, unknown> = Record<string, unknown>,\n  Bag extends BagTemplate = BagTemplate,\n>(options: AnyStreamCustomOptions<StateType, Bag>) {\n  type UpdateType = GetUpdateType<Bag, StateType>;\n  type CustomType = GetCustomEventType<Bag>;\n  type InterruptType = GetInterruptType<Bag>;\n  type ConfigurableType = GetConfigurableType<Bag>;\n  type ToolCallType = GetToolCallsType<StateType>;\n\n  const messageManager = new MessageTupleManager();\n  const stream = new StreamManager<StateType, Bag>(messageManager, {\n    throttle: options.throttle ?? false,\n    subagentToolNames: options.subagentToolNames,\n    filterSubagentMessages: options.filterSubagentMessages,\n    toMessage: toMessageClass,\n  });\n\n  const streamValues = signal<StateType | null>(stream.values);\n  const streamError = signal<unknown>(stream.error);\n  const isLoading = signal(stream.isLoading);\n\n  effect((onCleanup) => {\n    const unsubscribe = stream.subscribe(() => {\n      streamValues.set(stream.values);\n      streamError.set(stream.error);\n      isLoading.set(stream.isLoading);\n    });\n\n    onCleanup(() => unsubscribe());\n  });\n\n  let threadId: string | null = options.threadId ?? null;\n\n  const getMessages = (value: StateType): Message[] => {\n    const messagesKey = options.messagesKey ?? \"messages\";\n    return Array.isArray(value[messagesKey])\n      ? (value[messagesKey] as Message[])\n      : [];\n  };\n\n  const setMessages = (current: StateType, messages: Message[]): StateType => {\n    const messagesKey = options.messagesKey ?? \"messages\";\n    return { ...current, [messagesKey]: messages };\n  };\n\n  const historyValues = options.initialValues ?? ({} as StateType);\n\n  const historyMessages = getMessages(historyValues);\n  const shouldReconstructSubagents =\n    options.filterSubagentMessages &&\n    !stream.isLoading &&\n    historyMessages.length > 0;\n\n  effect(() => {\n    const loading = isLoading();\n    const hvMessages = getMessages(historyValues);\n    const should =\n      options.filterSubagentMessages && !loading && hvMessages.length > 0;\n    if (should) {\n      stream.reconstructSubagents(hvMessages, { skipIfPopulated: true });\n    }\n  });\n\n  if (shouldReconstructSubagents) {\n    stream.reconstructSubagents(historyMessages, { skipIfPopulated: true });\n  }\n\n  function switchThread(newThreadId: string | null) {\n    if (newThreadId !== threadId) {\n      threadId = newThreadId;\n      stream.clear();\n    }\n  }\n\n  function stop() {\n    return stream.stop(historyValues, { onStop: options.onStop });\n  }\n\n  async function submitDirect(\n    values: UpdateType | null | undefined,\n    submitOptions?: CustomSubmitOptions<StateType, ConfigurableType>,\n  ) {\n    const currentThreadId = options.threadId ?? null;\n    if (currentThreadId !== threadId) {\n      threadId = currentThreadId;\n      stream.clear();\n    }\n\n    let usableThreadId = threadId;\n\n    stream.setStreamValues(() => {\n      if (submitOptions?.optimisticValues != null) {\n        return {\n          ...historyValues,\n          ...(typeof submitOptions.optimisticValues === \"function\"\n            ? submitOptions.optimisticValues(historyValues)\n            : submitOptions.optimisticValues),\n        };\n      }\n\n      return { ...historyValues };\n    });\n\n    await stream.start(\n      async (signal: AbortSignal) => {\n        if (!usableThreadId) {\n          usableThreadId = crypto.randomUUID();\n          threadId = usableThreadId;\n          options.onThreadId?.(usableThreadId);\n        }\n\n        if (!usableThreadId) {\n          throw new Error(\"Failed to obtain valid thread ID.\");\n        }\n\n        return options.transport.stream({\n          input: values,\n          context: submitOptions?.context,\n          command: submitOptions?.command,\n          signal,\n          config: {\n            ...submitOptions?.config,\n            configurable: {\n              thread_id: usableThreadId,\n              ...submitOptions?.config?.configurable,\n            } as unknown as GetConfigurableType<Bag>,\n          },\n        }) as Promise<\n          AsyncGenerator<EventStreamEvent<StateType, UpdateType, CustomType>>\n        >;\n      },\n      {\n        getMessages,\n        setMessages,\n\n        initialValues: {} as StateType,\n        callbacks: options,\n\n        onSuccess: () => undefined,\n        onError(error) {\n          options.onError?.(error, undefined);\n        },\n      },\n    );\n  }\n\n  async function submit(\n    values: UpdateType | null | undefined,\n    submitOptions?: CustomSubmitOptions<StateType, ConfigurableType>,\n  ) {\n    await submitDirect(values, submitOptions);\n  }\n\n  const values = computed(() => streamValues() ?? ({} as StateType));\n\n  return {\n    values,\n    error: streamError,\n    isLoading,\n\n    stop,\n    submit,\n    switchThread,\n\n    queue: {\n      entries: signal([]),\n      size: signal(0),\n      async cancel() {\n        return false;\n      },\n      async clear() {},\n    },\n\n    get interrupts(): Interrupt<InterruptType>[] {\n      const vals = streamValues();\n      if (\n        vals != null &&\n        \"__interrupt__\" in vals &&\n        Array.isArray(vals.__interrupt__)\n      ) {\n        const valueInterrupts = vals.__interrupt__;\n        if (valueInterrupts.length === 0) return [{ when: \"breakpoint\" }];\n        return valueInterrupts;\n      }\n\n      return [];\n    },\n\n    get interrupt(): Interrupt<InterruptType> | undefined {\n      return extractInterrupts<InterruptType>(streamValues());\n    },\n\n    get messages(): Message<ToolCallType>[] {\n      const vals = streamValues();\n      if (!vals) return [];\n      return getMessages(vals);\n    },\n\n    get toolCalls() {\n      const vals = streamValues();\n      if (!vals) return [];\n      const msgs = getMessages(vals);\n      return getToolCallsWithResults<ToolCallType>(msgs);\n    },\n\n    getToolCalls(message: Message<ToolCallType>) {\n      const vals = streamValues();\n      if (!vals) return [];\n      const msgs = getMessages(vals);\n      const allToolCalls = getToolCallsWithResults<ToolCallType>(msgs);\n      return allToolCalls.filter((tc) => tc.aiMessage.id === message.id);\n    },\n\n    get subagents() {\n      return stream.getSubagents();\n    },\n\n    get activeSubagents() {\n      return stream.getActiveSubagents();\n    },\n\n    getSubagent(toolCallId: string) {\n      return stream.getSubagent(toolCallId);\n    },\n\n    getSubagentsByType(type: string) {\n      return stream.getSubagentsByType(type);\n    },\n\n    getSubagentsByMessage(messageId: string) {\n      return stream.getSubagentsByMessage(messageId);\n    },\n  };\n}\n"],"mappings":";;;;;AAkBA,SAAgB,gBAGd,SAAiD;CAQjD,MAAM,SAAS,IAAI,cADI,IAAI,qBAAqB,EACiB;EAC/D,UAAU,QAAQ,YAAY;EAC9B,mBAAmB,QAAQ;EAC3B,wBAAwB,QAAQ;EAChC,WAAW;EACZ,CAAC;CAEF,MAAM,eAAe,OAAyB,OAAO,OAAO;CAC5D,MAAM,cAAc,OAAgB,OAAO,MAAM;CACjD,MAAM,YAAY,OAAO,OAAO,UAAU;AAE1C,SAAQ,cAAc;EACpB,MAAM,cAAc,OAAO,gBAAgB;AACzC,gBAAa,IAAI,OAAO,OAAO;AAC/B,eAAY,IAAI,OAAO,MAAM;AAC7B,aAAU,IAAI,OAAO,UAAU;IAC/B;AAEF,kBAAgB,aAAa,CAAC;GAC9B;CAEF,IAAI,WAA0B,QAAQ,YAAY;CAElD,MAAM,eAAe,UAAgC;EACnD,MAAM,cAAc,QAAQ,eAAe;AAC3C,SAAO,MAAM,QAAQ,MAAM,aAAa,GACnC,MAAM,eACP,EAAE;;CAGR,MAAM,eAAe,SAAoB,aAAmC;EAC1E,MAAM,cAAc,QAAQ,eAAe;AAC3C,SAAO;GAAE,GAAG;IAAU,cAAc;GAAU;;CAGhD,MAAM,gBAAgB,QAAQ,iBAAkB,EAAE;CAElD,MAAM,kBAAkB,YAAY,cAAc;CAClD,MAAM,6BACJ,QAAQ,0BACR,CAAC,OAAO,aACR,gBAAgB,SAAS;AAE3B,cAAa;EACX,MAAM,UAAU,WAAW;EAC3B,MAAM,aAAa,YAAY,cAAc;AAG7C,MADE,QAAQ,0BAA0B,CAAC,WAAW,WAAW,SAAS,EAElE,QAAO,qBAAqB,YAAY,EAAE,iBAAiB,MAAM,CAAC;GAEpE;AAEF,KAAI,2BACF,QAAO,qBAAqB,iBAAiB,EAAE,iBAAiB,MAAM,CAAC;CAGzE,SAAS,aAAa,aAA4B;AAChD,MAAI,gBAAgB,UAAU;AAC5B,cAAW;AACX,UAAO,OAAO;;;CAIlB,SAAS,OAAO;AACd,SAAO,OAAO,KAAK,eAAe,EAAE,QAAQ,QAAQ,QAAQ,CAAC;;CAG/D,eAAe,aACb,QACA,eACA;EACA,MAAM,kBAAkB,QAAQ,YAAY;AAC5C,MAAI,oBAAoB,UAAU;AAChC,cAAW;AACX,UAAO,OAAO;;EAGhB,IAAI,iBAAiB;AAErB,SAAO,sBAAsB;AAC3B,OAAI,eAAe,oBAAoB,KACrC,QAAO;IACL,GAAG;IACH,GAAI,OAAO,cAAc,qBAAqB,aAC1C,cAAc,iBAAiB,cAAc,GAC7C,cAAc;IACnB;AAGH,UAAO,EAAE,GAAG,eAAe;IAC3B;AAEF,QAAM,OAAO,MACX,OAAO,WAAwB;AAC7B,OAAI,CAAC,gBAAgB;AACnB,qBAAiB,OAAO,YAAY;AACpC,eAAW;AACX,YAAQ,aAAa,eAAe;;AAGtC,OAAI,CAAC,eACH,OAAM,IAAI,MAAM,oCAAoC;AAGtD,UAAO,QAAQ,UAAU,OAAO;IAC9B,OAAO;IACP,SAAS,eAAe;IACxB,SAAS,eAAe;IACxB;IACA,QAAQ;KACN,GAAG,eAAe;KAClB,cAAc;MACZ,WAAW;MACX,GAAG,eAAe,QAAQ;MAC3B;KACF;IACF,CAAC;KAIJ;GACE;GACA;GAEA,eAAe,EAAE;GACjB,WAAW;GAEX,iBAAiB;GACjB,QAAQ,OAAO;AACb,YAAQ,UAAU,OAAO,OAAU;;GAEtC,CACF;;CAGH,eAAe,OACb,QACA,eACA;AACA,QAAM,aAAa,QAAQ,cAAc;;AAK3C,QAAO;EACL,QAHa,eAAe,cAAc,IAAK,EAAE,CAAe;EAIhE,OAAO;EACP;EAEA;EACA;EACA;EAEA,OAAO;GACL,SAAS,OAAO,EAAE,CAAC;GACnB,MAAM,OAAO,EAAE;GACf,MAAM,SAAS;AACb,WAAO;;GAET,MAAM,QAAQ;GACf;EAED,IAAI,aAAyC;GAC3C,MAAM,OAAO,cAAc;AAC3B,OACE,QAAQ,QACR,mBAAmB,QACnB,MAAM,QAAQ,KAAK,cAAc,EACjC;IACA,MAAM,kBAAkB,KAAK;AAC7B,QAAI,gBAAgB,WAAW,EAAG,QAAO,CAAC,EAAE,MAAM,cAAc,CAAC;AACjE,WAAO;;AAGT,UAAO,EAAE;;EAGX,IAAI,YAAkD;AACpD,UAAO,kBAAiC,cAAc,CAAC;;EAGzD,IAAI,WAAoC;GACtC,MAAM,OAAO,cAAc;AAC3B,OAAI,CAAC,KAAM,QAAO,EAAE;AACpB,UAAO,YAAY,KAAK;;EAG1B,IAAI,YAAY;GACd,MAAM,OAAO,cAAc;AAC3B,OAAI,CAAC,KAAM,QAAO,EAAE;AAEpB,UAAO,wBADM,YAAY,KAAK,CACoB;;EAGpD,aAAa,SAAgC;GAC3C,MAAM,OAAO,cAAc;AAC3B,OAAI,CAAC,KAAM,QAAO,EAAE;AAGpB,UADqB,wBADR,YAAY,KAAK,CACkC,CAC5C,QAAQ,OAAO,GAAG,UAAU,OAAO,QAAQ,GAAG;;EAGpE,IAAI,YAAY;AACd,UAAO,OAAO,cAAc;;EAG9B,IAAI,kBAAkB;AACpB,UAAO,OAAO,oBAAoB;;EAGpC,YAAY,YAAoB;AAC9B,UAAO,OAAO,YAAY,WAAW;;EAGvC,mBAAmB,MAAc;AAC/B,UAAO,OAAO,mBAAmB,KAAK;;EAGxC,sBAAsB,WAAmB;AACvC,UAAO,OAAO,sBAAsB,UAAU;;EAEjD"}
+{"version":3,"file":"stream.custom.js","names":[],"sources":["../src/stream.custom.ts"],"sourcesContent":["import { signal, computed, effect } from \"@angular/core\";\nimport {\n  StreamManager,\n  MessageTupleManager,\n  extractInterrupts,\n  toMessageClass,\n  ensureMessageInstances,\n  type EventStreamEvent,\n  type GetUpdateType,\n  type GetCustomEventType,\n  type GetInterruptType,\n  type GetConfigurableType,\n  type GetToolCallsType,\n  type AnyStreamCustomOptions,\n  type CustomSubmitOptions,\n  type MessageMetadata,\n} from \"@langchain/langgraph-sdk/ui\";\nimport { getToolCallsWithResults } from \"@langchain/langgraph-sdk/utils\";\nimport type { BagTemplate, Message, Interrupt } from \"@langchain/langgraph-sdk\";\n\nexport function useStreamCustom<\n  StateType extends Record<string, unknown> = Record<string, unknown>,\n  Bag extends BagTemplate = BagTemplate,\n>(options: AnyStreamCustomOptions<StateType, Bag>) {\n  type UpdateType = GetUpdateType<Bag, StateType>;\n  type CustomType = GetCustomEventType<Bag>;\n  type InterruptType = GetInterruptType<Bag>;\n  type ConfigurableType = GetConfigurableType<Bag>;\n  type ToolCallType = GetToolCallsType<StateType>;\n\n  const messageManager = new MessageTupleManager();\n  const stream = new StreamManager<StateType, Bag>(messageManager, {\n    throttle: options.throttle ?? false,\n    subagentToolNames: options.subagentToolNames,\n    filterSubagentMessages: options.filterSubagentMessages,\n    toMessage: toMessageClass,\n  });\n\n  const streamValues = signal<StateType | null>(stream.values);\n  const streamError = signal<unknown>(stream.error);\n  const isLoading = signal(stream.isLoading);\n\n  const subagentVersion = signal(0);\n\n  effect((onCleanup) => {\n    const unsubscribe = stream.subscribe(() => {\n      streamValues.set(stream.values);\n      streamError.set(stream.error);\n      isLoading.set(stream.isLoading);\n      subagentVersion.update((v) => v + 1);\n    });\n\n    onCleanup(() => unsubscribe());\n  });\n\n  let threadId: string | null = options.threadId ?? null;\n\n  const branch = signal<string>(\"\");\n\n  const getMessages = (value: StateType): Message[] => {\n    const messagesKey = options.messagesKey ?? \"messages\";\n    return Array.isArray(value[messagesKey])\n      ? (value[messagesKey] as Message[])\n      : [];\n  };\n\n  const setMessages = (current: StateType, messages: Message[]): StateType => {\n    const messagesKey = options.messagesKey ?? \"messages\";\n    return { ...current, [messagesKey]: messages };\n  };\n\n  const historyValues = options.initialValues ?? ({} as StateType);\n\n  const historyMessages = getMessages(historyValues);\n  const shouldReconstructSubagents =\n    options.filterSubagentMessages &&\n    !stream.isLoading &&\n    historyMessages.length > 0;\n\n  effect(() => {\n    const loading = isLoading();\n    const hvMessages = getMessages(historyValues);\n    const should =\n      options.filterSubagentMessages && !loading && hvMessages.length > 0;\n    if (should) {\n      stream.reconstructSubagents(hvMessages, { skipIfPopulated: true });\n    }\n  });\n\n  if (shouldReconstructSubagents) {\n    stream.reconstructSubagents(historyMessages, { skipIfPopulated: true });\n  }\n\n  function switchThread(newThreadId: string | null) {\n    if (newThreadId !== threadId) {\n      threadId = newThreadId;\n      stream.clear();\n    }\n  }\n\n  function stop() {\n    return stream.stop(historyValues, { onStop: options.onStop });\n  }\n\n  async function submitDirect(\n    values: UpdateType | null | undefined,\n    submitOptions?: CustomSubmitOptions<StateType, ConfigurableType>,\n  ) {\n    const currentThreadId = options.threadId ?? null;\n    if (currentThreadId !== threadId) {\n      threadId = currentThreadId;\n      stream.clear();\n    }\n\n    let usableThreadId = threadId;\n\n    stream.setStreamValues(() => {\n      if (submitOptions?.optimisticValues != null) {\n        return {\n          ...historyValues,\n          ...(typeof submitOptions.optimisticValues === \"function\"\n            ? submitOptions.optimisticValues(historyValues)\n            : submitOptions.optimisticValues),\n        };\n      }\n\n      return { ...historyValues };\n    });\n\n    await stream.start(\n      async (signal: AbortSignal) => {\n        if (!usableThreadId) {\n          usableThreadId = crypto.randomUUID();\n          threadId = usableThreadId;\n          options.onThreadId?.(usableThreadId);\n        }\n\n        if (!usableThreadId) {\n          throw new Error(\"Failed to obtain valid thread ID.\");\n        }\n\n        return options.transport.stream({\n          input: values,\n          context: submitOptions?.context,\n          command: submitOptions?.command,\n          signal,\n          config: {\n            ...submitOptions?.config,\n            configurable: {\n              thread_id: usableThreadId,\n              ...submitOptions?.config?.configurable,\n            } as unknown as GetConfigurableType<Bag>,\n          },\n        }) as Promise<\n          AsyncGenerator<EventStreamEvent<StateType, UpdateType, CustomType>>\n        >;\n      },\n      {\n        getMessages,\n        setMessages,\n\n        initialValues: {} as StateType,\n        callbacks: options,\n\n        onSuccess: () => undefined,\n        onError(error) {\n          options.onError?.(error, undefined);\n          submitOptions?.onError?.(error, undefined);\n        },\n      },\n    );\n  }\n\n  async function submit(\n    values: UpdateType | null | undefined,\n    submitOptions?: CustomSubmitOptions<StateType, ConfigurableType>,\n  ) {\n    await submitDirect(values, submitOptions);\n  }\n\n  const values = computed(() => streamValues() ?? ({} as StateType));\n\n  function setBranch(value: string) {\n    branch.set(value);\n  }\n\n  function getMessagesMetadata(\n    message: Message<ToolCallType>,\n    index?: number,\n  ): MessageMetadata<StateType> | undefined {\n    const streamMetadata = messageManager.get(message.id)?.metadata;\n    if (streamMetadata != null) {\n      return {\n        messageId: message.id ?? String(index),\n        firstSeenState: undefined,\n        branch: undefined,\n        branchOptions: undefined,\n        streamMetadata,\n      } as MessageMetadata<StateType>;\n    }\n    return undefined;\n  }\n\n  return {\n    values,\n    error: streamError,\n    isLoading,\n\n    stop,\n    submit,\n    switchThread,\n\n    branch,\n    setBranch,\n    getMessagesMetadata,\n\n    queue: {\n      entries: signal([]),\n      size: signal(0),\n      async cancel() {\n        return false;\n      },\n      async clear() {},\n    },\n\n    interrupts: computed((): Interrupt<InterruptType>[] => {\n      const vals = streamValues();\n      if (\n        vals != null &&\n        \"__interrupt__\" in vals &&\n        Array.isArray(vals.__interrupt__)\n      ) {\n        const valueInterrupts = vals.__interrupt__;\n        if (valueInterrupts.length === 0) return [{ when: \"breakpoint\" }];\n        return valueInterrupts;\n      }\n\n      return [];\n    }),\n\n    interrupt: computed((): Interrupt<InterruptType> | undefined => {\n      return extractInterrupts<InterruptType>(streamValues());\n    }),\n\n    messages: computed(() => {\n      const vals = streamValues();\n      if (!vals) return [];\n      return ensureMessageInstances(getMessages(vals));\n    }),\n\n    toolCalls: computed(() => {\n      const vals = streamValues();\n      if (!vals) return [];\n      const msgs = getMessages(vals);\n      return getToolCallsWithResults<ToolCallType>(msgs);\n    }),\n\n    getToolCalls(message: Message<ToolCallType>) {\n      const vals = streamValues();\n      if (!vals) return [];\n      const msgs = getMessages(vals);\n      const allToolCalls = getToolCallsWithResults<ToolCallType>(msgs);\n      return allToolCalls.filter((tc) => tc.aiMessage.id === message.id);\n    },\n\n    get subagents() {\n      void subagentVersion();\n      return stream.getSubagents();\n    },\n\n    get activeSubagents() {\n      void subagentVersion();\n      return stream.getActiveSubagents();\n    },\n\n    getSubagent(toolCallId: string) {\n      return stream.getSubagent(toolCallId);\n    },\n\n    getSubagentsByType(type: string) {\n      return stream.getSubagentsByType(type);\n    },\n\n    getSubagentsByMessage(messageId: string) {\n      return stream.getSubagentsByMessage(messageId);\n    },\n  };\n}\n"],"mappings":";;;;;AAoBA,SAAgB,gBAGd,SAAiD;CAOjD,MAAM,iBAAiB,IAAI,qBAAqB;CAChD,MAAM,SAAS,IAAI,cAA8B,gBAAgB;EAC/D,UAAU,QAAQ,YAAY;EAC9B,mBAAmB,QAAQ;EAC3B,wBAAwB,QAAQ;EAChC,WAAW;EACZ,CAAC;CAEF,MAAM,eAAe,OAAyB,OAAO,OAAO;CAC5D,MAAM,cAAc,OAAgB,OAAO,MAAM;CACjD,MAAM,YAAY,OAAO,OAAO,UAAU;CAE1C,MAAM,kBAAkB,OAAO,EAAE;AAEjC,SAAQ,cAAc;EACpB,MAAM,cAAc,OAAO,gBAAgB;AACzC,gBAAa,IAAI,OAAO,OAAO;AAC/B,eAAY,IAAI,OAAO,MAAM;AAC7B,aAAU,IAAI,OAAO,UAAU;AAC/B,mBAAgB,QAAQ,MAAM,IAAI,EAAE;IACpC;AAEF,kBAAgB,aAAa,CAAC;GAC9B;CAEF,IAAI,WAA0B,QAAQ,YAAY;CAElD,MAAM,SAAS,OAAe,GAAG;CAEjC,MAAM,eAAe,UAAgC;EACnD,MAAM,cAAc,QAAQ,eAAe;AAC3C,SAAO,MAAM,QAAQ,MAAM,aAAa,GACnC,MAAM,eACP,EAAE;;CAGR,MAAM,eAAe,SAAoB,aAAmC;EAC1E,MAAM,cAAc,QAAQ,eAAe;AAC3C,SAAO;GAAE,GAAG;IAAU,cAAc;GAAU;;CAGhD,MAAM,gBAAgB,QAAQ,iBAAkB,EAAE;CAElD,MAAM,kBAAkB,YAAY,cAAc;CAClD,MAAM,6BACJ,QAAQ,0BACR,CAAC,OAAO,aACR,gBAAgB,SAAS;AAE3B,cAAa;EACX,MAAM,UAAU,WAAW;EAC3B,MAAM,aAAa,YAAY,cAAc;AAG7C,MADE,QAAQ,0BAA0B,CAAC,WAAW,WAAW,SAAS,EAElE,QAAO,qBAAqB,YAAY,EAAE,iBAAiB,MAAM,CAAC;GAEpE;AAEF,KAAI,2BACF,QAAO,qBAAqB,iBAAiB,EAAE,iBAAiB,MAAM,CAAC;CAGzE,SAAS,aAAa,aAA4B;AAChD,MAAI,gBAAgB,UAAU;AAC5B,cAAW;AACX,UAAO,OAAO;;;CAIlB,SAAS,OAAO;AACd,SAAO,OAAO,KAAK,eAAe,EAAE,QAAQ,QAAQ,QAAQ,CAAC;;CAG/D,eAAe,aACb,QACA,eACA;EACA,MAAM,kBAAkB,QAAQ,YAAY;AAC5C,MAAI,oBAAoB,UAAU;AAChC,cAAW;AACX,UAAO,OAAO;;EAGhB,IAAI,iBAAiB;AAErB,SAAO,sBAAsB;AAC3B,OAAI,eAAe,oBAAoB,KACrC,QAAO;IACL,GAAG;IACH,GAAI,OAAO,cAAc,qBAAqB,aAC1C,cAAc,iBAAiB,cAAc,GAC7C,cAAc;IACnB;AAGH,UAAO,EAAE,GAAG,eAAe;IAC3B;AAEF,QAAM,OAAO,MACX,OAAO,WAAwB;AAC7B,OAAI,CAAC,gBAAgB;AACnB,qBAAiB,OAAO,YAAY;AACpC,eAAW;AACX,YAAQ,aAAa,eAAe;;AAGtC,OAAI,CAAC,eACH,OAAM,IAAI,MAAM,oCAAoC;AAGtD,UAAO,QAAQ,UAAU,OAAO;IAC9B,OAAO;IACP,SAAS,eAAe;IACxB,SAAS,eAAe;IACxB;IACA,QAAQ;KACN,GAAG,eAAe;KAClB,cAAc;MACZ,WAAW;MACX,GAAG,eAAe,QAAQ;MAC3B;KACF;IACF,CAAC;KAIJ;GACE;GACA;GAEA,eAAe,EAAE;GACjB,WAAW;GAEX,iBAAiB;GACjB,QAAQ,OAAO;AACb,YAAQ,UAAU,OAAO,OAAU;AACnC,mBAAe,UAAU,OAAO,OAAU;;GAE7C,CACF;;CAGH,eAAe,OACb,QACA,eACA;AACA,QAAM,aAAa,QAAQ,cAAc;;CAG3C,MAAM,SAAS,eAAe,cAAc,IAAK,EAAE,CAAe;CAElE,SAAS,UAAU,OAAe;AAChC,SAAO,IAAI,MAAM;;CAGnB,SAAS,oBACP,SACA,OACwC;EACxC,MAAM,iBAAiB,eAAe,IAAI,QAAQ,GAAG,EAAE;AACvD,MAAI,kBAAkB,KACpB,QAAO;GACL,WAAW,QAAQ,MAAM,OAAO,MAAM;GACtC,gBAAgB;GAChB,QAAQ;GACR,eAAe;GACf;GACD;;AAKL,QAAO;EACL;EACA,OAAO;EACP;EAEA;EACA;EACA;EAEA;EACA;EACA;EAEA,OAAO;GACL,SAAS,OAAO,EAAE,CAAC;GACnB,MAAM,OAAO,EAAE;GACf,MAAM,SAAS;AACb,WAAO;;GAET,MAAM,QAAQ;GACf;EAED,YAAY,eAA2C;GACrD,MAAM,OAAO,cAAc;AAC3B,OACE,QAAQ,QACR,mBAAmB,QACnB,MAAM,QAAQ,KAAK,cAAc,EACjC;IACA,MAAM,kBAAkB,KAAK;AAC7B,QAAI,gBAAgB,WAAW,EAAG,QAAO,CAAC,EAAE,MAAM,cAAc,CAAC;AACjE,WAAO;;AAGT,UAAO,EAAE;IACT;EAEF,WAAW,eAAqD;AAC9D,UAAO,kBAAiC,cAAc,CAAC;IACvD;EAEF,UAAU,eAAe;GACvB,MAAM,OAAO,cAAc;AAC3B,OAAI,CAAC,KAAM,QAAO,EAAE;AACpB,UAAO,uBAAuB,YAAY,KAAK,CAAC;IAChD;EAEF,WAAW,eAAe;GACxB,MAAM,OAAO,cAAc;AAC3B,OAAI,CAAC,KAAM,QAAO,EAAE;AAEpB,UAAO,wBADM,YAAY,KAAK,CACoB;IAClD;EAEF,aAAa,SAAgC;GAC3C,MAAM,OAAO,cAAc;AAC3B,OAAI,CAAC,KAAM,QAAO,EAAE;AAGpB,UADqB,wBADR,YAAY,KAAK,CACkC,CAC5C,QAAQ,OAAO,GAAG,UAAU,OAAO,QAAQ,GAAG;;EAGpE,IAAI,YAAY;AACd,GAAK,iBAAiB;AACtB,UAAO,OAAO,cAAc;;EAG9B,IAAI,kBAAkB;AACpB,GAAK,iBAAiB;AACtB,UAAO,OAAO,oBAAoB;;EAGpC,YAAY,YAAoB;AAC9B,UAAO,OAAO,YAAY,WAAW;;EAGvC,mBAAmB,MAAc;AAC/B,UAAO,OAAO,mBAAmB,KAAK;;EAGxC,sBAAsB,WAAmB;AACvC,UAAO,OAAO,sBAAsB,UAAU;;EAEjD"}

package/package.json

@@ -1,11 +1,23 @@
 {
   "name": "@langchain/angular",
-  "version": "0.0.1",
+  "version": "0.1.0",
   "description": "Angular integration for LangGraph & LangChain",
   "type": "module",
+  "scripts": {
+    "clean": "rm -rf dist/ dist-cjs/",
+    "build": "pnpm turbo build:internal --filter=@langchain/angular",
+    "build:internal": "pnpm --filter @langchain/build compile @langchain/angular",
+    "prepublish": "pnpm build",
+    "format:check": "prettier --check src",
+    "format": "prettier --write src",
+    "lint:eslint": "NODE_OPTIONS=--max-old-space-size=4096 eslint --cache --ext .ts,.js,.jsx,.tsx src/",
+    "lint": "pnpm lint:eslint",
+    "lint:fix": "pnpm lint:eslint --fix",
+    "test": "vitest run"
+  },
   "license": "MIT",
   "dependencies": {
-    "@langchain/langgraph-sdk": "^1.6.5"
+    "@langchain/langgraph-sdk": "workspace:^"
   },
   "devDependencies": {
     "@analogjs/vite-plugin-angular": "^2.2.3",
@@ -17,13 +29,17 @@
     "@angular/router": "^20.3.10",
     "@hono/node-server": "^1.14.1",
     "@langchain/core": "^1.1.27",
+    "@langchain/langgraph": "workspace:^",
+    "@langchain/langgraph-api": "workspace:^",
+    "@langchain/langgraph-checkpoint": "workspace:^",
     "@langchain/scripts": "^0.1.4",
     "@tsconfig/recommended": "^1.0.2",
     "@types/node": "^18.15.11",
     "@types/uuid": "^9.0.1",
     "@typescript-eslint/eslint-plugin": "^6.12.0",
     "@typescript-eslint/parser": "^6.12.0",
-    "@vitest/browser": "^3.2.4",
+    "@vitest/browser": "^4.0.18",
+    "@vitest/browser-webdriverio": "^4.0.18",
     "deepagents": "^1.8.0",
     "eslint": "^8.33.0",
     "eslint-config-airbnb-base": "^15.0.0",
@@ -38,13 +54,10 @@
     "tslib": "^2.6.0",
     "typescript": "^4.9.5 || ^5.4.5",
     "vite": "^7.3.1",
-    "vitest": "^3.2.4",
+    "vitest": "^4.0.18",
     "vitest-browser-angular": "^0.3.0",
     "webdriverio": "^9.15.0",
-    "zod": "^4.3.6",
-    "@langchain/langgraph-api": "^1.1.14",
-    "@langchain/langgraph-checkpoint": "^1.0.0",
-    "@langchain/langgraph": "^1.2.0"
+    "zod": "^4.3.6"
   },
   "peerDependencies": {
     "@angular/core": "^18.0.0 || ^19.0.0 || ^20.0.0",
@@ -69,17 +82,5 @@
   },
   "files": [
     "dist/"
-  ],
-  "scripts": {
-    "clean": "rm -rf dist/ dist-cjs/",
-    "build": "pnpm turbo build:internal --filter=@langchain/angular",
-    "build:internal": "pnpm --filter @langchain/build compile @langchain/angular",
-    "prepublish": "pnpm build",
-    "format:check": "prettier --check src",
-    "format": "prettier --write src",
-    "lint:eslint": "NODE_OPTIONS=--max-old-space-size=4096 eslint --cache --ext .ts,.js,.jsx,.tsx src/",
-    "lint": "pnpm lint:eslint",
-    "lint:fix": "pnpm lint:eslint --fix",
-    "test": "vitest run"
-  }
-}
+  ]
+}