langgraph-ui-components 0.0.27 → 0.0.28

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

@@ -1 +1 @@
-{"version":3,"file":"Stream.es.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 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      if (enrichedMessages.length > 0) {\n        options.mutate((prev) => {\n          const prevMessages = (prev.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) return prev;\n\n          return { ...prev, messages: updatedMessages };\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","prev","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":";;;;;;;;;;AAmCA,MAAMA,KAAiBC,GAyDjBC,IAAgBC,EAA6C,MAAS;AAE5E,eAAeC,GAAiBC,GAAgBC,GAAsC;AACpF,MAAI;AAIF,YAHY,MAAM,MAAM,GAAGD,CAAM,SAAS;AAAA,MACxC,SAASC,IAAY,EAAE,eAAe,UAAUA,CAAS,OAAO;AAAA,IAAA,CACjE,GACU;AAAA,EACb,QAAQ;AACN,IAAAC,EAAO,MAAM,sCAAsC;AAAA,EACrD;AACF;AAEA,eAAeC,GAAaH,GAAgBC,GAAsC;AAChF,MAAI;AACF,UAAMG,IAAM,MAAM,MAAM,GAAGJ,CAAM,mBAAmB;AAAA,MAClD,SAASC,IAAY,EAAE,eAAe,UAAUA,CAAS,OAAO;AAAA,IAAA,CACjE;AACD,WAAIG,EAAI,KACO,MAAMA,EAAI,KAAA,IAGlB;AAAA,EACT,QACM;AACJ,WAAAF,EAAO,MAAM,mCAAmC,GACzC;AAAA,EACT;AACF;AAEA,eAAeG,GAAMC,IAAK,KAAM;AAC9B,SAAO,IAAI,QAAQ,CAACC,MAAY,WAAWA,GAASD,CAAE,CAAC;AACzD;AAEA,MAAME,KAAgB,CAAC,EAAE,UAAAC,QAAwC;AAC/D,QAAM,EAAE,QAAAT,GAAQ,aAAAU,GAAa,UAAAC,EAAA,IAAaC,EAAA,GACpC,EAAE,MAAAC,GAAM,UAAAC,GAAU,aAAAC,GAAa,YAAAC,GAAY,YAAAC,GAAY,eAAAC,EAAA,IAAkBC,EAAA,GAGzE,CAACC,GAAeC,CAAgB,IAAIC,EAAoB,CAAA,CAAE,GAE1D,CAACC,GAASC,CAAU,IAAIF,EAAsB,CAAA,CAAE,GAEhD,CAACG,GAAgBC,CAAiB,IAAIJ,EAAS,EAAK,GACpD,CAACK,GAAcC,CAAe,IAAIN,EAAkB,IAAI,GAExDO,IAAclC,GAAe;AAAA,IACjC,QAAAK;AAAA,IACA,aAAAU;AAAA,IACA,UAAUG,MAAS,UAAUC,IAAW;AAAA,IACxC,gBAAgBH,GAAU,YACtB,EAAE,eAAe,UAAUA,GAAU,SAAS,GAAA,IAC9C;AAAA,IACJ,mBAAmB;AAAA,IACnB,eAAe,CAACmB,GAAMC,MAAY;AAIhC,YAAMC,IAA8B,CAAA,GAE9BC,IAAO,CAACC,MAAiB;AAC7B,YAAI,CAACA,KAAO,OAAOA,KAAQ,SAAU;AACrC,cAAMC,IAAID;AACV,YAAI,MAAM,QAAQC,EAAE,QAAQ;AAC1B,qBAAWC,KAAKD,EAAE,UAAuB;AACvC,kBAAME,IAAMD,EAA8B;AAC1C,gBAAIA,GAAG,SAAS,QAAQA,EAAE,MAAMC,GAAI;AAClC,oBAAMC,IAAe,OAAOD,EAAG,qBAAsB,YAAYA,EAAG,kBAAkB,SAAS,GACzFE,IAAgB,MAAM,QAAQF,EAAG,WAAW,KAAMA,EAAG,YAA0B,SAAS;AAC9F,eAAIC,KAAgBC,MAClBP,EAAiB,KAAKI,CAAC;AAAA,YAE3B;AAAA,UACF;AAEF,mBAAWI,KAAO,OAAO,OAAOL,CAAC;AAC/B,UAAIK,KAAO,OAAOA,KAAQ,YAAY,CAAC,MAAM,QAAQA,CAAG,KACtDP,EAAKO,CAAG;AAAA,MAGd;AAEA,MAAAP,EAAKH,CAAI,GAELE,EAAiB,SAAS,KAC5BD,EAAQ,OAAO,CAACU,MAAS;AACvB,cAAMC,IAAgBD,EAAK,YAAY,CAAA,GACjCE,wBAAmB,IAAA;AACzB,mBAAWP,KAAKJ;AACd,UAAII,EAAE,MAAIO,EAAa,IAAIP,EAAE,IAAIA,CAAC;AAGpC,YAAIQ,IAAU;AACd,cAAMC,IAAkBH,EAAa,IAAI,CAACN,MAAM;AAC9C,cAAI,CAACA,EAAE,MAAM,CAACO,EAAa,IAAIP,EAAE,EAAE,EAAG,QAAOA;AAC7C,gBAAMU,IAAWH,EAAa,IAAIP,EAAE,EAAE;AACtC,UAAAO,EAAa,OAAOP,EAAE,EAAE,GACxBQ,IAAU;AACV,gBAAMG,IAAkBD,EAAqC;AAC7D,iBAAO;AAAA,YACL,GAAGV;AAAA,YACH,mBAAmB;AAAA,cACjB,GAAIA,EAAE,qBAAqB,CAAA;AAAA,cAC3B,GAAIW,KAAkB,CAAA;AAAA,YAAC;AAAA,UACzB;AAAA,QAEJ,CAAC;AAED,eAAKH,IAEE,EAAE,GAAGH,GAAM,UAAUI,EAAA,IAFPJ;AAAA,MAGvB,CAAC;AAAA,IAEL;AAAA,IACA,eAAe,CAACO,GAAOjB,MAAY;AACjC,OAAIkB,EAAYD,CAAK,KAAKE,EAAkBF,CAAK,MAC/CjB,EAAQ,OAAO,CAACU,MAAS;AACvB,QAAAvC,EAAO,MAAM,wCAAwC8C,CAAK;AAC1D,cAAMG,IAAKC,EAAiBX,EAAK,MAAM,CAAA,GAAIO,CAAK;AAChD,eAAO,EAAE,GAAGP,GAAM,IAAAU,EAAA;AAAA,MACpB,CAAC;AAAA,IAEL;AAAA,IAEA,YAAY,CAACE,MAAO;AAClB,MAAKA,MACDxC,MAAS,WAENC,KAAUC,EAAYsC,CAAE,KAG7BtC,EAAYsC,CAAE,GAGdhD,GAAA,EAAQ,KAAK,MAAMW,EAAA,EAAa,KAAKC,CAAU,EAAE,MAAM,CAACqC,MAAQpD,EAAO,MAAM,8BAA8BoD,CAAG,CAAC,CAAC;AAAA,IAEpH;AAAA,EAAA,CACD,GAOKC,IAAcC;AAAA,IAClB,OACEC,GACA1B,MAYG;AAEH,YAAM2B,IAAY3B,GAAS,MAAM4B,GAAA,GAE3BC,IACJ,OAAOH,KAAY,WACd;AAAA,QACD,IAAIC;AAAA;AAAA,QACJ,MAAM3B,GAAS,QAAQ;AAAA,QACvB,SAAS0B;AAAA,QACT,GAAI1B,GAAS,QAAQ,EAAE,MAAMA,EAAQ,KAAA;AAAA,QACrC,GAAIA,GAAS,cAAc,EAAE,YAAYA,EAAQ,WAAA;AAAA,QACjD,GAAIA,GAAS,sBAAsB,EAAE,oBAAoBA,EAAQ,mBAAA;AAAA,QACjE,GAAIA,GAAS,gBAAgB,EAAE,cAAcA,EAAQ,aAAA;AAAA,QACrD,IAAKA,GAAS,qBAAqBA,GAAS,WAAW;AAAA,UACrD,mBAAmB;AAAA,YACjB,GAAIA,GAAS,UAAU,EAAE,QAAQ,GAAA;AAAA,YACjC,GAAIA,GAAS,qBAAqB,CAAA;AAAA,UAAC;AAAA,QACrC;AAAA,MACF,IAEA;AAAA,QACA,GAAG0B;AAAA,QACH,IAAIC;AAAA;AAAA;AAAA,QAEJ,GAAI3B,GAAS,cAAc,EAAE,YAAYA,EAAQ,WAAA;AAAA,QACjD,GAAIA,GAAS,qBAAqB;AAAA,UAChC,mBAAmB;AAAA,YACjB,GAAG0B,EAAQ;AAAA,YACX,GAAG1B,EAAQ;AAAA,UAAA;AAAA,QACb;AAAA,MACF;AAQN,UAJA7B,EAAO,MAAM,oCAAoC0D,CAAU,GAIvD7B,GAAS,SAAS,MAAM;AAI1B,YAHAV,EAAiB,CAACoB,MAAS,CAAC,GAAGA,GAAMmB,CAAU,CAAC,GAChD1D,EAAO,MAAM,yCAAyC6B,GAAS,EAAE,GAE7DA,EAAQ,MAAMA,EAAQ,GAAG,SAAS,GAAG;AACvC,gBAAM8B,IAAkB9B,EAAQ,GAAG,IAAI,CAACoB,GAAIW,OAAW;AAAA,YACrD,GAAGX;AAAA,YACH,IAAIO;AAAA;AAAA;AAAA,YAEJ,MAAOP,EAAqC,QAAQ,GAAGO,CAAS,OAAOI,CAAK;AAAA,UAAA,EAC5E;AACF,UAAAtC,EAAW,CAACiB,MAAS,CAAC,GAAGA,GAAM,GAAGoB,CAAe,CAAC;AAAA,QACpD;AACA;AAAA,MACF;AAGA,YAAME,IAAkBlC,EAAY,YAAY,CAAA,GAE1CmC,IAAgBjC,GAAS,UAC3B,EAAE,GAAGpB,GAAU,GAAGoB,EAAQ,SAAS,GAAGb,EAAA,IACtCP,KAAY,CAAA;AAEhB,YAAMkB,EAAY;AAAA,QAChB,EAAE,UAAU,CAAC,GAAGkC,GAAiBH,CAAU,EAAA;AAAA,QAC3C;AAAA,UACE,SAASI;AAAA,UACT,YAAY,CAAC,UAAU,kBAAkB,WAAW,QAAQ;AAAA,QAAA;AAAA,MAC9D;AAAA,IAEJ;AAAA,IACA,CAACnC,GAAalB,GAAUO,GAAeG,GAAkBG,CAAU;AAAA,EAAA,GAI/DyC,IAAmBC,EAAQ,MAAM;AACrC,UAAMC,IAAc,CAAC,GAAG/C,GAAe,GAAIS,EAAY,YAAY,EAAG,GAEhEuC,wBAAW,IAAA;AACjB,WAAOD,EAAY,OAAO,CAAAE,MACnBA,EAAI,KACLD,EAAK,IAAIC,EAAI,EAAE,IAAU,MAC7BD,EAAK,IAAIC,EAAI,EAAE,GACR,MAHa,EAIrB;AAAA,EACH,GAAG,CAACjD,GAAeS,EAAY,QAAQ,CAAC,GAGlCyC,IAAaJ,EAAQ,MAAM;AAC/B,IAAAhE,EAAO,MAAM,sCAAsCqB,GAASM,EAAY,QAAQ,EAAE;AAClF,UAAM0C,IAAQ,CAAC,GAAGhD,GAAS,GAAIM,EAAY,QAAQ,MAAM,EAAG,GAEtDuC,wBAAW,IAAA;AACjB,WAAOG,EAAM,OAAO,CAAApB,MAAM;AACxB,YAAMqB,IAAOrB,EAAqC,QAAQ,GAAGA,EAAG,EAAE,IAAIA,EAAG,IAAI;AAC7E,aAAIiB,EAAK,IAAII,CAAG,IAAU,MAC1BJ,EAAK,IAAII,CAAG,GACL;AAAA,IACT,CAAC;AAAA,EACH,GAAG,CAACjD,GAASM,EAAY,QAAQ,EAAE,CAAC,GAE9B4C,IAAgBjB;AAAA,IACpB,OACEC,GACA1B,MAMG;AAEH,YAAM2C,IAAqBT,KAAoB,CAAA;AAE/C,YAAMpC,EAAY;AAAA,QAChB,EAAE,UAAU,CAAC,GAAG6C,GAAoBjB,CAAO,EAAA;AAAA,QAC3C;AAAA,UACE,SAAS,EAAE,GAAG9C,GAAU,GAAGoB,GAAS,cAAA;AAAA,UACpC,YAAYA,GAAS,cAAc,CAAC,UAAU,kBAAkB,WAAW,QAAQ;AAAA,UACnF,iBAAiBA,GAAS,mBAAmB;AAAA,UAC7C,iBAAiBA,GAAS,mBAAmB;AAAA,UAC7C,kBAAkB,CAACU,OAAU;AAAA,YAC3B,GAAGA;AAAA,YACH,UAAU,CAAC,GAAGiC,GAAoBjB,CAAO;AAAA,UAAA;AAAA,QAC3C;AAAA,MACF;AAAA,IAEJ;AAAA,IACA,CAAC5B,GAAalB,GAAUsD,CAAgB;AAAA,EAAA,GAOpCU,IAAoBnB;AAAA,IACxB,OAAOE,MAAsB;AAC3B,YAAMS,IAAcF,KAAoB,CAAA,GAClCW,IAAeT,EAAY,UAAU,CAACE,MAAQA,EAAI,OAAOX,CAAS;AAExE,UAAIkB,MAAiB,IAAI;AACvB,QAAA1E,EAAO,MAAM,uCAAuCwD,CAAS;AAC7D;AAAA,MACF;AAGA,UAAImB,IAAiBD,IAAe;AACpC,aAAOC,KAAkB,KAAKV,EAAYU,CAAc,EAAE,SAAS;AACjE,QAAAA;AAGF,UAAIA,MAAmB,IAAI;AACzB,QAAA3E,EAAO,MAAM,6CAA6CwD,CAAS;AACnE;AAAA,MACF;AAGA,YAAMoB,IAAiBX,EAAY,MAAM,GAAGU,IAAiB,CAAC,GAGxDE,IAAsBD,EAAe,MAAM,GAAG,EAAE,GAChDE,IAAeF,EAAeA,EAAe,SAAS,CAAC;AAG7D,MAAAzD;AAAA,QAAiB,CAACoB,MAChBA,EAAK,OAAO,CAAC4B,MACMF,EAAY,UAAU,CAAC/B,MAAMA,EAAE,OAAOiC,EAAI,EAAE,IAC3CQ,CACnB;AAAA,MAAA,GAIHrD;AAAA,QAAW,CAACiB,MACVA,EAAK,OAAO,CAACU,MACMgB,EAAY,UAAU,CAAC/B,MAAMA,EAAE,OAAOe,EAAG,EAAE,IAC1C0B,CACnB;AAAA,MAAA,GAIH,MAAMhD,EAAY;AAAA,QAChB,EAAE,UAAU,CAAC,GAAGkD,GAAqBC,CAAY,EAAA;AAAA,QACjD;AAAA,UACE,YAAY,CAAC,QAAQ;AAAA,UACrB,iBAAiB;AAAA,UACjB,iBAAiB;AAAA,QAAA;AAAA,MACnB;AAAA,IAEJ;AAAA,IACA,CAACf,GAAkBpC,GAAalB,GAAUO,GAAeG,GAAkBG,CAAU;AAAA,EAAA;AAGvF,EAAAyD,EAAU,MAAM;AACd,IAAAlF,GAAiBC,GAAQW,GAAU,SAAS,EAAE,KAAK,CAACuE,MAAO;AACzD,MAAKA,KACHC,EAAM,MAAM,yCAAyC;AAAA,QACnD,aAAa,mBAAmBnF,CAAM;AAAA,QACtC,UAAU;AAAA,MAAA,CACX;AAAA,IAEL,CAAC;AAAA,EACH,GAAG,CAACA,GAAQW,GAAU,SAAS,CAAC;AAEhC,QAAMyE,IAAuB5B,EAAY,YAAY;AAEnD,QAAI/B,KAAkBE,MAAiB;AACrC,aAAOA;AAIT,QAAI,CAACF,GAAgB;AACnB,MAAAC,EAAkB,EAAI;AACtB,YAAMI,IAAO,MAAM3B,GAAaH,GAAQW,GAAU,SAAS;AAC3D,aAAAiB,EAAgBE,CAAI,GACbA;AAAA,IACT;AAEA,WAAOH;AAAA,EACT,GAAG,CAAC3B,GAAQW,GAAU,WAAWc,GAAgBE,CAAY,CAAC,GAExD0D,IAAQnB;AAAA,IACZ,OAAO;AAAA,MACL,GAAGrC;AAAA,MACH,UAAUoC;AAAA;AAAA,MACV,QAAQ;AAAA,QACN,GAAGpC,EAAY;AAAA,QACf,IAAIyC;AAAA;AAAA,MAAA;AAAA,MAEN,aAAAf;AAAA,MACA,eAAAkB;AAAA,MACA,mBAAAE;AAAA,MACA,cAAcS;AAAA,IAAA;AAAA,IAEhB,CAACvD,GAAaoC,GAAkBK,GAAYf,GAAakB,GAAeE,GAAmBS,CAAoB;AAAA,EAAA;AAGjH,SACE,gBAAAE,EAACzF,EAAc,UAAd,EAAuB,OAAAwF,GAAe,UAAA5E,EAAA,CAAS;AAEpD;AAaO,SAAS8E,GAAe,EAAE,UAAA9E,KAAqC;AACpE,SAAO,gBAAA6E,EAAC9E,MAAe,UAAAC,GAAS;AAClC;AAaO,SAAS+E,KAAsC;AACpD,QAAMC,IAAMC,EAAW7F,CAAa;AACpC,MAAI,CAAC4F;AACH,UAAM,IAAI,MAAM,qDAAqD;AAEvE,SAAOA;AACT;"}
+{"version":3,"file":"Stream.es.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":";;;;;;;;;;AA+CA,MAAMA,KAAiBC,GAkDjBC,IAAgBC,EAA6C,MAAS;AAE5E,eAAeC,GAAiBC,GAAgBC,GAAsC;AACpF,MAAI;AAIF,YAHY,MAAM,MAAM,GAAGD,CAAM,SAAS;AAAA,MACxC,SAASC,IAAY,EAAE,eAAe,UAAUA,CAAS,OAAO;AAAA,IAAA,CACjE,GACU;AAAA,EACb,QAAQ;AACN,IAAAC,EAAO,MAAM,sCAAsC;AAAA,EACrD;AACF;AAEA,eAAeC,GAAaH,GAAgBC,GAAsC;AAChF,MAAI;AACF,UAAMG,IAAM,MAAM,MAAM,GAAGJ,CAAM,mBAAmB;AAAA,MAClD,SAASC,IAAY,EAAE,eAAe,UAAUA,CAAS,OAAO;AAAA,IAAA,CACjE;AACD,WAAIG,EAAI,KACO,MAAMA,EAAI,KAAA,IAGlB;AAAA,EACT,QACM;AACJ,WAAAF,EAAO,MAAM,mCAAmC,GACzC;AAAA,EACT;AACF;AAEA,eAAeG,GAAMC,IAAK,KAAM;AAC9B,SAAO,IAAI,QAAQ,CAACC,MAAY,WAAWA,GAASD,CAAE,CAAC;AACzD;AAEA,MAAME,KAAgB,CAAC,EAAE,UAAAC,QAAwC;AAC/D,QAAM,EAAE,QAAAT,GAAQ,aAAAU,GAAa,UAAAC,EAAA,IAAaC,EAAA,GACpC,EAAE,MAAAC,GAAM,UAAAC,GAAU,aAAAC,GAAa,YAAAC,GAAY,YAAAC,GAAY,eAAAC,EAAA,IAAkBC,GAAA,GAGzE,CAACC,GAAeC,CAAgB,IAAIC,EAAoB,CAAA,CAAE,GAE1D,CAACC,GAASC,CAAU,IAAIF,EAAsB,CAAA,CAAE,GAEhD,CAACG,GAAgBC,CAAiB,IAAIJ,EAAS,EAAK,GACpD,CAACK,GAAcC,CAAe,IAAIN,EAAkB,IAAI,GAExDO,IAAclC,GAAe;AAAA,IACjC,QAAAK;AAAA,IACA,aAAAU;AAAA,IACA,UAAUG,MAAS,UAAUC,IAAW;AAAA,IACxC,gBAAgBH,GAAU,YACtB,EAAE,eAAe,UAAUA,GAAU,SAAS,GAAA,IAC9C;AAAA,IACJ,mBAAmB;AAAA,IACnB,eAAe,CAACmB,GAAMC,MAAY;AAIhC,YAAMC,IAA8B,CAAA,GAE9BC,IAAO,CAACC,MAAiB;AAC7B,YAAI,CAACA,KAAO,OAAOA,KAAQ,SAAU;AACrC,cAAMC,IAAID;AACV,YAAI,MAAM,QAAQC,EAAE,QAAQ;AAC1B,qBAAWC,KAAKD,EAAE,UAAuB;AACvC,kBAAME,IAAMD,EAA8B;AAC1C,gBAAIA,GAAG,SAAS,QAAQA,EAAE,MAAMC,GAAI;AAClC,oBAAMC,IAAe,OAAOD,EAAG,qBAAsB,YAAYA,EAAG,kBAAkB,SAAS,GACzFE,IAAgB,MAAM,QAAQF,EAAG,WAAW,KAAMA,EAAG,YAA0B,SAAS;AAC9F,eAAIC,KAAgBC,MAClBP,EAAiB,KAAKI,CAAC;AAAA,YAE3B;AAAA,UACF;AAEF,mBAAWI,KAAO,OAAO,OAAOL,CAAC;AAC/B,UAAIK,KAAO,OAAOA,KAAQ,YAAY,CAAC,MAAM,QAAQA,CAAG,KACtDP,EAAKO,CAAG;AAAA,MAGd;AAEA,MAAAP,EAAKH,CAAI;AAQT,UAAIW;AACJ,YAAMC,IAAY,CAACR,MAAiB;AAClC,YAAI,CAACA,KAAO,OAAOA,KAAQ,SAAU;AACrC,cAAMC,IAAID;AACV,QAAI,WAAWC,KAAK,MAAM,QAAQA,EAAE,KAAK,MAEnCA,EAAE,MAAM,SAAS,KAAKM,MAAe,YACvCA,IAAaN,EAAE;AAGnB,mBAAWK,KAAO,OAAO,OAAOL,CAAC;AAC/B,UAAIK,KAAO,OAAOA,KAAQ,YAAY,CAAC,MAAM,QAAQA,CAAG,KACtDE,EAAUF,CAAG;AAAA,MAGnB;AACA,MAAAE,EAAUZ,CAAI,IAEVE,EAAiB,SAAS,KAAKS,MAAe,WAChDV,EAAQ,OAAO,CAACY,MAAS;AACvB,YAAIC,IAASD;AAEb,YAAIX,EAAiB,SAAS,GAAG;AAC/B,gBAAMa,IAAgBD,EAAO,YAAY,CAAA,GACnCE,wBAAmB,IAAA;AACzB,qBAAW,KAAKd;AACd,YAAI,EAAE,MAAIc,EAAa,IAAI,EAAE,IAAI,CAAC;AAGpC,cAAIC,IAAU;AACd,gBAAMC,IAAkBH,EAAa,IAAI,CAAC,MAAM;AAC9C,gBAAI,CAAC,EAAE,MAAM,CAACC,EAAa,IAAI,EAAE,EAAE,EAAG,QAAO;AAC7C,kBAAMG,IAAWH,EAAa,IAAI,EAAE,EAAE;AACtC,YAAAA,EAAa,OAAO,EAAE,EAAE,GACxBC,IAAU;AACV,kBAAMG,IAAkBD,EAAqC;AAC7D,mBAAO;AAAA,cACL,GAAG;AAAA,cACH,mBAAmB;AAAA,gBACjB,GAAI,EAAE,qBAAqB,CAAA;AAAA,gBAC3B,GAAIC,KAAkB,CAAA;AAAA,cAAC;AAAA,YACzB;AAAA,UAEJ,CAAC;AAED,UAAIH,MACFH,IAAS,EAAE,GAAGA,GAAQ,UAAUI,EAAA;AAAA,QAEpC;AAEA,eAAIP,MAAe,UAGbA,EAAW,SAAS,MACtBG,IAAS,EAAE,GAAGA,GAAQ,OAAOH,EAAA,IAK1BG,MAAWD,IAAOA,IAAOC;AAAA,MAClC,CAAC;AAAA,IAEL;AAAA,IACA,eAAe,CAACO,GAAOpB,MAAY;AACjC,OAAIqB,EAAYD,CAAK,KAAKE,EAAkBF,CAAK,MAC/CpB,EAAQ,OAAO,CAACY,MAAS;AACvB,QAAAzC,EAAO,MAAM,wCAAwCiD,CAAK;AAC1D,cAAMG,IAAKC,EAAiBZ,EAAK,MAAM,CAAA,GAAIQ,CAAK;AAChD,eAAO,EAAE,GAAGR,GAAM,IAAAW,EAAA;AAAA,MACpB,CAAC;AAAA,IAEL;AAAA,IAEA,YAAY,CAACE,MAAO;AAClB,MAAKA,MACD3C,MAAS,WAENC,KAAUC,EAAYyC,CAAE,KAG7BzC,EAAYyC,CAAE,GAGdnD,GAAA,EAAQ,KAAK,MAAMW,EAAA,EAAa,KAAKC,CAAU,EAAE,MAAM,CAACwC,MAAQvD,EAAO,MAAM,8BAA8BuD,CAAG,CAAC,CAAC;AAAA,IAEpH;AAAA,EAAA,CACD,GAOKC,IAAcC;AAAA,IAClB,OACEC,GACA7B,MAYG;AAEH,YAAM8B,IAAY9B,GAAS,MAAM+B,GAAA,GAE3BC,IACJ,OAAOH,KAAY,WACd;AAAA,QACD,IAAIC;AAAA;AAAA,QACJ,MAAM9B,GAAS,QAAQ;AAAA,QACvB,SAAS6B;AAAA,QACT,GAAI7B,GAAS,QAAQ,EAAE,MAAMA,EAAQ,KAAA;AAAA,QACrC,GAAIA,GAAS,cAAc,EAAE,YAAYA,EAAQ,WAAA;AAAA,QACjD,GAAIA,GAAS,sBAAsB,EAAE,oBAAoBA,EAAQ,mBAAA;AAAA,QACjE,GAAIA,GAAS,gBAAgB,EAAE,cAAcA,EAAQ,aAAA;AAAA,QACrD,IAAKA,GAAS,qBAAqBA,GAAS,WAAW;AAAA,UACrD,mBAAmB;AAAA,YACjB,GAAIA,GAAS,UAAU,EAAE,QAAQ,GAAA;AAAA,YACjC,GAAIA,GAAS,qBAAqB,CAAA;AAAA,UAAC;AAAA,QACrC;AAAA,MACF,IAEA;AAAA,QACA,GAAG6B;AAAA,QACH,IAAIC;AAAA;AAAA;AAAA,QAEJ,GAAI9B,GAAS,cAAc,EAAE,YAAYA,EAAQ,WAAA;AAAA,QACjD,GAAIA,GAAS,qBAAqB;AAAA,UAChC,mBAAmB;AAAA,YACjB,GAAG6B,EAAQ;AAAA,YACX,GAAG7B,EAAQ;AAAA,UAAA;AAAA,QACb;AAAA,MACF;AAQN,UAJA7B,EAAO,MAAM,oCAAoC6D,CAAU,GAIvDhC,GAAS,SAAS,MAAM;AAI1B,YAHAV,EAAiB,CAACsB,MAAS,CAAC,GAAGA,GAAMoB,CAAU,CAAC,GAChD7D,EAAO,MAAM,yCAAyC6B,GAAS,EAAE,GAE7DA,EAAQ,MAAMA,EAAQ,GAAG,SAAS,GAAG;AACvC,gBAAMiC,IAAkBjC,EAAQ,GAAG,IAAI,CAACuB,GAAIW,OAAW;AAAA,YACrD,GAAGX;AAAA,YACH,IAAIO;AAAA;AAAA;AAAA,YAEJ,MAAOP,EAAqC,QAAQ,GAAGO,CAAS,OAAOI,CAAK;AAAA,UAAA,EAC5E;AACF,UAAAzC,EAAW,CAACmB,MAAS,CAAC,GAAGA,GAAM,GAAGqB,CAAe,CAAC;AAAA,QACpD;AACA;AAAA,MACF;AAGA,YAAME,IAAkBrC,EAAY,YAAY,CAAA,GAE1CsC,IAAgBpC,GAAS,UAC3B,EAAE,GAAGpB,GAAU,GAAGoB,EAAQ,SAAS,GAAGb,EAAA,IACtCP,KAAY,CAAA;AAEhB,YAAMkB,EAAY;AAAA,QAChB,EAAE,UAAU,CAAC,GAAGqC,GAAiBH,CAAU,EAAA;AAAA,QAC3C;AAAA,UACE,SAASI;AAAA,UACT,YAAY,CAAC,UAAU,kBAAkB,WAAW,QAAQ;AAAA,QAAA;AAAA,MAC9D;AAAA,IAEJ;AAAA,IACA,CAACtC,GAAalB,GAAUO,GAAeG,GAAkBG,CAAU;AAAA,EAAA,GAI/D4C,IAAmBC,EAAQ,MAAM;AACrC,UAAMC,IAAc,CAAC,GAAGlD,GAAe,GAAIS,EAAY,YAAY,EAAG,GAEhE0C,wBAAW,IAAA;AACjB,WAAOD,EAAY,OAAO,CAAAE,MACnBA,EAAI,KACLD,EAAK,IAAIC,EAAI,EAAE,IAAU,MAC7BD,EAAK,IAAIC,EAAI,EAAE,GACR,MAHa,EAIrB;AAAA,EACH,GAAG,CAACpD,GAAeS,EAAY,QAAQ,CAAC,GAGlC4C,IAAaJ,EAAQ,MAAM;AAC/B,IAAAnE,EAAO,MAAM,sCAAsCqB,GAASM,EAAY,QAAQ,EAAE;AAClF,UAAM6C,IAAQ,CAAC,GAAGnD,GAAS,GAAIM,EAAY,QAAQ,MAAM,EAAG,GAEtD0C,wBAAW,IAAA;AACjB,WAAOG,EAAM,OAAO,CAAApB,MAAM;AACxB,YAAMqB,IAAOrB,EAAqC,QAAQ,GAAGA,EAAG,EAAE,IAAIA,EAAG,IAAI;AAC7E,aAAIiB,EAAK,IAAII,CAAG,IAAU,MAC1BJ,EAAK,IAAII,CAAG,GACL;AAAA,IACT,CAAC;AAAA,EACH,GAAG,CAACpD,GAASM,EAAY,QAAQ,EAAE,CAAC,GAE9B+C,IAAgBjB;AAAA,IACpB,OACEC,GACA7B,MAMG;AAEH,YAAM8C,IAAqBT,KAAoB,CAAA;AAE/C,YAAMvC,EAAY;AAAA,QAChB,EAAE,UAAU,CAAC,GAAGgD,GAAoBjB,CAAO,EAAA;AAAA,QAC3C;AAAA,UACE,SAAS,EAAE,GAAGjD,GAAU,GAAGoB,GAAS,cAAA;AAAA,UACpC,YAAYA,GAAS,cAAc,CAAC,UAAU,kBAAkB,WAAW,QAAQ;AAAA,UACnF,iBAAiBA,GAAS,mBAAmB;AAAA,UAC7C,iBAAiBA,GAAS,mBAAmB;AAAA,UAC7C,kBAAkB,CAACY,OAAU;AAAA,YAC3B,GAAGA;AAAA,YACH,UAAU,CAAC,GAAGkC,GAAoBjB,CAAO;AAAA,UAAA;AAAA,QAC3C;AAAA,MACF;AAAA,IAEJ;AAAA,IACA,CAAC/B,GAAalB,GAAUyD,CAAgB;AAAA,EAAA,GAOpCU,IAAoBnB;AAAA,IACxB,OAAOE,MAAsB;AAC3B,YAAMS,IAAcF,KAAoB,CAAA,GAClCW,IAAeT,EAAY,UAAU,CAACE,MAAQA,EAAI,OAAOX,CAAS;AAExE,UAAIkB,MAAiB,IAAI;AACvB,QAAA7E,EAAO,MAAM,uCAAuC2D,CAAS;AAC7D;AAAA,MACF;AAGA,UAAImB,IAAiBD,IAAe;AACpC,aAAOC,KAAkB,KAAKV,EAAYU,CAAc,EAAE,SAAS;AACjE,QAAAA;AAGF,UAAIA,MAAmB,IAAI;AACzB,QAAA9E,EAAO,MAAM,6CAA6C2D,CAAS;AACnE;AAAA,MACF;AAGA,YAAMoB,IAAiBX,EAAY,MAAM,GAAGU,IAAiB,CAAC,GAGxDE,IAAsBD,EAAe,MAAM,GAAG,EAAE,GAChDE,IAAeF,EAAeA,EAAe,SAAS,CAAC;AAG7D,MAAA5D;AAAA,QAAiB,CAACsB,MAChBA,EAAK,OAAO,CAAC6B,MACMF,EAAY,UAAU,CAAClC,MAAMA,EAAE,OAAOoC,EAAI,EAAE,IAC3CQ,CACnB;AAAA,MAAA,GAIHxD;AAAA,QAAW,CAACmB,MACVA,EAAK,OAAO,CAACW,MACMgB,EAAY,UAAU,CAAClC,MAAMA,EAAE,OAAOkB,EAAG,EAAE,IAC1C0B,CACnB;AAAA,MAAA,GAIH,MAAMnD,EAAY;AAAA,QAChB,EAAE,UAAU,CAAC,GAAGqD,GAAqBC,CAAY,EAAA;AAAA,QACjD;AAAA,UACE,YAAY,CAAC,QAAQ;AAAA,UACrB,iBAAiB;AAAA,UACjB,iBAAiB;AAAA,QAAA;AAAA,MACnB;AAAA,IAEJ;AAAA,IACA,CAACf,GAAkBvC,GAAalB,GAAUO,GAAeG,GAAkBG,CAAU;AAAA,EAAA;AAGvF,EAAA4D,GAAU,MAAM;AACd,IAAArF,GAAiBC,GAAQW,GAAU,SAAS,EAAE,KAAK,CAAC0E,MAAO;AACzD,MAAKA,KACHC,GAAM,MAAM,yCAAyC;AAAA,QACnD,aAAa,mBAAmBtF,CAAM;AAAA,QACtC,UAAU;AAAA,MAAA,CACX;AAAA,IAEL,CAAC;AAAA,EACH,GAAG,CAACA,GAAQW,GAAU,SAAS,CAAC;AAEhC,QAAM4E,IAAuB5B,EAAY,YAAY;AAEnD,QAAIlC,KAAkBE,MAAiB;AACrC,aAAOA;AAIT,QAAI,CAACF,GAAgB;AACnB,MAAAC,EAAkB,EAAI;AACtB,YAAMI,IAAO,MAAM3B,GAAaH,GAAQW,GAAU,SAAS;AAC3D,aAAAiB,EAAgBE,CAAI,GACbA;AAAA,IACT;AAEA,WAAOH;AAAA,EACT,GAAG,CAAC3B,GAAQW,GAAU,WAAWc,GAAgBE,CAAY,CAAC,GAExD6D,IAAQnB;AAAA,IACZ,OAAO;AAAA,MACL,GAAGxC;AAAA,MACH,UAAUuC;AAAA;AAAA,MACV,QAAQ;AAAA,QACN,GAAGvC,EAAY;AAAA,QACf,IAAI4C;AAAA;AAAA,MAAA;AAAA,MAEN,aAAAf;AAAA,MACA,eAAAkB;AAAA,MACA,mBAAAE;AAAA,MACA,cAAcS;AAAA,IAAA;AAAA,IAEhB,CAAC1D,GAAauC,GAAkBK,GAAYf,GAAakB,GAAeE,GAAmBS,CAAoB;AAAA,EAAA;AAGjH,SACE,gBAAAE,EAAC5F,EAAc,UAAd,EAAuB,OAAA2F,GAAe,UAAA/E,EAAA,CAAS;AAEpD;AAaO,SAASiF,GAAe,EAAE,UAAAjF,KAAqC;AACpE,SAAO,gBAAAgF,EAACjF,MAAe,UAAAC,GAAS;AAClC;AAaO,SAASkF,KAAsC;AACpD,QAAMC,IAAMC,EAAWhG,CAAa;AACpC,MAAI,CAAC+F;AACH,UAAM,IAAI,MAAM,qDAAqD;AAEvE,SAAOA;AACT;"}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "langgraph-ui-components",
   "private": false,
-  "version": "0.0.27",
+  "version": "0.0.28",
   "type": "module",
   "main": "./dist/index.cjs.js",
   "module": "./dist/index.es.js",
@@ -121,6 +121,22 @@
     "vite": "^7.2.4",
     "vite-plugin-dts": "^4.5.4"
   },
+  "keywords": [
+    "langgraph",
+    "langgraph-ui",
+    "react",
+    "chat-ui",
+    "ai-components",
+    "chatbot",
+    "sidebar",
+    "streaming",
+    "langchain",
+    "assistant",
+    "chatgpt-ui",
+    "human-in-the-loop",
+    "hitl",
+    "generative-ui"
+  ],
   "sideEffects": [
     "**/*.css"
   ],

package/src/App.tsx

@@ -27,8 +27,8 @@ function ChatWrapper({ children }: { children?: React.ReactNode }) {
   }
 
   return (<>
-    {/* <Sidebar supportChatHistory={true} textToSpeechVoice={textToSpeechVoice} /> */}
-    <Chat textToSpeechVoice={textToSpeechVoice} />
+    <Sidebar supportChatHistory={true} textToSpeechVoice={textToSpeechVoice} />
+    {/* <Chat textToSpeechVoice={textToSpeechVoice} /> */}
     {children}
   </>
   )
@@ -40,12 +40,12 @@ function App() {
     user_id: "68f3b3ba4ce23e5b582b7780",
     org_id: "1",
     quote_id: "695b5cfcc4d07dc56f093d92",
-    authToken: "LgCCT6e4CteXOO3842u-oqJpv-Cnb83-lTchzrJ9Lhw",
+    authToken: "xQpz7LPUqh5nArwn8Boam6JpJX2BOXf4NDcJ9kEEuWc",
   });
   return (
     <React.Suspense fallback={<div>Loading (layout)...</div>}>
       <ChatRuntimeProvider
-        apiUrl={"https://agent-api.gauravshivaprasad.com"}
+        apiUrl={"https://agents.3ya.io"}
         assistantId={"chat_agent"}
         // apiUrl={"https://agents.3ya.io"}
         // assistantId={"v3ya_external_agent"}

package/src/components/ChatBody.tsx

@@ -1,10 +1,10 @@
-import { parsePartialJson } from "@langchain/core/output_parsers";
 import { useStreamContext } from "@/providers/Stream";
 import { useCustomComponents } from "@/providers/CustomComponentProvider";
 import type { chatBodyProps } from "@/types/ChatProps";
 import { logger } from "@/utils/logger";
 import { isAiWithToolCalls, isToolMessage } from "@/utils/utils";
-import type { AIMessage, Message } from "@langchain/langgraph-sdk";
+import type { Message } from "@langchain/langgraph-sdk";
+import type { TodoItem } from "@/providers/Stream";
 import React, { useCallback, useEffect, useLayoutEffect, useMemo, useRef, useState } from "react";
 import AgentMessage from "./messages/AgentMessage";
 import CustomComponentRender from "./messages/CustomComponentRender";
@@ -26,9 +26,67 @@ export default function ChatBody({ setIsFirstMessage, enableToolCallIndicator =
   // Memoize messages with stable reference
   const memoMessages = useMemo(() => messages ?? [], [messages]);
 
+  const uiMessageIds = useMemo(() => {
+    const ids = new Set<string>();
+    const uiMessages = stream.values?.ui;
+    if (!Array.isArray(uiMessages)) return ids;
+
+    for (const ui of uiMessages) {
+      const uiObj = ui as unknown as Record<string, unknown>;
+      const metadata =
+        uiObj.metadata && typeof uiObj.metadata === "object"
+          ? (uiObj.metadata as Record<string, unknown>)
+          : undefined;
+
+      const candidates = [
+        typeof uiObj.id === "string" ? uiObj.id : undefined,
+        typeof metadata?.id === "string" ? metadata.id : undefined,
+        typeof metadata?.message_id === "string" ? metadata.message_id : undefined,
+      ];
+
+      for (const candidate of candidates) {
+        if (candidate) ids.add(candidate);
+      }
+    }
+
+    return ids;
+  }, [stream.values?.ui]);
+
   const containerRef = useRef<HTMLDivElement | null>(null);
   const prevMessageCountRef = useRef(0);
   const shouldAutoScrollRef = useRef(true);
+  const frozenTodosByGroupRef = useRef<Map<string, TodoItem[]>>(new Map());
+
+  const normalizeTodos = useCallback((todos: unknown): TodoItem[] => {
+    if (!Array.isArray(todos)) return [];
+
+    return todos
+      .map((todo, index) => {
+        if (!todo || typeof todo !== "object") return null;
+        const t = todo as Record<string, unknown>;
+        const status = t.status;
+        if (
+          typeof t.content !== "string" ||
+          (status !== "pending" && status !== "in_progress" && status !== "completed")
+        ) {
+          return null;
+        }
+
+        return {
+          id: t.id && typeof t.id === "string" ? t.id : `todo-${index}`,
+          content: t.content,
+          status,
+          updatedAt: t.updatedAt instanceof Date || typeof t.updatedAt === "string" || typeof t.updatedAt === "number"
+            ? (t.updatedAt as string | number | Date)
+            : undefined,
+          run_id: typeof t.run_id === "string" ? t.run_id : undefined,
+          runId: typeof t.runId === "string" ? t.runId : undefined,
+          messageId: typeof t.messageId === "string" ? t.messageId : undefined,
+          checkpoint: typeof t.checkpoint === "string" ? t.checkpoint : undefined,
+        } as TodoItem;
+      })
+      .filter((todo): todo is TodoItem => todo !== null);
+  }, []);
 
   // Handler for message feedback
   const handleFeedback = useCallback((messageId: string, feedback: MessageFeedback) => {
@@ -62,7 +120,11 @@ export default function ChatBody({ setIsFirstMessage, enableToolCallIndicator =
 
   // Handler for HITL interrupt responses
   const handleInterruptRespond = useCallback((response: any) => {
-    stream.submit(null, { command: { resume: response } });
+    stream.submit(null, {
+      command: { resume: response },
+      streamMode: ["values", "messages-tuple", "updates", "custom"],
+      streamSubgraphs: true,
+    });
   }, [stream]);
 
   // Build interrupt actions for custom renderers
@@ -102,34 +164,6 @@ export default function ChatBody({ setIsFirstMessage, enableToolCallIndicator =
     };
   }, [stream.interrupt, handleInterruptRespond]);
 
-  const getToolCallsFromContent = useCallback((content: Message["content"]): AIMessage["tool_calls"] => {
-    if (!Array.isArray(content)) return [];
-    const toolCallContents = content.filter(
-      (c) => typeof c === "object" && c !== null && (c as any).type === "tool_use" && (c as any).id,
-    ) as Array<Record<string, any>>;
-
-    return toolCallContents.map((tc) => {
-      let args: Record<string, any> = {};
-      if (tc?.input) {
-        if (typeof tc.input === "string") {
-          try {
-            args = parsePartialJson(tc.input) ?? {};
-          } catch {
-            args = {};
-          }
-        } else if (typeof tc.input === "object") {
-          args = tc.input as Record<string, any>;
-        }
-      }
-      return {
-        name: tc.name ?? "",
-        id: tc.id ?? "",
-        args,
-        type: "tool_call",
-      };
-    });
-  }, []);
-
   // Get the parent scroll container
   const getScrollContainer = useCallback(() => {
     if (containerRef.current) {
@@ -163,6 +197,8 @@ export default function ChatBody({ setIsFirstMessage, enableToolCallIndicator =
 
     // Use message id or fallback to index for key
     const msgKey = msg.id ?? `msg-${index}`;
+    const hasCustomComponentForMessage = (message: Message) =>
+      typeof message.id === "string" && uiMessageIds.has(message.id);
 
     if (msg.type === "human") {
       return <HumanMessage key={msgKey} message={msg} fontSize={chatBodyProps?.fontSize} />;
@@ -182,8 +218,12 @@ export default function ChatBody({ setIsFirstMessage, enableToolCallIndicator =
       }
 
       if (prevIndex >= 0 && messagesArray[prevIndex].type === "ai") {
-        // This is a consecutive AI message, skip it (it will be combined with the previous one)
-        return null;
+        // Only render separately if THIS message owns custom UI (needs its own group).
+        // Otherwise it was already consumed by the previous group's forward scan.
+        const currentHasCustomUi = hasCustomComponentForMessage(msg);
+        if (!currentHasCustomUi) {
+          return null;
+        }
       }
     }
 
@@ -197,6 +237,11 @@ export default function ChatBody({ setIsFirstMessage, enableToolCallIndicator =
         groupedTimelineMessages.push(nextMsg);
         nextIndex++;
       } else if (nextMsg.type === "ai") {
+        if (hasCustomComponentForMessage(nextMsg)) {
+          // Sub-agent/custom UI messages should be rendered in their own section
+          // to avoid parent-bubble flicker and reparenting during streaming.
+          break;
+        }
         groupedTimelineMessages.push(nextMsg);
         nextIndex++;
       } else {
@@ -206,23 +251,42 @@ export default function ChatBody({ setIsFirstMessage, enableToolCallIndicator =
     }
 
     const groupedAiMessages = groupedTimelineMessages.filter((m) => m.type === "ai");
-    const toolMessages = groupedTimelineMessages.filter((m) => isToolMessage(m));
 
-    // Merge content from all AI messages, preserving r/reasoning blocks
+    // Merge content from all AI messages, preserving reasoning blocks.
+    // Deduplicate identical text content (subgraph messages can duplicate parent messages).
     const hasArrayContent = groupedAiMessages.some((m) => Array.isArray(m.content));
+    const seenTexts = new Set<string>();
     const mergedContent: Message["content"] = hasArrayContent
       ? groupedAiMessages.flatMap((m) => {
-          if (Array.isArray(m.content)) return m.content as any[];
-          if (typeof m.content === "string" && m.content.length > 0)
+          if (Array.isArray(m.content)) {
+            return (m.content as any[]).filter((block: any) => {
+              if (block.type === "text" && typeof block.text === "string") {
+                if (seenTexts.has(block.text)) return false;
+                seenTexts.add(block.text);
+              }
+              return true;
+            });
+          }
+          if (typeof m.content === "string" && m.content.length > 0) {
+            if (seenTexts.has(m.content)) return [];
+            seenTexts.add(m.content);
             return [{ type: "text", text: m.content }];
+          }
           return [];
         })
-      : groupedAiMessages
-          .map((m) => (typeof m.content === "string" ? m.content : ""))
-          .filter(Boolean)
-          .join("\n\n");
+      : (() => {
+          const unique: string[] = [];
+          for (const m of groupedAiMessages) {
+            const text = typeof m.content === "string" ? m.content : "";
+            if (text && !seenTexts.has(text)) {
+              seenTexts.add(text);
+              unique.push(text);
+            }
+          }
+          return unique.join("\n\n");
+        })();
 
-    // Extract plain-text portion for legacy checks (tool-call extraction, etc.)
+    // Extract plain-text portion for legacy checks
     const combinedContent = hasArrayContent
       ? (mergedContent as any[])
           .map((c: any) => (c.type === "text" ? (c.text as string) : ""))
@@ -230,13 +294,6 @@ export default function ChatBody({ setIsFirstMessage, enableToolCallIndicator =
           .join(" ")
       : (mergedContent as string);
 
-    const toolCallsFromContent = groupedAiMessages
-      .flatMap((m) => {
-        const calls = getToolCallsFromContent(m.content);
-        return calls || [];
-      })
-      .filter((tc): tc is NonNullable<typeof tc> => tc != null && tc !== undefined);
-
     // Create a combined message object preserving full content (including thinking blocks)
     const combinedMessage = {
       ...msg,
@@ -259,26 +316,64 @@ export default function ChatBody({ setIsFirstMessage, enableToolCallIndicator =
         const ak = (m as Record<string, unknown>).additional_kwargs as Record<string, unknown> | undefined;
         return typeof ak?.reasoning_content === "string" && (ak.reasoning_content as string).length > 0;
       });
+
     const hasKwargsToolStatus = groupedAiMessages.some((m) => {
       const ak = (m as Record<string, unknown>).additional_kwargs as Record<string, unknown> | undefined;
       return Array.isArray(ak?.tool_status) && (ak.tool_status as unknown[]).length > 0;
     });
-
-    const hasToolCalls =
+    const hasToolMessages = groupedTimelineMessages.some((m) => isToolMessage(m));
+    const hasToolCalls = enableToolCallIndicator && (
       groupedAiMessages.some((m) => isAiWithToolCalls(m)) ||
-      toolCallsFromContent.length > 0 ||
-      toolMessages.length > 0 ||
-      hasKwargsToolStatus;
-
+      hasKwargsToolStatus ||
+      hasToolMessages
+    );
     const hasAnyDisplayableContent =
       !!combinedContent ||
       hasThinkingContent ||
       hasKwargsReasoning ||
-      (enableToolCallIndicator && hasToolCalls);
+      hasToolCalls;
 
     const isLastMessageGroup = nextIndex >= messagesArray.length;
     const isStreamingThisMessage = isLoading && isLastMessageGroup;
 
+    const groupKey = groupedAiMessages
+      .map((m) => m.id)
+      .filter((id): id is string => typeof id === "string" && id.length > 0)
+      .join("|") || msgKey;
+
+    const todosForGroup = (() => {
+      const frozen = frozenTodosByGroupRef.current.get(groupKey);
+      if (!isStreamingThisMessage && frozen && frozen.length > 0) {
+        return frozen;
+      }
+
+      if (isStreamingThisMessage) {
+        const live = normalizeTodos(stream.values?.todos);
+        if (live.length > 0) {
+          frozenTodosByGroupRef.current.set(groupKey, live);
+          return live;
+        }
+      }
+
+      for (let i = groupedAiMessages.length - 1; i >= 0; i--) {
+        const aiMeta = stream.getMessagesMetadata(groupedAiMessages[i]);
+        const aiValues = aiMeta?.firstSeenState?.values as { todos?: unknown } | undefined;
+        const snapshot = normalizeTodos(aiValues?.todos);
+        if (snapshot.length > 0) {
+          if (!frozen || frozen.length === 0) {
+            frozenTodosByGroupRef.current.set(groupKey, snapshot);
+          }
+          return frozen ?? snapshot;
+        }
+      }
+
+      if (frozen && frozen.length > 0) {
+        return frozen;
+      }
+
+      return [];
+    })();
+
     // Show Thinking animation whenever streaming with no displayable content yet (plain chat or tool-call)
     const showThinkingIndicator = isStreamingThisMessage && !hasAnyDisplayableContent;
 
@@ -293,7 +388,6 @@ export default function ChatBody({ setIsFirstMessage, enableToolCallIndicator =
           <AgentMessage
             agentName={chatBodyProps?.agentName}
             fontSize={chatBodyProps?.fontSize}
-            // agentAvatarUrl={chatBodyProps?.agentAvatarUrl}
             message={combinedMessage}
             groupedMessages={groupedTimelineMessages}
             showToolActivity={enableToolCallIndicator}
@@ -304,6 +398,7 @@ export default function ChatBody({ setIsFirstMessage, enableToolCallIndicator =
             branch={meta?.branch}
             branchOptions={meta?.branchOptions}
             onBranchSelect={handleBranchSelect}
+            todos={todosForGroup}
           />
         )}
         {/* 3. Custom component (from all messages in the group) */}
@@ -312,7 +407,7 @@ export default function ChatBody({ setIsFirstMessage, enableToolCallIndicator =
         ))}
       </React.Fragment>
     );
-  }, [isLoading, stream, enableToolCallIndicator, handleRegenerate, messageFeedback, handleFeedback, handleBranchSelect]);
+  }, [isLoading, stream, normalizeTodos, enableToolCallIndicator, handleRegenerate, messageFeedback, handleFeedback, handleBranchSelect, uiMessageIds]);
 
   // Memoize the rendered messages array
   const renderedMessages = useMemo(() => {
@@ -409,6 +504,11 @@ export default function ChatBody({ setIsFirstMessage, enableToolCallIndicator =
         <>
           {renderedMessages}
           {/* Show HITL interrupt card when agent is paused for approval */}
+          {/* Show thinking indicator when loading and last message is human (no AI response yet) */}
+          {isLoading && memoMessages.length > 0 &&
+            memoMessages[memoMessages.length - 1].type === "human" && (
+              <Thinking />
+            )}
           {!isLoading && stream.interrupt && interruptActions && (() => {
             const payload = stream.interrupt.value as any;
             const toolName = payload?.actionRequests?.[0]?.name;
@@ -416,15 +516,6 @@ export default function ChatBody({ setIsFirstMessage, enableToolCallIndicator =
             if (!InterruptComponent) return null;
             return <InterruptComponent interrupt={payload} actions={interruptActions} />;
           })()}
-          {/* Show thinking indicator when loading and no AI response has started yet */}
-          {isLoading && memoMessages.length > 0 && (() => {
-            const lastMsg = memoMessages[memoMessages.length - 1];
-            // Show if last message is human or if last AI has no content (only tool calls)
-            return lastMsg.type === "human" ||
-              (lastMsg.type === "ai" && !lastMsg.content && isAiWithToolCalls(lastMsg));
-          })() && (
-              <Thinking />
-            )}
         </>
       )}
 

package/src/components/TodoList.tsx

@@ -0,0 +1,72 @@
+import type { TodoItem } from "@/providers/Stream";
+import { CheckCircle2, Circle, Loader2, Sparkles } from "lucide-react";
+
+const statusConfig = {
+  completed: {
+    icon: CheckCircle2,
+    iconClass: "text-emerald-400",
+    textClass: "text-zinc-400 line-through",
+  },
+  in_progress: {
+    icon: Loader2,
+    iconClass: "text-blue-400 animate-spin",
+    textClass: "text-zinc-100",
+  },
+  pending: {
+    icon: Circle,
+    iconClass: "text-zinc-500",
+    textClass: "text-zinc-400",
+  },
+} as const;
+
+export default function TodoList({ todos }: { todos: TodoItem[] }) {
+  if (!todos.length) return null;
+
+  const completed = todos.filter((t) => t.status === "completed").length;
+  const inProgress = todos.find((t) => t.status === "in_progress");
+  const percentage = Math.round((completed / todos.length) * 100);
+
+  return (
+    <div className="mb-3 w-full max-w-full rounded-2xl border border-zinc-700/60 bg-zinc-950/35 px-4 py-3.5 shadow-[0_10px_30px_rgba(0,0,0,0.25)] backdrop-blur-sm">
+      <div className="mb-3 flex items-start justify-between gap-3">
+        <div className="min-w-0">
+          <div className="flex items-center gap-2 text-zinc-200">
+            <Sparkles className="h-3.5 w-3.5 text-cyan-300/90" />
+            <span className="text-xs font-semibold uppercase tracking-[0.14em] text-zinc-300">Agent Plan</span>
+          </div>
+          <p className="mt-1 truncate text-xs text-zinc-400">
+            {inProgress ? `Now: ${inProgress.content}` : "Wrapping up remaining steps"}
+          </p>
+        </div>
+        <div className="rounded-lg border border-zinc-700/80 bg-zinc-900/70 px-2.5 py-1 text-[11px] font-medium text-zinc-300">
+          {completed}/{todos.length} done
+        </div>
+      </div>
+
+      <div className="mb-3 h-1.5 overflow-hidden rounded-full bg-zinc-800/80">
+        <div
+          className="h-full rounded-full bg-linear-to-r from-cyan-400 to-emerald-400 transition-all duration-500"
+          style={{ width: `${percentage}%` }}
+        />
+      </div>
+
+      <ul className="space-y-1.5">
+        {todos.map((todo) => {
+          const config = statusConfig[todo.status];
+          const Icon = config.icon;
+          return (
+            <li
+              key={todo.id}
+              className="group flex items-center gap-2 rounded-lg border border-zinc-800/80 bg-zinc-900/45 px-2.5 py-2 transition-colors duration-200 hover:border-zinc-700/80"
+            >
+              <Icon className={`h-3.5 w-3.5 flex-shrink-0 ${config.iconClass}`} />
+              <span className={`line-clamp-1 text-xs ${config.textClass}`}>
+                {todo.content}
+              </span>
+            </li>
+          );
+        })}
+      </ul>
+    </div>
+  );
+}