@docyrus/ui-pro-ai-assistant 0.5.8 → 0.6.1

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';
@@ -14539,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",
@@ -14609,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);
@@ -14637,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) {
@@ -31870,6 +31960,7 @@ var DocyAssistant = ({
   initialFeatures,
   initialFiles,
   hostEnvironment,
+  clientTools,
   ...props
 }) => {
   const config3 = useAssistantConfig();
@@ -31915,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]);
@@ -31948,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 } : {}
           }
         };
@@ -31999,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;
@@ -32867,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,
@@ -32880,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: [