@docyrus/ui-pro-ai-assistant 0.5.7 → 0.6.0

package/dist/components/input-area.d.ts

@@ -48,6 +48,7 @@ interface AIInputAreaProps {
     tenantAiProjectId?: string;
     initialModelId?: string;
     initialFeatures?: InitialFeatureFlags;
+    onManageMemories?: () => void;
 }
 export declare const AIInputArea: FC<AIInputAreaProps>;
 export {};

package/dist/docy-assistant.d.ts

@@ -1,5 +1,5 @@
 import { type RefObject } from 'react';
 import { type DocyAssistantProps } from './types';
-export declare const DocyAssistant: ({ ref, isOpen, onClose, supportWebSearch, supportThinking, supportFiles, supportDocumentSearch, supportDeepResearch, supportMultiModels, supportWorkCanvas, apiEndpoint, title: titleProp, description: descriptionProp, placeholder: placeholderProp, logo, footerText: footerTextProp, variant, renderMode, enableSidebar, enableNavDropdown, enableVoice, enableMicrophone, enableWelcomePage, tenantAiAgentId, onMessageSend, onVoiceStart, onVoiceEnd, className, defaultFullscreen, hideExpand, hideCloseButton, hideAgentSelector, hideBorder, showHeader, hideHeaderOnWelcome, agentSelectorUrl, baseAgentSelectorUrl, onAgentChange, enableSharing, onShare: onShareProp, initialPrompt, initialModelId, initialFeatures, initialFiles, hostEnvironment, ...props }: DocyAssistantProps & {
+export declare const DocyAssistant: ({ ref, isOpen, onClose, supportWebSearch, supportThinking, supportFiles, supportDocumentSearch, supportDeepResearch, supportMultiModels, supportWorkCanvas, apiEndpoint, title: titleProp, description: descriptionProp, placeholder: placeholderProp, logo, footerText: footerTextProp, variant, renderMode, enableSidebar, enableNavDropdown, enableVoice, enableMicrophone, enableWelcomePage, tenantAiAgentId, onMessageSend, onVoiceStart, onVoiceEnd, className, defaultFullscreen, hideExpand, hideCloseButton, hideAgentSelector, hideBorder, showHeader, hideHeaderOnWelcome, agentSelectorUrl, baseAgentSelectorUrl, onAgentChange, enableSharing, onShare: onShareProp, initialPrompt, initialModelId, initialFeatures, initialFiles, hostEnvironment, clientTools, ...props }: DocyAssistantProps & {
     ref?: RefObject<HTMLDivElement | null>;
 }) => import("react/jsx-runtime").JSX.Element;

package/dist/index.d.ts

@@ -3,5 +3,5 @@ export { type AssistantConfig, AssistantProvider, type AssistantUser, type DataS
 export { AssistantI18nProvider, useAssistantTranslation } from './i18n';
 export { AgentTriggerWidget, type AgentTriggerWidgetProps } from './components/agent-trigger-widget';
 export { type PromptFeatureFlags } from './components/agent-trigger-widget/agent-trigger-prompt';
-export type { InitialFeatureFlags, Principal, PrincipalType, Project, ShareItem, Work, WorkTypes } from './types';
+export type { ClientTool, ClientToolExecuteContext, InitialFeatureFlags, Principal, PrincipalType, Project, ShareItem, Work, WorkTypes } from './types';
 export { BASE_DATA_SOURCE_ID, SHARE_PERMISSIONS, SHARE_PERMISSION_LEVELS } from './types';

package/dist/index.js

@@ -9,7 +9,7 @@ import { Input } from '@docyrus/ui-pro-shared/components/input';
 import { Tabs as Tabs$1, TabsList as TabsList$1, TabsTrigger as TabsTrigger$1, TabsContent as TabsContent$1 } from '@docyrus/ui-pro-shared/components/tabs';
 import { cn } from '@docyrus/ui-pro-shared/lib/utils';
 import { DefaultChatTransport, lastAssistantMessageIsCompleteWithToolCalls } from 'ai';
-import { Pencil, PilcrowIcon, Heading1Icon, Heading2Icon, Heading3Icon, SquareIcon, ListIcon, ListOrderedIcon, ChevronDownIcon, Code2Icon, QuoteIcon, LightbulbIcon, Columns3Icon, GripVertical, FileUpIcon, TableIcon, ImageIcon, FilmIcon, AudioLinesIcon, TableOfContentsIcon, RadicalIcon, RectangleVerticalIcon, CalendarIcon, PlusIcon, FileCodeIcon, MinusIcon, ChevronRightIcon, ListTree, PenToolIcon, Link2Icon, Check, X, Copy, FilesIcon, Link, Text, ExternalLink, Unlink, Bold, Italic, Underline, Strikethrough, Code2, MoreHorizontal, ArrowLeftIcon, ArrowRightIcon, Minus, Plus, CircleArrowDown, Minimize2, Trash2, FileUp, CheckCircle2, FileText, Loader2, CornerDownLeftIcon, PencilLineIcon, MessageSquareTextIcon, MessagesSquareIcon, ArrowUpIcon, CheckIcon, CaptionsIcon, ZoomInIcon, CircleArrowDownIcon, MoveUpRightIcon, MoreHorizontalIcon, Eye, PlusCircle, HelpCircle, Maximize2, Download, FileSpreadsheet, ChevronDown, ChevronLeft, ChevronRight, CheckCircle, XCircle, Search, Globe, User, ArrowRight, MapPin, CalendarClock, List, RefreshCw, FilePlus, XIcon, CornerUpLeftIcon, AlbumIcon, FeatherIcon, ListMinusIcon, ListPlusIcon, ListEnd, Wand, LanguagesIcon, BadgeHelpIcon, PenLineIcon, SearchIcon, MusicIcon, CompassIcon, SmileIcon, LeafIcon, ClockIcon, AppleIcon, FlagIcon, StarIcon, DeleteIcon, AlignLeft, AlignCenter, AlignRight, RotateCcw, AlertTriangle, AlertCircle, ArrowUpDown, ArrowDownToLine, PenLine, PencilIcon, TrashIcon, FileIcon, Star, ChevronsUpDown, FolderIcon, MessageSquare, Hash, Table, CheckSquare, Calendar as Calendar$1, Clock, Phone, Mail, Send, RefreshCcw, Undo2, ChevronsUpDownIcon, ChevronsRight, ChevronsLeft, CalendarDays, RefreshCwIcon, PaintRoller, MessageSquareText, ArrowLeft, Users, Bot, Brain, Lightbulb, BookOpen, PenTool, SlidersHorizontal, Wand2, Mic, Code, Trash, Edit, Share, FolderInput, Archive, Building2, ShieldCheck, Sparkles, PanelLeft, NotebookText, CirclePlus, MoreVertical, FileSearch, Microscope, Ruler, AudioLines, BrainCircuit, File as File$1, Plug, Inbox, Menu, FolderOpen, LayoutDashboard, Table2, WandSparklesIcon, ArrowUpToLineIcon, BoldIcon, ItalicIcon, UnderlineIcon, StrikethroughIcon, HighlighterIcon, Undo2Icon, Redo2Icon, ArrowDownToLineIcon, AlignLeftIcon, AlignCenterIcon, AlignRightIcon, AlignJustifyIcon, ListOrdered, ListTodoIcon, ListCollapseIcon, Grid3x3Icon, Combine, Ungroup, ArrowUp, ArrowDown, Trash2Icon, LinkIcon, WrapText, OutdentIcon, IndentIcon, EyeIcon, PenIcon } from 'lucide-react';
+import { Pencil, PilcrowIcon, Heading1Icon, Heading2Icon, Heading3Icon, SquareIcon, ListIcon, ListOrderedIcon, ChevronDownIcon, Code2Icon, QuoteIcon, LightbulbIcon, Columns3Icon, GripVertical, FileUpIcon, TableIcon, ImageIcon, FilmIcon, AudioLinesIcon, TableOfContentsIcon, RadicalIcon, RectangleVerticalIcon, CalendarIcon, PlusIcon, FileCodeIcon, MinusIcon, ChevronRightIcon, ListTree, PenToolIcon, Link2Icon, Check, X, Copy, FilesIcon, Link, Text, ExternalLink, Unlink, Bold, Italic, Underline, Strikethrough, Code2, MoreHorizontal, ArrowLeftIcon, ArrowRightIcon, Minus, Plus, CircleArrowDown, Minimize2, Trash2, FileUp, CheckCircle2, FileText, Loader2, Wrench, CornerDownLeftIcon, PencilLineIcon, MessageSquareTextIcon, MessagesSquareIcon, ArrowUpIcon, CheckIcon, CaptionsIcon, ZoomInIcon, CircleArrowDownIcon, MoveUpRightIcon, MoreHorizontalIcon, Eye, PlusCircle, HelpCircle, Maximize2, Download, FileSpreadsheet, ChevronDown, ChevronLeft, ChevronRight, CheckCircle, XCircle, Search, Globe, User, ArrowRight, MapPin, CalendarClock, List, RefreshCw, FilePlus, XIcon, CornerUpLeftIcon, AlbumIcon, FeatherIcon, ListMinusIcon, ListPlusIcon, ListEnd, Wand, LanguagesIcon, BadgeHelpIcon, PenLineIcon, SearchIcon, MusicIcon, CompassIcon, SmileIcon, LeafIcon, ClockIcon, AppleIcon, FlagIcon, StarIcon, DeleteIcon, AlignLeft, AlignCenter, AlignRight, RotateCcw, AlertTriangle, AlertCircle, ArrowUpDown, ArrowDownToLine, PenLine, PencilIcon, TrashIcon, FileIcon, Star, ChevronsUpDown, FolderIcon, MessageSquare, Hash, Table, CheckSquare, Calendar as Calendar$1, Clock, Phone, Mail, Send, RefreshCcw, Undo2, ChevronsUpDownIcon, ChevronsRight, ChevronsLeft, CalendarDays, RefreshCwIcon, PaintRoller, MessageSquareText, ArrowLeft, Users, Bot, Brain, Lightbulb, BookOpen, PenTool, SlidersHorizontal, Wand2, Mic, Code, Trash, Edit, Share, FolderInput, Archive, Building2, ShieldCheck, Sparkles, PanelLeft, NotebookText, CirclePlus, MoreVertical, FileSearch, Microscope, Ruler, AudioLines, BrainCircuit, File as File$1, Plug, Inbox, Menu, FolderOpen, LayoutDashboard, Table2, WandSparklesIcon, ArrowUpToLineIcon, BoldIcon, ItalicIcon, UnderlineIcon, StrikethroughIcon, HighlighterIcon, Undo2Icon, Redo2Icon, ArrowDownToLineIcon, AlignLeftIcon, AlignCenterIcon, AlignRightIcon, AlignJustifyIcon, ListOrdered, ListTodoIcon, ListCollapseIcon, Grid3x3Icon, Combine, Ungroup, ArrowUp, ArrowDown, Trash2Icon, LinkIcon, WrapText, OutdentIcon, IndentIcon, EyeIcon, PenIcon } from 'lucide-react';
 import { jsx, jsxs, Fragment } from 'react/jsx-runtime';
 import { ScrollArea as ScrollArea$1 } from '@docyrus/ui-pro-shared/components/scroll-area';
 import { AsyncTokenManager, RestApiClient } from '@docyrus/api-client';
@@ -5522,7 +5522,8 @@ var AIInputArea = ({
   hideFooter = false,
   tenantAiProjectId,
   initialModelId,
-  initialFeatures
+  initialFeatures,
+  onManageMemories
 }) => {
   const { t } = useAssistantTranslation();
   const apiClient = useApiClient();
@@ -6237,20 +6238,38 @@ var AIInputArea = ({
     ),
     !hideFooter && /* @__PURE__ */ jsxs("div", { className: "relative mt-2 flex items-center px-1", children: [
       footerText && /* @__PURE__ */ jsx("span", { className: "flex-1 text-center text-xs text-muted-foreground", children: footerText }),
-      /* @__PURE__ */ jsxs(Tooltip$2, { children: [
-        /* @__PURE__ */ jsx(TooltipTrigger$1, { asChild: true, children: /* @__PURE__ */ jsx(
-          Button,
-          {
-            type: "button",
-            variant: "ghost",
-            size: "icon",
-            className: "absolute right-0 h-6 w-6 text-muted-foreground",
-            disabled: !threadId || !hasMessages,
-            onClick: handleExtractMemories,
-            children: /* @__PURE__ */ jsx(BrainCircuit, { className: "h-3.5 w-3.5" })
-          }
-        ) }),
-        /* @__PURE__ */ jsx(TooltipContent$1, { side: "top", children: t("tools.extract_memories") })
+      /* @__PURE__ */ jsxs(DropdownMenu$1, { children: [
+        /* @__PURE__ */ jsxs(Tooltip$2, { children: [
+          /* @__PURE__ */ jsx(TooltipTrigger$1, { asChild: true, children: /* @__PURE__ */ jsx(DropdownMenuTrigger, { asChild: true, children: /* @__PURE__ */ jsx(
+            Button,
+            {
+              type: "button",
+              variant: "ghost",
+              size: "icon",
+              className: "absolute right-0 h-6 w-6 text-muted-foreground",
+              children: /* @__PURE__ */ jsx(BrainCircuit, { className: "h-3.5 w-3.5" })
+            }
+          ) }) }),
+          /* @__PURE__ */ jsx(TooltipContent$1, { side: "top", children: t("tools.extract_memories") })
+        ] }),
+        /* @__PURE__ */ jsxs(DropdownMenuContent, { align: "end", side: "top", className: "w-60", children: [
+          /* @__PURE__ */ jsx(
+            DropdownMenuItem,
+            {
+              disabled: !threadId || !hasMessages,
+              onClick: handleExtractMemories,
+              children: "Extract Memories from Session"
+            }
+          ),
+          /* @__PURE__ */ jsx(
+            DropdownMenuItem,
+            {
+              disabled: !onManageMemories,
+              onClick: () => onManageMemories?.(),
+              children: "Manage Memories"
+            }
+          )
+        ] })
       ] })
     ] }),
     /* @__PURE__ */ jsx(
@@ -14520,6 +14539,78 @@ function GenerativeUIObject({
   }
   return null;
 }
+var EMPTY_MAP = /* @__PURE__ */ new Map();
+var ClientToolsContext = createContext(EMPTY_MAP);
+function ClientToolsProvider({
+  tools,
+  children
+}) {
+  return /* @__PURE__ */ jsx(ClientToolsContext, { value: tools, children });
+}
+function useClientToolMap() {
+  return use(ClientToolsContext);
+}
+function buildClientToolMap(tools) {
+  const map = /* @__PURE__ */ new Map();
+  for (const tool of tools ?? []) {
+    if (tool?.name && !map.has(tool.name)) {
+      map.set(tool.name, tool);
+    }
+  }
+  return map;
+}
+function serializeClientTools(tools) {
+  return (tools ?? []).filter((tool) => !!tool?.name).map((tool) => ({
+    name: tool.name,
+    description: tool.description,
+    parameters: tool.parameters ?? {
+      $schema: "http://json-schema.org/draft-07/schema#",
+      type: "object",
+      additionalProperties: false,
+      properties: {}
+    }
+  }));
+}
+async function runClientTool(tool, input, context) {
+  try {
+    const result = await tool.execute(input, context);
+    return { success: true, result: result ?? null };
+  } catch (error) {
+    return {
+      success: false,
+      error: error instanceof Error ? error.message : String(error)
+    };
+  }
+}
+function resolveStatus(part) {
+  if (part.state === "output-error") return "error";
+  if (part.state === "output-available") {
+    return part.output && part.output.success === false ? "error" : "success";
+  }
+  return "running";
+}
+function ClientToolCall({ tool, part }) {
+  const status = resolveStatus(part);
+  const title = status === "running" ? tool.progressMessage ?? `Running ${tool.name}\u2026` : status === "success" ? tool.successMessage ?? `${tool.name} completed` : tool.failMessage ?? `${tool.name} failed`;
+  const errorDetail = status === "error" ? part.output?.error || part.errorText || void 0 : void 0;
+  return /* @__PURE__ */ jsxs(
+    "div",
+    {
+      className: cn(
+        "my-2 flex items-center gap-2.5 rounded-lg border px-3 py-2 text-sm",
+        status === "error" ? "border-destructive/40 bg-destructive/5" : "border-border bg-muted/40"
+      ),
+      children: [
+        /* @__PURE__ */ jsx("span", { className: "flex h-5 w-5 shrink-0 items-center justify-center text-muted-foreground", children: status === "running" ? /* @__PURE__ */ jsx(Spinner, { className: "h-4 w-4" }) : status === "success" ? /* @__PURE__ */ jsx(Check, { className: "h-4 w-4 text-emerald-600 dark:text-emerald-500" }) : /* @__PURE__ */ jsx(X, { className: "h-4 w-4 text-destructive" }) }),
+        /* @__PURE__ */ jsx("span", { className: "flex h-5 w-5 shrink-0 items-center justify-center text-muted-foreground", children: tool.icon ?? /* @__PURE__ */ jsx(Wrench, { className: "h-4 w-4" }) }),
+        /* @__PURE__ */ jsxs("div", { className: "min-w-0 flex-1", children: [
+          /* @__PURE__ */ jsx("div", { className: "truncate font-medium text-foreground", children: title }),
+          errorDetail ? /* @__PURE__ */ jsx("div", { className: "truncate text-xs text-destructive", children: errorDetail }) : null
+        ] })
+      ]
+    }
+  );
+}
 var isToolPart = (part) => part.type?.startsWith("tool-") || part.type === "dynamic-tool";
 var DISPLAY_ONLY_TOOLS = /* @__PURE__ */ new Set([
   "generateChart",
@@ -14590,6 +14681,7 @@ var MessageItem = memo(({
   onEditPrompt
 }) => {
   const { t } = useAssistantTranslation();
+  const clientToolMap = useClientToolMap();
   const isAssistantMessage = message.role === "assistant";
   const isUserMessage = message.role === "user";
   const [isEditing, setIsEditing] = useState(false);
@@ -14618,6 +14710,23 @@ var MessageItem = memo(({
   }, [message.parts]);
   const renderToolPart = (part, messageId, idx) => {
     const toolName = part.type === "dynamic-tool" ? part.toolName : part.type?.startsWith("tool-") ? part.type.replace(/^tool-/, "") : part.toolName || "Unknown Tool";
+    const clientTool = clientToolMap.get(toolName);
+    if (clientTool) {
+      return /* @__PURE__ */ jsx(
+        ClientToolCall,
+        {
+          tool: clientTool,
+          part: {
+            state: part.state || "input-available",
+            input: part.args || part.input,
+            output: part.output,
+            errorText: part.errorText,
+            toolCallId: part.toolCallId || `${messageId}-${idx}`
+          }
+        },
+        `${messageId}-clienttool-${idx}`
+      );
+    }
     const isSubagentTool = toolName.startsWith("call") && toolName.endsWith("Agent");
     const isGenerativeTool = DISPLAY_ONLY_TOOLS.has(toolName) || INTERACTIVE_GENERATIVE_TOOLS.has(toolName) && !!onToolAction || isSubagentTool && (part.state === "input-available" || part.state === "input-streaming" || part.state === "output-available");
     if (isGenerativeTool) {
@@ -14875,7 +14984,8 @@ function ChatPanel({
   threadId,
   initialModelId,
   initialFeatures,
-  subagents
+  subagents,
+  onManageMemories
 }) {
   const renderInputArea = (hideFooter = false) => /* @__PURE__ */ jsx(
     AIInputArea,
@@ -14907,7 +15017,8 @@ function ChatPanel({
       hasMessages: messages.length > 0,
       hideFooter,
       initialModelId,
-      initialFeatures
+      initialFeatures,
+      onManageMemories
     }
   );
   const isWelcome = showWelcome && messages.length === 0;
@@ -30332,21 +30443,7 @@ var AssistantView = ({ ref, ...props }) => {
                   children: /* @__PURE__ */ jsx(PanelLeft, { className: "w-4 h-4" })
                 }
               ),
-              onTabChange && /* @__PURE__ */ jsx(
-                Button,
-                {
-                  variant: "ghost",
-                  size: "icon",
-                  onClick: () => onTabChange(activeTab === 4 ? 0 : 4),
-                  className: cn(
-                    "h-8 w-8 text-foreground hover:bg-accent hover:text-accent-foreground rounded-md",
-                    activeTab === 4 && "bg-background text-foreground shadow"
-                  ),
-                  title: t("tabs.memories"),
-                  children: /* @__PURE__ */ jsx(Brain, { className: "w-4 h-4" })
-                }
-              ),
-              isFullscreen && onNewChat && /* @__PURE__ */ jsx(
+              onNewChat && /* @__PURE__ */ jsx(
                 Button,
                 {
                   variant: "ghost",
@@ -30397,32 +30494,16 @@ var AssistantView = ({ ref, ...props }) => {
                   isSidebarOpen ? "w-72" : "w-0"
                 ),
                 children: /* @__PURE__ */ jsxs("div", { className: "w-72 h-full flex flex-col relative", children: [
-                  isSidebarOpen && /* @__PURE__ */ jsxs("div", { className: "absolute top-2 right-2 z-10 flex flex-col gap-1", children: [
-                    onToggleSidebar && /* @__PURE__ */ jsx(
-                      Button,
-                      {
-                        variant: "ghost",
-                        size: "icon",
-                        onClick: onToggleSidebar,
-                        className: "h-8 w-8 text-muted-foreground hover:bg-accent hover:text-accent-foreground rounded-md",
-                        children: /* @__PURE__ */ jsx(PanelLeft, { className: "w-4 h-4" })
-                      }
-                    ),
-                    onTabChange && /* @__PURE__ */ jsx(
-                      Button,
-                      {
-                        variant: "ghost",
-                        size: "icon",
-                        onClick: () => onTabChange(activeTab === 4 ? 0 : 4),
-                        className: cn(
-                          "h-8 w-8 text-muted-foreground hover:bg-accent hover:text-accent-foreground rounded-md",
-                          activeTab === 4 && "bg-accent text-accent-foreground"
-                        ),
-                        title: t("tabs.memories"),
-                        children: /* @__PURE__ */ jsx(Brain, { className: "w-4 h-4" })
-                      }
-                    )
-                  ] }),
+                  isSidebarOpen && /* @__PURE__ */ jsx("div", { className: "absolute top-2 right-2 z-10 flex flex-col gap-1", children: onToggleSidebar && /* @__PURE__ */ jsx(
+                    Button,
+                    {
+                      variant: "ghost",
+                      size: "icon",
+                      onClick: onToggleSidebar,
+                      className: "h-8 w-8 text-muted-foreground hover:bg-accent hover:text-accent-foreground rounded-md",
+                      children: /* @__PURE__ */ jsx(PanelLeft, { className: "w-4 h-4" })
+                    }
+                  ) }),
                   /* @__PURE__ */ jsxs("div", { className: "flex-1 p-4 flex flex-col min-h-0 overflow-hidden", children: [
                     !showHeader && agentSelectorElement && /* @__PURE__ */ jsx("div", { className: "mb-3 shrink-0", children: agentSelectorElement }),
                     renderSidebar(false)
@@ -30539,6 +30620,7 @@ var AssistantView = ({ ref, ...props }) => {
                   initialModelId: commonProps.initialModelId,
                   initialFeatures: commonProps.initialFeatures,
                   subagents: commonProps.subagents,
+                  onManageMemories: onTabChange ? () => onTabChange(4) : void 0,
                   compactToolbar: !isFullscreen
                 }
               )
@@ -31878,6 +31960,7 @@ var DocyAssistant = ({
   initialFeatures,
   initialFiles,
   hostEnvironment,
+  clientTools,
   ...props
 }) => {
   const config3 = useAssistantConfig();
@@ -31923,6 +32006,14 @@ var DocyAssistant = ({
   const authTokenRef = useRef("");
   const messageOptionsRef = useRef(null);
   const initialPromptSentRef = useRef(false);
+  const clientToolMap = useMemo(() => buildClientToolMap(clientTools), [clientTools]);
+  const clientToolDefs = useMemo(() => serializeClientTools(clientTools), [clientTools]);
+  const clientToolMapRef = useRef(clientToolMap);
+  clientToolMapRef.current = clientToolMap;
+  const clientToolDefsRef = useRef(clientToolDefs);
+  clientToolDefsRef.current = clientToolDefs;
+  const addToolOutputRef = useRef(null);
+  const executedClientToolCallsRef = useRef(/* @__PURE__ */ new Set());
   useEffect(() => {
     selectedSessionIdRef.current = sessionState.selectedSessionId;
   }, [sessionState.selectedSessionId]);
@@ -31956,6 +32047,7 @@ var DocyAssistant = ({
             supportThinking: options3?.supportThinking,
             supportWorkCanvas: options3?.supportWorkCanvas,
             ...hostEnvironment ? { hostEnvironment } : {},
+            ...clientToolDefsRef.current.length ? { clientTools: clientToolDefsRef.current } : {},
             ...options3?.filePaths?.length ? { files: options3.filePaths } : {}
           }
         };
@@ -32007,8 +32099,20 @@ var DocyAssistant = ({
     },
     onError: (error) => {
       console.error("[AI] Chat error:", error);
+    },
+    onToolCall: async ({ toolCall }) => {
+      const name = toolCall?.toolName;
+      const tool = name ? clientToolMapRef.current.get(name) : void 0;
+      if (!tool) return;
+      const { toolCallId } = toolCall;
+      if (executedClientToolCallsRef.current.has(toolCallId)) return;
+      executedClientToolCallsRef.current.add(toolCallId);
+      const input2 = toolCall.input ?? toolCall.args ?? {};
+      const output = await runClientTool(tool, input2, { toolCallId });
+      addToolOutputRef.current?.({ tool: name, toolCallId, output });
     }
   });
+  addToolOutputRef.current = addToolOutput;
   const isStreaming = status === "streaming";
   const isSubmitting = status === "submitted";
   const isLoading = isStreaming || isSubmitting;
@@ -32875,7 +32979,7 @@ var DocyAssistant = ({
     )
   ] });
   if (isInlineMode) {
-    return /* @__PURE__ */ jsxs(Fragment, { children: [
+    return /* @__PURE__ */ jsxs(ClientToolsProvider, { tools: clientToolMap, children: [
       renderInlineViewContent({
         isFullscreen: isInlineFullscreen,
         enableSidebar: isInlineFullscreen ? true : enableSidebar,
@@ -32888,7 +32992,7 @@ var DocyAssistant = ({
       assistantDialogs
     ] });
   }
-  return /* @__PURE__ */ jsxs(Fragment, { children: [
+  return /* @__PURE__ */ jsxs(ClientToolsProvider, { tools: clientToolMap, children: [
     /* @__PURE__ */ jsxs(Dialog$1, { open: isOpen, onOpenChange: (o) => {
       if (!o) onClose?.();
     }, children: [