@copilotkit/react-core 1.3.2-pre.0 → 1.3.3-pre.0

package/dist/index.js

@@ -87,7 +87,7 @@ __export(src_exports, {
   defaultCopilotContextCategories: () => defaultCopilotContextCategories,
   extract: () => extract,
   useCoAgent: () => useCoAgent,
-  useCoAgentAction: () => useCoAgentAction,
+  useCoagentStateRender: () => useCoagentStateRender,
   useCopilotAction: () => useCopilotAction,
   useCopilotChat: () => useCopilotChat,
   useCopilotContext: () => useCopilotContext,
@@ -612,6 +612,11 @@ function useChat(options) {
   const abortControllerRef = (0, import_react5.useRef)();
   const threadIdRef = (0, import_react5.useRef)(null);
   const runIdRef = (0, import_react5.useRef)(null);
+  const runChatCompletionRef = (0, import_react5.useRef)();
+  const coagentStatesRef = (0, import_react5.useRef)(coagentStates);
+  coagentStatesRef.current = coagentStates;
+  const agentSessionRef = (0, import_react5.useRef)(agentSession);
+  agentSessionRef.current = agentSession;
   const publicApiKey = copilotConfig.publicApiKey;
   const headers = __spreadValues(__spreadValues({}, copilotConfig.headers || {}), publicApiKey ? { [import_shared4.COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: publicApiKey } : {});
   const runtimeClient = new import_runtime_client_gql.CopilotRuntimeClient({
@@ -638,7 +643,7 @@ function useChat(options) {
       runtimeClient.generateCopilotResponse({
         data: __spreadProps(__spreadValues(__spreadProps(__spreadValues({
           frontend: {
-            actions: actions.map((action) => ({
+            actions: actions.filter((action) => !action.disabled).map((action) => ({
               name: action.name,
               description: action.description || "",
               jsonSchema: JSON.stringify((0, import_shared4.actionParametersToJsonSchema)(action.parameters || []))
@@ -661,10 +666,10 @@ function useChat(options) {
           metadata: {
             requestType: import_runtime_client_gql.CopilotRequestType.Chat
           }
-        }), agentSession ? {
-          agentSession
+        }), agentSessionRef.current ? {
+          agentSession: agentSessionRef.current
         } : {}), {
-          agentStates: Object.values(coagentStates).map((state) => ({
+          agentStates: Object.values(coagentStatesRef.current).map((state) => ({
             agentName: state.name,
             state: JSON.stringify(state.state)
           }))
@@ -711,6 +716,7 @@ function useChat(options) {
                   break;
                 }
                 try {
+                  setMessages([...previousMessages, ...newMessages]);
                   const result = yield onFunctionCall({
                     messages: previousMessages,
                     name: message.name,
@@ -776,7 +782,7 @@ function useChat(options) {
         newMessages.length && newMessages[newMessages.length - 1] instanceof import_runtime_client_gql.ResultMessage
       ) {
         yield new Promise((resolve) => setTimeout(resolve, 10));
-        return yield runChatCompletion([...previousMessages, ...newMessages]);
+        return yield runChatCompletionRef.current([...previousMessages, ...newMessages]);
       } else {
         return newMessages.slice();
       }
@@ -784,8 +790,9 @@ function useChat(options) {
       setIsLoading(false);
     }
   });
+  runChatCompletionRef.current = runChatCompletion;
   const runChatCompletionAndHandleFunctionCall = (messages2) => __async(this, null, function* () {
-    yield runChatCompletion(messages2);
+    yield runChatCompletionRef.current(messages2);
   });
   const append = (message) => __async(this, null, function* () {
     if (isLoading) {
@@ -966,6 +973,31 @@ var import_shared5 = require("@copilotkit/shared");
 function useCopilotAction(action, dependencies) {
   const { setAction, removeAction, actions, chatComponentsCache } = useCopilotContext();
   const idRef = (0, import_react7.useRef)((0, import_shared5.randomId)());
+  const renderAndWaitRef = (0, import_react7.useRef)(null);
+  action = __spreadValues({}, action);
+  if (action.renderAndWait) {
+    const renderAndWait = action.renderAndWait;
+    action.renderAndWait = void 0;
+    action.handler = () => __async(this, null, function* () {
+      let resolve;
+      let reject;
+      const promise = new Promise((resolvePromise, rejectPromise) => {
+        resolve = resolvePromise;
+        reject = rejectPromise;
+      });
+      renderAndWaitRef.current = { promise, resolve, reject };
+      return yield promise;
+    });
+    action.render = (props) => {
+      const waitProps = {
+        status: props.status,
+        args: props.args,
+        result: props.result,
+        handler: props.status === "executing" ? renderAndWaitRef.current.resolve : void 0
+      };
+      return renderAndWait(waitProps);
+    };
+  }
   if (dependencies === void 0) {
     if (actions[idRef.current]) {
       actions[idRef.current].handler = action.handler;
@@ -977,9 +1009,6 @@ function useCopilotAction(action, dependencies) {
     }
   }
   (0, import_react7.useEffect)(() => {
-    if (action.disabled) {
-      return;
-    }
     setAction(idRef.current, action);
     if (chatComponentsCache.current !== null && action.render !== void 0) {
       chatComponentsCache.current.actions[action.name] = action.render;
@@ -1003,10 +1032,10 @@ function useCopilotAction(action, dependencies) {
   ]);
 }
 
-// src/hooks/use-coagent-action.ts
+// src/hooks/use-coagent-state-render.ts
 var import_react8 = require("react");
 var import_shared6 = require("@copilotkit/shared");
-function useCoAgentAction(action, dependencies) {
+function useCoagentStateRender(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"}`;
@@ -1171,26 +1200,12 @@ function runAgent(name, context, appendMessage, hint) {
       }
     }
     let state = ((_b = (_a = context.coagentStates) == null ? void 0 : _a[name]) == null ? void 0 : _b.state) || {};
-    let content = "The state of the agent has been updated\n";
-    if (previousState !== null) {
-      content += `The previous state was:
-${JSON.stringify(previousState, null, 2)}
-
-`;
-    }
-    content += `The current state is:
-${JSON.stringify(state, null, 2)}`;
     if (hint) {
-      content += `
-
-${hint}`;
+      const hintMessage = hint({ previousState, currentState: state });
+      if (hintMessage) {
+        yield appendMessage(hintMessage);
+      }
     }
-    return yield appendMessage(
-      new import_runtime_client_gql3.TextMessage({
-        role: import_runtime_client_gql3.Role.System,
-        content
-      })
-    );
   });
 }
 
@@ -1434,7 +1449,7 @@ Any additional messages provided are for providing context only and should not b
   defaultCopilotContextCategories,
   extract,
   useCoAgent,
-  useCoAgentAction,
+  useCoagentStateRender,
   useCopilotAction,
   useCopilotChat,
   useCopilotContext,