@copilotkit/react-core 1.3.6 → 1.3.7

package/dist/hooks/index.js

@@ -89,7 +89,7 @@ __export(hooks_exports, {
 module.exports = __toCommonJS(hooks_exports);
 
 // src/hooks/use-copilot-chat.ts
-var import_react4 = require("react");
+var import_react5 = require("react");
 
 // src/context/copilot-context.tsx
 var import_react = __toESM(require("react"));
@@ -111,8 +111,6 @@ var emptyCopilotContext = {
   },
   getFunctionCallHandler: () => returnAndThrowInDebug(() => __async(void 0, null, function* () {
   })),
-  messages: [],
-  setMessages: () => returnAndThrowInDebug([]),
   isLoading: false,
   setIsLoading: () => returnAndThrowInDebug(false),
   chatInstructions: "",
@@ -284,7 +282,7 @@ function useChat(options) {
         } else {
           for (const message of messages2) {
             newMessages.push(message);
-            if (message instanceof import_runtime_client_gql.ActionExecutionMessage && message.status.code !== import_runtime_client_gql.MessageStatusCode.Pending && message.scope === "client" && onFunctionCall) {
+            if (message.isActionExecutionMessage() && message.status.code !== import_runtime_client_gql.MessageStatusCode.Pending && message.scope === "client" && onFunctionCall) {
               if (!(message.id in actionResults)) {
                 if (guardrailsEnabled && value.generateCopilotResponse.status === void 0) {
                   break;
@@ -314,7 +312,7 @@ function useChat(options) {
                 })
               );
             }
-            if (message instanceof import_runtime_client_gql.AgentStateMessage && !message.active && !executedCoAgentStateRenders.includes(message.id) && onCoAgentStateRender) {
+            if (message.isAgentStateMessage() && !message.active && !executedCoAgentStateRenders.includes(message.id) && onCoAgentStateRender) {
               if (guardrailsEnabled && value.generateCopilotResponse.status === void 0) {
                 break;
               }
@@ -326,7 +324,7 @@ function useChat(options) {
               executedCoAgentStateRenders.push(message.id);
             }
           }
-          const lastAgentStateMessage = [...messages2].reverse().find((message) => message instanceof import_runtime_client_gql.AgentStateMessage);
+          const lastAgentStateMessage = [...messages2].reverse().find((message) => message.isAgentStateMessage());
           if (lastAgentStateMessage) {
             if (lastAgentStateMessage.running) {
               setCoagentStates((prevAgentStates) => __spreadProps(__spreadValues({}, prevAgentStates), {
@@ -358,7 +356,7 @@ function useChat(options) {
         // if followUp is not explicitly false
         followUp !== false && // if we have client side results
         (Object.values(actionResults).length || // or the last message we received is a result
-        newMessages.length && newMessages[newMessages.length - 1] instanceof import_runtime_client_gql.ResultMessage)
+        newMessages.length && newMessages[newMessages.length - 1].isResultMessage())
       ) {
         yield new Promise((resolve) => setTimeout(resolve, 10));
         return yield runChatCompletionRef.current([...previousMessages, ...newMessages]);
@@ -387,7 +385,7 @@ function useChat(options) {
     }
     let newMessages = [...messages];
     const lastMessage = messages[messages.length - 1];
-    if (lastMessage instanceof import_runtime_client_gql.TextMessage && lastMessage.role === "assistant") {
+    if (lastMessage.isTextMessage() && lastMessage.role === "assistant") {
       newMessages = newMessages.slice(0, -1);
     }
     setMessages(newMessages);
@@ -405,9 +403,28 @@ function useChat(options) {
 }
 
 // src/components/copilot-provider/copilotkit.tsx
-var import_react3 = require("react");
+var import_react4 = require("react");
 var import_react_dom = require("react-dom");
 var import_shared2 = require("@copilotkit/shared");
+
+// src/context/copilot-messages-context.tsx
+var import_react3 = __toESM(require("react"));
+var emptyCopilotContext2 = {
+  messages: [],
+  setMessages: () => []
+};
+var CopilotMessagesContext = import_react3.default.createContext(emptyCopilotContext2);
+function useCopilotMessagesContext() {
+  const context = import_react3.default.useContext(CopilotMessagesContext);
+  if (context === emptyCopilotContext2) {
+    throw new Error(
+      "A messages consuming component was not wrapped with `<CopilotMessages> {...} </CopilotMessages>`"
+    );
+  }
+  return context;
+}
+
+// src/components/copilot-provider/copilotkit.tsx
 var import_jsx_runtime = require("react/jsx-runtime");
 var defaultCopilotContextCategories = ["global"];
 
@@ -422,8 +439,6 @@ function useCopilotChat(_a = {}) {
     getContextString,
     getFunctionCallHandler,
     copilotApiConfig,
-    messages,
-    setMessages,
     isLoading,
     setIsLoading,
     chatInstructions,
@@ -434,14 +449,15 @@ function useCopilotChat(_a = {}) {
     agentSession,
     setAgentSession
   } = useCopilotContext();
+  const { messages, setMessages } = useCopilotMessagesContext();
   const latestGetContextString = useUpdatedRef(getContextString);
-  const deleteMessage = (0, import_react4.useCallback)(
+  const deleteMessage = (0, import_react5.useCallback)(
     (messageId) => {
       setMessages((prev) => prev.filter((message) => message.id !== messageId));
     },
     [setMessages]
   );
-  const makeSystemMessageCallback = (0, import_react4.useCallback)(() => {
+  const makeSystemMessageCallback = (0, import_react5.useCallback)(() => {
     const systemMessageMaker = makeSystemMessage || defaultSystemMessage;
     const contextString = latestGetContextString.current([], defaultCopilotContextCategories);
     return new import_runtime_client_gql2.TextMessage({
@@ -449,7 +465,7 @@ function useCopilotChat(_a = {}) {
       role: import_runtime_client_gql2.Role.System
     });
   }, [getContextString, makeSystemMessage, chatInstructions]);
-  const onCoAgentStateRender = (0, import_react4.useCallback)(
+  const onCoAgentStateRender = (0, import_react5.useCallback)(
     (args) => __async(this, null, function* () {
       var _a2;
       const { name, nodeName, state } = args;
@@ -484,29 +500,29 @@ function useCopilotChat(_a = {}) {
     setAgentSession
   }));
   const latestAppend = useUpdatedRef(append);
-  const latestAppendFunc = (0, import_react4.useCallback)(
+  const latestAppendFunc = (0, import_react5.useCallback)(
     (message) => {
       return latestAppend.current(message);
     },
     [latestAppend]
   );
   const latestReload = useUpdatedRef(reload);
-  const latestReloadFunc = (0, import_react4.useCallback)(() => {
+  const latestReloadFunc = (0, import_react5.useCallback)(() => {
     return latestReload.current();
   }, [latestReload]);
   const latestStop = useUpdatedRef(stop);
-  const latestStopFunc = (0, import_react4.useCallback)(() => {
+  const latestStopFunc = (0, import_react5.useCallback)(() => {
     return latestStop.current();
   }, [latestStop]);
   const latestDelete = useUpdatedRef(deleteMessage);
-  const latestDeleteFunc = (0, import_react4.useCallback)(
+  const latestDeleteFunc = (0, import_react5.useCallback)(
     (messageId) => {
       return latestDelete.current(messageId);
     },
     [latestDelete]
   );
   const latestSetMessages = useUpdatedRef(setMessages);
-  const latestSetMessagesFunc = (0, import_react4.useCallback)(
+  const latestSetMessagesFunc = (0, import_react5.useCallback)(
     (messages2) => {
       return latestSetMessages.current(messages2);
     },
@@ -523,8 +539,8 @@ function useCopilotChat(_a = {}) {
   };
 }
 function useUpdatedRef(value) {
-  const ref = (0, import_react4.useRef)(value);
-  (0, import_react4.useEffect)(() => {
+  const ref = (0, import_react5.useRef)(value);
+  (0, import_react5.useEffect)(() => {
     ref.current = value;
   }, [value]);
   return ref;
@@ -554,12 +570,12 @@ ${additionalInstructions}` : "");
 }
 
 // src/hooks/use-copilot-action.ts
-var import_react5 = require("react");
+var import_react6 = require("react");
 var import_shared3 = require("@copilotkit/shared");
 function useCopilotAction(action, dependencies) {
   const { setAction, removeAction, actions, chatComponentsCache } = useCopilotContext();
-  const idRef = (0, import_react5.useRef)((0, import_shared3.randomId)());
-  const renderAndWaitRef = (0, import_react5.useRef)(null);
+  const idRef = (0, import_react6.useRef)((0, import_shared3.randomId)());
+  const renderAndWaitRef = (0, import_react6.useRef)(null);
   action = __spreadValues({}, action);
   if (action.renderAndWait) {
     const renderAndWait = action.renderAndWait;
@@ -594,7 +610,7 @@ function useCopilotAction(action, dependencies) {
       }
     }
   }
-  (0, import_react5.useEffect)(() => {
+  (0, import_react6.useEffect)(() => {
     setAction(idRef.current, action);
     if (chatComponentsCache.current !== null && action.render !== void 0) {
       chatComponentsCache.current.actions[action.name] = action.render;
@@ -619,7 +635,7 @@ function useCopilotAction(action, dependencies) {
 }
 
 // src/hooks/use-coagent-state-render.ts
-var import_react6 = require("react");
+var import_react7 = require("react");
 var import_shared4 = require("@copilotkit/shared");
 function useCoAgentStateRender(action, dependencies) {
   const {
@@ -627,8 +643,8 @@ function useCoAgentStateRender(action, dependencies) {
     removeCoAgentStateRender,
     coAgentStateRenders,
     chatComponentsCache
-  } = (0, import_react6.useContext)(CopilotContext);
-  const idRef = (0, import_react6.useRef)((0, import_shared4.randomId)());
+  } = (0, import_react7.useContext)(CopilotContext);
+  const idRef = (0, import_react7.useRef)((0, import_shared4.randomId)());
   const key = `${action.name}-${action.nodeName || "global"}`;
   if (dependencies === void 0) {
     if (coAgentStateRenders[idRef.current]) {
@@ -640,7 +656,7 @@ function useCoAgentStateRender(action, dependencies) {
       }
     }
   }
-  (0, import_react6.useEffect)(() => {
+  (0, import_react7.useEffect)(() => {
     setCoAgentStateRender(idRef.current, action);
     if (chatComponentsCache.current !== null && action.render !== void 0) {
       chatComponentsCache.current.coAgentStateRenders[key] = action.render;
@@ -660,11 +676,11 @@ function useCoAgentStateRender(action, dependencies) {
 }
 
 // src/hooks/use-make-copilot-document-readable.ts
-var import_react7 = require("react");
+var import_react8 = require("react");
 function useMakeCopilotDocumentReadable(document, categories, dependencies = []) {
   const { addDocumentContext, removeDocumentContext } = useCopilotContext();
-  const idRef = (0, import_react7.useRef)();
-  (0, import_react7.useEffect)(() => {
+  const idRef = (0, import_react8.useRef)();
+  (0, import_react8.useEffect)(() => {
     const id = addDocumentContext(document, categories);
     idRef.current = id;
     return () => {
@@ -675,16 +691,16 @@ function useMakeCopilotDocumentReadable(document, categories, dependencies = [])
 }
 
 // src/hooks/use-copilot-readable.ts
-var import_react8 = require("react");
+var import_react9 = require("react");
 function convertToJSON(description, value) {
   return `${description}: ${typeof value === "string" ? value : JSON.stringify(value)}`;
 }
 function useCopilotReadable({ description, value, parentId, categories, convert }, dependencies) {
   const { addContext, removeContext } = useCopilotContext();
-  const idRef = (0, import_react8.useRef)();
+  const idRef = (0, import_react9.useRef)();
   convert = convert || convertToJSON;
   const information = convert(description, value);
-  (0, import_react8.useEffect)(() => {
+  (0, import_react9.useEffect)(() => {
     const id = addContext(information, parentId, categories);
     idRef.current = id;
     return () => {
@@ -695,8 +711,7 @@ function useCopilotReadable({ description, value, parentId, categories, convert
 }
 
 // src/hooks/use-coagent.ts
-var import_react9 = require("react");
-var import_runtime_client_gql3 = require("@copilotkit/runtime-client-gql");
+var import_react10 = require("react");
 function useCoAgent(options) {
   const isExternalStateManagement = (options2) => {
     return "state" in options2 && "setState" in options2;
@@ -705,7 +720,9 @@ function useCoAgent(options) {
   const isInternalStateManagementWithInitial = (options2) => {
     return "initialState" in options2;
   };
-  const context = useCopilotContext();
+  const generalContext = useCopilotContext();
+  const messagesContext = useCopilotMessagesContext();
+  const context = __spreadValues(__spreadValues({}, generalContext), messagesContext);
   const { coagentStates, setCoagentStates } = context;
   const { appendMessage } = useCopilotChat();
   const getCoagentState = (coagentStates2, name2) => {
@@ -736,7 +753,7 @@ function useCoAgent(options) {
   };
   const coagentState = getCoagentState(coagentStates, name);
   const state = isExternalStateManagement(options) ? options.state : coagentState.state;
-  (0, import_react9.useEffect)(() => {
+  (0, import_react10.useEffect)(() => {
     if (isExternalStateManagement(options)) {
       setState(options.state);
     } else if (coagentStates[name] === void 0) {
@@ -786,7 +803,7 @@ function runAgent(name, context, appendMessage, hint) {
     let previousState = null;
     for (let i = context.messages.length - 1; i >= 0; i--) {
       const message = context.messages[i];
-      if (message instanceof import_runtime_client_gql3.AgentStateMessage && message.agentName === name) {
+      if (message.isAgentStateMessage() && message.agentName === name) {
         previousState = message.state;
       }
     }