langgraph-ui-components 0.0.25 → 0.0.28

package/dist/providers/Stream.cjs.js.map

@@ -1 +1 @@
-{"version":3,"file":"Stream.cjs.js","sources":["../../src/providers/Stream.tsx"],"sourcesContent":["/* @refresh reset */\nimport { useChatRuntime } from \"@/providers/ChatRuntime\";\nimport { logger } from \"@/utils/logger\";\nimport { type AIMessage, type Message } from \"@langchain/langgraph-sdk\";\n\n// Derive the tool-call types from AIMessage since the SDK doesn't re-export them\ntype ToolCall = NonNullable<AIMessage[\"tool_calls\"]>[number];\ntype InvalidToolCall = NonNullable<AIMessage[\"invalid_tool_calls\"]>[number];\nimport { useStream, type UseStream } from \"@langchain/langgraph-sdk/react\";\nimport {\n  isRemoveUIMessage,\n  isUIMessage,\n  type RemoveUIMessage,\n  type UIMessage,\n  uiMessageReducer,\n} from \"@langchain/langgraph-sdk/react-ui\";\nimport {\n  createContext,\n  type ReactNode,\n  useCallback,\n  useContext,\n  useEffect,\n  useMemo,\n  useState,\n} from \"react\";\nimport { toast } from \"sonner\";\nimport { v4 as uuidv4 } from \"uuid\";\nimport { useThread } from \"./Thread\";\n\nexport type StateType = {\n  messages: Message[];\n  ui?: UIMessage[];\n  suggestions?: string[];\n};\n\nconst useTypedStream = useStream<\n  StateType,\n  {\n    UpdateType: {\n      messages?: Message[] | Message | string;\n      ui?: (UIMessage | RemoveUIMessage)[] | UIMessage | RemoveUIMessage;\n    };\n    CustomEventType: UIMessage | RemoveUIMessage;\n  }\n>;\n\ntype StreamContextType = UseStream<StateType, {\n  UpdateType: {\n\n    messages?: Message[] | Message | string;\n    ui?: (UIMessage | RemoveUIMessage)[] | UIMessage | RemoveUIMessage;\n  };\n  CustomEventType: UIMessage | RemoveUIMessage;\n}> & {\n  sendMessage: (\n    message: Message | string,\n    options?: {\n      /** If true, message is meant for agent only (not user-visible) */\n      type?: Message[\"type\"];\n      /** Name field required for function/tool messages */\n      name?: string;\n      /** If true, message is hidden from user UI */\n      hidden?: boolean;\n      /** Tool calls associated with this message */\n      tool_calls?: ToolCall[];\n      /** Invalid tool calls associated with this message */\n      invalid_tool_calls?: InvalidToolCall[];\n      /** If provided, use this ID for the message instead of generating one */\n      tool_call_id?: string;\n      /** Additional kwargs to attach to the message */\n      additional_kwargs?: Record<string, unknown>;\n      /** UI components to display alongside the message */\n      ui?: UIMessage[];\n      /** Custom ID for the message (will also be used for UI components) */\n      id?: string;\n      /** Custom context to override/merge with identity */\n      context?: Record<string, unknown>;\n    }\n  ) => Promise<void>;\n  submitMessage: (\n    message: Message,\n    options?: {\n      streamMode?: (\"values\" | \"updates\" | \"messages\" | \"custom\" | \"debug\")[];\n      streamSubgraphs?: boolean;\n      streamResumable?: boolean;\n      contextValues?: Record<string, unknown>;\n    }\n  ) => Promise<void>;\n  regenerateMessage: (messageId: string) => Promise<void>;\n  fetchCatalog: () => Promise<unknown>;\n};\n\nconst StreamContext = createContext<StreamContextType | undefined>(undefined);\n\nasync function checkGraphStatus(apiUrl: string, authToken: string | null | undefined) {\n  try {\n    const res = await fetch(`${apiUrl}/info`, {\n      headers: authToken ? { Authorization: `Bearer ${authToken}` } : undefined,\n    });\n    return res.ok;\n  } catch {\n    logger.error(\"Failed to initialize LangGraph Agent\");\n  }\n}\n\nasync function fetchCatalog(apiUrl: string, authToken: string | null | undefined) {\n  try {\n    const res = await fetch(`${apiUrl}/agents/catalog`, {\n      headers: authToken ? { Authorization: `Bearer ${authToken}` } : undefined,\n    });\n    if (res.ok) {\n      const data = await res.json();\n      return data;\n    }\n    return null;\n  }\n  catch {\n    logger.error(\"Failed to fetch LangGraph Catalog\");\n    return null;\n  }\n}\n\nasync function sleep(ms = 6000) {\n  return new Promise((resolve) => setTimeout(resolve, ms));\n}\n\nconst StreamSession = ({ children }: { children: ReactNode }) => {\n  const { apiUrl, assistantId, identity } = useChatRuntime();\n  const { mode, threadId, setThreadId, getThreads, setThreads, configuration } = useThread();\n\n  // Store local-only AI messages that shouldn't trigger backend calls\n  const [localMessages, setLocalMessages] = useState<Message[]>([]);\n  // Store local-only UI components\n  const [localUI, setLocalUI] = useState<UIMessage[]>([]);\n  // Track if catalog has been fetched to prevent duplicate calls\n  const [catalogFetched, setCatalogFetched] = useState(false);\n  const [catalogCache, setCatalogCache] = useState<unknown>(null);\n\n  const streamValue = useTypedStream({\n    apiUrl,\n    assistantId,\n    threadId: mode === \"multi\" ? threadId : null,\n    defaultHeaders: identity?.authToken\n      ? { Authorization: `Bearer ${identity?.authToken}` }\n      : undefined,\n    fetchStateHistory: true,\n    onUpdateEvent: (data, options) => {\n      // Scan update data (potentially nested by node name) for AI messages\n      // with reasoning_content. Inject them into state early so they're\n      // visible during streaming of the final text response.\n      const reasoningMessages: Message[] = [];\n\n      const scan = (obj: unknown) => {\n        if (!obj || typeof obj !== \"object\") return;\n        const o = obj as Record<string, unknown>;\n        if (Array.isArray(o.messages)) {\n          for (const m of o.messages as Message[]) {\n            const ak = (m as Record<string, unknown>).additional_kwargs as Record<string, unknown> | undefined;\n            if (\n              m?.type === \"ai\" &&\n              m.id &&\n              typeof ak?.reasoning_content === \"string\" &&\n              ak.reasoning_content.length > 0\n            ) {\n              reasoningMessages.push(m);\n            }\n          }\n        }\n        for (const val of Object.values(o)) {\n          if (val && typeof val === \"object\" && !Array.isArray(val)) {\n            scan(val);\n          }\n        }\n      };\n\n      scan(data);\n\n      if (reasoningMessages.length > 0) {\n        options.mutate((prev) => {\n          const prevMessages = (prev.messages ?? []) as Message[];\n          const prevIds = new Set(prevMessages.map((m) => m.id).filter(Boolean));\n          const newMessages = reasoningMessages.filter((m) => !prevIds.has(m.id!));\n          if (newMessages.length === 0) return prev;\n          return { ...prev, messages: [...prevMessages, ...newMessages] };\n        });\n      }\n    },\n    onCustomEvent: (event, options) => {\n      if (isUIMessage(event) || isRemoveUIMessage(event)) {\n        options.mutate((prev) => {\n          logger.debug(\"UI Event received in StreamProvider:\", event);\n          const ui = uiMessageReducer(prev.ui ?? [], event);\n          return { ...prev, ui };\n        });\n      }\n    },\n\n    onThreadId: (id) => {\n      if (!id) return;\n      if (mode === \"single\") {\n        // lock once\n        if (!threadId) setThreadId(id);\n      } else {\n        // switch freely\n        setThreadId(id);\n        // Refetch threads list when thread ID changes.\n        // Wait for some seconds before fetching so we're able to get the new thread that was created.\n        sleep().then(() => getThreads().then(setThreads).catch((err) => logger.error(\"Failed to refresh threads:\", err)));\n      }\n    },\n  });\n\n  /**\n   * Send a message to the agent programmatically\n   * Useful for triggering agent actions without user-visible messages\n   * e.g., \"login completed\", \"card clicked: {id}\"\n   */\n  const sendMessage = useCallback(\n    async (\n      message: Message | string,\n      options?: {\n        type?: Message[\"type\"];\n        name?: string;\n        hidden?: boolean;\n        tool_calls?: ToolCall[];\n        invalid_tool_calls?: InvalidToolCall[];\n        tool_call_id?: string;\n        additional_kwargs?: Record<string, unknown>;\n        ui?: UIMessage[];\n        id?: string; // Allow passing custom ID\n        context?: Record<string, unknown>; // Allow passing custom context to override identity\n      }\n    ) => {\n      // Use provided ID or generate new one\n      const messageId = options?.id || uuidv4();\n\n      const messageObj: Message =\n        typeof message === \"string\"\n          ? ({\n            id: messageId, // Use the determined ID\n            type: options?.type ?? \"human\",\n            content: message,\n            ...(options?.name && { name: options.name }),\n            ...(options?.tool_calls && { tool_calls: options.tool_calls }),\n            ...(options?.invalid_tool_calls && { invalid_tool_calls: options.invalid_tool_calls }),\n            ...(options?.tool_call_id && { tool_call_id: options.tool_call_id }),\n            ...((options?.additional_kwargs || options?.hidden) && {\n              additional_kwargs: {\n                ...(options?.hidden && { hidden: true }),\n                ...(options?.additional_kwargs ?? {}),\n              }\n            }),\n          } as Message)\n          : {\n            ...message,\n            id: messageId, // Override with determined ID\n            // Allow overriding message fields with options\n            ...(options?.tool_calls && { tool_calls: options.tool_calls }),\n            ...(options?.additional_kwargs && {\n              additional_kwargs: {\n                ...message.additional_kwargs,\n                ...options.additional_kwargs,\n              }\n            }),\n          };\n\n      // console.log(\"Sending message via sendMessage:\", messageObj); → converted to debug log\n      logger.debug(\"Sending message via sendMessage:\", messageObj);\n\n      // If message type is \"ai\", just append to local state without submitting to agent\n      // This is useful for injecting initial messages or system messages that don't need agent processing\n      if (options?.type === \"ai\") {\n        setLocalMessages((prev) => [...prev, messageObj]);\n        logger.debug(\"Appended AI message to localMessages:\", options?.ui);\n        // Also store any UI components, linking them to the message id\n        if (options.ui && options.ui.length > 0) {\n          const uiWithMessageId = options.ui.map((ui, index) => ({\n            ...ui,\n            id: messageId, // Link to the message ID so CustomComponentRender can find it\n            // Ensure each UI component has a unique identifier for React keys\n            _key: (ui as UIMessage & { _key?: string })._key || `${messageId}-ui-${index}`,\n          })) as UIMessage[];\n          setLocalUI((prev) => [...prev, ...uiWithMessageId]);\n        }\n        return;\n      }\n\n      // For non-AI messages, submit to the agent\n      const currentMessages = streamValue.messages || [];\n      // Merge identity with custom context, custom context takes precedence\n      const mergedContext = options?.context\n        ? { ...identity, ...options.context, ...configuration }\n        : identity || {};\n\n      await streamValue.submit(\n        { messages: [...currentMessages, messageObj] },\n        {\n          context: mergedContext,\n          streamMode: [\"values\", \"messages-tuple\", \"updates\", \"custom\"],\n        }\n      );\n    },\n    [streamValue, identity, configuration, setLocalMessages, setLocalUI]\n  );\n\n  // Combine stream messages with local AI messages, deduplicating by ID\n  const combinedMessages = useMemo(() => {\n    const allMessages = [...localMessages, ...(streamValue.messages || [])];\n    // Deduplicate by message ID, keeping the first occurrence\n    const seen = new Set<string>();\n    return allMessages.filter(msg => {\n      if (!msg.id) return true; // Keep messages without IDs\n      if (seen.has(msg.id)) return false;\n      seen.add(msg.id);\n      return true;\n    });\n  }, [localMessages, streamValue.messages]);\n\n  // Combine local UI with stream UI, deduplicating by unique key\n  const combinedUI = useMemo(() => {\n    logger.debug(\"Combining local UI with stream UI:\", localUI, streamValue.values?.ui);\n    const allUI = [...localUI, ...(streamValue.values?.ui || [])];\n    // Deduplicate by _key or id+name combination\n    const seen = new Set<string>();\n    return allUI.filter(ui => {\n      const key = (ui as UIMessage & { _key?: string })._key || `${ui.id}-${ui.name}`;\n      if (seen.has(key)) return false;\n      seen.add(key);\n      return true;\n    });\n  }, [localUI, streamValue.values?.ui]);\n\n  const submitMessage = useCallback(\n    async (\n      message: Message,\n      options?: {\n        streamMode?: (\"values\" | \"updates\" | \"messages\" | \"custom\" | \"debug\")[];\n        streamSubgraphs?: boolean;\n        streamResumable?: boolean;\n        contextValues?: Record<string, unknown>;\n      }\n    ) => {\n      // Get ALL current messages (including local AI messages)\n      const allCurrentMessages = combinedMessages || [];\n\n      await streamValue.submit(\n        { messages: [...allCurrentMessages, message] },\n        {\n          context: { ...identity, ...options?.contextValues },\n          streamMode: options?.streamMode || [\"values\", \"messages-tuple\", \"updates\", \"custom\"],\n          streamSubgraphs: options?.streamSubgraphs ?? true,\n          streamResumable: options?.streamResumable ?? true,\n          optimisticValues: (prev) => ({\n            ...prev,\n            messages: [...allCurrentMessages, message],\n          }),\n        }\n      );\n    },\n    [streamValue, identity, combinedMessages]\n  );\n\n  /**\n   * Regenerate an AI response by finding the last human message before the specified AI message\n   * and resubmitting from that point\n   */\n  const regenerateMessage = useCallback(\n    async (messageId: string) => {\n      const allMessages = combinedMessages || [];\n      const messageIndex = allMessages.findIndex((msg) => msg.id === messageId);\n\n      if (messageIndex === -1) {\n        logger.error(\"Message not found for regeneration:\", messageId);\n        return;\n      }\n\n      // Find the last human message before this AI message\n      let lastHumanIndex = messageIndex - 1;\n      while (lastHumanIndex >= 0 && allMessages[lastHumanIndex].type !== \"human\") {\n        lastHumanIndex--;\n      }\n\n      if (lastHumanIndex === -1) {\n        logger.error(\"No human message found before AI message:\", messageId);\n        return;\n      }\n\n      // Get all messages up to and including the last human message\n      const messagesToKeep = allMessages.slice(0, lastHumanIndex + 1);\n\n      // Remove the last human message from the list since we'll resubmit it\n      const messagesBeforeHuman = messagesToKeep.slice(0, -1);\n      const humanMessage = messagesToKeep[messagesToKeep.length - 1];\n\n      // Clear local messages that came after the human message\n      setLocalMessages((prev) =>\n        prev.filter((msg) => {\n          const msgIndex = allMessages.findIndex((m) => m.id === msg.id);\n          return msgIndex < lastHumanIndex;\n        })\n      );\n\n      // Clear local UI that came after the human message  \n      setLocalUI((prev) =>\n        prev.filter((ui) => {\n          const msgIndex = allMessages.findIndex((m) => m.id === ui.id);\n          return msgIndex < lastHumanIndex;\n        })\n      );\n\n      // Resubmit from the human message\n      await streamValue.submit(\n        { messages: [...messagesBeforeHuman, humanMessage] },\n        {\n          streamMode: [\"values\"],\n          streamSubgraphs: true,\n          streamResumable: true,\n        }\n      );\n    },\n    [combinedMessages, streamValue, identity, configuration, setLocalMessages, setLocalUI]\n  );\n\n  useEffect(() => {\n    checkGraphStatus(apiUrl, identity?.authToken).then((ok) => {\n      if (!ok) {\n        toast.error(\"Failed to connect to LangGraph server\", {\n          description: `Unable to reach ${apiUrl}`,\n          duration: 10000,\n        });\n      }\n    });\n  }, [apiUrl, identity?.authToken]);\n\n  const fetchCatalogMemoized = useCallback(async () => {\n    // Return cached data if already fetched\n    if (catalogFetched && catalogCache !== null) {\n      return catalogCache;\n    }\n\n    // Fetch catalog only once\n    if (!catalogFetched) {\n      setCatalogFetched(true);\n      const data = await fetchCatalog(apiUrl, identity?.authToken);\n      setCatalogCache(data);\n      return data;\n    }\n\n    return catalogCache;\n  }, [apiUrl, identity?.authToken, catalogFetched, catalogCache]);\n\n  const value = useMemo(\n    () => ({\n      ...streamValue,\n      messages: combinedMessages, // Override with combined messages\n      values: {\n        ...streamValue.values,\n        ui: combinedUI, // Override with combined UI\n      },\n      sendMessage,\n      submitMessage,\n      regenerateMessage,\n      fetchCatalog: fetchCatalogMemoized,\n    }),\n    [streamValue, combinedMessages, combinedUI, sendMessage, submitMessage, regenerateMessage, fetchCatalogMemoized]\n  );\n\n  return (\n    <StreamContext.Provider value={value}>{children}</StreamContext.Provider>\n  );\n};\n\n/**\n * Provides streaming message functionality for real-time AI responses.\n * Manages message state, handles streaming updates, and provides submit/sendMessage functions.\n * \n * @example\n * ```tsx\n * <StreamProvider>\n *   <ChatInterface />\n * </StreamProvider>\n * ```\n */\nexport function StreamProvider({ children }: { children: ReactNode }) {\n  return <StreamSession>{children}</StreamSession>;\n}\n\n/**\n * Hook to access the streaming context.\n * Provides access to messages, loading state, and functions to submit messages.\n * \n * @throws {Error} If used outside of StreamProvider\n * \n * @example\n * ```tsx\n * const { messages, isLoading, submit, sendMessage } = useStreamContext();\n * ```\n */\nexport function useStreamContext(): StreamContextType {\n  const ctx = useContext(StreamContext);\n  if (!ctx) {\n    throw new Error(\"useStreamContext must be used within StreamProvider\");\n  }\n  return ctx;\n}"],"names":["useTypedStream","useStream","StreamContext","createContext","checkGraphStatus","apiUrl","authToken","logger","fetchCatalog","res","sleep","ms","resolve","StreamSession","children","assistantId","identity","useChatRuntime","mode","threadId","setThreadId","getThreads","setThreads","configuration","useThread","localMessages","setLocalMessages","useState","localUI","setLocalUI","catalogFetched","setCatalogFetched","catalogCache","setCatalogCache","streamValue","data","options","reasoningMessages","scan","obj","o","m","ak","val","prev","prevMessages","prevIds","newMessages","event","isUIMessage","isRemoveUIMessage","ui","uiMessageReducer","id","err","sendMessage","useCallback","message","messageId","uuidv4","messageObj","uiWithMessageId","index","currentMessages","mergedContext","combinedMessages","useMemo","allMessages","seen","msg","combinedUI","allUI","key","submitMessage","allCurrentMessages","regenerateMessage","messageIndex","lastHumanIndex","messagesToKeep","messagesBeforeHuman","humanMessage","useEffect","ok","toast","fetchCatalogMemoized","value","jsx","StreamProvider","useStreamContext","ctx","useContext"],"mappings":"odAmCMA,EAAiBC,EAAAA,UAyDjBC,EAAgBC,EAAAA,cAA6C,MAAS,EAE5E,eAAeC,EAAiBC,EAAgBC,EAAsC,CACpF,GAAI,CAIF,OAHY,MAAM,MAAM,GAAGD,CAAM,QAAS,CACxC,QAASC,EAAY,CAAE,cAAe,UAAUA,CAAS,IAAO,MAAA,CACjE,GACU,EACb,MAAQ,CACNC,EAAAA,OAAO,MAAM,sCAAsC,CACrD,CACF,CAEA,eAAeC,EAAaH,EAAgBC,EAAsC,CAChF,GAAI,CACF,MAAMG,EAAM,MAAM,MAAM,GAAGJ,CAAM,kBAAmB,CAClD,QAASC,EAAY,CAAE,cAAe,UAAUA,CAAS,IAAO,MAAA,CACjE,EACD,OAAIG,EAAI,GACO,MAAMA,EAAI,KAAA,EAGlB,IACT,MACM,CACJF,OAAAA,EAAAA,OAAO,MAAM,mCAAmC,EACzC,IACT,CACF,CAEA,eAAeG,EAAMC,EAAK,IAAM,CAC9B,OAAO,IAAI,QAASC,GAAY,WAAWA,EAASD,CAAE,CAAC,CACzD,CAEA,MAAME,EAAgB,CAAC,CAAE,SAAAC,KAAwC,CAC/D,KAAM,CAAE,OAAAT,EAAQ,YAAAU,EAAa,SAAAC,CAAA,EAAaC,EAAAA,eAAA,EACpC,CAAE,KAAAC,EAAM,SAAAC,EAAU,YAAAC,EAAa,WAAAC,EAAY,WAAAC,EAAY,cAAAC,CAAA,EAAkBC,YAAA,EAGzE,CAACC,EAAeC,CAAgB,EAAIC,EAAAA,SAAoB,CAAA,CAAE,EAE1D,CAACC,EAASC,CAAU,EAAIF,EAAAA,SAAsB,CAAA,CAAE,EAEhD,CAACG,EAAgBC,CAAiB,EAAIJ,EAAAA,SAAS,EAAK,EACpD,CAACK,EAAcC,CAAe,EAAIN,EAAAA,SAAkB,IAAI,EAExDO,EAAclC,EAAe,CACjC,OAAAK,EACA,YAAAU,EACA,SAAUG,IAAS,QAAUC,EAAW,KACxC,eAAgBH,GAAU,UACtB,CAAE,cAAe,UAAUA,GAAU,SAAS,EAAA,EAC9C,OACJ,kBAAmB,GACnB,cAAe,CAACmB,EAAMC,IAAY,CAIhC,MAAMC,EAA+B,CAAA,EAE/BC,EAAQC,GAAiB,CAC7B,GAAI,CAACA,GAAO,OAAOA,GAAQ,SAAU,OACrC,MAAMC,EAAID,EACV,GAAI,MAAM,QAAQC,EAAE,QAAQ,EAC1B,UAAWC,KAAKD,EAAE,SAAuB,CACvC,MAAME,EAAMD,EAA8B,kBAExCA,GAAG,OAAS,MACZA,EAAE,IACF,OAAOC,GAAI,mBAAsB,UACjCA,EAAG,kBAAkB,OAAS,GAE9BL,EAAkB,KAAKI,CAAC,CAE5B,CAEF,UAAWE,KAAO,OAAO,OAAOH,CAAC,EAC3BG,GAAO,OAAOA,GAAQ,UAAY,CAAC,MAAM,QAAQA,CAAG,GACtDL,EAAKK,CAAG,CAGd,EAEAL,EAAKH,CAAI,EAELE,EAAkB,OAAS,GAC7BD,EAAQ,OAAQQ,GAAS,CACvB,MAAMC,EAAgBD,EAAK,UAAY,CAAA,EACjCE,EAAU,IAAI,IAAID,EAAa,IAAK,GAAM,EAAE,EAAE,EAAE,OAAO,OAAO,CAAC,EAC/DE,EAAcV,EAAkB,OAAQ,GAAM,CAACS,EAAQ,IAAI,EAAE,EAAG,CAAC,EACvE,OAAIC,EAAY,SAAW,EAAUH,EAC9B,CAAE,GAAGA,EAAM,SAAU,CAAC,GAAGC,EAAc,GAAGE,CAAW,CAAA,CAC9D,CAAC,CAEL,EACA,cAAe,CAACC,EAAOZ,IAAY,EAC7Ba,EAAAA,YAAYD,CAAK,GAAKE,EAAAA,kBAAkBF,CAAK,IAC/CZ,EAAQ,OAAQQ,GAAS,CACvBrC,SAAO,MAAM,uCAAwCyC,CAAK,EAC1D,MAAMG,EAAKC,EAAAA,iBAAiBR,EAAK,IAAM,CAAA,EAAII,CAAK,EAChD,MAAO,CAAE,GAAGJ,EAAM,GAAAO,CAAA,CACpB,CAAC,CAEL,EAEA,WAAaE,GAAO,CACbA,IACDnC,IAAS,SAENC,GAAUC,EAAYiC,CAAE,GAG7BjC,EAAYiC,CAAE,EAGd3C,EAAA,EAAQ,KAAK,IAAMW,EAAA,EAAa,KAAKC,CAAU,EAAE,MAAOgC,GAAQ/C,EAAAA,OAAO,MAAM,6BAA8B+C,CAAG,CAAC,CAAC,GAEpH,CAAA,CACD,EAOKC,EAAcC,EAAAA,YAClB,MACEC,EACArB,IAYG,CAEH,MAAMsB,EAAYtB,GAAS,IAAMuB,EAAA,EAE3BC,EACJ,OAAOH,GAAY,SACd,CACD,GAAIC,EACJ,KAAMtB,GAAS,MAAQ,QACvB,QAASqB,EACT,GAAIrB,GAAS,MAAQ,CAAE,KAAMA,EAAQ,IAAA,EACrC,GAAIA,GAAS,YAAc,CAAE,WAAYA,EAAQ,UAAA,EACjD,GAAIA,GAAS,oBAAsB,CAAE,mBAAoBA,EAAQ,kBAAA,EACjE,GAAIA,GAAS,cAAgB,CAAE,aAAcA,EAAQ,YAAA,EACrD,IAAKA,GAAS,mBAAqBA,GAAS,SAAW,CACrD,kBAAmB,CACjB,GAAIA,GAAS,QAAU,CAAE,OAAQ,EAAA,EACjC,GAAIA,GAAS,mBAAqB,CAAA,CAAC,CACrC,CACF,EAEA,CACA,GAAGqB,EACH,GAAIC,EAEJ,GAAItB,GAAS,YAAc,CAAE,WAAYA,EAAQ,UAAA,EACjD,GAAIA,GAAS,mBAAqB,CAChC,kBAAmB,CACjB,GAAGqB,EAAQ,kBACX,GAAGrB,EAAQ,iBAAA,CACb,CACF,EAQN,GAJA7B,SAAO,MAAM,mCAAoCqD,CAAU,EAIvDxB,GAAS,OAAS,KAAM,CAI1B,GAHAV,EAAkBkB,GAAS,CAAC,GAAGA,EAAMgB,CAAU,CAAC,EAChDrD,EAAAA,OAAO,MAAM,wCAAyC6B,GAAS,EAAE,EAE7DA,EAAQ,IAAMA,EAAQ,GAAG,OAAS,EAAG,CACvC,MAAMyB,EAAkBzB,EAAQ,GAAG,IAAI,CAACe,EAAIW,KAAW,CACrD,GAAGX,EACH,GAAIO,EAEJ,KAAOP,EAAqC,MAAQ,GAAGO,CAAS,OAAOI,CAAK,EAAA,EAC5E,EACFjC,EAAYe,GAAS,CAAC,GAAGA,EAAM,GAAGiB,CAAe,CAAC,CACpD,CACA,MACF,CAGA,MAAME,EAAkB7B,EAAY,UAAY,CAAA,EAE1C8B,EAAgB5B,GAAS,QAC3B,CAAE,GAAGpB,EAAU,GAAGoB,EAAQ,QAAS,GAAGb,CAAA,EACtCP,GAAY,CAAA,EAEhB,MAAMkB,EAAY,OAChB,CAAE,SAAU,CAAC,GAAG6B,EAAiBH,CAAU,CAAA,EAC3C,CACE,QAASI,EACT,WAAY,CAAC,SAAU,iBAAkB,UAAW,QAAQ,CAAA,CAC9D,CAEJ,EACA,CAAC9B,EAAalB,EAAUO,EAAeG,EAAkBG,CAAU,CAAA,EAI/DoC,EAAmBC,EAAAA,QAAQ,IAAM,CACrC,MAAMC,EAAc,CAAC,GAAG1C,EAAe,GAAIS,EAAY,UAAY,EAAG,EAEhEkC,MAAW,IACjB,OAAOD,EAAY,OAAOE,GACnBA,EAAI,GACLD,EAAK,IAAIC,EAAI,EAAE,EAAU,IAC7BD,EAAK,IAAIC,EAAI,EAAE,EACR,IAHa,EAIrB,CACH,EAAG,CAAC5C,EAAeS,EAAY,QAAQ,CAAC,EAGlCoC,EAAaJ,EAAAA,QAAQ,IAAM,CAC/B3D,EAAAA,OAAO,MAAM,qCAAsCqB,EAASM,EAAY,QAAQ,EAAE,EAClF,MAAMqC,EAAQ,CAAC,GAAG3C,EAAS,GAAIM,EAAY,QAAQ,IAAM,EAAG,EAEtDkC,MAAW,IACjB,OAAOG,EAAM,OAAOpB,GAAM,CACxB,MAAMqB,EAAOrB,EAAqC,MAAQ,GAAGA,EAAG,EAAE,IAAIA,EAAG,IAAI,GAC7E,OAAIiB,EAAK,IAAII,CAAG,EAAU,IAC1BJ,EAAK,IAAII,CAAG,EACL,GACT,CAAC,CACH,EAAG,CAAC5C,EAASM,EAAY,QAAQ,EAAE,CAAC,EAE9BuC,EAAgBjB,EAAAA,YACpB,MACEC,EACArB,IAMG,CAEH,MAAMsC,EAAqBT,GAAoB,CAAA,EAE/C,MAAM/B,EAAY,OAChB,CAAE,SAAU,CAAC,GAAGwC,EAAoBjB,CAAO,CAAA,EAC3C,CACE,QAAS,CAAE,GAAGzC,EAAU,GAAGoB,GAAS,aAAA,EACpC,WAAYA,GAAS,YAAc,CAAC,SAAU,iBAAkB,UAAW,QAAQ,EACnF,gBAAiBA,GAAS,iBAAmB,GAC7C,gBAAiBA,GAAS,iBAAmB,GAC7C,iBAAmBQ,IAAU,CAC3B,GAAGA,EACH,SAAU,CAAC,GAAG8B,EAAoBjB,CAAO,CAAA,EAC3C,CACF,CAEJ,EACA,CAACvB,EAAalB,EAAUiD,CAAgB,CAAA,EAOpCU,EAAoBnB,EAAAA,YACxB,MAAOE,GAAsB,CAC3B,MAAMS,EAAcF,GAAoB,CAAA,EAClCW,EAAeT,EAAY,UAAWE,GAAQA,EAAI,KAAOX,CAAS,EAExE,GAAIkB,IAAiB,GAAI,CACvBrE,SAAO,MAAM,sCAAuCmD,CAAS,EAC7D,MACF,CAGA,IAAImB,EAAiBD,EAAe,EACpC,KAAOC,GAAkB,GAAKV,EAAYU,CAAc,EAAE,OAAS,SACjEA,IAGF,GAAIA,IAAmB,GAAI,CACzBtE,SAAO,MAAM,4CAA6CmD,CAAS,EACnE,MACF,CAGA,MAAMoB,EAAiBX,EAAY,MAAM,EAAGU,EAAiB,CAAC,EAGxDE,EAAsBD,EAAe,MAAM,EAAG,EAAE,EAChDE,EAAeF,EAAeA,EAAe,OAAS,CAAC,EAG7DpD,EAAkBkB,GAChBA,EAAK,OAAQyB,GACMF,EAAY,UAAW1B,GAAMA,EAAE,KAAO4B,EAAI,EAAE,EAC3CQ,CACnB,CAAA,EAIHhD,EAAYe,GACVA,EAAK,OAAQO,GACMgB,EAAY,UAAW1B,GAAMA,EAAE,KAAOU,EAAG,EAAE,EAC1C0B,CACnB,CAAA,EAIH,MAAM3C,EAAY,OAChB,CAAE,SAAU,CAAC,GAAG6C,EAAqBC,CAAY,CAAA,EACjD,CACE,WAAY,CAAC,QAAQ,EACrB,gBAAiB,GACjB,gBAAiB,EAAA,CACnB,CAEJ,EACA,CAACf,EAAkB/B,EAAalB,EAAUO,EAAeG,EAAkBG,CAAU,CAAA,EAGvFoD,EAAAA,UAAU,IAAM,CACd7E,EAAiBC,EAAQW,GAAU,SAAS,EAAE,KAAMkE,GAAO,CACpDA,GACHC,EAAAA,MAAM,MAAM,wCAAyC,CACnD,YAAa,mBAAmB9E,CAAM,GACtC,SAAU,GAAA,CACX,CAEL,CAAC,CACH,EAAG,CAACA,EAAQW,GAAU,SAAS,CAAC,EAEhC,MAAMoE,EAAuB5B,EAAAA,YAAY,SAAY,CAEnD,GAAI1B,GAAkBE,IAAiB,KACrC,OAAOA,EAIT,GAAI,CAACF,EAAgB,CACnBC,EAAkB,EAAI,EACtB,MAAMI,EAAO,MAAM3B,EAAaH,EAAQW,GAAU,SAAS,EAC3D,OAAAiB,EAAgBE,CAAI,EACbA,CACT,CAEA,OAAOH,CACT,EAAG,CAAC3B,EAAQW,GAAU,UAAWc,EAAgBE,CAAY,CAAC,EAExDqD,EAAQnB,EAAAA,QACZ,KAAO,CACL,GAAGhC,EACH,SAAU+B,EACV,OAAQ,CACN,GAAG/B,EAAY,OACf,GAAIoC,CAAA,EAEN,YAAAf,EACA,cAAAkB,EACA,kBAAAE,EACA,aAAcS,CAAA,GAEhB,CAAClD,EAAa+B,EAAkBK,EAAYf,EAAakB,EAAeE,EAAmBS,CAAoB,CAAA,EAGjH,OACEE,EAAAA,IAACpF,EAAc,SAAd,CAAuB,MAAAmF,EAAe,SAAAvE,CAAA,CAAS,CAEpD,EAaO,SAASyE,EAAe,CAAE,SAAAzE,GAAqC,CACpE,OAAOwE,MAACzE,GAAe,SAAAC,EAAS,CAClC,CAaO,SAAS0E,GAAsC,CACpD,MAAMC,EAAMC,EAAAA,WAAWxF,CAAa,EACpC,GAAI,CAACuF,EACH,MAAM,IAAI,MAAM,qDAAqD,EAEvE,OAAOA,CACT"}
+{"version":3,"file":"Stream.cjs.js","sources":["../../src/providers/Stream.tsx"],"sourcesContent":["/* @refresh reset */\nimport { useChatRuntime } from \"@/providers/ChatRuntime\";\nimport { logger } from \"@/utils/logger\";\nimport { type AIMessage, type Message } from \"@langchain/langgraph-sdk\";\n\n// Derive the tool-call types from AIMessage since the SDK doesn't re-export them\ntype ToolCall = NonNullable<AIMessage[\"tool_calls\"]>[number];\ntype InvalidToolCall = NonNullable<AIMessage[\"invalid_tool_calls\"]>[number];\nimport { useStream } from \"@langchain/langgraph-sdk/react\";\nimport {\n  isRemoveUIMessage,\n  isUIMessage,\n  type RemoveUIMessage,\n  type UIMessage,\n  uiMessageReducer,\n} from \"@langchain/langgraph-sdk/react-ui\";\nimport {\n  createContext,\n  type ReactNode,\n  useCallback,\n  useContext,\n  useEffect,\n  useMemo,\n  useState,\n} from \"react\";\nimport { toast } from \"sonner\";\nimport { v4 as uuidv4 } from \"uuid\";\nimport { useThread } from \"./Thread\";\n\nexport type TodoItem = {\n  id: string;\n  content: string;\n  status: \"pending\" | \"in_progress\" | \"completed\";\n  updatedAt?: string | number | Date;\n  run_id?: string;\n  runId?: string;\n  messageId?: string;\n  checkpoint?: string;\n};\n\nexport type StateType = {\n  messages: Message[];\n  ui?: UIMessage[];\n  suggestions?: string[];\n  todos?: TodoItem[];\n};\n\nconst useTypedStream = useStream<\n  StateType,\n  {\n    UpdateType: {\n      messages?: Message[] | Message | string;\n      ui?: (UIMessage | RemoveUIMessage)[] | UIMessage | RemoveUIMessage;\n    };\n    CustomEventType: UIMessage | RemoveUIMessage;\n  }\n>;\n\ntype StreamContextType = ReturnType<typeof useTypedStream> & {\n  sendMessage: (\n    message: Message | string,\n    options?: {\n      /** If true, message is meant for agent only (not user-visible) */\n      type?: Message[\"type\"];\n      /** Name field required for function/tool messages */\n      name?: string;\n      /** If true, message is hidden from user UI */\n      hidden?: boolean;\n      /** Tool calls associated with this message */\n      tool_calls?: ToolCall[];\n      /** Invalid tool calls associated with this message */\n      invalid_tool_calls?: InvalidToolCall[];\n      /** If provided, use this ID for the message instead of generating one */\n      tool_call_id?: string;\n      /** Additional kwargs to attach to the message */\n      additional_kwargs?: Record<string, unknown>;\n      /** UI components to display alongside the message */\n      ui?: UIMessage[];\n      /** Custom ID for the message (will also be used for UI components) */\n      id?: string;\n      /** Custom context to override/merge with identity */\n      context?: Record<string, unknown>;\n    }\n  ) => Promise<void>;\n  submitMessage: (\n    message: Message,\n    options?: {\n      streamMode?: (\"values\" | \"updates\" | \"messages\" | \"custom\" | \"debug\")[];\n      streamSubgraphs?: boolean;\n      streamResumable?: boolean;\n      contextValues?: Record<string, unknown>;\n    }\n  ) => Promise<void>;\n  regenerateMessage: (messageId: string) => Promise<void>;\n  fetchCatalog: () => Promise<unknown>;\n};\n\nconst StreamContext = createContext<StreamContextType | undefined>(undefined);\n\nasync function checkGraphStatus(apiUrl: string, authToken: string | null | undefined) {\n  try {\n    const res = await fetch(`${apiUrl}/info`, {\n      headers: authToken ? { Authorization: `Bearer ${authToken}` } : undefined,\n    });\n    return res.ok;\n  } catch {\n    logger.error(\"Failed to initialize LangGraph Agent\");\n  }\n}\n\nasync function fetchCatalog(apiUrl: string, authToken: string | null | undefined) {\n  try {\n    const res = await fetch(`${apiUrl}/agents/catalog`, {\n      headers: authToken ? { Authorization: `Bearer ${authToken}` } : undefined,\n    });\n    if (res.ok) {\n      const data = await res.json();\n      return data;\n    }\n    return null;\n  }\n  catch {\n    logger.error(\"Failed to fetch LangGraph Catalog\");\n    return null;\n  }\n}\n\nasync function sleep(ms = 6000) {\n  return new Promise((resolve) => setTimeout(resolve, ms));\n}\n\nconst StreamSession = ({ children }: { children: ReactNode }) => {\n  const { apiUrl, assistantId, identity } = useChatRuntime();\n  const { mode, threadId, setThreadId, getThreads, setThreads, configuration } = useThread();\n\n  // Store local-only AI messages that shouldn't trigger backend calls\n  const [localMessages, setLocalMessages] = useState<Message[]>([]);\n  // Store local-only UI components\n  const [localUI, setLocalUI] = useState<UIMessage[]>([]);\n  // Track if catalog has been fetched to prevent duplicate calls\n  const [catalogFetched, setCatalogFetched] = useState(false);\n  const [catalogCache, setCatalogCache] = useState<unknown>(null);\n\n  const streamValue = useTypedStream({\n    apiUrl,\n    assistantId,\n    threadId: mode === \"multi\" ? threadId : null,\n    defaultHeaders: identity?.authToken\n      ? { Authorization: `Bearer ${identity?.authToken}` }\n      : undefined,\n    fetchStateHistory: true,\n    onUpdateEvent: (data, options) => {\n      // Scan update data (potentially nested by node name) for AI messages\n      // with reasoning_content or tool_status in additional_kwargs.\n      // Merge these into existing messages so they're available during streaming.\n      const enrichedMessages: Message[] = [];\n\n      const scan = (obj: unknown) => {\n        if (!obj || typeof obj !== \"object\") return;\n        const o = obj as Record<string, unknown>;\n        if (Array.isArray(o.messages)) {\n          for (const m of o.messages as Message[]) {\n            const ak = (m as Record<string, unknown>).additional_kwargs as Record<string, unknown> | undefined;\n            if (m?.type === \"ai\" && m.id && ak) {\n              const hasReasoning = typeof ak.reasoning_content === \"string\" && ak.reasoning_content.length > 0;\n              const hasToolStatus = Array.isArray(ak.tool_status) && (ak.tool_status as unknown[]).length > 0;\n              if (hasReasoning || hasToolStatus) {\n                enrichedMessages.push(m);\n              }\n            }\n          }\n        }\n        for (const val of Object.values(o)) {\n          if (val && typeof val === \"object\" && !Array.isArray(val)) {\n            scan(val);\n          }\n        }\n      };\n\n      scan(data);\n\n      // Scan update data for todos. We track whether the field was present at all\n      // (foundTodos !== undefined) vs absent (foundTodos === undefined), so we can\n      // distinguish \"event has todos: []\" (preserve previous) from \"event has no\n      // todos field\" (don't touch).  A non-empty array updates the stored todos.\n      // To explicitly clear todos, the backend should send a dedicated signal\n      // (e.g. a custom event) rather than relying on an empty array.\n      let foundTodos: unknown[] | undefined;\n      const scanTodos = (obj: unknown) => {\n        if (!obj || typeof obj !== \"object\") return;\n        const o = obj as Record<string, unknown>;\n        if (\"todos\" in o && Array.isArray(o.todos)) {\n          // Prefer non-empty arrays; keep existing foundTodos if we already found one.\n          if (o.todos.length > 0 || foundTodos === undefined) {\n            foundTodos = o.todos;\n          }\n        }\n        for (const val of Object.values(o)) {\n          if (val && typeof val === \"object\" && !Array.isArray(val)) {\n            scanTodos(val);\n          }\n        }\n      };\n      scanTodos(data);\n\n      if (enrichedMessages.length > 0 || foundTodos !== undefined) {\n        options.mutate((prev) => {\n          let result = prev;\n\n          if (enrichedMessages.length > 0) {\n            const prevMessages = (result.messages ?? []) as Message[];\n            const enrichedById = new Map<string, Message>();\n            for (const m of enrichedMessages) {\n              if (m.id) enrichedById.set(m.id, m);\n            }\n\n            let changed = false;\n            const updatedMessages = prevMessages.map((m) => {\n              if (!m.id || !enrichedById.has(m.id)) return m;\n              const enriched = enrichedById.get(m.id)!;\n              enrichedById.delete(m.id);\n              changed = true;\n              const enrichedKwargs = (enriched as Record<string, unknown>).additional_kwargs as Record<string, unknown> | undefined;\n              return {\n                ...m,\n                additional_kwargs: {\n                  ...(m.additional_kwargs ?? {}),\n                  ...(enrichedKwargs ?? {}),\n                },\n              };\n            });\n\n            if (changed) {\n              result = { ...result, messages: updatedMessages };\n            }\n          }\n\n          if (foundTodos !== undefined) {\n            // If the event delivered a non-empty array, update; if empty, preserve\n            // the previous value so subgraph events that don't own todos don't wipe them.\n            if (foundTodos.length > 0) {\n              result = { ...result, todos: foundTodos as TodoItem[] };\n            }\n            // foundTodos.length === 0: field was present but empty — keep prev.todos\n          }\n\n          return result === prev ? prev : result;\n        });\n      }\n    },\n    onCustomEvent: (event, options) => {\n      if (isUIMessage(event) || isRemoveUIMessage(event)) {\n        options.mutate((prev) => {\n          logger.debug(\"UI Event received in StreamProvider:\", event);\n          const ui = uiMessageReducer(prev.ui ?? [], event);\n          return { ...prev, ui };\n        });\n      }\n    },\n\n    onThreadId: (id) => {\n      if (!id) return;\n      if (mode === \"single\") {\n        // lock once\n        if (!threadId) setThreadId(id);\n      } else {\n        // switch freely\n        setThreadId(id);\n        // Refetch threads list when thread ID changes.\n        // Wait for some seconds before fetching so we're able to get the new thread that was created.\n        sleep().then(() => getThreads().then(setThreads).catch((err) => logger.error(\"Failed to refresh threads:\", err)));\n      }\n    },\n  });\n\n  /**\n   * Send a message to the agent programmatically\n   * Useful for triggering agent actions without user-visible messages\n   * e.g., \"login completed\", \"card clicked: {id}\"\n   */\n  const sendMessage = useCallback(\n    async (\n      message: Message | string,\n      options?: {\n        type?: Message[\"type\"];\n        name?: string;\n        hidden?: boolean;\n        tool_calls?: ToolCall[];\n        invalid_tool_calls?: InvalidToolCall[];\n        tool_call_id?: string;\n        additional_kwargs?: Record<string, unknown>;\n        ui?: UIMessage[];\n        id?: string; // Allow passing custom ID\n        context?: Record<string, unknown>; // Allow passing custom context to override identity\n      }\n    ) => {\n      // Use provided ID or generate new one\n      const messageId = options?.id || uuidv4();\n\n      const messageObj: Message =\n        typeof message === \"string\"\n          ? ({\n            id: messageId, // Use the determined ID\n            type: options?.type ?? \"human\",\n            content: message,\n            ...(options?.name && { name: options.name }),\n            ...(options?.tool_calls && { tool_calls: options.tool_calls }),\n            ...(options?.invalid_tool_calls && { invalid_tool_calls: options.invalid_tool_calls }),\n            ...(options?.tool_call_id && { tool_call_id: options.tool_call_id }),\n            ...((options?.additional_kwargs || options?.hidden) && {\n              additional_kwargs: {\n                ...(options?.hidden && { hidden: true }),\n                ...(options?.additional_kwargs ?? {}),\n              }\n            }),\n          } as Message)\n          : {\n            ...message,\n            id: messageId, // Override with determined ID\n            // Allow overriding message fields with options\n            ...(options?.tool_calls && { tool_calls: options.tool_calls }),\n            ...(options?.additional_kwargs && {\n              additional_kwargs: {\n                ...message.additional_kwargs,\n                ...options.additional_kwargs,\n              }\n            }),\n          };\n\n      // console.log(\"Sending message via sendMessage:\", messageObj); → converted to debug log\n      logger.debug(\"Sending message via sendMessage:\", messageObj);\n\n      // If message type is \"ai\", just append to local state without submitting to agent\n      // This is useful for injecting initial messages or system messages that don't need agent processing\n      if (options?.type === \"ai\") {\n        setLocalMessages((prev) => [...prev, messageObj]);\n        logger.debug(\"Appended AI message to localMessages:\", options?.ui);\n        // Also store any UI components, linking them to the message id\n        if (options.ui && options.ui.length > 0) {\n          const uiWithMessageId = options.ui.map((ui, index) => ({\n            ...ui,\n            id: messageId, // Link to the message ID so CustomComponentRender can find it\n            // Ensure each UI component has a unique identifier for React keys\n            _key: (ui as UIMessage & { _key?: string })._key || `${messageId}-ui-${index}`,\n          })) as UIMessage[];\n          setLocalUI((prev) => [...prev, ...uiWithMessageId]);\n        }\n        return;\n      }\n\n      // For non-AI messages, submit to the agent\n      const currentMessages = streamValue.messages || [];\n      // Merge identity with custom context, custom context takes precedence\n      const mergedContext = options?.context\n        ? { ...identity, ...options.context, ...configuration }\n        : identity || {};\n\n      await streamValue.submit(\n        { messages: [...currentMessages, messageObj] },\n        {\n          context: mergedContext,\n          streamMode: [\"values\", \"messages-tuple\", \"updates\", \"custom\"],\n        }\n      );\n    },\n    [streamValue, identity, configuration, setLocalMessages, setLocalUI]\n  );\n\n  // Combine stream messages with local AI messages, deduplicating by ID\n  const combinedMessages = useMemo(() => {\n    const allMessages = [...localMessages, ...(streamValue.messages || [])];\n    // Deduplicate by message ID, keeping the first occurrence\n    const seen = new Set<string>();\n    return allMessages.filter(msg => {\n      if (!msg.id) return true; // Keep messages without IDs\n      if (seen.has(msg.id)) return false;\n      seen.add(msg.id);\n      return true;\n    });\n  }, [localMessages, streamValue.messages]);\n\n  // Combine local UI with stream UI, deduplicating by unique key\n  const combinedUI = useMemo(() => {\n    logger.debug(\"Combining local UI with stream UI:\", localUI, streamValue.values?.ui);\n    const allUI = [...localUI, ...(streamValue.values?.ui || [])];\n    // Deduplicate by _key or id+name combination\n    const seen = new Set<string>();\n    return allUI.filter(ui => {\n      const key = (ui as UIMessage & { _key?: string })._key || `${ui.id}-${ui.name}`;\n      if (seen.has(key)) return false;\n      seen.add(key);\n      return true;\n    });\n  }, [localUI, streamValue.values?.ui]);\n\n  const submitMessage = useCallback(\n    async (\n      message: Message,\n      options?: {\n        streamMode?: (\"values\" | \"updates\" | \"messages\" | \"custom\" | \"debug\")[];\n        streamSubgraphs?: boolean;\n        streamResumable?: boolean;\n        contextValues?: Record<string, unknown>;\n      }\n    ) => {\n      // Get ALL current messages (including local AI messages)\n      const allCurrentMessages = combinedMessages || [];\n\n      await streamValue.submit(\n        { messages: [...allCurrentMessages, message] },\n        {\n          context: { ...identity, ...options?.contextValues },\n          streamMode: options?.streamMode || [\"values\", \"messages-tuple\", \"updates\", \"custom\"],\n          streamSubgraphs: options?.streamSubgraphs ?? true,\n          streamResumable: options?.streamResumable ?? true,\n          optimisticValues: (prev) => ({\n            ...prev,\n            messages: [...allCurrentMessages, message],\n          }),\n        }\n      );\n    },\n    [streamValue, identity, combinedMessages]\n  );\n\n  /**\n   * Regenerate an AI response by finding the last human message before the specified AI message\n   * and resubmitting from that point\n   */\n  const regenerateMessage = useCallback(\n    async (messageId: string) => {\n      const allMessages = combinedMessages || [];\n      const messageIndex = allMessages.findIndex((msg) => msg.id === messageId);\n\n      if (messageIndex === -1) {\n        logger.error(\"Message not found for regeneration:\", messageId);\n        return;\n      }\n\n      // Find the last human message before this AI message\n      let lastHumanIndex = messageIndex - 1;\n      while (lastHumanIndex >= 0 && allMessages[lastHumanIndex].type !== \"human\") {\n        lastHumanIndex--;\n      }\n\n      if (lastHumanIndex === -1) {\n        logger.error(\"No human message found before AI message:\", messageId);\n        return;\n      }\n\n      // Get all messages up to and including the last human message\n      const messagesToKeep = allMessages.slice(0, lastHumanIndex + 1);\n\n      // Remove the last human message from the list since we'll resubmit it\n      const messagesBeforeHuman = messagesToKeep.slice(0, -1);\n      const humanMessage = messagesToKeep[messagesToKeep.length - 1];\n\n      // Clear local messages that came after the human message\n      setLocalMessages((prev) =>\n        prev.filter((msg) => {\n          const msgIndex = allMessages.findIndex((m) => m.id === msg.id);\n          return msgIndex < lastHumanIndex;\n        })\n      );\n\n      // Clear local UI that came after the human message  \n      setLocalUI((prev) =>\n        prev.filter((ui) => {\n          const msgIndex = allMessages.findIndex((m) => m.id === ui.id);\n          return msgIndex < lastHumanIndex;\n        })\n      );\n\n      // Resubmit from the human message\n      await streamValue.submit(\n        { messages: [...messagesBeforeHuman, humanMessage] },\n        {\n          streamMode: [\"values\"],\n          streamSubgraphs: true,\n          streamResumable: true,\n        }\n      );\n    },\n    [combinedMessages, streamValue, identity, configuration, setLocalMessages, setLocalUI]\n  );\n\n  useEffect(() => {\n    checkGraphStatus(apiUrl, identity?.authToken).then((ok) => {\n      if (!ok) {\n        toast.error(\"Failed to connect to LangGraph server\", {\n          description: `Unable to reach ${apiUrl}`,\n          duration: 10000,\n        });\n      }\n    });\n  }, [apiUrl, identity?.authToken]);\n\n  const fetchCatalogMemoized = useCallback(async () => {\n    // Return cached data if already fetched\n    if (catalogFetched && catalogCache !== null) {\n      return catalogCache;\n    }\n\n    // Fetch catalog only once\n    if (!catalogFetched) {\n      setCatalogFetched(true);\n      const data = await fetchCatalog(apiUrl, identity?.authToken);\n      setCatalogCache(data);\n      return data;\n    }\n\n    return catalogCache;\n  }, [apiUrl, identity?.authToken, catalogFetched, catalogCache]);\n\n  const value = useMemo(\n    () => ({\n      ...streamValue,\n      messages: combinedMessages, // Override with combined messages\n      values: {\n        ...streamValue.values,\n        ui: combinedUI, // Override with combined UI\n      },\n      sendMessage,\n      submitMessage,\n      regenerateMessage,\n      fetchCatalog: fetchCatalogMemoized,\n    }),\n    [streamValue, combinedMessages, combinedUI, sendMessage, submitMessage, regenerateMessage, fetchCatalogMemoized]\n  );\n\n  return (\n    <StreamContext.Provider value={value}>{children}</StreamContext.Provider>\n  );\n};\n\n/**\n * Provides streaming message functionality for real-time AI responses.\n * Manages message state, handles streaming updates, and provides submit/sendMessage functions.\n * \n * @example\n * ```tsx\n * <StreamProvider>\n *   <ChatInterface />\n * </StreamProvider>\n * ```\n */\nexport function StreamProvider({ children }: { children: ReactNode }) {\n  return <StreamSession>{children}</StreamSession>;\n}\n\n/**\n * Hook to access the streaming context.\n * Provides access to messages, loading state, and functions to submit messages.\n * \n * @throws {Error} If used outside of StreamProvider\n * \n * @example\n * ```tsx\n * const { messages, isLoading, submit, sendMessage } = useStreamContext();\n * ```\n */\nexport function useStreamContext(): StreamContextType {\n  const ctx = useContext(StreamContext);\n  if (!ctx) {\n    throw new Error(\"useStreamContext must be used within StreamProvider\");\n  }\n  return ctx;\n}"],"names":["useTypedStream","useStream","StreamContext","createContext","checkGraphStatus","apiUrl","authToken","logger","fetchCatalog","res","sleep","ms","resolve","StreamSession","children","assistantId","identity","useChatRuntime","mode","threadId","setThreadId","getThreads","setThreads","configuration","useThread","localMessages","setLocalMessages","useState","localUI","setLocalUI","catalogFetched","setCatalogFetched","catalogCache","setCatalogCache","streamValue","data","options","enrichedMessages","scan","obj","o","m","ak","hasReasoning","hasToolStatus","val","foundTodos","scanTodos","prev","result","prevMessages","enrichedById","changed","updatedMessages","enriched","enrichedKwargs","event","isUIMessage","isRemoveUIMessage","ui","uiMessageReducer","id","err","sendMessage","useCallback","message","messageId","uuidv4","messageObj","uiWithMessageId","index","currentMessages","mergedContext","combinedMessages","useMemo","allMessages","seen","msg","combinedUI","allUI","key","submitMessage","allCurrentMessages","regenerateMessage","messageIndex","lastHumanIndex","messagesToKeep","messagesBeforeHuman","humanMessage","useEffect","ok","toast","fetchCatalogMemoized","value","jsx","StreamProvider","useStreamContext","ctx","useContext"],"mappings":"odA+CMA,EAAiBC,EAAAA,UAkDjBC,EAAgBC,EAAAA,cAA6C,MAAS,EAE5E,eAAeC,EAAiBC,EAAgBC,EAAsC,CACpF,GAAI,CAIF,OAHY,MAAM,MAAM,GAAGD,CAAM,QAAS,CACxC,QAASC,EAAY,CAAE,cAAe,UAAUA,CAAS,IAAO,MAAA,CACjE,GACU,EACb,MAAQ,CACNC,EAAAA,OAAO,MAAM,sCAAsC,CACrD,CACF,CAEA,eAAeC,EAAaH,EAAgBC,EAAsC,CAChF,GAAI,CACF,MAAMG,EAAM,MAAM,MAAM,GAAGJ,CAAM,kBAAmB,CAClD,QAASC,EAAY,CAAE,cAAe,UAAUA,CAAS,IAAO,MAAA,CACjE,EACD,OAAIG,EAAI,GACO,MAAMA,EAAI,KAAA,EAGlB,IACT,MACM,CACJF,OAAAA,EAAAA,OAAO,MAAM,mCAAmC,EACzC,IACT,CACF,CAEA,eAAeG,GAAMC,EAAK,IAAM,CAC9B,OAAO,IAAI,QAASC,GAAY,WAAWA,EAASD,CAAE,CAAC,CACzD,CAEA,MAAME,GAAgB,CAAC,CAAE,SAAAC,KAAwC,CAC/D,KAAM,CAAE,OAAAT,EAAQ,YAAAU,EAAa,SAAAC,CAAA,EAAaC,EAAAA,eAAA,EACpC,CAAE,KAAAC,EAAM,SAAAC,EAAU,YAAAC,EAAa,WAAAC,EAAY,WAAAC,EAAY,cAAAC,CAAA,EAAkBC,YAAA,EAGzE,CAACC,EAAeC,CAAgB,EAAIC,EAAAA,SAAoB,CAAA,CAAE,EAE1D,CAACC,EAASC,CAAU,EAAIF,EAAAA,SAAsB,CAAA,CAAE,EAEhD,CAACG,EAAgBC,CAAiB,EAAIJ,EAAAA,SAAS,EAAK,EACpD,CAACK,EAAcC,CAAe,EAAIN,EAAAA,SAAkB,IAAI,EAExDO,EAAclC,EAAe,CACjC,OAAAK,EACA,YAAAU,EACA,SAAUG,IAAS,QAAUC,EAAW,KACxC,eAAgBH,GAAU,UACtB,CAAE,cAAe,UAAUA,GAAU,SAAS,EAAA,EAC9C,OACJ,kBAAmB,GACnB,cAAe,CAACmB,EAAMC,IAAY,CAIhC,MAAMC,EAA8B,CAAA,EAE9BC,EAAQC,GAAiB,CAC7B,GAAI,CAACA,GAAO,OAAOA,GAAQ,SAAU,OACrC,MAAMC,EAAID,EACV,GAAI,MAAM,QAAQC,EAAE,QAAQ,EAC1B,UAAWC,KAAKD,EAAE,SAAuB,CACvC,MAAME,EAAMD,EAA8B,kBAC1C,GAAIA,GAAG,OAAS,MAAQA,EAAE,IAAMC,EAAI,CAClC,MAAMC,EAAe,OAAOD,EAAG,mBAAsB,UAAYA,EAAG,kBAAkB,OAAS,EACzFE,EAAgB,MAAM,QAAQF,EAAG,WAAW,GAAMA,EAAG,YAA0B,OAAS,GAC1FC,GAAgBC,IAClBP,EAAiB,KAAKI,CAAC,CAE3B,CACF,CAEF,UAAWI,KAAO,OAAO,OAAOL,CAAC,EAC3BK,GAAO,OAAOA,GAAQ,UAAY,CAAC,MAAM,QAAQA,CAAG,GACtDP,EAAKO,CAAG,CAGd,EAEAP,EAAKH,CAAI,EAQT,IAAIW,EACJ,MAAMC,EAAaR,GAAiB,CAClC,GAAI,CAACA,GAAO,OAAOA,GAAQ,SAAU,OACrC,MAAMC,EAAID,EACN,UAAWC,GAAK,MAAM,QAAQA,EAAE,KAAK,IAEnCA,EAAE,MAAM,OAAS,GAAKM,IAAe,UACvCA,EAAaN,EAAE,OAGnB,UAAWK,KAAO,OAAO,OAAOL,CAAC,EAC3BK,GAAO,OAAOA,GAAQ,UAAY,CAAC,MAAM,QAAQA,CAAG,GACtDE,EAAUF,CAAG,CAGnB,EACAE,EAAUZ,CAAI,GAEVE,EAAiB,OAAS,GAAKS,IAAe,SAChDV,EAAQ,OAAQY,GAAS,CACvB,IAAIC,EAASD,EAEb,GAAIX,EAAiB,OAAS,EAAG,CAC/B,MAAMa,EAAgBD,EAAO,UAAY,CAAA,EACnCE,MAAmB,IACzB,UAAW,KAAKd,EACV,EAAE,IAAIc,EAAa,IAAI,EAAE,GAAI,CAAC,EAGpC,IAAIC,EAAU,GACd,MAAMC,EAAkBH,EAAa,IAAK,GAAM,CAC9C,GAAI,CAAC,EAAE,IAAM,CAACC,EAAa,IAAI,EAAE,EAAE,EAAG,OAAO,EAC7C,MAAMG,EAAWH,EAAa,IAAI,EAAE,EAAE,EACtCA,EAAa,OAAO,EAAE,EAAE,EACxBC,EAAU,GACV,MAAMG,EAAkBD,EAAqC,kBAC7D,MAAO,CACL,GAAG,EACH,kBAAmB,CACjB,GAAI,EAAE,mBAAqB,CAAA,EAC3B,GAAIC,GAAkB,CAAA,CAAC,CACzB,CAEJ,CAAC,EAEGH,IACFH,EAAS,CAAE,GAAGA,EAAQ,SAAUI,CAAA,EAEpC,CAEA,OAAIP,IAAe,QAGbA,EAAW,OAAS,IACtBG,EAAS,CAAE,GAAGA,EAAQ,MAAOH,CAAA,GAK1BG,IAAWD,EAAOA,EAAOC,CAClC,CAAC,CAEL,EACA,cAAe,CAACO,EAAOpB,IAAY,EAC7BqB,EAAAA,YAAYD,CAAK,GAAKE,EAAAA,kBAAkBF,CAAK,IAC/CpB,EAAQ,OAAQY,GAAS,CACvBzC,SAAO,MAAM,uCAAwCiD,CAAK,EAC1D,MAAMG,EAAKC,EAAAA,iBAAiBZ,EAAK,IAAM,CAAA,EAAIQ,CAAK,EAChD,MAAO,CAAE,GAAGR,EAAM,GAAAW,CAAA,CACpB,CAAC,CAEL,EAEA,WAAaE,GAAO,CACbA,IACD3C,IAAS,SAENC,GAAUC,EAAYyC,CAAE,GAG7BzC,EAAYyC,CAAE,EAGdnD,GAAA,EAAQ,KAAK,IAAMW,EAAA,EAAa,KAAKC,CAAU,EAAE,MAAOwC,GAAQvD,EAAAA,OAAO,MAAM,6BAA8BuD,CAAG,CAAC,CAAC,GAEpH,CAAA,CACD,EAOKC,EAAcC,EAAAA,YAClB,MACEC,EACA7B,IAYG,CAEH,MAAM8B,EAAY9B,GAAS,IAAM+B,EAAA,EAE3BC,EACJ,OAAOH,GAAY,SACd,CACD,GAAIC,EACJ,KAAM9B,GAAS,MAAQ,QACvB,QAAS6B,EACT,GAAI7B,GAAS,MAAQ,CAAE,KAAMA,EAAQ,IAAA,EACrC,GAAIA,GAAS,YAAc,CAAE,WAAYA,EAAQ,UAAA,EACjD,GAAIA,GAAS,oBAAsB,CAAE,mBAAoBA,EAAQ,kBAAA,EACjE,GAAIA,GAAS,cAAgB,CAAE,aAAcA,EAAQ,YAAA,EACrD,IAAKA,GAAS,mBAAqBA,GAAS,SAAW,CACrD,kBAAmB,CACjB,GAAIA,GAAS,QAAU,CAAE,OAAQ,EAAA,EACjC,GAAIA,GAAS,mBAAqB,CAAA,CAAC,CACrC,CACF,EAEA,CACA,GAAG6B,EACH,GAAIC,EAEJ,GAAI9B,GAAS,YAAc,CAAE,WAAYA,EAAQ,UAAA,EACjD,GAAIA,GAAS,mBAAqB,CAChC,kBAAmB,CACjB,GAAG6B,EAAQ,kBACX,GAAG7B,EAAQ,iBAAA,CACb,CACF,EAQN,GAJA7B,SAAO,MAAM,mCAAoC6D,CAAU,EAIvDhC,GAAS,OAAS,KAAM,CAI1B,GAHAV,EAAkBsB,GAAS,CAAC,GAAGA,EAAMoB,CAAU,CAAC,EAChD7D,EAAAA,OAAO,MAAM,wCAAyC6B,GAAS,EAAE,EAE7DA,EAAQ,IAAMA,EAAQ,GAAG,OAAS,EAAG,CACvC,MAAMiC,EAAkBjC,EAAQ,GAAG,IAAI,CAACuB,EAAIW,KAAW,CACrD,GAAGX,EACH,GAAIO,EAEJ,KAAOP,EAAqC,MAAQ,GAAGO,CAAS,OAAOI,CAAK,EAAA,EAC5E,EACFzC,EAAYmB,GAAS,CAAC,GAAGA,EAAM,GAAGqB,CAAe,CAAC,CACpD,CACA,MACF,CAGA,MAAME,EAAkBrC,EAAY,UAAY,CAAA,EAE1CsC,EAAgBpC,GAAS,QAC3B,CAAE,GAAGpB,EAAU,GAAGoB,EAAQ,QAAS,GAAGb,CAAA,EACtCP,GAAY,CAAA,EAEhB,MAAMkB,EAAY,OAChB,CAAE,SAAU,CAAC,GAAGqC,EAAiBH,CAAU,CAAA,EAC3C,CACE,QAASI,EACT,WAAY,CAAC,SAAU,iBAAkB,UAAW,QAAQ,CAAA,CAC9D,CAEJ,EACA,CAACtC,EAAalB,EAAUO,EAAeG,EAAkBG,CAAU,CAAA,EAI/D4C,EAAmBC,EAAAA,QAAQ,IAAM,CACrC,MAAMC,EAAc,CAAC,GAAGlD,EAAe,GAAIS,EAAY,UAAY,EAAG,EAEhE0C,MAAW,IACjB,OAAOD,EAAY,OAAOE,GACnBA,EAAI,GACLD,EAAK,IAAIC,EAAI,EAAE,EAAU,IAC7BD,EAAK,IAAIC,EAAI,EAAE,EACR,IAHa,EAIrB,CACH,EAAG,CAACpD,EAAeS,EAAY,QAAQ,CAAC,EAGlC4C,EAAaJ,EAAAA,QAAQ,IAAM,CAC/BnE,EAAAA,OAAO,MAAM,qCAAsCqB,EAASM,EAAY,QAAQ,EAAE,EAClF,MAAM6C,EAAQ,CAAC,GAAGnD,EAAS,GAAIM,EAAY,QAAQ,IAAM,EAAG,EAEtD0C,MAAW,IACjB,OAAOG,EAAM,OAAOpB,GAAM,CACxB,MAAMqB,EAAOrB,EAAqC,MAAQ,GAAGA,EAAG,EAAE,IAAIA,EAAG,IAAI,GAC7E,OAAIiB,EAAK,IAAII,CAAG,EAAU,IAC1BJ,EAAK,IAAII,CAAG,EACL,GACT,CAAC,CACH,EAAG,CAACpD,EAASM,EAAY,QAAQ,EAAE,CAAC,EAE9B+C,EAAgBjB,EAAAA,YACpB,MACEC,EACA7B,IAMG,CAEH,MAAM8C,EAAqBT,GAAoB,CAAA,EAE/C,MAAMvC,EAAY,OAChB,CAAE,SAAU,CAAC,GAAGgD,EAAoBjB,CAAO,CAAA,EAC3C,CACE,QAAS,CAAE,GAAGjD,EAAU,GAAGoB,GAAS,aAAA,EACpC,WAAYA,GAAS,YAAc,CAAC,SAAU,iBAAkB,UAAW,QAAQ,EACnF,gBAAiBA,GAAS,iBAAmB,GAC7C,gBAAiBA,GAAS,iBAAmB,GAC7C,iBAAmBY,IAAU,CAC3B,GAAGA,EACH,SAAU,CAAC,GAAGkC,EAAoBjB,CAAO,CAAA,EAC3C,CACF,CAEJ,EACA,CAAC/B,EAAalB,EAAUyD,CAAgB,CAAA,EAOpCU,EAAoBnB,EAAAA,YACxB,MAAOE,GAAsB,CAC3B,MAAMS,EAAcF,GAAoB,CAAA,EAClCW,EAAeT,EAAY,UAAWE,GAAQA,EAAI,KAAOX,CAAS,EAExE,GAAIkB,IAAiB,GAAI,CACvB7E,SAAO,MAAM,sCAAuC2D,CAAS,EAC7D,MACF,CAGA,IAAImB,EAAiBD,EAAe,EACpC,KAAOC,GAAkB,GAAKV,EAAYU,CAAc,EAAE,OAAS,SACjEA,IAGF,GAAIA,IAAmB,GAAI,CACzB9E,SAAO,MAAM,4CAA6C2D,CAAS,EACnE,MACF,CAGA,MAAMoB,EAAiBX,EAAY,MAAM,EAAGU,EAAiB,CAAC,EAGxDE,EAAsBD,EAAe,MAAM,EAAG,EAAE,EAChDE,EAAeF,EAAeA,EAAe,OAAS,CAAC,EAG7D5D,EAAkBsB,GAChBA,EAAK,OAAQ6B,GACMF,EAAY,UAAWlC,GAAMA,EAAE,KAAOoC,EAAI,EAAE,EAC3CQ,CACnB,CAAA,EAIHxD,EAAYmB,GACVA,EAAK,OAAQW,GACMgB,EAAY,UAAWlC,GAAMA,EAAE,KAAOkB,EAAG,EAAE,EAC1C0B,CACnB,CAAA,EAIH,MAAMnD,EAAY,OAChB,CAAE,SAAU,CAAC,GAAGqD,EAAqBC,CAAY,CAAA,EACjD,CACE,WAAY,CAAC,QAAQ,EACrB,gBAAiB,GACjB,gBAAiB,EAAA,CACnB,CAEJ,EACA,CAACf,EAAkBvC,EAAalB,EAAUO,EAAeG,EAAkBG,CAAU,CAAA,EAGvF4D,EAAAA,UAAU,IAAM,CACdrF,EAAiBC,EAAQW,GAAU,SAAS,EAAE,KAAM0E,GAAO,CACpDA,GACHC,EAAAA,MAAM,MAAM,wCAAyC,CACnD,YAAa,mBAAmBtF,CAAM,GACtC,SAAU,GAAA,CACX,CAEL,CAAC,CACH,EAAG,CAACA,EAAQW,GAAU,SAAS,CAAC,EAEhC,MAAM4E,EAAuB5B,EAAAA,YAAY,SAAY,CAEnD,GAAIlC,GAAkBE,IAAiB,KACrC,OAAOA,EAIT,GAAI,CAACF,EAAgB,CACnBC,EAAkB,EAAI,EACtB,MAAMI,EAAO,MAAM3B,EAAaH,EAAQW,GAAU,SAAS,EAC3D,OAAAiB,EAAgBE,CAAI,EACbA,CACT,CAEA,OAAOH,CACT,EAAG,CAAC3B,EAAQW,GAAU,UAAWc,EAAgBE,CAAY,CAAC,EAExD6D,EAAQnB,EAAAA,QACZ,KAAO,CACL,GAAGxC,EACH,SAAUuC,EACV,OAAQ,CACN,GAAGvC,EAAY,OACf,GAAI4C,CAAA,EAEN,YAAAf,EACA,cAAAkB,EACA,kBAAAE,EACA,aAAcS,CAAA,GAEhB,CAAC1D,EAAauC,EAAkBK,EAAYf,EAAakB,EAAeE,EAAmBS,CAAoB,CAAA,EAGjH,OACEE,EAAAA,IAAC5F,EAAc,SAAd,CAAuB,MAAA2F,EAAe,SAAA/E,CAAA,CAAS,CAEpD,EAaO,SAASiF,GAAe,CAAE,SAAAjF,GAAqC,CACpE,OAAOgF,MAACjF,IAAe,SAAAC,EAAS,CAClC,CAaO,SAASkF,IAAsC,CACpD,MAAMC,EAAMC,EAAAA,WAAWhG,CAAa,EACpC,GAAI,CAAC+F,EACH,MAAM,IAAI,MAAM,qDAAqD,EAEvE,OAAOA,CACT"}

package/dist/providers/Stream.d.ts

@@ -1,21 +1,53 @@
 import { AIMessage, Message } from '@langchain/langgraph-sdk';
-import { UseStream } from '@langchain/langgraph-sdk/react';
 import { RemoveUIMessage, UIMessage } from '@langchain/langgraph-sdk/react-ui';
 import { ReactNode } from 'react';
 type ToolCall = NonNullable<AIMessage["tool_calls"]>[number];
 type InvalidToolCall = NonNullable<AIMessage["invalid_tool_calls"]>[number];
+export type TodoItem = {
+    id: string;
+    content: string;
+    status: "pending" | "in_progress" | "completed";
+    updatedAt?: string | number | Date;
+    run_id?: string;
+    runId?: string;
+    messageId?: string;
+    checkpoint?: string;
+};
 export type StateType = {
     messages: Message[];
     ui?: UIMessage[];
     suggestions?: string[];
+    todos?: TodoItem[];
+};
+declare const useTypedStream: {
+    (options: import('@langchain/langgraph-sdk/react').UseStreamOptions<StateType, {
+        UpdateType: {
+            messages?: Message[] | Message | string;
+            ui?: (UIMessage | RemoveUIMessage)[] | UIMessage | RemoveUIMessage;
+        };
+        CustomEventType: UIMessage | RemoveUIMessage;
+    }>): import('@langchain/langgraph-sdk').BaseStream<StateType, import('@langchain/langgraph-sdk').DefaultToolCall, {
+        UpdateType: {
+            messages?: Message[] | Message | string;
+            ui?: (UIMessage | RemoveUIMessage)[] | UIMessage | RemoveUIMessage;
+        };
+        CustomEventType: UIMessage | RemoveUIMessage;
+    }>;
+    (options: import('@langchain/langgraph-sdk/react').UseStreamCustomOptions<StateType, {
+        UpdateType: {
+            messages?: Message[] | Message | string;
+            ui?: (UIMessage | RemoveUIMessage)[] | UIMessage | RemoveUIMessage;
+        };
+        CustomEventType: UIMessage | RemoveUIMessage;
+    }>): import('@langchain/langgraph-sdk').BaseStream<StateType, import('@langchain/langgraph-sdk').DefaultToolCall, {
+        UpdateType: {
+            messages?: Message[] | Message | string;
+            ui?: (UIMessage | RemoveUIMessage)[] | UIMessage | RemoveUIMessage;
+        };
+        CustomEventType: UIMessage | RemoveUIMessage;
+    }>;
 };
-type StreamContextType = UseStream<StateType, {
-    UpdateType: {
-        messages?: Message[] | Message | string;
-        ui?: (UIMessage | RemoveUIMessage)[] | UIMessage | RemoveUIMessage;
-    };
-    CustomEventType: UIMessage | RemoveUIMessage;
-}> & {
+type StreamContextType = ReturnType<typeof useTypedStream> & {
     sendMessage: (message: Message | string, options?: {
         /** If true, message is meant for agent only (not user-visible) */
         type?: Message["type"];

package/dist/providers/Stream.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Stream.d.ts","sourceRoot":"","sources":["../../src/providers/Stream.tsx"],"names":[],"mappings":"AAGA,OAAO,EAAE,KAAK,SAAS,EAAE,KAAK,OAAO,EAAE,MAAM,0BAA0B,CAAC;AAGxE,KAAK,QAAQ,GAAG,WAAW,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;AAC7D,KAAK,eAAe,GAAG,WAAW,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;AAC5E,OAAO,EAAa,KAAK,SAAS,EAAE,MAAM,gCAAgC,CAAC;AAC3E,OAAO,EAGL,KAAK,eAAe,EACpB,KAAK,SAAS,EAEf,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EAEL,KAAK,SAAS,EAMf,MAAM,OAAO,CAAC;AAKf,MAAM,MAAM,SAAS,GAAG;IACtB,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC;IACjB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;CACxB,CAAC;AAaF,KAAK,iBAAiB,GAAG,SAAS,CAAC,SAAS,EAAE;IAC5C,UAAU,EAAE;QAEV,QAAQ,CAAC,EAAE,OAAO,EAAE,GAAG,OAAO,GAAG,MAAM,CAAC;QACxC,EAAE,CAAC,EAAE,CAAC,SAAS,GAAG,eAAe,CAAC,EAAE,GAAG,SAAS,GAAG,eAAe,CAAC;KACpE,CAAC;IACF,eAAe,EAAE,SAAS,GAAG,eAAe,CAAC;CAC9C,CAAC,GAAG;IACH,WAAW,EAAE,CACX,OAAO,EAAE,OAAO,GAAG,MAAM,EACzB,OAAO,CAAC,EAAE;QACR,kEAAkE;QAClE,IAAI,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QACvB,qDAAqD;QACrD,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,8CAA8C;QAC9C,MAAM,CAAC,EAAE,OAAO,CAAC;QACjB,8CAA8C;QAC9C,UAAU,CAAC,EAAE,QAAQ,EAAE,CAAC;QACxB,sDAAsD;QACtD,kBAAkB,CAAC,EAAE,eAAe,EAAE,CAAC;QACvC,yEAAyE;QACzE,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,iDAAiD;QACjD,iBAAiB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC5C,qDAAqD;QACrD,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC;QACjB,sEAAsE;QACtE,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,qDAAqD;QACrD,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACnC,KACE,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,aAAa,EAAE,CACb,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE;QACR,UAAU,CAAC,EAAE,CAAC,QAAQ,GAAG,SAAS,GAAG,UAAU,GAAG,QAAQ,GAAG,OAAO,CAAC,EAAE,CAAC;QACxE,eAAe,CAAC,EAAE,OAAO,CAAC;QAC1B,eAAe,CAAC,EAAE,OAAO,CAAC;QAC1B,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACzC,KACE,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,iBAAiB,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACxD,YAAY,EAAE,MAAM,OAAO,CAAC,OAAO,CAAC,CAAC;CACtC,CAAC;AA8XF;;;;;;;;;;GAUG;AACH,wBAAgB,cAAc,CAAC,EAAE,QAAQ,EAAE,EAAE;IAAE,QAAQ,EAAE,SAAS,CAAA;CAAE,2CAEnE;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,gBAAgB,IAAI,iBAAiB,CAMpD"}
+{"version":3,"file":"Stream.d.ts","sourceRoot":"","sources":["../../src/providers/Stream.tsx"],"names":[],"mappings":"AAGA,OAAO,EAAE,KAAK,SAAS,EAAE,KAAK,OAAO,EAAE,MAAM,0BAA0B,CAAC;AAGxE,KAAK,QAAQ,GAAG,WAAW,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;AAC7D,KAAK,eAAe,GAAG,WAAW,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;AAE5E,OAAO,EAGL,KAAK,eAAe,EACpB,KAAK,SAAS,EAEf,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EAEL,KAAK,SAAS,EAMf,MAAM,OAAO,CAAC;AAKf,MAAM,MAAM,QAAQ,GAAG;IACrB,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,SAAS,GAAG,aAAa,GAAG,WAAW,CAAC;IAChD,SAAS,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAC;IACnC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,SAAS,GAAG;IACtB,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC;IACjB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,KAAK,CAAC,EAAE,QAAQ,EAAE,CAAC;CACpB,CAAC;AAEF,QAAA,MAAM,cAAc;;oBAGJ;YACV,QAAQ,CAAC,EAAE,OAAO,EAAE,GAAG,OAAO,GAAG,MAAM,CAAC;YACxC,EAAE,CAAC,EAAE,CAAC,SAAS,GAAG,eAAe,CAAC,EAAE,GAAG,SAAS,GAAG,eAAe,CAAC;SACpE;yBACgB,SAAS,GAAG,eAAe;;oBAJhC;YACV,QAAQ,CAAC,EAAE,OAAO,EAAE,GAAG,OAAO,GAAG,MAAM,CAAC;YACxC,EAAE,CAAC,EAAE,CAAC,SAAS,GAAG,eAAe,CAAC,EAAE,GAAG,SAAS,GAAG,eAAe,CAAC;SACpE;yBACgB,SAAS,GAAG,eAAe;;;oBAJhC;YACV,QAAQ,CAAC,EAAE,OAAO,EAAE,GAAG,OAAO,GAAG,MAAM,CAAC;YACxC,EAAE,CAAC,EAAE,CAAC,SAAS,GAAG,eAAe,CAAC,EAAE,GAAG,SAAS,GAAG,eAAe,CAAC;SACpE;yBACgB,SAAS,GAAG,eAAe;;oBAJhC;YACV,QAAQ,CAAC,EAAE,OAAO,EAAE,GAAG,OAAO,GAAG,MAAM,CAAC;YACxC,EAAE,CAAC,EAAE,CAAC,SAAS,GAAG,eAAe,CAAC,EAAE,GAAG,SAAS,GAAG,eAAe,CAAC;SACpE;yBACgB,SAAS,GAAG,eAAe;;CAE/C,CAAC;AAEF,KAAK,iBAAiB,GAAG,UAAU,CAAC,OAAO,cAAc,CAAC,GAAG;IAC3D,WAAW,EAAE,CACX,OAAO,EAAE,OAAO,GAAG,MAAM,EACzB,OAAO,CAAC,EAAE;QACR,kEAAkE;QAClE,IAAI,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QACvB,qDAAqD;QACrD,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,8CAA8C;QAC9C,MAAM,CAAC,EAAE,OAAO,CAAC;QACjB,8CAA8C;QAC9C,UAAU,CAAC,EAAE,QAAQ,EAAE,CAAC;QACxB,sDAAsD;QACtD,kBAAkB,CAAC,EAAE,eAAe,EAAE,CAAC;QACvC,yEAAyE;QACzE,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,iDAAiD;QACjD,iBAAiB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC5C,qDAAqD;QACrD,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC;QACjB,sEAAsE;QACtE,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,qDAAqD;QACrD,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACnC,KACE,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,aAAa,EAAE,CACb,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE;QACR,UAAU,CAAC,EAAE,CAAC,QAAQ,GAAG,SAAS,GAAG,UAAU,GAAG,QAAQ,GAAG,OAAO,CAAC,EAAE,CAAC;QACxE,eAAe,CAAC,EAAE,OAAO,CAAC;QAC1B,eAAe,CAAC,EAAE,OAAO,CAAC;QAC1B,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACzC,KACE,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,iBAAiB,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACxD,YAAY,EAAE,MAAM,OAAO,CAAC,OAAO,CAAC,CAAC;CACtC,CAAC;AAwbF;;;;;;;;;;GAUG;AACH,wBAAgB,cAAc,CAAC,EAAE,QAAQ,EAAE,EAAE;IAAE,QAAQ,EAAE,SAAS,CAAA;CAAE,2CAEnE;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,gBAAgB,IAAI,iBAAiB,CAMpD"}

package/dist/providers/Stream.es.js

@@ -1,73 +1,105 @@
-import { jsx as B } from "react/jsx-runtime";
-import { useChatRuntime as O } from "./ChatRuntime.es.js";
-import { logger as g } from "../utils/logger.es.js";
+import { jsx as L } from "react/jsx-runtime";
+import { useChatRuntime as q } from "./ChatRuntime.es.js";
+import { logger as f } from "../utils/logger.es.js";
 import "@langchain/langgraph-sdk";
-import { useStream as K } from "@langchain/langgraph-sdk/react";
-import { isUIMessage as N, isRemoveUIMessage as W, uiMessageReducer as q } from "@langchain/langgraph-sdk/react-ui";
-import { useContext as D, createContext as J, useState as v, useCallback as I, useMemo as S, useEffect as Q } from "react";
-import { toast as X } from "sonner";
-import { useThread as Y } from "./Thread.es.js";
-import Z from "../node_modules/.pnpm/uuid@13.0.0/node_modules/uuid/dist/v4.es.js";
-const ee = K, E = J(void 0);
-async function te(u, i) {
+import { useStream as D } from "@langchain/langgraph-sdk/react";
+import { isUIMessage as J, isRemoveUIMessage as Q, uiMessageReducer as X } from "@langchain/langgraph-sdk/react-ui";
+import { useContext as Y, createContext as Z, useState as b, useCallback as k, useMemo as A, useEffect as ee } from "react";
+import { toast as te } from "sonner";
+import { useThread as se } from "./Thread.es.js";
+import ae from "../node_modules/.pnpm/uuid@13.0.0/node_modules/uuid/dist/v4.es.js";
+const re = D, P = Z(void 0);
+async function ne(c, l) {
   try {
-    return (await fetch(`${u}/info`, {
-      headers: i ? { Authorization: `Bearer ${i}` } : void 0
+    return (await fetch(`${c}/info`, {
+      headers: l ? { Authorization: `Bearer ${l}` } : void 0
     })).ok;
   } catch {
-    g.error("Failed to initialize LangGraph Agent");
+    f.error("Failed to initialize LangGraph Agent");
   }
 }
-async function ae(u, i) {
+async function oe(c, l) {
   try {
-    const h = await fetch(`${u}/agents/catalog`, {
-      headers: i ? { Authorization: `Bearer ${i}` } : void 0
+    const v = await fetch(`${c}/agents/catalog`, {
+      headers: l ? { Authorization: `Bearer ${l}` } : void 0
     });
-    return h.ok ? await h.json() : null;
+    return v.ok ? await v.json() : null;
   } catch {
-    return g.error("Failed to fetch LangGraph Catalog"), null;
+    return f.error("Failed to fetch LangGraph Catalog"), null;
   }
 }
-async function se(u = 6e3) {
-  return new Promise((i) => setTimeout(i, u));
+async function ie(c = 6e3) {
+  return new Promise((l) => setTimeout(l, c));
 }
-const re = ({ children: u }) => {
-  const { apiUrl: i, assistantId: h, identity: l } = O(), { mode: C, threadId: p, setThreadId: T, getThreads: L, setThreads: P, configuration: w } = Y(), [U, _] = v([]), [b, y] = v([]), [x, j] = v(!1), [M, G] = v(null), o = ee({
-    apiUrl: i,
-    assistantId: h,
-    threadId: C === "multi" ? p : null,
-    defaultHeaders: l?.authToken ? { Authorization: `Bearer ${l?.authToken}` } : void 0,
+const le = ({ children: c }) => {
+  const { apiUrl: l, assistantId: v, identity: d } = q(), { mode: T, threadId: U, setThreadId: $, getThreads: G, setThreads: H, configuration: x } = se(), [R, w] = b([]), [S, M] = b([]), [C, O] = b(!1), [I, V] = b(null), i = re({
+    apiUrl: l,
+    assistantId: v,
+    threadId: T === "multi" ? U : null,
+    defaultHeaders: d?.authToken ? { Authorization: `Bearer ${d?.authToken}` } : void 0,
     fetchStateHistory: !0,
     onUpdateEvent: (t, e) => {
-      const a = [], s = (c) => {
-        if (!c || typeof c != "object") return;
-        const d = c;
-        if (Array.isArray(d.messages))
-          for (const r of d.messages) {
-            const n = r.additional_kwargs;
-            r?.type === "ai" && r.id && typeof n?.reasoning_content == "string" && n.reasoning_content.length > 0 && a.push(r);
+      const a = [], n = (o) => {
+        if (!o || typeof o != "object") return;
+        const s = o;
+        if (Array.isArray(s.messages))
+          for (const r of s.messages) {
+            const g = r.additional_kwargs;
+            if (r?.type === "ai" && r.id && g) {
+              const h = typeof g.reasoning_content == "string" && g.reasoning_content.length > 0, p = Array.isArray(g.tool_status) && g.tool_status.length > 0;
+              (h || p) && a.push(r);
+            }
           }
-        for (const r of Object.values(d))
-          r && typeof r == "object" && !Array.isArray(r) && s(r);
+        for (const r of Object.values(s))
+          r && typeof r == "object" && !Array.isArray(r) && n(r);
       };
-      s(t), a.length > 0 && e.mutate((c) => {
-        const d = c.messages ?? [], r = new Set(d.map((m) => m.id).filter(Boolean)), n = a.filter((m) => !r.has(m.id));
-        return n.length === 0 ? c : { ...c, messages: [...d, ...n] };
+      n(t);
+      let u;
+      const _ = (o) => {
+        if (!o || typeof o != "object") return;
+        const s = o;
+        "todos" in s && Array.isArray(s.todos) && (s.todos.length > 0 || u === void 0) && (u = s.todos);
+        for (const r of Object.values(s))
+          r && typeof r == "object" && !Array.isArray(r) && _(r);
+      };
+      _(t), (a.length > 0 || u !== void 0) && e.mutate((o) => {
+        let s = o;
+        if (a.length > 0) {
+          const r = s.messages ?? [], g = /* @__PURE__ */ new Map();
+          for (const m of a)
+            m.id && g.set(m.id, m);
+          let h = !1;
+          const p = r.map((m) => {
+            if (!m.id || !g.has(m.id)) return m;
+            const N = g.get(m.id);
+            g.delete(m.id), h = !0;
+            const W = N.additional_kwargs;
+            return {
+              ...m,
+              additional_kwargs: {
+                ...m.additional_kwargs ?? {},
+                ...W ?? {}
+              }
+            };
+          });
+          h && (s = { ...s, messages: p });
+        }
+        return u !== void 0 && u.length > 0 && (s = { ...s, todos: u }), s === o ? o : s;
       });
     },
     onCustomEvent: (t, e) => {
-      (N(t) || W(t)) && e.mutate((a) => {
-        g.debug("UI Event received in StreamProvider:", t);
-        const s = q(a.ui ?? [], t);
-        return { ...a, ui: s };
+      (J(t) || Q(t)) && e.mutate((a) => {
+        f.debug("UI Event received in StreamProvider:", t);
+        const n = X(a.ui ?? [], t);
+        return { ...a, ui: n };
       });
     },
     onThreadId: (t) => {
-      t && (C === "single" ? p || T(t) : (T(t), se().then(() => L().then(P).catch((e) => g.error("Failed to refresh threads:", e)))));
+      t && (T === "single" ? U || $(t) : ($(t), ie().then(() => G().then(H).catch((e) => f.error("Failed to refresh threads:", e)))));
     }
-  }), A = I(
+  }), F = k(
     async (t, e) => {
-      const a = e?.id || Z(), s = typeof t == "string" ? {
+      const a = e?.id || ae(), n = typeof t == "string" ? {
         id: a,
         // Use the determined ID
         type: e?.type ?? "human",
@@ -95,78 +127,78 @@ const re = ({ children: u }) => {
           }
         }
       };
-      if (g.debug("Sending message via sendMessage:", s), e?.type === "ai") {
-        if (_((r) => [...r, s]), g.debug("Appended AI message to localMessages:", e?.ui), e.ui && e.ui.length > 0) {
-          const r = e.ui.map((n, m) => ({
-            ...n,
+      if (f.debug("Sending message via sendMessage:", n), e?.type === "ai") {
+        if (w((o) => [...o, n]), f.debug("Appended AI message to localMessages:", e?.ui), e.ui && e.ui.length > 0) {
+          const o = e.ui.map((s, r) => ({
+            ...s,
             id: a,
             // Link to the message ID so CustomComponentRender can find it
             // Ensure each UI component has a unique identifier for React keys
-            _key: n._key || `${a}-ui-${m}`
+            _key: s._key || `${a}-ui-${r}`
           }));
-          y((n) => [...n, ...r]);
+          M((s) => [...s, ...o]);
         }
         return;
       }
-      const c = o.messages || [], d = e?.context ? { ...l, ...e.context, ...w } : l || {};
-      await o.submit(
-        { messages: [...c, s] },
+      const u = i.messages || [], _ = e?.context ? { ...d, ...e.context, ...x } : d || {};
+      await i.submit(
+        { messages: [...u, n] },
         {
-          context: d,
+          context: _,
           streamMode: ["values", "messages-tuple", "updates", "custom"]
         }
       );
     },
-    [o, l, w, _, y]
-  ), f = S(() => {
-    const t = [...U, ...o.messages || []], e = /* @__PURE__ */ new Set();
+    [i, d, x, w, M]
+  ), y = A(() => {
+    const t = [...R, ...i.messages || []], e = /* @__PURE__ */ new Set();
     return t.filter((a) => a.id ? e.has(a.id) ? !1 : (e.add(a.id), !0) : !0);
-  }, [U, o.messages]), $ = S(() => {
-    g.debug("Combining local UI with stream UI:", b, o.values?.ui);
-    const t = [...b, ...o.values?.ui || []], e = /* @__PURE__ */ new Set();
+  }, [R, i.messages]), j = A(() => {
+    f.debug("Combining local UI with stream UI:", S, i.values?.ui);
+    const t = [...S, ...i.values?.ui || []], e = /* @__PURE__ */ new Set();
     return t.filter((a) => {
-      const s = a._key || `${a.id}-${a.name}`;
-      return e.has(s) ? !1 : (e.add(s), !0);
+      const n = a._key || `${a.id}-${a.name}`;
+      return e.has(n) ? !1 : (e.add(n), !0);
     });
-  }, [b, o.values?.ui]), F = I(
+  }, [S, i.values?.ui]), z = k(
     async (t, e) => {
-      const a = f || [];
-      await o.submit(
+      const a = y || [];
+      await i.submit(
         { messages: [...a, t] },
         {
-          context: { ...l, ...e?.contextValues },
+          context: { ...d, ...e?.contextValues },
           streamMode: e?.streamMode || ["values", "messages-tuple", "updates", "custom"],
           streamSubgraphs: e?.streamSubgraphs ?? !0,
           streamResumable: e?.streamResumable ?? !0,
-          optimisticValues: (s) => ({
-            ...s,
+          optimisticValues: (n) => ({
+            ...n,
             messages: [...a, t]
           })
         }
       );
     },
-    [o, l, f]
-  ), R = I(
+    [i, d, y]
+  ), B = k(
     async (t) => {
-      const e = f || [], a = e.findIndex((n) => n.id === t);
+      const e = y || [], a = e.findIndex((s) => s.id === t);
       if (a === -1) {
-        g.error("Message not found for regeneration:", t);
+        f.error("Message not found for regeneration:", t);
         return;
       }
-      let s = a - 1;
-      for (; s >= 0 && e[s].type !== "human"; )
-        s--;
-      if (s === -1) {
-        g.error("No human message found before AI message:", t);
+      let n = a - 1;
+      for (; n >= 0 && e[n].type !== "human"; )
+        n--;
+      if (n === -1) {
+        f.error("No human message found before AI message:", t);
         return;
       }
-      const c = e.slice(0, s + 1), d = c.slice(0, -1), r = c[c.length - 1];
-      _(
-        (n) => n.filter((m) => e.findIndex((k) => k.id === m.id) < s)
-      ), y(
-        (n) => n.filter((m) => e.findIndex((k) => k.id === m.id) < s)
-      ), await o.submit(
-        { messages: [...d, r] },
+      const u = e.slice(0, n + 1), _ = u.slice(0, -1), o = u[u.length - 1];
+      w(
+        (s) => s.filter((r) => e.findIndex((h) => h.id === r.id) < n)
+      ), M(
+        (s) => s.filter((r) => e.findIndex((h) => h.id === r.id) < n)
+      ), await i.submit(
+        { messages: [..._, o] },
         {
           streamMode: ["values"],
           streamSubgraphs: !0,
@@ -174,55 +206,55 @@ const re = ({ children: u }) => {
         }
       );
     },
-    [f, o, l, w, _, y]
+    [y, i, d, x, w, M]
   );
-  Q(() => {
-    te(i, l?.authToken).then((t) => {
-      t || X.error("Failed to connect to LangGraph server", {
-        description: `Unable to reach ${i}`,
+  ee(() => {
+    ne(l, d?.authToken).then((t) => {
+      t || te.error("Failed to connect to LangGraph server", {
+        description: `Unable to reach ${l}`,
         duration: 1e4
       });
     });
-  }, [i, l?.authToken]);
-  const z = I(async () => {
-    if (x && M !== null)
-      return M;
-    if (!x) {
-      j(!0);
-      const t = await ae(i, l?.authToken);
-      return G(t), t;
+  }, [l, d?.authToken]);
+  const E = k(async () => {
+    if (C && I !== null)
+      return I;
+    if (!C) {
+      O(!0);
+      const t = await oe(l, d?.authToken);
+      return V(t), t;
     }
-    return M;
-  }, [i, l?.authToken, x, M]), H = S(
+    return I;
+  }, [l, d?.authToken, C, I]), K = A(
     () => ({
-      ...o,
-      messages: f,
+      ...i,
+      messages: y,
       // Override with combined messages
       values: {
-        ...o.values,
-        ui: $
+        ...i.values,
+        ui: j
         // Override with combined UI
       },
-      sendMessage: A,
-      submitMessage: F,
-      regenerateMessage: R,
-      fetchCatalog: z
+      sendMessage: F,
+      submitMessage: z,
+      regenerateMessage: B,
+      fetchCatalog: E
     }),
-    [o, f, $, A, F, R, z]
+    [i, y, j, F, z, B, E]
   );
-  return /* @__PURE__ */ B(E.Provider, { value: H, children: u });
+  return /* @__PURE__ */ L(P.Provider, { value: K, children: c });
 };
-function he({ children: u }) {
-  return /* @__PURE__ */ B(re, { children: u });
+function we({ children: c }) {
+  return /* @__PURE__ */ L(le, { children: c });
 }
-function _e() {
-  const u = D(E);
-  if (!u)
+function Me() {
+  const c = Y(P);
+  if (!c)
     throw new Error("useStreamContext must be used within StreamProvider");
-  return u;
+  return c;
 }
 export {
-  he as StreamProvider,
-  _e as useStreamContext
+  we as StreamProvider,
+  Me as useStreamContext
 };
 //# sourceMappingURL=Stream.es.js.map