@copilotkit/react-core 1.8.7 → 1.8.8-next.1

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

@@ -406,7 +406,7 @@ function useChat(options) {
   });
   const runChatCompletion = useAsyncCallback(
     (previousMessages) => __async(this, null, function* () {
-      var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n;
+      var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o;
       setIsLoading(true);
       const interruptEvent = langGraphInterruptAction == null ? void 0 : langGraphInterruptAction.event;
       if ((interruptEvent == null ? void 0 : interruptEvent.name) === import_runtime_client_gql3.MetaEventName.LangGraphInterruptEvent && (interruptEvent == null ? void 0 : interruptEvent.value) && !(interruptEvent == null ? void 0 : interruptEvent.response) && agentSessionRef.current) {
@@ -427,8 +427,15 @@ function useChat(options) {
       const systemMessage = makeSystemMessageCallback();
       const messagesWithContext = [systemMessage, ...initialMessages || [], ...previousMessages];
       const finalProperties = __spreadValues({}, copilotConfig.properties || {});
-      if (copilotConfig.mcpEndpoints && copilotConfig.mcpEndpoints.length > 0) {
-        finalProperties.mcpEndpoints = copilotConfig.mcpEndpoints;
+      let mcpServersToUse = null;
+      if (copilotConfig.mcpServers && Array.isArray(copilotConfig.mcpServers) && copilotConfig.mcpServers.length > 0) {
+        mcpServersToUse = copilotConfig.mcpServers;
+      } else if (((_a = copilotConfig.properties) == null ? void 0 : _a.mcpServers) && Array.isArray(copilotConfig.properties.mcpServers) && copilotConfig.properties.mcpServers.length > 0) {
+        mcpServersToUse = copilotConfig.properties.mcpServers;
+      }
+      if (mcpServersToUse) {
+        finalProperties.mcpServers = mcpServersToUse;
+        copilotConfig.mcpServers = mcpServersToUse;
       }
       const isAgentRun = agentSessionRef.current !== null;
       const stream = runtimeClient.asStream(
@@ -444,7 +451,7 @@ function useChat(options) {
             metaEvents: composeAndFlushMetaEventsInput([langGraphInterruptAction == null ? void 0 : langGraphInterruptAction.event]),
             messages: (0, import_runtime_client_gql3.convertMessagesToGqlInput)((0, import_runtime_client_gql3.filterAgentStateMessages)(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) ? {
+            cloud: __spreadValues({}, ((_d = (_c = (_b = copilotConfig.cloud.guardrails) == null ? void 0 : _b.input) == null ? void 0 : _c.restrictToTopic) == null ? void 0 : _d.enabled) ? {
               guardrails: {
                 inputValidationRules: {
                   allowList: copilotConfig.cloud.guardrails.input.restrictToTopic.validTopics,
@@ -470,10 +477,10 @@ function useChat(options) {
             forwardedParameters: options.forwardedParameters || {}
           }),
           properties: finalProperties,
-          signal: (_d = chatAbortControllerRef.current) == null ? void 0 : _d.signal
+          signal: (_e = chatAbortControllerRef.current) == null ? void 0 : _e.signal
         })
       );
-      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 guardrailsEnabled = ((_h = (_g = (_f = copilotConfig.cloud) == null ? void 0 : _f.guardrails) == null ? void 0 : _g.input) == null ? void 0 : _h.restrictToTopic.enabled) || false;
       const reader = stream.getReader();
       let executedCoAgentStateRenders = [];
       let followUp = void 0;
@@ -506,7 +513,7 @@ function useChat(options) {
           setRunId(runIdRef.current);
           setExtensions(extensionsRef.current);
           let rawMessagesResponse = value.generateCopilotResponse.messages;
-          const metaEvents = (_i = (_h = value.generateCopilotResponse) == null ? void 0 : _h.metaEvents) != null ? _i : [];
+          const metaEvents = (_j = (_i = value.generateCopilotResponse) == null ? void 0 : _i.metaEvents) != null ? _j : [];
           (metaEvents != null ? metaEvents : []).forEach((ev) => {
             if (ev.name === import_runtime_client_gql3.MetaEventName.LangGraphInterruptEvent) {
               let eventValue = (0, import_runtime_client_gql3.langGraphInterruptEvent)(ev).value;
@@ -533,11 +540,11 @@ function useChat(options) {
             continue;
           }
           newMessages = [];
-          if (((_j = value.generateCopilotResponse.status) == null ? void 0 : _j.__typename) === "FailedResponseStatus" && value.generateCopilotResponse.status.reason === "GUARDRAILS_VALIDATION_FAILED") {
+          if (((_k = value.generateCopilotResponse.status) == null ? void 0 : _k.__typename) === "FailedResponseStatus" && value.generateCopilotResponse.status.reason === "GUARDRAILS_VALIDATION_FAILED") {
             newMessages = [
               new import_runtime_client_gql3.TextMessage({
                 role: import_runtime_client_gql3.MessageRole.Assistant,
-                content: ((_k = value.generateCopilotResponse.status.details) == null ? void 0 : _k.guardrailsReason) || ""
+                content: ((_l = value.generateCopilotResponse.status.details) == null ? void 0 : _l.guardrailsReason) || ""
               })
             ];
             setMessages([...previousMessages, ...newMessages]);
@@ -672,11 +679,11 @@ function useChat(options) {
           followUp !== false && // and we executed an action
           (didExecuteAction || // the last message is a server side result
           !isAgentRun && finalMessages.length && finalMessages[finalMessages.length - 1].isResultMessage()) && // the user did not stop generation
-          !((_l = chatAbortControllerRef.current) == null ? void 0 : _l.signal.aborted)
+          !((_m = chatAbortControllerRef.current) == null ? void 0 : _m.signal.aborted)
         ) {
           yield new Promise((resolve) => setTimeout(resolve, 10));
           return yield runChatCompletionRef.current(finalMessages);
-        } else if ((_m = chatAbortControllerRef.current) == null ? void 0 : _m.signal.aborted) {
+        } else if ((_n = chatAbortControllerRef.current) == null ? void 0 : _n.signal.aborted) {
           const repairedMessages = finalMessages.filter((message, actionExecutionIndex) => {
             if (message.isActionExecutionMessage()) {
               return finalMessages.find(
@@ -687,7 +694,7 @@ function useChat(options) {
           });
           const repairedMessageIds = repairedMessages.map((message) => message.id);
           setMessages(repairedMessages);
-          if ((_n = agentSessionRef.current) == null ? void 0 : _n.nodeName) {
+          if ((_o = agentSessionRef.current) == null ? void 0 : _o.nodeName) {
             setAgentSession({
               threadId: agentSessionRef.current.threadId,
               agentName: agentSessionRef.current.agentName,
@@ -936,21 +943,20 @@ function useCopilotChat(_a = {}) {
     setLangGraphInterruptAction
   } = useCopilotContext();
   const { messages, setMessages } = useCopilotMessagesContext();
-  const latestGetContextString = useUpdatedRef(getContextString);
-  const deleteMessage = (0, import_react8.useCallback)(
-    (messageId) => {
-      setMessages((prev) => prev.filter((message) => message.id !== messageId));
-    },
-    [setMessages]
-  );
-  const makeSystemMessageCallback = (0, import_react8.useCallback)(() => {
-    const systemMessageMaker = makeSystemMessage || defaultSystemMessage;
-    const contextString = latestGetContextString.current([], defaultCopilotContextCategories);
-    return new import_runtime_client_gql4.TextMessage({
-      content: systemMessageMaker(contextString, chatInstructions),
-      role: import_runtime_client_gql4.Role.System
-    });
-  }, [getContextString, makeSystemMessage, chatInstructions]);
+  const [mcpServers, setLocalMcpServers] = (0, import_react8.useState)([]);
+  (0, import_react8.useEffect)(() => {
+    if (mcpServers.length > 0) {
+      const serversCopy = [...mcpServers];
+      copilotApiConfig.mcpServers = serversCopy;
+      if (!copilotApiConfig.properties) {
+        copilotApiConfig.properties = {};
+      }
+      copilotApiConfig.properties.mcpServers = serversCopy;
+    }
+  }, [mcpServers, copilotApiConfig]);
+  const setMcpServers = (0, import_react8.useCallback)((servers) => {
+    setLocalMcpServers(servers);
+  }, []);
   const onCoAgentStateRender = useAsyncCallback(
     (args) => __async(this, null, function* () {
       var _a2;
@@ -969,6 +975,20 @@ function useCopilotChat(_a = {}) {
     }),
     [coAgentStateRenders]
   );
+  const makeSystemMessageCallback = (0, import_react8.useCallback)(() => {
+    const systemMessageMaker = makeSystemMessage || defaultSystemMessage;
+    const contextString = getContextString([], defaultCopilotContextCategories);
+    return new import_runtime_client_gql4.TextMessage({
+      content: systemMessageMaker(contextString, chatInstructions),
+      role: import_runtime_client_gql4.Role.System
+    });
+  }, [getContextString, makeSystemMessage, chatInstructions]);
+  const deleteMessage = (0, import_react8.useCallback)(
+    (messageId) => {
+      setMessages((prev) => prev.filter((message) => message.id !== messageId));
+    },
+    [setMessages]
+  );
   const { append, reload, stop, runChatCompletion } = useChat(__spreadProps(__spreadValues({}, options), {
     actions: Object.values(actions),
     copilotConfig: copilotApiConfig,
@@ -1065,7 +1085,9 @@ function useCopilotChat(_a = {}) {
     reset: latestResetFunc,
     deleteMessage: latestDeleteFunc,
     runChatCompletion: latestRunChatCompletionFunc,
-    isLoading
+    isLoading,
+    mcpServers,
+    setMcpServers
   };
 }
 function useUpdatedRef(value) {