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

package/dist/ui/index.d.cts

@@ -1,12 +1,14 @@
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import * as React$1 from 'react';
 import React__default from 'react';
+import { LinkSafetyConfig } from 'streamdown';
+export { LinkSafetyConfig } from 'streamdown';
 import * as class_variance_authority_types from 'class-variance-authority/types';
 import { VariantProps } from 'class-variance-authority';
 import * as use_stick_to_bottom from 'use-stick-to-bottom';
 import { d as WebSearchResult, f as WebSearchResponse } from '../types-ZguuKEs_.cjs';
 import { T as Thread, b as ThreadStorageAdapter, A as AsyncThreadStorageAdapter } from '../types-DG2ya08y.cjs';
-import { T as ToolDefinition } from '../tools-DDWrco4h.cjs';
+import { T as ToolDefinition } from '../tools-Clyufshc.cjs';
 import { d as MCPUIIntent, n as MCPUIFrameProps, M as MCPUIResource } from '../types-B20VCJXL.cjs';
 import { ClassValue } from 'clsx';
 
@@ -23,8 +25,10 @@ type MarkdownProps = {
     id?: string;
     className?: string;
     isStreaming?: boolean;
+    /** Link safety modal configuration. Disabled by default. */
+    linkSafety?: LinkSafetyConfig;
 };
-declare function MarkdownComponent({ children, className, isStreaming, }: MarkdownProps): react_jsx_runtime.JSX.Element;
+declare function MarkdownComponent({ children, className, isStreaming, linkSafety, }: MarkdownProps): react_jsx_runtime.JSX.Element;
 declare const Markdown: React$1.MemoExoticComponent<typeof MarkdownComponent>;
 
 type CodeBlockProps = {
@@ -1211,6 +1215,10 @@ type ToolExecutionStatus = "pending" | "executing" | "completed" | "error" | "fa
  * Tool approval status (for human-in-the-loop)
  */
 type ToolApprovalStatus = "none" | "required" | "approved" | "rejected";
+/**
+ * Tool execution source - where the tool came from
+ */
+type ToolExecutionSource = "mcp" | "native" | "custom";
 /**
  * Tool execution data
  */
@@ -1219,6 +1227,13 @@ interface ToolExecutionData {
     name: string;
     args: Record<string, unknown>;
     status: ToolExecutionStatus;
+    /**
+     * Source of the tool - helps distinguish MCP tools from native tools
+     * - "mcp": Tool from an MCP server
+     * - "native": Built-in SDK tool
+     * - "custom": User-defined tool
+     */
+    source?: ToolExecutionSource;
     result?: {
         success: boolean;
         message?: string;
@@ -1377,6 +1392,8 @@ interface ToolRendererProps {
         approvalStatus?: ToolApprovalStatus;
         /** Data passed from user's approval action */
         approvalData?: Record<string, unknown>;
+        /** Tool source (mcp, native, custom) - useful for MCP tool identification */
+        source?: "mcp" | "native" | "custom";
     };
     /**
      * Approval callbacks - only provided when approvalStatus is 'required'.
@@ -1554,6 +1571,23 @@ type ChatProps = {
      * ```
      */
     toolRenderers?: ToolRenderers;
+    /**
+     * Catch-all renderer for MCP tools (tools with source: "mcp").
+     * Applied to all MCP tools that don't have a specific renderer in `toolRenderers`.
+     *
+     * Priority: toolRenderers[name] > mcpToolRenderer > tool.render > default
+     *
+     * @example
+     * ```tsx
+     * <Chat
+     *   mcpToolRenderer={MCPToolCard}  // Applies to all MCP tools
+     *   toolRenderers={{
+     *     "mcp360_special_tool": SpecialCard,  // Override for specific tool
+     *   }}
+     * />
+     * ```
+     */
+    mcpToolRenderer?: React__default.ComponentType<ToolRendererProps>;
     /**
      * Called when user approves a tool execution.
      * @param executionId - The tool execution ID
@@ -1709,7 +1743,7 @@ declare function BackButton({ className, children, disabled, "aria-label": ariaL
  */
 type ThreadPickerCompoundProps = Omit<ThreadPickerProps, "value" | "threads" | "onSelect" | "onNewThread" | "onDeleteThread" | "disabled">;
 declare function ThreadPickerCompound(props: ThreadPickerCompoundProps): react_jsx_runtime.JSX.Element | null;
-declare function ChatComponent({ messages, onSendMessage, onStop, isLoading, children, placeholder, welcomeMessage, title, showHeader, header, threadPicker, logo, name, onClose, showPoweredBy, showUserAvatar, userAvatar: userAvatarProp, assistantAvatar: assistantAvatarProp, loaderVariant, fontSize, maxFileSize, allowedFileTypes, attachmentsEnabled, attachmentsDisabledTooltip, processAttachment: processAttachmentProp, suggestions, onSuggestionClick, welcome, recentThreads, onSelectThread, onDeleteThread, onViewMoreThreads, isProcessing, registeredTools, toolRenderers, onApproveToolExecution, onRejectToolExecution, showFollowUps, followUpClassName, followUpButtonClassName, citations, renderMessage, renderInput, renderHeader, className, classNames, onNewChat, threads, currentThreadId, onSwitchThread, isThreadBusy, }: ChatProps): react_jsx_runtime.JSX.Element;
+declare function ChatComponent({ messages, onSendMessage, onStop, isLoading, children, placeholder, welcomeMessage, title, showHeader, header, threadPicker, logo, name, onClose, showPoweredBy, showUserAvatar, userAvatar: userAvatarProp, assistantAvatar: assistantAvatarProp, loaderVariant, fontSize, maxFileSize, allowedFileTypes, attachmentsEnabled, attachmentsDisabledTooltip, processAttachment: processAttachmentProp, suggestions, onSuggestionClick, welcome, recentThreads, onSelectThread, onDeleteThread, onViewMoreThreads, isProcessing, registeredTools, toolRenderers, mcpToolRenderer, onApproveToolExecution, onRejectToolExecution, showFollowUps, followUpClassName, followUpButtonClassName, citations, renderMessage, renderInput, renderHeader, className, classNames, onNewChat, threads, currentThreadId, onSwitchThread, isThreadBusy, }: ChatProps): react_jsx_runtime.JSX.Element;
 /**
  * Chat component with compound component pattern.
  *

package/dist/ui/index.d.ts

@@ -1,12 +1,14 @@
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import * as React$1 from 'react';
 import React__default from 'react';
+import { LinkSafetyConfig } from 'streamdown';
+export { LinkSafetyConfig } from 'streamdown';
 import * as class_variance_authority_types from 'class-variance-authority/types';
 import { VariantProps } from 'class-variance-authority';
 import * as use_stick_to_bottom from 'use-stick-to-bottom';
 import { d as WebSearchResult, f as WebSearchResponse } from '../types-ZguuKEs_.js';
 import { T as Thread, b as ThreadStorageAdapter, A as AsyncThreadStorageAdapter } from '../types-DG2ya08y.js';
-import { T as ToolDefinition } from '../tools-DDWrco4h.js';
+import { T as ToolDefinition } from '../tools-Clyufshc.js';
 import { d as MCPUIIntent, n as MCPUIFrameProps, M as MCPUIResource } from '../types-B20VCJXL.js';
 import { ClassValue } from 'clsx';
 
@@ -23,8 +25,10 @@ type MarkdownProps = {
     id?: string;
     className?: string;
     isStreaming?: boolean;
+    /** Link safety modal configuration. Disabled by default. */
+    linkSafety?: LinkSafetyConfig;
 };
-declare function MarkdownComponent({ children, className, isStreaming, }: MarkdownProps): react_jsx_runtime.JSX.Element;
+declare function MarkdownComponent({ children, className, isStreaming, linkSafety, }: MarkdownProps): react_jsx_runtime.JSX.Element;
 declare const Markdown: React$1.MemoExoticComponent<typeof MarkdownComponent>;
 
 type CodeBlockProps = {
@@ -1211,6 +1215,10 @@ type ToolExecutionStatus = "pending" | "executing" | "completed" | "error" | "fa
  * Tool approval status (for human-in-the-loop)
  */
 type ToolApprovalStatus = "none" | "required" | "approved" | "rejected";
+/**
+ * Tool execution source - where the tool came from
+ */
+type ToolExecutionSource = "mcp" | "native" | "custom";
 /**
  * Tool execution data
  */
@@ -1219,6 +1227,13 @@ interface ToolExecutionData {
     name: string;
     args: Record<string, unknown>;
     status: ToolExecutionStatus;
+    /**
+     * Source of the tool - helps distinguish MCP tools from native tools
+     * - "mcp": Tool from an MCP server
+     * - "native": Built-in SDK tool
+     * - "custom": User-defined tool
+     */
+    source?: ToolExecutionSource;
     result?: {
         success: boolean;
         message?: string;
@@ -1377,6 +1392,8 @@ interface ToolRendererProps {
         approvalStatus?: ToolApprovalStatus;
         /** Data passed from user's approval action */
         approvalData?: Record<string, unknown>;
+        /** Tool source (mcp, native, custom) - useful for MCP tool identification */
+        source?: "mcp" | "native" | "custom";
     };
     /**
      * Approval callbacks - only provided when approvalStatus is 'required'.
@@ -1554,6 +1571,23 @@ type ChatProps = {
      * ```
      */
     toolRenderers?: ToolRenderers;
+    /**
+     * Catch-all renderer for MCP tools (tools with source: "mcp").
+     * Applied to all MCP tools that don't have a specific renderer in `toolRenderers`.
+     *
+     * Priority: toolRenderers[name] > mcpToolRenderer > tool.render > default
+     *
+     * @example
+     * ```tsx
+     * <Chat
+     *   mcpToolRenderer={MCPToolCard}  // Applies to all MCP tools
+     *   toolRenderers={{
+     *     "mcp360_special_tool": SpecialCard,  // Override for specific tool
+     *   }}
+     * />
+     * ```
+     */
+    mcpToolRenderer?: React__default.ComponentType<ToolRendererProps>;
     /**
      * Called when user approves a tool execution.
      * @param executionId - The tool execution ID
@@ -1709,7 +1743,7 @@ declare function BackButton({ className, children, disabled, "aria-label": ariaL
  */
 type ThreadPickerCompoundProps = Omit<ThreadPickerProps, "value" | "threads" | "onSelect" | "onNewThread" | "onDeleteThread" | "disabled">;
 declare function ThreadPickerCompound(props: ThreadPickerCompoundProps): react_jsx_runtime.JSX.Element | null;
-declare function ChatComponent({ messages, onSendMessage, onStop, isLoading, children, placeholder, welcomeMessage, title, showHeader, header, threadPicker, logo, name, onClose, showPoweredBy, showUserAvatar, userAvatar: userAvatarProp, assistantAvatar: assistantAvatarProp, loaderVariant, fontSize, maxFileSize, allowedFileTypes, attachmentsEnabled, attachmentsDisabledTooltip, processAttachment: processAttachmentProp, suggestions, onSuggestionClick, welcome, recentThreads, onSelectThread, onDeleteThread, onViewMoreThreads, isProcessing, registeredTools, toolRenderers, onApproveToolExecution, onRejectToolExecution, showFollowUps, followUpClassName, followUpButtonClassName, citations, renderMessage, renderInput, renderHeader, className, classNames, onNewChat, threads, currentThreadId, onSwitchThread, isThreadBusy, }: ChatProps): react_jsx_runtime.JSX.Element;
+declare function ChatComponent({ messages, onSendMessage, onStop, isLoading, children, placeholder, welcomeMessage, title, showHeader, header, threadPicker, logo, name, onClose, showPoweredBy, showUserAvatar, userAvatar: userAvatarProp, assistantAvatar: assistantAvatarProp, loaderVariant, fontSize, maxFileSize, allowedFileTypes, attachmentsEnabled, attachmentsDisabledTooltip, processAttachment: processAttachmentProp, suggestions, onSuggestionClick, welcome, recentThreads, onSelectThread, onDeleteThread, onViewMoreThreads, isProcessing, registeredTools, toolRenderers, mcpToolRenderer, onApproveToolExecution, onRejectToolExecution, showFollowUps, followUpClassName, followUpButtonClassName, citations, renderMessage, renderInput, renderHeader, className, classNames, onNewChat, threads, currentThreadId, onSwitchThread, isThreadBusy, }: ChatProps): react_jsx_runtime.JSX.Element;
 /**
  * Chat component with compound component pattern.
  *

package/dist/ui/index.js

@@ -1,9 +1,9 @@
-import { useCopilot, useThreadManager } from '../chunk-IXFV6AW6.js';
-import { createServerAdapter } from '../chunk-7XFFRV7D.js';
+import { useCopilot, useThreadManager } from '../chunk-3SP6V22Y.js';
+import { createServerAdapter } from '../chunk-7PJ4GAJR.js';
 import { DEFAULT_MCP_UI_SANDBOX, parseMCPUIMessage } from '../chunk-G4SF2PNQ.js';
-import '../chunk-BH7MNDWW.js';
-import '../chunk-ZPYQDMUX.js';
-import '../chunk-BKO7DSPU.js';
+import '../chunk-EWVQWTNV.js';
+import '../chunk-7YZVSG62.js';
+import '../chunk-BRUDSHCV.js';
 import '../chunk-XGITAEXU.js';
 import '../chunk-7W7QLZNC.js';
 import '../chunk-CEOMTQTP.js';
@@ -328,7 +328,8 @@ var headingComponents = {
 function MarkdownComponent({
   children,
   className,
-  isStreaming = false
+  isStreaming = false,
+  linkSafety = { enabled: false }
 }) {
   return /* @__PURE__ */ jsx("div", { className, children: /* @__PURE__ */ jsx(
     Streamdown,
@@ -336,6 +337,7 @@ function MarkdownComponent({
       plugins: { code },
       isAnimating: isStreaming,
       components: headingComponents,
+      linkSafety,
       children
     }
   ) });
@@ -1903,8 +1905,8 @@ function ToolStep({
         "aria-hidden": "true"
       }
     ),
-    /* @__PURE__ */ jsxs("div", { className: "flex items-start gap-2", children: [
-      !isExecuting && /* @__PURE__ */ jsx(StatusIndicator, { status: step.status, className: "mt-0.5" }),
+    /* @__PURE__ */ jsxs("div", { className: "flex items-center gap-2", children: [
+      !isExecuting && /* @__PURE__ */ jsx(StatusIndicator, { status: step.status }),
       /* @__PURE__ */ jsxs("div", { className: "flex-1 min-w-0", children: [
         /* @__PURE__ */ jsxs(
           "button",
@@ -4226,6 +4228,7 @@ function DefaultMessage({
   loaderVariant = "typing",
   registeredTools,
   toolRenderers,
+  mcpToolRenderer,
   onApproveToolExecution,
   onRejectToolExecution,
   showFollowUps = true,
@@ -4347,17 +4350,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) => ({
@@ -4407,7 +4412,7 @@ function DefaultMessage({
               children: cleanContent
             }
           ),
-          toolsWithCustomRender && toolsWithCustomRender.length > 0 && /* @__PURE__ */ jsx("div", { className: "mt-2 space-y-2", children: toolsWithCustomRender.map((exec) => {
+          toolsWithCustomRender && toolsWithCustomRender.length > 0 && /* @__PURE__ */ jsx("div", { className: cn("space-y-2", cleanContent?.trim() && "mt-2"), children: toolsWithCustomRender.map((exec) => {
             const Renderer = toolRenderers?.[exec.name];
             if (Renderer) {
               return /* @__PURE__ */ jsx(
@@ -4420,7 +4425,8 @@ function DefaultMessage({
                     status: exec.status,
                     result: exec.result,
                     error: exec.error,
-                    approvalStatus: exec.approvalStatus
+                    approvalStatus: exec.approvalStatus,
+                    source: exec.source
                   }
                 },
                 exec.id
@@ -4429,7 +4435,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__ */ 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";
@@ -4443,14 +4468,23 @@ function DefaultMessage({
                 toolCallId: exec.id,
                 toolName: exec.name
               };
-              const output = toolDef.render(
+              const output = toolDefForRender.render(
                 renderProps
               );
               return /* @__PURE__ */ jsx(React20.Fragment, { children: output }, exec.id);
             }
             return null;
           }) }),
-          toolSteps && toolSteps.length > 0 && /* @__PURE__ */ jsx("div", { className: "mt-2 rounded-lg bg-muted/50 px-3 py-2", children: /* @__PURE__ */ jsx(ToolSteps, { steps: toolSteps }) }),
+          toolSteps && toolSteps.length > 0 && /* @__PURE__ */ jsx(
+            "div",
+            {
+              className: cn(
+                "rounded-lg bg-muted/50 px-3 py-2",
+                cleanContent?.trim() && "mt-2"
+              ),
+              children: /* @__PURE__ */ 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;
@@ -5229,6 +5263,7 @@ function ChatComponent({
   isProcessing = false,
   registeredTools,
   toolRenderers,
+  mcpToolRenderer,
   onApproveToolExecution,
   onRejectToolExecution,
   // Follow-up Questions
@@ -5556,6 +5591,7 @@ function ChatComponent({
                             loaderVariant,
                             registeredTools,
                             toolRenderers,
+                            mcpToolRenderer,
                             onApproveToolExecution,
                             onRejectToolExecution,
                             showFollowUps,
@@ -5963,9 +5999,6 @@ function useInternalThreadManager(config = {}) {
     });
   }, [clearCurrentThread, setMessages, onThreadChange]);
   useEffect(() => {
-    if (!adapter) {
-      return;
-    }
     if (hasInitializedRef.current || !currentThread) {
       return;
     }
@@ -5997,9 +6030,6 @@ function useInternalThreadManager(config = {}) {
     });
   }, [adapter, currentThread, setMessages, getMessageSnapshot, onThreadChange]);
   useEffect(() => {
-    if (!adapter) {
-      return;
-    }
     if (isLoadingMessagesRef.current) {
       return;
     }