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

package/dist/hooks/use-copilot-chat_internal.js

@@ -83,7 +83,7 @@ __export(use_copilot_chat_internal_exports, {
   useCopilotChatInternal: () => useCopilotChatInternal
 });
 module.exports = __toCommonJS(use_copilot_chat_internal_exports);
-var import_react12 = 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: () => {
   },
@@ -332,24 +332,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") {
@@ -358,12 +394,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;
         }
       }
     };
@@ -372,7 +417,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: {
@@ -381,15 +426,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) => {
@@ -397,7 +442,9 @@ function useLangGraphInterruptRender(agent) {
         return true;
       return action.enabled({
         eventValue: currentQueuedEvent.event.value,
-        agentMetadata: agentSession
+        agentMetadata: __spreadProps(__spreadValues({}, agentSession), {
+          nodeName
+        })
       });
     });
     if (!matchingAction)
@@ -415,7 +462,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,
@@ -425,14 +472,14 @@ function useLangGraphInterruptRender(agent) {
 }
 
 // src/hooks/use-copilot-chat_internal.ts
-var import_react13 = require("@copilotkitnext/react");
+var import_react16 = 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))
@@ -453,34 +500,36 @@ function useLazyToolRenderer() {
 }
 
 // src/hooks/use-configure-chat-suggestions.tsx
-var import_react8 = require("@copilotkitnext/react");
-var import_react9 = require("react");
+var import_react11 = require("@copilotkitnext/react");
+var import_react12 = require("react");
 function useConfigureChatSuggestions(config, dependencies = []) {
-  const { agentSession } = useCopilotContext();
-  const { copilotkit } = (0, import_react8.useCopilotKit)();
+  var _a;
+  const existingConfig = (0, import_react11.useCopilotChatConfiguration)();
+  const resolvedAgentId = (_a = existingConfig == null ? void 0 : existingConfig.agentId) != null ? _a : "default";
+  const { copilotkit } = (0, import_react11.useCopilotKit)();
   const available = config.available === "enabled" ? "always" : config.available;
   const finalSuggestionConfig = __spreadProps(__spreadValues({}, config), {
     available,
-    consumerAgentId: agentSession == null ? void 0 : agentSession.agentName
+    consumerAgentId: resolvedAgentId
     // 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)(() => {
+  (0, import_react11.useConfigureSuggestions)(finalSuggestionConfig, dependencies);
+  const result = (0, import_react11.useSuggestions)({ agentId: resolvedAgentId });
+  (0, import_react12.useEffect)(() => {
     if (finalSuggestionConfig.available === "disabled")
       return;
     const subscription = copilotkit.subscribe({
       onAgentsChanged: () => {
-        const agent = copilotkit.getAgent(agentSession == null ? void 0 : agentSession.agentName);
+        const agent = copilotkit.getAgent(resolvedAgentId);
         if (agent && !agent.isRunning && !result.suggestions.length) {
-          copilotkit.reloadSuggestions(agentSession == null ? void 0 : agentSession.agentName);
+          copilotkit.reloadSuggestions(resolvedAgentId);
         }
       }
     });
     return () => {
       subscription.unsubscribe();
     };
-  }, []);
+  }, [resolvedAgentId]);
   return result;
 }
 
@@ -890,10 +939,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_react13 = require("@copilotkitnext/react");
+var import_react14 = 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;
 }
@@ -916,11 +967,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_react13.useAgent)({ agentId });
+  const [nodeName, setNodeName] = (0, import_react14.useState)(void 0);
   const runId = (_a = props.runId) != null ? _a : message.runId;
   const effectiveRunId = runId || "pending";
-  (0, import_react11.useEffect)(() => {
+  (0, import_react14.useEffect)(() => {
     if (!agent)
       return;
     const subscriber = {
@@ -940,10 +991,7 @@ function useCoagentStateRenderBridge(agentId, props) {
       unsubscribe();
     };
   }, [agentId, nodeName]);
-  if (messageIndexInRun !== 0) {
-    return null;
-  }
-  const getStateRender = (0, import_react11.useCallback)(
+  const getStateRender = (0, import_react14.useCallback)(
     (messageId) => {
       return Object.entries(coAgentStateRenders).find(([stateRenderId, stateRender]) => {
         if (claimsRef.current[messageId]) {
@@ -970,7 +1018,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)) {
@@ -985,8 +1033,11 @@ function useCoagentStateRenderBridge(agentId, props) {
     claimsRef.current[messageId] = { stateRenderId, runId: runId2 };
     return true;
   };
-  return (0, import_react11.useMemo)(() => {
+  return (0, import_react14.useMemo)(() => {
     var _a2, _b, _c;
+    if (messageIndexInRun !== 0) {
+      return null;
+    }
     const [stateRenderId, stateRender] = (_a2 = getStateRender(message.id)) != null ? _a2 : [];
     if (!stateRender || !stateRenderId) {
       return null;
@@ -1033,7 +1084,8 @@ function useCoagentStateRenderBridge(agentId, props) {
     agent == null ? void 0 : agent.isRunning,
     nodeName,
     effectiveRunId,
-    message.id
+    message.id,
+    messageIndexInRun
   ]);
 }
 function CoAgentStateRenderBridge(props) {
@@ -1061,15 +1113,15 @@ function useConfigureSuggestions2(suggestions) {
 function useCopilotChatInternal({
   suggestions
 } = {}) {
-  var _a, _b, _c, _d;
-  const { copilotkit } = (0, import_react13.useCopilotKit)();
+  var _a, _b, _c;
+  const { copilotkit } = (0, import_react16.useCopilotKit)();
   const { threadId, agentSession } = useCopilotContext();
-  const existingConfig = (0, import_react13.useCopilotChatConfiguration)();
-  const [agentAvailable, setAgentAvailable] = (0, import_react12.useState)(false);
+  const existingConfig = (0, import_react16.useCopilotChatConfiguration)();
+  const [agentAvailable, setAgentAvailable] = (0, import_react15.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 resolvedAgentId = (_a = existingConfig == null ? void 0 : existingConfig.agentId) != null ? _a : "default";
+  const { agent } = (0, import_react16.useAgent)({ agentId: resolvedAgentId });
+  (0, import_react15.useEffect)(() => {
     const connect = (agent2) => __async(this, null, function* () {
       setAgentAvailable(false);
       try {
@@ -1094,7 +1146,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_react15.useCallback)(
     (messageId) => {
       var _a2;
       const filteredMessages = ((_a2 = agent == null ? void 0 : agent.messages) != null ? _a2 : []).filter(
@@ -1105,13 +1157,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_react15.useCallback)(
     (messageId) => {
       return latestDelete.current(messageId);
     },
     [latestDelete]
   );
-  const currentSuggestions = (0, import_react13.useSuggestions)({ agentId: resolvedAgentId });
+  const currentSuggestions = (0, import_react16.useSuggestions)({ agentId: resolvedAgentId });
   const reload = useAsyncCallback(
     (reloadMessageId) => __async(this, null, function* () {
       var _a2;
@@ -1179,7 +1231,7 @@ function useCopilotChatInternal({
     [latestSendMessageFunc]
   );
   const latestSetMessages = useUpdatedRef(agent == null ? void 0 : agent.setMessages);
-  const latestSetMessagesFunc = (0, import_react12.useCallback)(
+  const latestSetMessagesFunc = (0, import_react15.useCallback)(
     (messages) => {
       var _a2, _b2;
       if (messages.every((message) => message instanceof import_runtime_client_gql2.Message)) {
@@ -1197,24 +1249,24 @@ function useCopilotChatInternal({
     [latestReload]
   );
   const latestStop = useUpdatedRef(agent == null ? void 0 : agent.abortRun);
-  const latestStopFunc = (0, import_react12.useCallback)(() => {
+  const latestStopFunc = (0, import_react15.useCallback)(() => {
     var _a2;
     return (_a2 = latestStop.current) == null ? void 0 : _a2.call(latestStop);
   }, [latestStop]);
   const latestReset = useUpdatedRef(reset);
-  const latestResetFunc = (0, import_react12.useCallback)(() => {
+  const latestResetFunc = (0, import_react15.useCallback)(() => {
     return latestReset.current();
   }, [latestReset]);
   const lazyToolRendered = useLazyToolRenderer();
-  const renderCustomMessage = (0, import_react13.useRenderCustomMessages)();
+  const renderCustomMessage = (0, import_react16.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_react15.useMemo)(() => {
     let processedMessages = allMessages.map((message) => {
       if (message.role !== "assistant") {
         return message;
@@ -1242,31 +1294,13 @@ 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
   ]);
   return {
@@ -1278,7 +1312,7 @@ function useCopilotChatInternal({
     stopGeneration: latestStopFunc,
     reset: latestResetFunc,
     deleteMessage: latestDeleteFunc,
-    isAvailable: !agentAvailable,
+    isAvailable: agentAvailable,
     isLoading: Boolean(agent == null ? void 0 : agent.isRunning),
     // mcpServers,
     // setMcpServers,
@@ -1295,8 +1329,8 @@ function useCopilotChatInternal({
   };
 }
 function useUpdatedRef(value) {
-  const ref = (0, import_react12.useRef)(value);
-  (0, import_react12.useEffect)(() => {
+  const ref = (0, import_react15.useRef)(value);
+  (0, import_react15.useEffect)(() => {
     ref.current = value;
   }, [value]);
   return ref;
@@ -1307,20 +1341,19 @@ function useLegacyCoagentRenderer({
   agentId,
   threadId
 }) {
-  return (0, import_react12.useMemo)(() => {
+  return (0, import_react15.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,
@@ -1331,7 +1364,7 @@ function useLegacyCoagentRenderer({
         agentId,
         stateSnapshot: message.state
       };
-      return (0, import_react12.createElement)(CoAgentStateRenderBridge, bridgeProps);
+      return (0, import_react15.createElement)(CoAgentStateRenderBridge, bridgeProps);
     };
   }, [agent, agentId, copilotkit, threadId]);
 }