@copilotkit/react-core 1.1.3-feat-runtime-remote-actions.2 → 1.2.0

package/dist/index.js

@@ -86,8 +86,6 @@ __export(src_exports, {
   CopilotTask: () => CopilotTask,
   defaultCopilotContextCategories: () => defaultCopilotContextCategories,
   extract: () => extract,
-  useCoAgent: () => useCoAgent,
-  useCoAgentAction: () => useCoAgentAction,
   useCopilotAction: () => useCopilotAction,
   useCopilotChat: () => useCopilotChat,
   useCopilotContext: () => useCopilotContext,
@@ -107,12 +105,7 @@ var emptyCopilotContext = {
   },
   removeAction: () => {
   },
-  coagentActions: {},
-  setCoagentAction: () => {
-  },
-  removeCoagentAction: () => {
-  },
-  chatComponentsCache: { current: { actions: {}, coagentActions: {} } },
+  chatComponentsCache: { current: {} },
   getContextString: (documents, categories) => returnAndThrowInDebug(""),
   addContext: () => "",
   removeContext: () => {
@@ -145,13 +138,7 @@ var emptyCopilotContext = {
   },
   removeChatSuggestionConfiguration: () => {
   },
-  showDevConsole: "auto",
-  coagentStates: {},
-  setCoagentStates: () => {
-  },
-  agentSession: null,
-  setAgentSession: () => {
-  }
+  showDevConsole: "auto"
 };
 var CopilotContext = import_react.default.createContext(emptyCopilotContext);
 function useCopilotContext() {
@@ -379,11 +366,7 @@ function CopilotKit(_a) {
   }
   const chatApiEndpoint = props.runtimeUrl || import_shared3.COPILOT_CLOUD_CHAT_URL;
   const [actions, setActions] = (0, import_react4.useState)({});
-  const [coagentActions, setCoagentActions] = (0, import_react4.useState)({});
-  const chatComponentsCache = (0, import_react4.useRef)({
-    actions: {},
-    coagentActions: {}
-  });
+  const chatComponentsCache = (0, import_react4.useRef)({});
   const { addElement, removeElement, printTree } = use_tree_default();
   const [messages, setMessages] = (0, import_react4.useState)([]);
   const [isLoading, setIsLoading] = (0, import_react4.useState)(false);
@@ -407,20 +390,6 @@ function CopilotKit(_a) {
       return newPoints;
     });
   }, []);
-  const setCoagentAction = (0, import_react4.useCallback)((id, action) => {
-    setCoagentActions((prevPoints) => {
-      return __spreadProps(__spreadValues({}, prevPoints), {
-        [id]: action
-      });
-    });
-  }, []);
-  const removeCoagentAction = (0, import_react4.useCallback)((id) => {
-    setCoagentActions((prevPoints) => {
-      const newPoints = __spreadValues({}, prevPoints);
-      delete newPoints[id];
-      return newPoints;
-    });
-  }, []);
   const getContextString = (0, import_react4.useCallback)(
     (documents, categories) => {
       const documentsString = documents.map((document) => {
@@ -511,14 +480,6 @@ ${nonDocumentStrings}`;
       return rest;
     });
   };
-  const [coagentStates, setCoagentStates] = (0, import_react4.useState)({});
-  let initialAgentSession = null;
-  if (props.agent) {
-    initialAgentSession = {
-      agentName: props.agent
-    };
-  }
-  const [agentSession, setAgentSession] = (0, import_react4.useState)(initialAgentSession);
   return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
     CopilotContext.Provider,
     {
@@ -528,9 +489,6 @@ ${nonDocumentStrings}`;
         getFunctionCallHandler,
         setAction,
         removeAction,
-        coagentActions,
-        setCoagentAction,
-        removeCoagentAction,
         getContextString,
         addContext,
         removeContext,
@@ -547,11 +505,7 @@ ${nonDocumentStrings}`;
         removeChatSuggestionConfiguration,
         chatInstructions,
         setChatInstructions,
-        showDevConsole: props.showDevConsole === void 0 ? "auto" : props.showDevConsole,
-        coagentStates,
-        setCoagentStates,
-        agentSession,
-        setAgentSession
+        showDevConsole: props.showDevConsole === void 0 ? "auto" : props.showDevConsole
       },
       children
     }
@@ -569,9 +523,8 @@ function entryPointsToFunctionCallHandler(actions) {
     if (action) {
       yield new Promise((resolve, reject) => {
         (0, import_react_dom.flushSync)(() => __async(this, null, function* () {
-          var _a;
           try {
-            result = yield (_a = action.handler) == null ? void 0 : _a.call(action, args);
+            result = yield action.handler(args);
             resolve();
           } catch (error) {
             reject(error);
@@ -602,12 +555,7 @@ function useChat(options) {
     initialMessages,
     isLoading,
     actions,
-    onFunctionCall,
-    onCoagentAction,
-    setCoagentStates,
-    coagentStates,
-    agentSession,
-    setAgentSession
+    onFunctionCall
   } = options;
   const abortControllerRef = (0, import_react5.useRef)();
   const threadIdRef = (0, import_react5.useRef)(null);
@@ -636,7 +584,7 @@ function useChat(options) {
     const messagesWithContext = [systemMessage, ...initialMessages || [], ...previousMessages];
     const stream = import_runtime_client_gql.CopilotRuntimeClient.asStream(
       runtimeClient.generateCopilotResponse({
-        data: __spreadProps(__spreadValues(__spreadProps(__spreadValues({
+        data: __spreadProps(__spreadValues({
           frontend: {
             actions: actions.map((action) => ({
               name: action.name,
@@ -647,7 +595,7 @@ function useChat(options) {
           },
           threadId: threadIdRef.current,
           runId: runIdRef.current,
-          messages: (0, import_runtime_client_gql.convertMessagesToGqlInput)((0, import_runtime_client_gql.filterAgentStateMessages)(messagesWithContext))
+          messages: (0, import_runtime_client_gql.convertMessagesToGqlInput)(messagesWithContext)
         }, copilotConfig.cloud ? {
           cloud: __spreadValues({}, ((_c = (_b = (_a = copilotConfig.cloud.guardrails) == null ? void 0 : _a.input) == null ? void 0 : _b.restrictToTopic) == null ? void 0 : _c.enabled) ? {
             guardrails: {
@@ -661,13 +609,6 @@ function useChat(options) {
           metadata: {
             requestType: import_runtime_client_gql.CopilotRequestType.Chat
           }
-        }), agentSession ? {
-          agentSession
-        } : {}), {
-          agentStates: Object.values(coagentStates).map((state) => ({
-            agentName: state.name,
-            state: JSON.stringify(state.state)
-          }))
         }),
         properties: copilotConfig.properties,
         signal: (_d = abortControllerRef.current) == null ? void 0 : _d.signal
@@ -675,8 +616,7 @@ function useChat(options) {
     );
     const guardrailsEnabled = ((_g = (_f = (_e = copilotConfig.cloud) == null ? void 0 : _e.guardrails) == null ? void 0 : _f.input) == null ? void 0 : _g.restrictToTopic.enabled) || false;
     const reader = stream.getReader();
-    let actionResults = {};
-    let executedCoagentActions = [];
+    let results = {};
     try {
       while (true) {
         const { done, value } = yield reader.read();
@@ -688,9 +628,7 @@ function useChat(options) {
         }
         threadIdRef.current = value.generateCopilotResponse.threadId || null;
         runIdRef.current = value.generateCopilotResponse.runId || null;
-        const messages2 = (0, import_runtime_client_gql.convertGqlOutputToMessages)(
-          (0, import_runtime_client_gql.filterAdjacentAgentStateMessages)(value.generateCopilotResponse.messages)
-        );
+        const messages2 = (0, import_runtime_client_gql.convertGqlOutputToMessages)(value.generateCopilotResponse.messages);
         if (messages2.length === 0) {
           continue;
         }
@@ -706,7 +644,7 @@ function useChat(options) {
           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.id in actionResults)) {
+              if (!(message.id in results)) {
                 if (guardrailsEnabled && value.generateCopilotResponse.status === void 0) {
                   break;
                 }
@@ -715,50 +653,16 @@ function useChat(options) {
                   name: message.name,
                   args: message.arguments
                 });
-                actionResults[message.id] = result;
+                results[message.id] = result;
               }
               newMessages.push(
                 new import_runtime_client_gql.ResultMessage({
-                  result: import_runtime_client_gql.ResultMessage.encodeResult(actionResults[message.id]),
+                  result: import_runtime_client_gql.ResultMessage.encodeResult(results[message.id]),
                   actionExecutionId: message.id,
                   actionName: message.name
                 })
               );
             }
-            if (message instanceof import_runtime_client_gql.AgentStateMessage && !message.active && !executedCoagentActions.includes(message.id) && onCoagentAction) {
-              if (guardrailsEnabled && value.generateCopilotResponse.status === void 0) {
-                break;
-              }
-              yield onCoagentAction({
-                name: message.agentName,
-                nodeName: message.nodeName,
-                state: message.state
-              });
-              executedCoagentActions.push(message.id);
-            }
-          }
-          const lastAgentStateMessage = [...messages2].reverse().find((message) => message instanceof import_runtime_client_gql.AgentStateMessage);
-          if (lastAgentStateMessage) {
-            if (lastAgentStateMessage.running) {
-              setCoagentStates((prevAgentStates) => __spreadProps(__spreadValues({}, prevAgentStates), {
-                [lastAgentStateMessage.agentName]: {
-                  name: lastAgentStateMessage.agentName,
-                  state: lastAgentStateMessage.state,
-                  running: lastAgentStateMessage.running,
-                  active: lastAgentStateMessage.active,
-                  threadId: lastAgentStateMessage.threadId,
-                  nodeName: lastAgentStateMessage.nodeName,
-                  runId: lastAgentStateMessage.runId
-                }
-              }));
-              setAgentSession({
-                threadId: lastAgentStateMessage.threadId,
-                agentName: lastAgentStateMessage.agentName,
-                nodeName: lastAgentStateMessage.nodeName
-              });
-            } else {
-              setAgentSession(null);
-            }
           }
         }
         if (newMessages.length > 0) {
@@ -767,7 +671,7 @@ function useChat(options) {
       }
       if (
         // if we have client side results
-        Object.values(actionResults).length || // or the last message we received is a result
+        Object.values(results).length || // or the last message we received is a result
         newMessages.length && newMessages[newMessages.length - 1] instanceof import_runtime_client_gql.ResultMessage
       ) {
         yield new Promise((resolve) => setTimeout(resolve, 10));
@@ -829,12 +733,7 @@ function useCopilotChat(_a = {}) {
     isLoading,
     setIsLoading,
     chatInstructions,
-    actions,
-    coagentStates,
-    setCoagentStates,
-    coagentActions,
-    agentSession,
-    setAgentSession
+    actions
   } = useCopilotContext();
   const latestGetContextString = useUpdatedRef(getContextString);
   const deleteMessage = (0, import_react6.useCallback)(
@@ -851,39 +750,16 @@ function useCopilotChat(_a = {}) {
       role: import_runtime_client_gql2.Role.System
     });
   }, [getContextString, makeSystemMessage2, chatInstructions]);
-  const onCoagentAction = (0, import_react6.useCallback)(
-    (args) => __async(this, null, function* () {
-      var _a2;
-      const { name, nodeName, state } = args;
-      let action = Object.values(coagentActions).find(
-        (action2) => action2.name === name && action2.nodeName === nodeName
-      );
-      if (!action) {
-        action = Object.values(coagentActions).find(
-          (action2) => action2.name === name && !action2.nodeName
-        );
-      }
-      if (action) {
-        yield (_a2 = action.handler) == null ? void 0 : _a2.call(action, { state, nodeName });
-      }
-    }),
-    [coagentActions]
-  );
   const { append, reload, stop } = useChat(__spreadProps(__spreadValues({}, options), {
     actions: Object.values(actions),
     copilotConfig: copilotApiConfig,
     initialMessages: options.initialMessages || [],
     onFunctionCall: getFunctionCallHandler(),
-    onCoagentAction,
     messages,
     setMessages,
     makeSystemMessageCallback,
     isLoading,
-    setIsLoading,
-    coagentStates,
-    setCoagentStates,
-    agentSession,
-    setAgentSession
+    setIsLoading
   }));
   const latestAppend = useUpdatedRef(append);
   const latestAppendFunc = (0, import_react6.useCallback)(
@@ -966,7 +842,7 @@ function useCopilotAction(action, dependencies) {
       actions[idRef.current].handler = action.handler;
       if (typeof action.render === "function") {
         if (chatComponentsCache.current !== null) {
-          chatComponentsCache.current.actions[action.name] = action.render;
+          chatComponentsCache.current[action.name] = action.render;
         }
       }
     }
@@ -977,7 +853,7 @@ function useCopilotAction(action, dependencies) {
     }
     setAction(idRef.current, action);
     if (chatComponentsCache.current !== null && action.render !== void 0) {
-      chatComponentsCache.current.actions[action.name] = action.render;
+      chatComponentsCache.current[action.name] = action.render;
     }
     return () => {
       removeAction(idRef.current);
@@ -998,48 +874,12 @@ function useCopilotAction(action, dependencies) {
   ]);
 }
 
-// src/hooks/use-coagent-action.ts
-var import_react8 = require("react");
-var import_shared6 = require("@copilotkit/shared");
-function useCoAgentAction(action, dependencies) {
-  const { setCoagentAction, removeCoagentAction, coagentActions, chatComponentsCache } = (0, import_react8.useContext)(CopilotContext);
-  const idRef = (0, import_react8.useRef)((0, import_shared6.randomId)());
-  const key = `${action.name}-${action.nodeName || "global"}`;
-  if (dependencies === void 0) {
-    if (coagentActions[idRef.current]) {
-      coagentActions[idRef.current].handler = action.handler;
-      if (typeof action.render === "function") {
-        if (chatComponentsCache.current !== null) {
-          chatComponentsCache.current.coagentActions[key] = action.render;
-        }
-      }
-    }
-  }
-  (0, import_react8.useEffect)(() => {
-    setCoagentAction(idRef.current, action);
-    if (chatComponentsCache.current !== null && action.render !== void 0) {
-      chatComponentsCache.current.coagentActions[key] = action.render;
-    }
-    return () => {
-      removeCoagentAction(idRef.current);
-    };
-  }, [
-    setCoagentAction,
-    removeCoagentAction,
-    action.name,
-    // include render only if it's a string
-    typeof action.render === "string" ? action.render : void 0,
-    // dependencies set by the developer
-    ...dependencies || []
-  ]);
-}
-
 // src/hooks/use-make-copilot-document-readable.ts
-var import_react9 = require("react");
+var import_react8 = require("react");
 function useMakeCopilotDocumentReadable(document, categories, dependencies = []) {
   const { addDocumentContext, removeDocumentContext } = useCopilotContext();
-  const idRef = (0, import_react9.useRef)();
-  (0, import_react9.useEffect)(() => {
+  const idRef = (0, import_react8.useRef)();
+  (0, import_react8.useEffect)(() => {
     const id = addDocumentContext(document, categories);
     idRef.current = id;
     return () => {
@@ -1050,16 +890,16 @@ function useMakeCopilotDocumentReadable(document, categories, dependencies = [])
 }
 
 // src/hooks/use-copilot-readable.ts
-var import_react10 = 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_react10.useRef)();
+  const idRef = (0, import_react9.useRef)();
   convert = convert || convertToJSON;
   const information = convert(description, value);
-  (0, import_react10.useEffect)(() => {
+  (0, import_react9.useEffect)(() => {
     const id = addContext(information, parentId, categories);
     idRef.current = id;
     return () => {
@@ -1069,84 +909,9 @@ function useCopilotReadable({ description, value, parentId, categories, convert
   return idRef.current;
 }
 
-// src/hooks/use-coagent.ts
-var import_react11 = require("react");
-function useCoAgent(options) {
-  const isExternalStateManagement = (options2) => {
-    return "state" in options2 && "setState" in options2;
-  };
-  const { name } = options;
-  const isInternalStateManagementWithInitial = (options2) => {
-    return "initialState" in options2;
-  };
-  const { coagentStates, setCoagentStates } = useCopilotContext();
-  const getCoagentState = (coagentStates2, name2) => {
-    if (coagentStates2[name2]) {
-      return coagentStates2[name2];
-    } else {
-      return {
-        name: name2,
-        state: isInternalStateManagementWithInitial(options) ? options.initialState : {},
-        running: false,
-        active: false,
-        threadId: void 0,
-        nodeName: void 0,
-        runId: void 0
-      };
-    }
-  };
-  const setState = (newState) => {
-    setCoagentStates((prevAgentStates) => {
-      let coagentState2 = getCoagentState(prevAgentStates, name);
-      const updatedState = typeof newState === "function" ? newState(coagentState2.state) : newState;
-      return __spreadProps(__spreadValues({}, prevAgentStates), {
-        [name]: __spreadProps(__spreadValues({}, coagentState2), {
-          state: updatedState
-        })
-      });
-    });
-  };
-  const coagentState = getCoagentState(coagentStates, name);
-  const state = isExternalStateManagement(options) ? options.state : coagentState.state;
-  (0, import_react11.useEffect)(() => {
-    if (isExternalStateManagement(options)) {
-      setState(options.state);
-    } else if (coagentStates[name] === void 0) {
-      setState(options.initialState === void 0 ? {} : options.initialState);
-    }
-  }, [isExternalStateManagement(options) ? JSON.stringify(options.state) : void 0]);
-  return {
-    name,
-    nodeName: coagentState.nodeName,
-    state,
-    setState,
-    running: coagentState.running,
-    start: () => {
-      startAgent(name);
-    },
-    stop: () => {
-      stopAgent(name);
-    }
-  };
-}
-function startAgent(name) {
-  const { setAgentSession } = useCopilotContext();
-  setAgentSession({
-    agentName: name
-  });
-}
-function stopAgent(name) {
-  const { agentSession, setAgentSession } = useCopilotContext();
-  if (agentSession && agentSession.agentName === name) {
-    setAgentSession(null);
-  } else {
-    console.warn(`No agent session found for ${name}`);
-  }
-}
-
 // src/lib/copilot-task.ts
 var import_runtime_client_gql3 = require("@copilotkit/runtime-client-gql");
-var import_shared7 = require("@copilotkit/shared");
+var import_shared6 = require("@copilotkit/shared");
 var CopilotTask = class {
   constructor(config) {
     this.instructions = config.instructions;
@@ -1190,11 +955,11 @@ var CopilotTask = class {
             actions: Object.values(actions).map((action) => ({
               name: action.name,
               description: action.description || "",
-              jsonSchema: JSON.stringify((0, import_shared7.actionParametersToJsonSchema)(action.parameters || []))
+              jsonSchema: JSON.stringify((0, import_shared6.actionParametersToJsonSchema)(action.parameters || []))
             })),
             url: window.location.href
           },
-          messages: (0, import_runtime_client_gql3.convertMessagesToGqlInput)((0, import_runtime_client_gql3.filterAgentStateMessages)(messages)),
+          messages: (0, import_runtime_client_gql3.convertMessagesToGqlInput)(messages),
           metadata: {
             requestType: import_runtime_client_gql3.CopilotRequestType.Task
           },
@@ -1245,7 +1010,7 @@ ${instructions}
 }
 
 // src/utils/extract.ts
-var import_shared8 = require("@copilotkit/shared");
+var import_shared7 = require("@copilotkit/shared");
 var import_runtime_client_gql4 = require("@copilotkit/runtime-client-gql");
 var import_runtime_client_gql5 = require("@copilotkit/runtime-client-gql");
 var import_runtime_client_gql6 = require("@copilotkit/runtime-client-gql");
@@ -1281,7 +1046,7 @@ function extract(_0) {
       content: makeSystemMessage(contextString, instructions),
       role: import_runtime_client_gql4.Role.System
     });
-    const headers = __spreadValues(__spreadValues({}, context.copilotApiConfig.headers || {}), context.copilotApiConfig.publicApiKey ? { [import_shared8.COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: context.copilotApiConfig.publicApiKey } : {});
+    const headers = __spreadValues(__spreadValues({}, context.copilotApiConfig.headers || {}), context.copilotApiConfig.publicApiKey ? { [import_shared7.COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: context.copilotApiConfig.publicApiKey } : {});
     const runtimeClient = new import_runtime_client_gql5.CopilotRuntimeClient({
       url: context.copilotApiConfig.chatApiEndpoint,
       publicApiKey: context.copilotApiConfig.publicApiKey,
@@ -1296,13 +1061,13 @@ function extract(_0) {
               {
                 name: action.name,
                 description: action.description || "",
-                jsonSchema: JSON.stringify((0, import_shared8.actionParametersToJsonSchema)(action.parameters || []))
+                jsonSchema: JSON.stringify((0, import_shared7.actionParametersToJsonSchema)(action.parameters || []))
               }
             ],
             url: window.location.href
           },
           messages: (0, import_runtime_client_gql6.convertMessagesToGqlInput)(
-            includeMessages ? [systemMessage, ...(0, import_runtime_client_gql6.filterAgentStateMessages)(messages)] : [systemMessage]
+            includeMessages ? [systemMessage, ...messages] : [systemMessage]
           ),
           metadata: {
             requestType
@@ -1383,8 +1148,6 @@ Any additional messages provided are for providing context only and should not b
   CopilotTask,
   defaultCopilotContextCategories,
   extract,
-  useCoAgent,
-  useCoAgentAction,
   useCopilotAction,
   useCopilotChat,
   useCopilotContext,