@copilotkit/react-core 1.50.0-beta.1 → 1.50.0-beta.11

package/dist/hooks/use-copilot-chat-headless_c.js

@@ -83,7 +83,7 @@ __export(use_copilot_chat_headless_c_exports, {
   useCopilotChatHeadless_c: () => useCopilotChatHeadless_c
 });
 module.exports = __toCommonJS(use_copilot_chat_headless_c_exports);
-var import_react14 = require("react");
+var import_react15 = require("react");
 
 // src/context/copilot-context.tsx
 var import_react = __toESM(require("react"));
@@ -161,7 +161,7 @@ var emptyCopilotContext = {
   interruptEventQueue: {},
   addInterruptEvent: () => {
   },
-  removeInterruptEvent: () => {
+  resolveInterruptEvent: () => {
   },
   onError: () => {
   },
@@ -187,7 +187,7 @@ function returnAndThrowInDebug(_value) {
 }
 
 // src/hooks/use-copilot-chat_internal.ts
-var import_react12 = require("react");
+var import_react13 = require("react");
 
 // src/components/error-boundary/error-utils.tsx
 var import_react3 = require("react");
@@ -335,24 +335,60 @@ function useCoAgentStateRenders() {
 }
 
 // src/hooks/use-langgraph-interrupt-render.ts
-var import_react5 = __toESM(require("react"));
+var import_react7 = __toESM(require("react"));
 var import_runtime_client_gql = require("@copilotkit/runtime-client-gql");
 var import_shared2 = require("@copilotkit/shared");
+
+// src/hooks/use-agent-nodename.ts
+var import_react5 = require("react");
+var import_react6 = require("@copilotkitnext/react");
+function useAgentNodeName(agentName) {
+  const { agent } = (0, import_react6.useAgent)({ agentId: agentName });
+  const nodeNameRef = (0, import_react5.useRef)("start");
+  (0, import_react5.useEffect)(() => {
+    if (!agent)
+      return;
+    const subscriber = {
+      onStepStartedEvent: ({ event }) => {
+        nodeNameRef.current = event.stepName;
+      },
+      onRunStartedEvent: () => {
+        nodeNameRef.current = "start";
+      },
+      onRunFinishedEvent: () => {
+        nodeNameRef.current = "end";
+      }
+    };
+    const subscription = agent.subscribe(subscriber);
+    return () => {
+      subscription.unsubscribe();
+    };
+  }, [agent]);
+  return nodeNameRef.current;
+}
+
+// src/hooks/use-langgraph-interrupt-render.ts
+var import_react8 = require("@copilotkitnext/react");
 var InterruptRenderer = ({ event, result, render, resolve }) => {
   return render({ event, result, resolve });
 };
 function useLangGraphInterruptRender(agent) {
+  var _a;
   const {
     interruptActions,
     agentSession,
     threadId,
     interruptEventQueue,
     addInterruptEvent,
-    removeInterruptEvent
+    resolveInterruptEvent
   } = useCopilotContext();
-  (0, import_react5.useEffect)(() => {
+  const existingConfig = (0, import_react8.useCopilotChatConfiguration)();
+  const resolvedAgentId = (_a = existingConfig == null ? void 0 : existingConfig.agentId) != null ? _a : "default";
+  const nodeName = useAgentNodeName(resolvedAgentId);
+  (0, import_react7.useEffect)(() => {
     if (!agent)
       return;
+    let localInterrupt = null;
     const subscriber = {
       onCustomEvent: ({ event }) => {
         if (event.name === "on_interrupt") {
@@ -361,12 +397,21 @@ function useLangGraphInterruptRender(agent) {
             type: event.type,
             value: (0, import_shared2.parseJson)(event.value, event.value)
           };
-          const eventId = (0, import_shared2.dataToUUID)(JSON.stringify(eventData), "interruptEvents");
-          addInterruptEvent({
+          const eventId = (0, import_shared2.dataToUUID)(eventData, "interruptEvents");
+          localInterrupt = {
             eventId,
             threadId,
             event: eventData
-          });
+          };
+        }
+      },
+      onRunStartedEvent: () => {
+        localInterrupt = null;
+      },
+      onRunFinalized: () => {
+        if (localInterrupt) {
+          addInterruptEvent(localInterrupt);
+          localInterrupt = null;
         }
       }
     };
@@ -375,7 +420,7 @@ function useLangGraphInterruptRender(agent) {
       unsubscribe();
     };
   }, [agent, threadId]);
-  const handleResolve = (0, import_react5.useCallback)(
+  const handleResolve = (0, import_react7.useCallback)(
     (eventId, response) => {
       agent == null ? void 0 : agent.runAgent({
         forwardedProps: {
@@ -384,15 +429,15 @@ function useLangGraphInterruptRender(agent) {
           }
         }
       });
-      removeInterruptEvent(threadId, eventId);
+      resolveInterruptEvent(threadId, eventId, response != null ? response : "");
     },
     // eslint-disable-next-line react-hooks/exhaustive-deps
     [agent, threadId]
   );
-  return (0, import_react5.useMemo)(() => {
+  return (0, import_react7.useMemo)(() => {
     const eventQueue = interruptEventQueue[threadId] || [];
     const currentQueuedEvent = eventQueue.find((qe) => !qe.event.response);
-    if (!currentQueuedEvent)
+    if (!currentQueuedEvent || !agentSession)
       return null;
     const allActions = Object.values(interruptActions);
     const matchingAction = allActions.find((action) => {
@@ -400,7 +445,9 @@ function useLangGraphInterruptRender(agent) {
         return true;
       return action.enabled({
         eventValue: currentQueuedEvent.event.value,
-        agentMetadata: agentSession
+        agentMetadata: __spreadProps(__spreadValues({}, agentSession), {
+          nodeName
+        })
       });
     });
     if (!matchingAction)
@@ -418,7 +465,7 @@ function useLangGraphInterruptRender(agent) {
     }
     if (!render)
       return null;
-    return import_react5.default.createElement(InterruptRenderer, {
+    return import_react7.default.createElement(InterruptRenderer, {
       event: currentQueuedEvent.event,
       result,
       render,
@@ -428,14 +475,14 @@ function useLangGraphInterruptRender(agent) {
 }
 
 // src/hooks/use-copilot-chat_internal.ts
-var import_react13 = require("@copilotkitnext/react");
+var import_react14 = require("@copilotkitnext/react");
 
 // src/hooks/use-lazy-tool-renderer.tsx
-var import_react6 = require("@copilotkitnext/react");
-var import_react7 = require("react");
+var import_react9 = require("@copilotkitnext/react");
+var import_react10 = require("react");
 function useLazyToolRenderer() {
-  const renderToolCall = (0, import_react6.useRenderToolCall)();
-  return (0, import_react7.useCallback)(
+  const renderToolCall = (0, import_react9.useRenderToolCall)();
+  return (0, import_react10.useCallback)(
     (message, messages) => {
       var _a;
       if (!((_a = message == null ? void 0 : message.toolCalls) == null ? void 0 : _a.length))
@@ -455,39 +502,7 @@ function useLazyToolRenderer() {
   );
 }
 
-// src/hooks/use-configure-chat-suggestions.tsx
-var import_react8 = require("@copilotkitnext/react");
-var import_react9 = require("react");
-function useConfigureChatSuggestions(config, dependencies = []) {
-  const { agentSession } = useCopilotContext();
-  const { copilotkit } = (0, import_react8.useCopilotKit)();
-  const available = config.available === "enabled" ? "always" : config.available;
-  const finalSuggestionConfig = __spreadProps(__spreadValues({}, config), {
-    available,
-    consumerAgentId: agentSession == null ? void 0 : agentSession.agentName
-    // Use chatConfig.agentId here
-  });
-  (0, import_react8.useConfigureSuggestions)(finalSuggestionConfig, { deps: dependencies });
-  const result = (0, import_react8.useSuggestions)({ agentId: agentSession == null ? void 0 : agentSession.agentName });
-  (0, import_react9.useEffect)(() => {
-    if (finalSuggestionConfig.available === "disabled")
-      return;
-    const subscription = copilotkit.subscribe({
-      onAgentsChanged: () => {
-        const agent = copilotkit.getAgent(agentSession == null ? void 0 : agentSession.agentName);
-        if (agent && !agent.isRunning && !result.suggestions.length) {
-          copilotkit.reloadSuggestions(agentSession == null ? void 0 : agentSession.agentName);
-        }
-      }
-    });
-    return () => {
-      subscription.unsubscribe();
-    };
-  }, []);
-  return result;
-}
-
-// ../../node_modules/.pnpm/@ag-ui+core@0.0.42-alpha.1/node_modules/@ag-ui/core/dist/index.mjs
+// ../../node_modules/.pnpm/@ag-ui+core@0.0.42/node_modules/@ag-ui/core/dist/index.mjs
 var import_zod = require("zod");
 var import_zod2 = require("zod");
 var FunctionCallSchema = import_zod.z.object({
@@ -893,10 +908,12 @@ var EventSchemas = import_zod2.z.discriminatedUnion("type", [
 ]);
 
 // src/hooks/use-coagent-state-render-bridge.tsx
-var import_react10 = require("@copilotkitnext/react");
-var import_react11 = require("react");
+var import_react11 = require("@copilotkitnext/react");
+var import_react12 = require("react");
 var import_shared3 = require("@copilotkit/shared");
 function getStateWithoutConstantKeys(state) {
+  if (!state)
+    return {};
   const _a = state, { messages, tools, copilotkit } = _a, stateWithoutConstantKeys = __objRest(_a, ["messages", "tools", "copilotkit"]);
   return stateWithoutConstantKeys;
 }
@@ -919,11 +936,11 @@ function useCoagentStateRenderBridge(agentId, props) {
   var _a;
   const { stateSnapshot, messageIndexInRun, message } = props;
   const { coAgentStateRenders, claimsRef } = useCoAgentStateRenders();
-  const { agent } = (0, import_react10.useAgent)({ agentId });
-  const [nodeName, setNodeName] = (0, import_react11.useState)(void 0);
+  const { agent } = (0, import_react11.useAgent)({ agentId });
+  const [nodeName, setNodeName] = (0, import_react12.useState)(void 0);
   const runId = (_a = props.runId) != null ? _a : message.runId;
   const effectiveRunId = runId || "pending";
-  (0, import_react11.useEffect)(() => {
+  (0, import_react12.useEffect)(() => {
     if (!agent)
       return;
     const subscriber = {
@@ -943,10 +960,7 @@ function useCoagentStateRenderBridge(agentId, props) {
       unsubscribe();
     };
   }, [agentId, nodeName]);
-  if (messageIndexInRun !== 0) {
-    return null;
-  }
-  const getStateRender = (0, import_react11.useCallback)(
+  const getStateRender = (0, import_react12.useCallback)(
     (messageId) => {
       return Object.entries(coAgentStateRenders).find(([stateRenderId, stateRender]) => {
         if (claimsRef.current[messageId]) {
@@ -973,7 +987,7 @@ function useCoagentStateRenderBridge(agentId, props) {
       return canRender;
     }
     const renderClaimedByOtherMessage = Object.values(claimsRef.current).find(
-      (c) => c.stateRenderId === stateRenderId && (0, import_shared3.dataToUUID)(JSON.stringify(getStateWithoutConstantKeys(c.stateSnapshot))) === (0, import_shared3.dataToUUID)(JSON.stringify(getStateWithoutConstantKeys(renderSnapshot)))
+      (c) => c.stateRenderId === stateRenderId && (0, import_shared3.dataToUUID)(getStateWithoutConstantKeys(c.stateSnapshot)) === (0, import_shared3.dataToUUID)(getStateWithoutConstantKeys(renderSnapshot))
     );
     if (renderClaimedByOtherMessage) {
       if (renderSnapshot && renderClaimedByOtherMessage.stateSnapshot && !areStatesEquals(renderClaimedByOtherMessage.stateSnapshot, renderSnapshot)) {
@@ -988,8 +1002,11 @@ function useCoagentStateRenderBridge(agentId, props) {
     claimsRef.current[messageId] = { stateRenderId, runId: runId2 };
     return true;
   };
-  return (0, import_react11.useMemo)(() => {
+  return (0, import_react12.useMemo)(() => {
     var _a2, _b, _c;
+    if (messageIndexInRun !== 0) {
+      return null;
+    }
     const [stateRenderId, stateRender] = (_a2 = getStateRender(message.id)) != null ? _a2 : [];
     if (!stateRender || !stateRenderId) {
       return null;
@@ -1036,7 +1053,8 @@ function useCoagentStateRenderBridge(agentId, props) {
     agent == null ? void 0 : agent.isRunning,
     nodeName,
     effectiveRunId,
-    message.id
+    message.id,
+    messageIndexInRun
   ]);
 }
 function CoAgentStateRenderBridge(props) {
@@ -1044,35 +1062,17 @@ function CoAgentStateRenderBridge(props) {
 }
 
 // src/hooks/use-copilot-chat_internal.ts
-function useConfigureSuggestions2(suggestions) {
-  let suggestionsConfig;
-  if (Array.isArray(suggestions)) {
-    suggestionsConfig = {
-      suggestions,
-      available: "always"
-    };
-  } else if (suggestions === "auto") {
-    suggestionsConfig = {
-      available: suggestions === "auto" ? "always" : "disabled",
-      instructions: "Suggest what the user could say next. Provide clear, highly relevant suggestions. Do not literally suggest function calls."
-    };
-  } else {
-    suggestionsConfig = { available: "disabled" };
-  }
-  useConfigureChatSuggestions(suggestionsConfig);
-}
 function useCopilotChatInternal({
   suggestions
 } = {}) {
-  var _a, _b, _c, _d;
-  const { copilotkit } = (0, import_react13.useCopilotKit)();
+  var _a, _b, _c;
+  const { copilotkit } = (0, import_react14.useCopilotKit)();
   const { threadId, agentSession } = useCopilotContext();
-  const existingConfig = (0, import_react13.useCopilotChatConfiguration)();
-  const [agentAvailable, setAgentAvailable] = (0, import_react12.useState)(false);
-  useConfigureSuggestions2(suggestions);
-  const resolvedAgentId = (_b = (_a = agentSession == null ? void 0 : agentSession.agentName) != null ? _a : existingConfig == null ? void 0 : existingConfig.agentId) != null ? _b : "default";
-  const { agent } = (0, import_react13.useAgent)({ agentId: resolvedAgentId });
-  (0, import_react12.useEffect)(() => {
+  const existingConfig = (0, import_react14.useCopilotChatConfiguration)();
+  const [agentAvailable, setAgentAvailable] = (0, import_react13.useState)(false);
+  const resolvedAgentId = (_a = existingConfig == null ? void 0 : existingConfig.agentId) != null ? _a : "default";
+  const { agent } = (0, import_react14.useAgent)({ agentId: resolvedAgentId });
+  (0, import_react13.useEffect)(() => {
     const connect = (agent2) => __async(this, null, function* () {
       setAgentAvailable(false);
       try {
@@ -1097,7 +1097,7 @@ function useCopilotChatInternal({
     agent == null ? void 0 : agent.setMessages([]);
     agent == null ? void 0 : agent.setState(null);
   };
-  const deleteMessage = (0, import_react12.useCallback)(
+  const deleteMessage = (0, import_react13.useCallback)(
     (messageId) => {
       var _a2;
       const filteredMessages = ((_a2 = agent == null ? void 0 : agent.messages) != null ? _a2 : []).filter(
@@ -1108,13 +1108,13 @@ function useCopilotChatInternal({
     [agent == null ? void 0 : agent.setMessages, agent == null ? void 0 : agent.messages]
   );
   const latestDelete = useUpdatedRef(deleteMessage);
-  const latestDeleteFunc = (0, import_react12.useCallback)(
+  const latestDeleteFunc = (0, import_react13.useCallback)(
     (messageId) => {
       return latestDelete.current(messageId);
     },
     [latestDelete]
   );
-  const currentSuggestions = (0, import_react13.useSuggestions)({ agentId: resolvedAgentId });
+  const currentSuggestions = (0, import_react14.useSuggestions)({ agentId: resolvedAgentId });
   const reload = useAsyncCallback(
     (reloadMessageId) => __async(this, null, function* () {
       var _a2;
@@ -1181,16 +1181,15 @@ function useCopilotChatInternal({
     }),
     [latestSendMessageFunc]
   );
-  const latestSetMessages = useUpdatedRef(agent == null ? void 0 : agent.setMessages);
-  const latestSetMessagesFunc = (0, import_react12.useCallback)(
+  const latestSetMessagesFunc = (0, import_react13.useCallback)(
     (messages) => {
       var _a2, _b2;
       if (messages.every((message) => message instanceof import_runtime_client_gql2.Message)) {
-        return (_a2 = latestSetMessages.current) == null ? void 0 : _a2.call(latestSetMessages, (0, import_runtime_client_gql2.gqlToAGUI)(messages));
+        return (_a2 = agent == null ? void 0 : agent.setMessages) == null ? void 0 : _a2.call(agent, (0, import_runtime_client_gql2.gqlToAGUI)(messages));
       }
-      return (_b2 = latestSetMessages.current) == null ? void 0 : _b2.call(latestSetMessages, messages);
+      return (_b2 = agent == null ? void 0 : agent.setMessages) == null ? void 0 : _b2.call(agent, messages);
     },
-    [latestSetMessages, agent]
+    [agent == null ? void 0 : agent.setMessages, agent]
   );
   const latestReload = useUpdatedRef(reload);
   const latestReloadFunc = useAsyncCallback(
@@ -1199,25 +1198,24 @@ function useCopilotChatInternal({
     }),
     [latestReload]
   );
-  const latestStop = useUpdatedRef(agent == null ? void 0 : agent.abortRun);
-  const latestStopFunc = (0, import_react12.useCallback)(() => {
+  const latestStopFunc = (0, import_react13.useCallback)(() => {
     var _a2;
-    return (_a2 = latestStop.current) == null ? void 0 : _a2.call(latestStop);
-  }, [latestStop]);
+    return (_a2 = agent == null ? void 0 : agent.abortRun) == null ? void 0 : _a2.call(agent);
+  }, [agent == null ? void 0 : agent.abortRun]);
   const latestReset = useUpdatedRef(reset);
-  const latestResetFunc = (0, import_react12.useCallback)(() => {
+  const latestResetFunc = (0, import_react13.useCallback)(() => {
     return latestReset.current();
   }, [latestReset]);
   const lazyToolRendered = useLazyToolRenderer();
-  const renderCustomMessage = (0, import_react13.useRenderCustomMessages)();
+  const renderCustomMessage = (0, import_react14.useRenderCustomMessages)();
   const legacyCustomMessageRenderer = useLegacyCoagentRenderer({
     copilotkit,
     agent,
     agentId: resolvedAgentId,
-    threadId: (_c = existingConfig == null ? void 0 : existingConfig.threadId) != null ? _c : threadId
+    threadId: (_b = existingConfig == null ? void 0 : existingConfig.threadId) != null ? _b : threadId
   });
-  const allMessages = (_d = agent == null ? void 0 : agent.messages) != null ? _d : [];
-  const resolvedMessages = (0, import_react12.useMemo)(() => {
+  const allMessages = (_c = agent == null ? void 0 : agent.messages) != null ? _c : [];
+  const resolvedMessages = (0, import_react13.useMemo)(() => {
     let processedMessages = allMessages.map((message) => {
       if (message.role !== "assistant") {
         return message;
@@ -1245,33 +1243,24 @@ function useCopilotChatInternal({
       return message;
     });
     const hasAssistantMessages = processedMessages.some((msg) => msg.role === "assistant");
-    if (legacyCustomMessageRenderer && !hasAssistantMessages) {
-      const placeholderId = `coagent-state-render-${resolvedAgentId}`;
-      const placeholderMessage = {
-        id: placeholderId,
-        role: "assistant",
-        content: "",
-        name: "coagent-state-render"
-      };
-      processedMessages = [
-        ...processedMessages,
-        __spreadProps(__spreadValues({}, placeholderMessage), {
-          generativeUI: () => legacyCustomMessageRenderer({
-            message: placeholderMessage,
-            position: "before"
-          })
-        })
-      ];
-    }
     return processedMessages;
   }, [
     agent == null ? void 0 : agent.messages,
     lazyToolRendered,
     allMessages,
     renderCustomMessage,
-    legacyCustomMessageRenderer,
+    // legacyCustomMessageRenderer,
     resolvedAgentId
   ]);
+  const renderedSuggestions = (0, import_react13.useMemo)(() => {
+    if (Array.isArray(suggestions)) {
+      return {
+        suggestions: suggestions.map((s) => __spreadProps(__spreadValues({}, s), { isLoading: false })),
+        isLoading: false
+      };
+    }
+    return currentSuggestions;
+  }, [suggestions, currentSuggestions]);
   return {
     messages: resolvedMessages,
     sendMessage: latestSendMessageFunc,
@@ -1281,25 +1270,25 @@ function useCopilotChatInternal({
     stopGeneration: latestStopFunc,
     reset: latestResetFunc,
     deleteMessage: latestDeleteFunc,
-    isAvailable: !agentAvailable,
+    isAvailable: agentAvailable,
     isLoading: Boolean(agent == null ? void 0 : agent.isRunning),
     // mcpServers,
     // setMcpServers,
-    suggestions: currentSuggestions.suggestions,
+    suggestions: renderedSuggestions.suggestions,
     setSuggestions: (suggestions2) => copilotkit.addSuggestionsConfig({ suggestions: suggestions2 }),
     generateSuggestions: () => __async(this, null, function* () {
       return copilotkit.reloadSuggestions(resolvedAgentId);
     }),
     resetSuggestions: () => copilotkit.clearSuggestions(resolvedAgentId),
-    isLoadingSuggestions: currentSuggestions.isLoading,
+    isLoadingSuggestions: renderedSuggestions.isLoading,
     interrupt,
     agent,
     threadId
   };
 }
 function useUpdatedRef(value) {
-  const ref = (0, import_react12.useRef)(value);
-  (0, import_react12.useEffect)(() => {
+  const ref = (0, import_react13.useRef)(value);
+  (0, import_react13.useEffect)(() => {
     ref.current = value;
   }, [value]);
   return ref;
@@ -1310,20 +1299,19 @@ function useLegacyCoagentRenderer({
   agentId,
   threadId
 }) {
-  return (0, import_react12.useMemo)(() => {
+  return (0, import_react13.useMemo)(() => {
     if (!copilotkit || !agent) {
       return null;
     }
     return ({ message, position }) => {
       var _a;
       const effectiveThreadId = (_a = threadId != null ? threadId : agent.threadId) != null ? _a : "default";
-      const existingRunId = copilotkit.getRunIdForMessage(
-        agentId,
-        effectiveThreadId,
-        message.id
-      );
+      const existingRunId = copilotkit.getRunIdForMessage(agentId, effectiveThreadId, message.id);
       const runId = existingRunId || `pending:${message.id}`;
-      const messageIndex = Math.max(agent.messages.findIndex((msg) => msg.id === message.id), 0);
+      const messageIndex = Math.max(
+        agent.messages.findIndex((msg) => msg.id === message.id),
+        0
+      );
       const bridgeProps = {
         message,
         position,
@@ -1334,7 +1322,7 @@ function useLegacyCoagentRenderer({
         agentId,
         stateSnapshot: message.state
       };
-      return (0, import_react12.createElement)(CoAgentStateRenderBridge, bridgeProps);
+      return (0, import_react13.createElement)(CoAgentStateRenderBridge, bridgeProps);
     };
   }, [agent, agentId, copilotkit, threadId]);
 }
@@ -1406,7 +1394,7 @@ function useCopilotChatHeadless_c(options = {}) {
   const { copilotApiConfig, setBannerError } = useCopilotContext();
   const hasPublicApiKey = Boolean(copilotApiConfig.publicApiKey);
   const internalResult = useCopilotChatInternal(options);
-  (0, import_react14.useEffect)(() => {
+  (0, import_react15.useEffect)(() => {
     if (!hasPublicApiKey) {
       setBannerError(
         new import_shared4.CopilotKitError({