@yourgpt/copilot-sdk 2.0.2-beta.2 → 2.1.0

package/dist/ui/index.cjs

@@ -1,11 +1,11 @@
 'use strict';
 
-var chunkH5XMKBBA_cjs = require('../chunk-H5XMKBBA.cjs');
-var chunkUOWLKFXK_cjs = require('../chunk-UOWLKFXK.cjs');
+var chunkMUZ2RYS2_cjs = require('../chunk-MUZ2RYS2.cjs');
+var chunk33JSFVGH_cjs = require('../chunk-33JSFVGH.cjs');
 var chunkJO4BHPAD_cjs = require('../chunk-JO4BHPAD.cjs');
-require('../chunk-7K7HZMP4.cjs');
-require('../chunk-UIWFYMAO.cjs');
-require('../chunk-CEKAYA2Q.cjs');
+require('../chunk-JGPDQDY4.cjs');
+require('../chunk-O33I4HBL.cjs');
+require('../chunk-HNRFRNMY.cjs');
 require('../chunk-ASV6JLYG.cjs');
 require('../chunk-CBAHCI4R.cjs');
 require('../chunk-MEBXW75C.cjs');
@@ -351,7 +351,8 @@ var headingComponents = {
 function MarkdownComponent({
   children,
   className,
-  isStreaming = false
+  isStreaming = false,
+  linkSafety = { enabled: false }
 }) {
   return /* @__PURE__ */ jsxRuntime.jsx("div", { className, children: /* @__PURE__ */ jsxRuntime.jsx(
     streamdown.Streamdown,
@@ -359,6 +360,7 @@ function MarkdownComponent({
       plugins: { code: code.code },
       isAnimating: isStreaming,
       components: headingComponents,
+      linkSafety,
       children
     }
   ) });
@@ -1926,8 +1928,8 @@ function ToolStep({
         "aria-hidden": "true"
       }
     ),
-    /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex items-start gap-2", children: [
-      !isExecuting && /* @__PURE__ */ jsxRuntime.jsx(StatusIndicator, { status: step.status, className: "mt-0.5" }),
+    /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex items-center gap-2", children: [
+      !isExecuting && /* @__PURE__ */ jsxRuntime.jsx(StatusIndicator, { status: step.status }),
       /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex-1 min-w-0", children: [
         /* @__PURE__ */ jsxRuntime.jsxs(
           "button",
@@ -4249,6 +4251,7 @@ function DefaultMessage({
   loaderVariant = "typing",
   registeredTools,
   toolRenderers,
+  mcpToolRenderer,
   onApproveToolExecution,
   onRejectToolExecution,
   showFollowUps = true,
@@ -4370,17 +4373,19 @@ function DefaultMessage({
   const completedTools = message.toolExecutions?.filter(
     (exec) => exec.approvalStatus !== "required" && !isToolHidden(exec.name)
   );
-  const hasCustomRender = (toolName) => {
+  const hasCustomRender = (toolName, execSource) => {
     if (toolRenderers?.[toolName]) return true;
     const toolDef = registeredTools?.find((t) => t.name === toolName);
+    if (mcpToolRenderer && (execSource === "mcp" || toolDef?.source === "mcp"))
+      return true;
     if (toolDef?.render) return true;
     return false;
   };
   const toolsWithCustomRender = completedTools?.filter(
-    (exec) => hasCustomRender(exec.name)
+    (exec) => hasCustomRender(exec.name, exec.source)
   );
   const toolsWithoutCustomRender = completedTools?.filter(
-    (exec) => !hasCustomRender(exec.name)
+    (exec) => !hasCustomRender(exec.name, exec.source)
   );
   const hasNativeCitations = !!message.metadata?.citations?.length;
   const toolSteps = toolsWithoutCustomRender?.filter((exec) => !(exec.name === "web_search" && hasNativeCitations)).map((exec) => ({
@@ -4430,7 +4435,7 @@ function DefaultMessage({
               children: cleanContent
             }
           ),
-          toolsWithCustomRender && toolsWithCustomRender.length > 0 && /* @__PURE__ */ jsxRuntime.jsx("div", { className: "mt-2 space-y-2", children: toolsWithCustomRender.map((exec) => {
+          toolsWithCustomRender && toolsWithCustomRender.length > 0 && /* @__PURE__ */ jsxRuntime.jsx("div", { className: cn("space-y-2", cleanContent?.trim() && "mt-2"), children: toolsWithCustomRender.map((exec) => {
             const Renderer = toolRenderers?.[exec.name];
             if (Renderer) {
               return /* @__PURE__ */ jsxRuntime.jsx(
@@ -4443,7 +4448,8 @@ function DefaultMessage({
                     status: exec.status,
                     result: exec.result,
                     error: exec.error,
-                    approvalStatus: exec.approvalStatus
+                    approvalStatus: exec.approvalStatus,
+                    source: exec.source
                   }
                 },
                 exec.id
@@ -4452,7 +4458,26 @@ function DefaultMessage({
             const toolDef = registeredTools?.find(
               (t) => t.name === exec.name
             );
-            if (toolDef?.render) {
+            if (mcpToolRenderer && (exec.source === "mcp" || toolDef?.source === "mcp")) {
+              const MCPRenderer = mcpToolRenderer;
+              return /* @__PURE__ */ jsxRuntime.jsx(
+                MCPRenderer,
+                {
+                  execution: {
+                    id: exec.id,
+                    name: exec.name,
+                    args: exec.args,
+                    status: exec.status,
+                    result: exec.result,
+                    error: exec.error,
+                    source: exec.source || toolDef?.source
+                  }
+                },
+                exec.id
+              );
+            }
+            const toolDefForRender = toolDef ?? registeredTools?.find((t) => t.name === exec.name);
+            if (toolDefForRender?.render) {
               let status = "pending";
               if (exec.status === "executing") status = "executing";
               else if (exec.status === "completed") status = "completed";
@@ -4466,14 +4491,23 @@ function DefaultMessage({
                 toolCallId: exec.id,
                 toolName: exec.name
               };
-              const output = toolDef.render(
+              const output = toolDefForRender.render(
                 renderProps
               );
               return /* @__PURE__ */ jsxRuntime.jsx(React20__namespace.Fragment, { children: output }, exec.id);
             }
             return null;
           }) }),
-          toolSteps && toolSteps.length > 0 && /* @__PURE__ */ jsxRuntime.jsx("div", { className: "mt-2 rounded-lg bg-muted/50 px-3 py-2", children: /* @__PURE__ */ jsxRuntime.jsx(ToolSteps, { steps: toolSteps }) }),
+          toolSteps && toolSteps.length > 0 && /* @__PURE__ */ jsxRuntime.jsx(
+            "div",
+            {
+              className: cn(
+                "rounded-lg bg-muted/50 px-3 py-2",
+                cleanContent?.trim() && "mt-2"
+              ),
+              children: /* @__PURE__ */ jsxRuntime.jsx(ToolSteps, { steps: toolSteps })
+            }
+          ),
           message.toolExecutions?.filter((exec) => !isToolHidden(exec.name)).map((exec) => {
             const uiResources = exec.result?._uiResources;
             if (!uiResources || uiResources.length === 0) return null;
@@ -5252,6 +5286,7 @@ function ChatComponent({
   isProcessing = false,
   registeredTools,
   toolRenderers,
+  mcpToolRenderer,
   onApproveToolExecution,
   onRejectToolExecution,
   // Follow-up Questions
@@ -5579,6 +5614,7 @@ function ChatComponent({
                             loaderVariant,
                             registeredTools,
                             toolRenderers,
+                            mcpToolRenderer,
                             onApproveToolExecution,
                             onRejectToolExecution,
                             showFollowUps,
@@ -5908,7 +5944,7 @@ function useInternalThreadManager(config = {}) {
     saveDebounce,
     autoRestoreLastThread
   };
-  const threadManager = chunkH5XMKBBA_cjs.useThreadManager(threadManagerConfig);
+  const threadManager = chunkMUZ2RYS2_cjs.useThreadManager(threadManagerConfig);
   const {
     currentThread,
     currentThreadId,
@@ -5918,7 +5954,7 @@ function useInternalThreadManager(config = {}) {
     clearCurrentThread,
     refreshThreads
   } = threadManager;
-  const { messages, setMessages, status, isLoading } = chunkH5XMKBBA_cjs.useCopilot();
+  const { messages, setMessages, status, isLoading } = chunkMUZ2RYS2_cjs.useCopilot();
   const isLoadingMessagesRef = React20.useRef(false);
   const savingToThreadRef = React20.useRef(null);
   const lastSavedSnapshotRef = React20.useRef("");
@@ -5986,9 +6022,6 @@ function useInternalThreadManager(config = {}) {
     });
   }, [clearCurrentThread, setMessages, onThreadChange]);
   React20.useEffect(() => {
-    if (!adapter) {
-      return;
-    }
     if (hasInitializedRef.current || !currentThread) {
       return;
     }
@@ -6020,9 +6053,6 @@ function useInternalThreadManager(config = {}) {
     });
   }, [adapter, currentThread, setMessages, getMessageSnapshot, onThreadChange]);
   React20.useEffect(() => {
-    if (!adapter) {
-      return;
-    }
     if (isLoadingMessagesRef.current) {
       return;
     }
@@ -6091,7 +6121,7 @@ function parsePersistenceConfig(persistence, onThreadChange) {
         };
       case "server":
         return {
-          adapter: chunkUOWLKFXK_cjs.createServerAdapter({
+          adapter: chunk33JSFVGH_cjs.createServerAdapter({
             endpoint: persistence.endpoint,
             headers: persistence.headers
           }),
@@ -6141,7 +6171,7 @@ function CopilotChatBase(props) {
     approveToolExecution,
     rejectToolExecution,
     registeredTools
-  } = chunkH5XMKBBA_cjs.useCopilot();
+  } = chunkMUZ2RYS2_cjs.useCopilot();
   const toolExecutions = rawToolExecutions.map(
     (exec) => ({
       id: exec.id,