@langchain/vue 0.0.1 → 0.1.2

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 { onUnmounted, shallowRef, watch } from \"vue\";\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 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\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 = shallowRef<StateType | null>(stream.values);\n  const streamError = shallowRef<unknown>(stream.error);\n  const isLoading = shallowRef(stream.isLoading);\n\n  const unsubscribe = stream.subscribe(() => {\n    streamValues.value = stream.values;\n    streamError.value = stream.error;\n    isLoading.value = stream.isLoading;\n  });\n\n  onUnmounted(() => {\n    unsubscribe();\n  });\n\n  let threadId: string | null = options.threadId ?? null;\n\n  watch(\n    () => options.threadId,\n    (newId) => {\n      const resolved = newId ?? null;\n      if (resolved !== threadId) {\n        threadId = resolved;\n        stream.clear();\n      }\n    },\n  );\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  // @ts-expect-error used in watch callback below\n  // eslint-disable-next-line @typescript-eslint/no-unused-vars\n  const shouldReconstructSubagents =\n    options.filterSubagentMessages &&\n    !stream.isLoading &&\n    historyMessages.length > 0;\n\n  watch(\n    () => ({\n      should:\n        options.filterSubagentMessages &&\n        !isLoading.value &&\n        getMessages(historyValues).length > 0,\n      len: getMessages(historyValues).length,\n    }),\n    ({ should }) => {\n      if (should) {\n        stream.reconstructSubagents(getMessages(historyValues), {\n          skipIfPopulated: true,\n        });\n      }\n    },\n    { immediate: 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    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 (!threadId) {\n          threadId = crypto.randomUUID();\n          options.onThreadId?.(threadId);\n        }\n\n        if (!threadId) {\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: threadId,\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  return {\n    get values() {\n      return streamValues.value ?? ({} as StateType);\n    },\n\n    error: streamError,\n    isLoading,\n\n    stop,\n    submit,\n    switchThread,\n\n    queue: {\n      entries: [],\n      size: 0,\n      async cancel() {\n        return false;\n      },\n      async clear() {},\n    },\n\n    get interrupts(): Interrupt<InterruptType>[] {\n      if (\n        streamValues.value != null &&\n        \"__interrupt__\" in streamValues.value &&\n        Array.isArray(streamValues.value.__interrupt__)\n      ) {\n        const valueInterrupts = streamValues.value.__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.value);\n    },\n\n    get messages(): Message[] {\n      if (!streamValues.value) return [];\n      return getMessages(streamValues.value);\n    },\n\n    get toolCalls() {\n      if (!streamValues.value) return [];\n      return getToolCallsWithResults(getMessages(streamValues.value));\n    },\n\n    getToolCalls(message: Message) {\n      if (!streamValues.value) return [];\n      const allToolCalls = getToolCallsWithResults(\n        getMessages(streamValues.value),\n      );\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":";;;;;AAiBA,SAAgB,gBAGd,SAAiD;CAOjD,MAAM,SAAS,IAAIA,0CADI,IAAIC,iDAAqB,EACiB;EAC/D,UAAU,QAAQ,YAAY;EAC9B,mBAAmB,QAAQ;EAC3B,wBAAwB,QAAQ;EAChC,WAAWC;EACZ,CAAC;CAEF,MAAM,mCAA4C,OAAO,OAAO;CAChE,MAAM,kCAAkC,OAAO,MAAM;CACrD,MAAM,gCAAuB,OAAO,UAAU;CAE9C,MAAM,cAAc,OAAO,gBAAgB;AACzC,eAAa,QAAQ,OAAO;AAC5B,cAAY,QAAQ,OAAO;AAC3B,YAAU,QAAQ,OAAO;GACzB;AAEF,4BAAkB;AAChB,eAAa;GACb;CAEF,IAAI,WAA0B,QAAQ,YAAY;AAElD,sBACQ,QAAQ,WACb,UAAU;EACT,MAAM,WAAW,SAAS;AAC1B,MAAI,aAAa,UAAU;AACzB,cAAW;AACX,UAAO,OAAO;;GAGnB;CAED,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;AAIhD,SAAQ,0BACR,CAAC,OAAO,aACR,gBAAgB;AAElB,uBACS;EACL,QACE,QAAQ,0BACR,CAAC,UAAU,SACX,YAAY,cAAc,CAAC,SAAS;EACtC,KAAK,YAAY,cAAc,CAAC;EACjC,IACA,EAAE,aAAa;AACd,MAAI,OACF,QAAO,qBAAqB,YAAY,cAAc,EAAE,EACtD,iBAAiB,MAClB,CAAC;IAGN,EAAE,WAAW,MAAM,CACpB;CAED,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;;AAGhB,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,UAAU;AACb,eAAW,OAAO,YAAY;AAC9B,YAAQ,aAAa,SAAS;;AAGhC,OAAI,CAAC,SACH,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;;AAG3C,QAAO;EACL,IAAI,SAAS;AACX,UAAO,aAAa,SAAU,EAAE;;EAGlC,OAAO;EACP;EAEA;EACA;EACA;EAEA,OAAO;GACL,SAAS,EAAE;GACX,MAAM;GACN,MAAM,SAAS;AACb,WAAO;;GAET,MAAM,QAAQ;GACf;EAED,IAAI,aAAyC;AAC3C,OACE,aAAa,SAAS,QACtB,mBAAmB,aAAa,SAChC,MAAM,QAAQ,aAAa,MAAM,cAAc,EAC/C;IACA,MAAM,kBAAkB,aAAa,MAAM;AAC3C,QAAI,gBAAgB,WAAW,EAAG,QAAO,CAAC,EAAE,MAAM,cAAc,CAAC;AACjE,WAAO;;AAGT,UAAO,EAAE;;EAGX,IAAI,YAAkD;AACpD,6DAAwC,aAAa,MAAM;;EAG7D,IAAI,WAAsB;AACxB,OAAI,CAAC,aAAa,MAAO,QAAO,EAAE;AAClC,UAAO,YAAY,aAAa,MAAM;;EAGxC,IAAI,YAAY;AACd,OAAI,CAAC,aAAa,MAAO,QAAO,EAAE;AAClC,sEAA+B,YAAY,aAAa,MAAM,CAAC;;EAGjE,aAAa,SAAkB;AAC7B,OAAI,CAAC,aAAa,MAAO,QAAO,EAAE;AAIlC,sEAFE,YAAY,aAAa,MAAM,CAChC,CACmB,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 { onScopeDispose, ref, shallowRef, watch } from \"vue\";\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 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\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 = shallowRef<StateType | null>(stream.values);\n  const streamError = shallowRef<unknown>(stream.error);\n  const isLoading = shallowRef(stream.isLoading);\n\n  const subagentVersion = shallowRef(0);\n\n  const unsubscribe = stream.subscribe(() => {\n    streamValues.value = stream.values;\n    streamError.value = stream.error;\n    isLoading.value = stream.isLoading;\n    subagentVersion.value += 1;\n  });\n\n  const branch = ref<string>(\"\");\n\n  let threadId: string | null = options.threadId ?? null;\n\n  watch(\n    () => options.threadId,\n    (newId) => {\n      const resolved = newId ?? null;\n      if (resolved !== threadId) {\n        threadId = resolved;\n        stream.clear();\n      }\n    },\n    { flush: \"sync\" },\n  );\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  onScopeDispose(() => {\n    unsubscribe();\n    void stream.stop(historyValues, { onStop: options.onStop });\n  });\n\n  const historyMessages = getMessages(historyValues);\n  // @ts-expect-error used in watch callback below\n  // eslint-disable-next-line @typescript-eslint/no-unused-vars\n  const shouldReconstructSubagents =\n    options.filterSubagentMessages &&\n    !stream.isLoading &&\n    historyMessages.length > 0;\n\n  watch(\n    () => ({\n      should:\n        options.filterSubagentMessages &&\n        !isLoading.value &&\n        getMessages(historyValues).length > 0,\n      len: getMessages(historyValues).length,\n    }),\n    ({ should }) => {\n      if (should) {\n        stream.reconstructSubagents(getMessages(historyValues), {\n          skipIfPopulated: true,\n        });\n      }\n    },\n    { immediate: 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    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 (!threadId) {\n          threadId = crypto.randomUUID();\n          options.onThreadId?.(threadId);\n        }\n\n        if (!threadId) {\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: threadId,\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  function setBranch(value: string) {\n    branch.value = value;\n  }\n\n  function getMessagesMetadata(\n    message: Message,\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    get values() {\n      return streamValues.value ?? ({} as StateType);\n    },\n\n    error: streamError,\n    isLoading,\n\n    stop,\n    submit,\n    switchThread,\n\n    branch,\n    setBranch,\n    getMessagesMetadata,\n\n    queue: {\n      entries: [],\n      size: 0,\n      async cancel() {\n        return false;\n      },\n      async clear() {},\n    },\n\n    get interrupts(): Interrupt<InterruptType>[] {\n      if (\n        streamValues.value != null &&\n        \"__interrupt__\" in streamValues.value &&\n        Array.isArray(streamValues.value.__interrupt__)\n      ) {\n        const valueInterrupts = streamValues.value.__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.value);\n    },\n\n    get messages() {\n      if (!streamValues.value) return [];\n      return ensureMessageInstances(getMessages(streamValues.value));\n    },\n\n    get toolCalls() {\n      if (!streamValues.value) return [];\n      return getToolCallsWithResults(getMessages(streamValues.value));\n    },\n\n    getToolCalls(message: Message) {\n      if (!streamValues.value) return [];\n      const allToolCalls = getToolCallsWithResults(\n        getMessages(streamValues.value),\n      );\n      return allToolCalls.filter((tc) => tc.aiMessage.id === message.id);\n    },\n\n    get subagents() {\n      void subagentVersion.value;\n      return stream.getSubagents();\n    },\n\n    get activeSubagents() {\n      void subagentVersion.value;\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":";;;;AAmBA,SAAgB,gBAGd,SAAiD;CAMjD,MAAM,iBAAiB,IAAIA,4BAAAA,qBAAqB;CAChD,MAAM,SAAS,IAAIC,4BAAAA,cAA8B,gBAAgB;EAC/D,UAAU,QAAQ,YAAY;EAC9B,mBAAmB,QAAQ;EAC3B,wBAAwB,QAAQ;EAChC,WAAWC,4BAAAA;EACZ,CAAC;CAEF,MAAM,gBAAA,GAAA,IAAA,YAA4C,OAAO,OAAO;CAChE,MAAM,eAAA,GAAA,IAAA,YAAkC,OAAO,MAAM;CACrD,MAAM,aAAA,GAAA,IAAA,YAAuB,OAAO,UAAU;CAE9C,MAAM,mBAAA,GAAA,IAAA,YAA6B,EAAE;CAErC,MAAM,cAAc,OAAO,gBAAgB;AACzC,eAAa,QAAQ,OAAO;AAC5B,cAAY,QAAQ,OAAO;AAC3B,YAAU,QAAQ,OAAO;AACzB,kBAAgB,SAAS;GACzB;CAEF,MAAM,UAAA,GAAA,IAAA,KAAqB,GAAG;CAE9B,IAAI,WAA0B,QAAQ,YAAY;AAElD,EAAA,GAAA,IAAA,aACQ,QAAQ,WACb,UAAU;EACT,MAAM,WAAW,SAAS;AAC1B,MAAI,aAAa,UAAU;AACzB,cAAW;AACX,UAAO,OAAO;;IAGlB,EAAE,OAAO,QAAQ,CAClB;CAED,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;AAElD,EAAA,GAAA,IAAA,sBAAqB;AACnB,eAAa;AACR,SAAO,KAAK,eAAe,EAAE,QAAQ,QAAQ,QAAQ,CAAC;GAC3D;CAEF,MAAM,kBAAkB,YAAY,cAAc;AAIhD,SAAQ,0BACR,CAAC,OAAO,aACR,gBAAgB;AAElB,EAAA,GAAA,IAAA,cACS;EACL,QACE,QAAQ,0BACR,CAAC,UAAU,SACX,YAAY,cAAc,CAAC,SAAS;EACtC,KAAK,YAAY,cAAc,CAAC;EACjC,IACA,EAAE,aAAa;AACd,MAAI,OACF,QAAO,qBAAqB,YAAY,cAAc,EAAE,EACtD,iBAAiB,MAClB,CAAC;IAGN,EAAE,WAAW,MAAM,CACpB;CAED,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;;AAGhB,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,UAAU;AACb,eAAW,OAAO,YAAY;AAC9B,YAAQ,aAAa,SAAS;;AAGhC,OAAI,CAAC,SACH,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,KAAA;GACjB,QAAQ,OAAO;AACb,YAAQ,UAAU,OAAO,KAAA,EAAU;AACnC,mBAAe,UAAU,OAAO,KAAA,EAAU;;GAE7C,CACF;;CAGH,eAAe,OACb,QACA,eACA;AACA,QAAM,aAAa,QAAQ,cAAc;;CAG3C,SAAS,UAAU,OAAe;AAChC,SAAO,QAAQ;;CAGjB,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,KAAA;GAChB,QAAQ,KAAA;GACR,eAAe,KAAA;GACf;GACD;;AAKL,QAAO;EACL,IAAI,SAAS;AACX,UAAO,aAAa,SAAU,EAAE;;EAGlC,OAAO;EACP;EAEA;EACA;EACA;EAEA;EACA;EACA;EAEA,OAAO;GACL,SAAS,EAAE;GACX,MAAM;GACN,MAAM,SAAS;AACb,WAAO;;GAET,MAAM,QAAQ;GACf;EAED,IAAI,aAAyC;AAC3C,OACE,aAAa,SAAS,QACtB,mBAAmB,aAAa,SAChC,MAAM,QAAQ,aAAa,MAAM,cAAc,EAC/C;IACA,MAAM,kBAAkB,aAAa,MAAM;AAC3C,QAAI,gBAAgB,WAAW,EAAG,QAAO,CAAC,EAAE,MAAM,cAAc,CAAC;AACjE,WAAO;;AAGT,UAAO,EAAE;;EAGX,IAAI,YAAkD;AACpD,WAAA,GAAA,4BAAA,mBAAwC,aAAa,MAAM;;EAG7D,IAAI,WAAW;AACb,OAAI,CAAC,aAAa,MAAO,QAAO,EAAE;AAClC,WAAA,GAAA,4BAAA,wBAA8B,YAAY,aAAa,MAAM,CAAC;;EAGhE,IAAI,YAAY;AACd,OAAI,CAAC,aAAa,MAAO,QAAO,EAAE;AAClC,WAAA,GAAA,+BAAA,yBAA+B,YAAY,aAAa,MAAM,CAAC;;EAGjE,aAAa,SAAkB;AAC7B,OAAI,CAAC,aAAa,MAAO,QAAO,EAAE;AAIlC,WAAA,GAAA,+BAAA,yBAFE,YAAY,aAAa,MAAM,CAChC,CACmB,QAAQ,OAAO,GAAG,UAAU,OAAO,QAAQ,GAAG;;EAGpE,IAAI,YAAY;AACT,mBAAgB;AACrB,UAAO,OAAO,cAAc;;EAG9B,IAAI,kBAAkB;AACf,mBAAgB;AACrB,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,10 @@
-import { onUnmounted, shallowRef, watch } from "vue";
-import { MessageTupleManager, StreamManager, extractInterrupts, toMessageClass } from "@langchain/langgraph-sdk/ui";
+import { onScopeDispose, ref, shallowRef, watch } from "vue";
+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,14 +13,14 @@ function useStreamCustom(options) {
 	const streamValues = shallowRef(stream.values);
 	const streamError = shallowRef(stream.error);
 	const isLoading = shallowRef(stream.isLoading);
+	const subagentVersion = shallowRef(0);
 	const unsubscribe = stream.subscribe(() => {
 		streamValues.value = stream.values;
 		streamError.value = stream.error;
 		isLoading.value = stream.isLoading;
+		subagentVersion.value += 1;
 	});
-	onUnmounted(() => {
-		unsubscribe();
-	});
+	const branch = ref("");
 	let threadId = options.threadId ?? null;
 	watch(() => options.threadId, (newId) => {
 		const resolved = newId ?? null;
@@ -28,7 +28,7 @@ function useStreamCustom(options) {
 			threadId = resolved;
 			stream.clear();
 		}
-	});
+	}, { flush: "sync" });
 	const getMessages = (value) => {
 		const messagesKey = options.messagesKey ?? "messages";
 		return Array.isArray(value[messagesKey]) ? value[messagesKey] : [];
@@ -41,6 +41,10 @@ function useStreamCustom(options) {
 		};
 	};
 	const historyValues = options.initialValues ?? {};
+	onScopeDispose(() => {
+		unsubscribe();
+		stream.stop(historyValues, { onStop: options.onStop });
+	});
 	const historyMessages = getMessages(historyValues);
 	options.filterSubagentMessages && !stream.isLoading && historyMessages.length;
 	watch(() => ({
@@ -98,12 +102,26 @@ 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);
 	}
+	function setBranch(value) {
+		branch.value = 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 {
 		get values() {
 			return streamValues.value ?? {};
@@ -113,6 +131,9 @@ function useStreamCustom(options) {
 		stop,
 		submit,
 		switchThread,
+		branch,
+		setBranch,
+		getMessagesMetadata,
 		queue: {
 			entries: [],
 			size: 0,
@@ -134,7 +155,7 @@ function useStreamCustom(options) {
 		},
 		get messages() {
 			if (!streamValues.value) return [];
-			return getMessages(streamValues.value);
+			return ensureMessageInstances(getMessages(streamValues.value));
 		},
 		get toolCalls() {
 			if (!streamValues.value) return [];
@@ -145,9 +166,11 @@ function useStreamCustom(options) {
 			return getToolCallsWithResults(getMessages(streamValues.value)).filter((tc) => tc.aiMessage.id === message.id);
 		},
 		get subagents() {
+			subagentVersion.value;
 			return stream.getSubagents();
 		},
 		get activeSubagents() {
+			subagentVersion.value;
 			return stream.getActiveSubagents();
 		},
 		getSubagent(toolCallId) {
@@ -161,7 +184,7 @@ function useStreamCustom(options) {
 		}
 	};
 }
-
 //#endregion
 export { useStreamCustom };
+
 //# sourceMappingURL=stream.custom.js.map

package/dist/stream.custom.js.map

@@ -1 +1 @@
-{"version":3,"file":"stream.custom.js","names":[],"sources":["../src/stream.custom.ts"],"sourcesContent":["import { onUnmounted, shallowRef, watch } from \"vue\";\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 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\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 = shallowRef<StateType | null>(stream.values);\n  const streamError = shallowRef<unknown>(stream.error);\n  const isLoading = shallowRef(stream.isLoading);\n\n  const unsubscribe = stream.subscribe(() => {\n    streamValues.value = stream.values;\n    streamError.value = stream.error;\n    isLoading.value = stream.isLoading;\n  });\n\n  onUnmounted(() => {\n    unsubscribe();\n  });\n\n  let threadId: string | null = options.threadId ?? null;\n\n  watch(\n    () => options.threadId,\n    (newId) => {\n      const resolved = newId ?? null;\n      if (resolved !== threadId) {\n        threadId = resolved;\n        stream.clear();\n      }\n    },\n  );\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  // @ts-expect-error used in watch callback below\n  // eslint-disable-next-line @typescript-eslint/no-unused-vars\n  const shouldReconstructSubagents =\n    options.filterSubagentMessages &&\n    !stream.isLoading &&\n    historyMessages.length > 0;\n\n  watch(\n    () => ({\n      should:\n        options.filterSubagentMessages &&\n        !isLoading.value &&\n        getMessages(historyValues).length > 0,\n      len: getMessages(historyValues).length,\n    }),\n    ({ should }) => {\n      if (should) {\n        stream.reconstructSubagents(getMessages(historyValues), {\n          skipIfPopulated: true,\n        });\n      }\n    },\n    { immediate: 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    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 (!threadId) {\n          threadId = crypto.randomUUID();\n          options.onThreadId?.(threadId);\n        }\n\n        if (!threadId) {\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: threadId,\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  return {\n    get values() {\n      return streamValues.value ?? ({} as StateType);\n    },\n\n    error: streamError,\n    isLoading,\n\n    stop,\n    submit,\n    switchThread,\n\n    queue: {\n      entries: [],\n      size: 0,\n      async cancel() {\n        return false;\n      },\n      async clear() {},\n    },\n\n    get interrupts(): Interrupt<InterruptType>[] {\n      if (\n        streamValues.value != null &&\n        \"__interrupt__\" in streamValues.value &&\n        Array.isArray(streamValues.value.__interrupt__)\n      ) {\n        const valueInterrupts = streamValues.value.__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.value);\n    },\n\n    get messages(): Message[] {\n      if (!streamValues.value) return [];\n      return getMessages(streamValues.value);\n    },\n\n    get toolCalls() {\n      if (!streamValues.value) return [];\n      return getToolCallsWithResults(getMessages(streamValues.value));\n    },\n\n    getToolCalls(message: Message) {\n      if (!streamValues.value) return [];\n      const allToolCalls = getToolCallsWithResults(\n        getMessages(streamValues.value),\n      );\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":";;;;;AAiBA,SAAgB,gBAGd,SAAiD;CAOjD,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,WAA6B,OAAO,OAAO;CAChE,MAAM,cAAc,WAAoB,OAAO,MAAM;CACrD,MAAM,YAAY,WAAW,OAAO,UAAU;CAE9C,MAAM,cAAc,OAAO,gBAAgB;AACzC,eAAa,QAAQ,OAAO;AAC5B,cAAY,QAAQ,OAAO;AAC3B,YAAU,QAAQ,OAAO;GACzB;AAEF,mBAAkB;AAChB,eAAa;GACb;CAEF,IAAI,WAA0B,QAAQ,YAAY;AAElD,aACQ,QAAQ,WACb,UAAU;EACT,MAAM,WAAW,SAAS;AAC1B,MAAI,aAAa,UAAU;AACzB,cAAW;AACX,UAAO,OAAO;;GAGnB;CAED,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;AAIhD,SAAQ,0BACR,CAAC,OAAO,aACR,gBAAgB;AAElB,cACS;EACL,QACE,QAAQ,0BACR,CAAC,UAAU,SACX,YAAY,cAAc,CAAC,SAAS;EACtC,KAAK,YAAY,cAAc,CAAC;EACjC,IACA,EAAE,aAAa;AACd,MAAI,OACF,QAAO,qBAAqB,YAAY,cAAc,EAAE,EACtD,iBAAiB,MAClB,CAAC;IAGN,EAAE,WAAW,MAAM,CACpB;CAED,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;;AAGhB,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,UAAU;AACb,eAAW,OAAO,YAAY;AAC9B,YAAQ,aAAa,SAAS;;AAGhC,OAAI,CAAC,SACH,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;;AAG3C,QAAO;EACL,IAAI,SAAS;AACX,UAAO,aAAa,SAAU,EAAE;;EAGlC,OAAO;EACP;EAEA;EACA;EACA;EAEA,OAAO;GACL,SAAS,EAAE;GACX,MAAM;GACN,MAAM,SAAS;AACb,WAAO;;GAET,MAAM,QAAQ;GACf;EAED,IAAI,aAAyC;AAC3C,OACE,aAAa,SAAS,QACtB,mBAAmB,aAAa,SAChC,MAAM,QAAQ,aAAa,MAAM,cAAc,EAC/C;IACA,MAAM,kBAAkB,aAAa,MAAM;AAC3C,QAAI,gBAAgB,WAAW,EAAG,QAAO,CAAC,EAAE,MAAM,cAAc,CAAC;AACjE,WAAO;;AAGT,UAAO,EAAE;;EAGX,IAAI,YAAkD;AACpD,UAAO,kBAAiC,aAAa,MAAM;;EAG7D,IAAI,WAAsB;AACxB,OAAI,CAAC,aAAa,MAAO,QAAO,EAAE;AAClC,UAAO,YAAY,aAAa,MAAM;;EAGxC,IAAI,YAAY;AACd,OAAI,CAAC,aAAa,MAAO,QAAO,EAAE;AAClC,UAAO,wBAAwB,YAAY,aAAa,MAAM,CAAC;;EAGjE,aAAa,SAAkB;AAC7B,OAAI,CAAC,aAAa,MAAO,QAAO,EAAE;AAIlC,UAHqB,wBACnB,YAAY,aAAa,MAAM,CAChC,CACmB,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 { onScopeDispose, ref, shallowRef, watch } from \"vue\";\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 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\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 = shallowRef<StateType | null>(stream.values);\n  const streamError = shallowRef<unknown>(stream.error);\n  const isLoading = shallowRef(stream.isLoading);\n\n  const subagentVersion = shallowRef(0);\n\n  const unsubscribe = stream.subscribe(() => {\n    streamValues.value = stream.values;\n    streamError.value = stream.error;\n    isLoading.value = stream.isLoading;\n    subagentVersion.value += 1;\n  });\n\n  const branch = ref<string>(\"\");\n\n  let threadId: string | null = options.threadId ?? null;\n\n  watch(\n    () => options.threadId,\n    (newId) => {\n      const resolved = newId ?? null;\n      if (resolved !== threadId) {\n        threadId = resolved;\n        stream.clear();\n      }\n    },\n    { flush: \"sync\" },\n  );\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  onScopeDispose(() => {\n    unsubscribe();\n    void stream.stop(historyValues, { onStop: options.onStop });\n  });\n\n  const historyMessages = getMessages(historyValues);\n  // @ts-expect-error used in watch callback below\n  // eslint-disable-next-line @typescript-eslint/no-unused-vars\n  const shouldReconstructSubagents =\n    options.filterSubagentMessages &&\n    !stream.isLoading &&\n    historyMessages.length > 0;\n\n  watch(\n    () => ({\n      should:\n        options.filterSubagentMessages &&\n        !isLoading.value &&\n        getMessages(historyValues).length > 0,\n      len: getMessages(historyValues).length,\n    }),\n    ({ should }) => {\n      if (should) {\n        stream.reconstructSubagents(getMessages(historyValues), {\n          skipIfPopulated: true,\n        });\n      }\n    },\n    { immediate: 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    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 (!threadId) {\n          threadId = crypto.randomUUID();\n          options.onThreadId?.(threadId);\n        }\n\n        if (!threadId) {\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: threadId,\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  function setBranch(value: string) {\n    branch.value = value;\n  }\n\n  function getMessagesMetadata(\n    message: Message,\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    get values() {\n      return streamValues.value ?? ({} as StateType);\n    },\n\n    error: streamError,\n    isLoading,\n\n    stop,\n    submit,\n    switchThread,\n\n    branch,\n    setBranch,\n    getMessagesMetadata,\n\n    queue: {\n      entries: [],\n      size: 0,\n      async cancel() {\n        return false;\n      },\n      async clear() {},\n    },\n\n    get interrupts(): Interrupt<InterruptType>[] {\n      if (\n        streamValues.value != null &&\n        \"__interrupt__\" in streamValues.value &&\n        Array.isArray(streamValues.value.__interrupt__)\n      ) {\n        const valueInterrupts = streamValues.value.__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.value);\n    },\n\n    get messages() {\n      if (!streamValues.value) return [];\n      return ensureMessageInstances(getMessages(streamValues.value));\n    },\n\n    get toolCalls() {\n      if (!streamValues.value) return [];\n      return getToolCallsWithResults(getMessages(streamValues.value));\n    },\n\n    getToolCalls(message: Message) {\n      if (!streamValues.value) return [];\n      const allToolCalls = getToolCallsWithResults(\n        getMessages(streamValues.value),\n      );\n      return allToolCalls.filter((tc) => tc.aiMessage.id === message.id);\n    },\n\n    get subagents() {\n      void subagentVersion.value;\n      return stream.getSubagents();\n    },\n\n    get activeSubagents() {\n      void subagentVersion.value;\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":";;;;AAmBA,SAAgB,gBAGd,SAAiD;CAMjD,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,WAA6B,OAAO,OAAO;CAChE,MAAM,cAAc,WAAoB,OAAO,MAAM;CACrD,MAAM,YAAY,WAAW,OAAO,UAAU;CAE9C,MAAM,kBAAkB,WAAW,EAAE;CAErC,MAAM,cAAc,OAAO,gBAAgB;AACzC,eAAa,QAAQ,OAAO;AAC5B,cAAY,QAAQ,OAAO;AAC3B,YAAU,QAAQ,OAAO;AACzB,kBAAgB,SAAS;GACzB;CAEF,MAAM,SAAS,IAAY,GAAG;CAE9B,IAAI,WAA0B,QAAQ,YAAY;AAElD,aACQ,QAAQ,WACb,UAAU;EACT,MAAM,WAAW,SAAS;AAC1B,MAAI,aAAa,UAAU;AACzB,cAAW;AACX,UAAO,OAAO;;IAGlB,EAAE,OAAO,QAAQ,CAClB;CAED,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;AAElD,sBAAqB;AACnB,eAAa;AACR,SAAO,KAAK,eAAe,EAAE,QAAQ,QAAQ,QAAQ,CAAC;GAC3D;CAEF,MAAM,kBAAkB,YAAY,cAAc;AAIhD,SAAQ,0BACR,CAAC,OAAO,aACR,gBAAgB;AAElB,cACS;EACL,QACE,QAAQ,0BACR,CAAC,UAAU,SACX,YAAY,cAAc,CAAC,SAAS;EACtC,KAAK,YAAY,cAAc,CAAC;EACjC,IACA,EAAE,aAAa;AACd,MAAI,OACF,QAAO,qBAAqB,YAAY,cAAc,EAAE,EACtD,iBAAiB,MAClB,CAAC;IAGN,EAAE,WAAW,MAAM,CACpB;CAED,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;;AAGhB,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,UAAU;AACb,eAAW,OAAO,YAAY;AAC9B,YAAQ,aAAa,SAAS;;AAGhC,OAAI,CAAC,SACH,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,KAAA;GACjB,QAAQ,OAAO;AACb,YAAQ,UAAU,OAAO,KAAA,EAAU;AACnC,mBAAe,UAAU,OAAO,KAAA,EAAU;;GAE7C,CACF;;CAGH,eAAe,OACb,QACA,eACA;AACA,QAAM,aAAa,QAAQ,cAAc;;CAG3C,SAAS,UAAU,OAAe;AAChC,SAAO,QAAQ;;CAGjB,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,KAAA;GAChB,QAAQ,KAAA;GACR,eAAe,KAAA;GACf;GACD;;AAKL,QAAO;EACL,IAAI,SAAS;AACX,UAAO,aAAa,SAAU,EAAE;;EAGlC,OAAO;EACP;EAEA;EACA;EACA;EAEA;EACA;EACA;EAEA,OAAO;GACL,SAAS,EAAE;GACX,MAAM;GACN,MAAM,SAAS;AACb,WAAO;;GAET,MAAM,QAAQ;GACf;EAED,IAAI,aAAyC;AAC3C,OACE,aAAa,SAAS,QACtB,mBAAmB,aAAa,SAChC,MAAM,QAAQ,aAAa,MAAM,cAAc,EAC/C;IACA,MAAM,kBAAkB,aAAa,MAAM;AAC3C,QAAI,gBAAgB,WAAW,EAAG,QAAO,CAAC,EAAE,MAAM,cAAc,CAAC;AACjE,WAAO;;AAGT,UAAO,EAAE;;EAGX,IAAI,YAAkD;AACpD,UAAO,kBAAiC,aAAa,MAAM;;EAG7D,IAAI,WAAW;AACb,OAAI,CAAC,aAAa,MAAO,QAAO,EAAE;AAClC,UAAO,uBAAuB,YAAY,aAAa,MAAM,CAAC;;EAGhE,IAAI,YAAY;AACd,OAAI,CAAC,aAAa,MAAO,QAAO,EAAE;AAClC,UAAO,wBAAwB,YAAY,aAAa,MAAM,CAAC;;EAGjE,aAAa,SAAkB;AAC7B,OAAI,CAAC,aAAa,MAAO,QAAO,EAAE;AAIlC,UAHqB,wBACnB,YAAY,aAAa,MAAM,CAChC,CACmB,QAAQ,OAAO,GAAG,UAAU,OAAO,QAAQ,GAAG;;EAGpE,IAAI,YAAY;AACT,mBAAgB;AACrB,UAAO,OAAO,cAAc;;EAG9B,IAAI,kBAAkB;AACf,mBAAgB;AACrB,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,43 +1,48 @@
 {
   "name": "@langchain/vue",
-  "version": "0.0.1",
+  "version": "0.1.2",
   "description": "Vue integration for LangGraph & LangChain",
   "type": "module",
   "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+ssh://git@github.com/langchain-ai/langgraphjs.git",
+    "directory": "libs/sdk-vue"
+  },
   "dependencies": {
-    "@langchain/langgraph-sdk": "^1.6.5"
+    "@langchain/langgraph-sdk": "^1.7.0"
   },
   "devDependencies": {
-    "@hono/node-server": "^1.14.1",
-    "@langchain/core": "^1.1.27",
+    "@hono/node-server": "^1.19.11",
+    "@langchain/core": "^1.1.31",
     "@langchain/scripts": "^0.1.4",
-    "@tsconfig/recommended": "^1.0.2",
-    "@types/node": "^18.15.11",
-    "@types/uuid": "^9.0.1",
+    "@tsconfig/recommended": "^1.0.13",
+    "@types/node": "^25.4.0",
     "@typescript-eslint/eslint-plugin": "^6.12.0",
     "@typescript-eslint/parser": "^6.12.0",
     "@vitejs/plugin-vue": "^6.0.4",
     "@vitejs/plugin-vue-jsx": "^5.1.4",
-    "@vitest/browser": "^3.2.4",
-    "deepagents": "^1.8.0",
+    "@vitest/browser": "^4.0.18",
+    "@vitest/browser-webdriverio": "^4.0.18",
+    "deepagents": "^1.8.1",
     "eslint": "^8.33.0",
     "eslint-config-airbnb-base": "^15.0.0",
     "eslint-config-prettier": "^8.6.0",
     "eslint-plugin-import": "^2.29.1",
     "eslint-plugin-no-instanceof": "^1.0.1",
     "eslint-plugin-prettier": "^4.2.1",
-    "hono": "^4.8.2",
-    "langchain": "^1.2.25",
+    "hono": "^4.12.7",
+    "langchain": "^1.2.30",
     "prettier": "^3.8.1",
-    "typescript": "^4.9.5 || ^5.4.5",
-    "vitest": "^3.2.4",
-    "vitest-browser-vue": "^0.1.0",
-    "vue": "^3.5.24",
-    "webdriverio": "^9.14.2",
+    "typescript": "^5.9.3",
+    "vitest": "^4.0.18",
+    "vitest-browser-vue": "^2.0.2",
+    "vue": "^3.5.30",
+    "webdriverio": "^9.25.0",
     "zod": "^4.3.6",
-    "@langchain/langgraph": "1.2.0",
-    "@langchain/langgraph-checkpoint": "1.0.0",
-    "@langchain/langgraph-api": "1.1.14"
+    "@langchain/langgraph": "1.2.2",
+    "@langchain/langgraph-api": "1.1.15",
+    "@langchain/langgraph-checkpoint": "1.0.0"
   },
   "peerDependencies": {
     "@langchain/core": "^1.1.27",