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

package/dist/components/mermaid-diagram.d.ts

@@ -4,5 +4,5 @@ interface MermaidDiagramProps {
     mermaidNotation: string;
     state?: string;
 }
-export declare function MermaidDiagram({ title, description, mermaidNotation, state }: MermaidDiagramProps): import("react/jsx-runtime").JSX.Element;
+export declare function MermaidDiagram({ title, description, mermaidNotation }: MermaidDiagramProps): import("react/jsx-runtime").JSX.Element;
 export {};

package/dist/index.js

@@ -12410,10 +12410,16 @@ function MermaidRender({ content, className }) {
     setZoom(1);
     setPan({ x: 0, y: 0 });
   };
-  const onWheel = useCallback((e) => {
-    e.preventDefault();
-    const delta = e.deltaY > 0 ? -0.1 : 0.1;
-    setZoom((z) => Math.min(Math.max(z + delta, 0.2), 4));
+  useEffect(() => {
+    const el = containerRef.current;
+    if (!el) return;
+    const handleWheel = (e) => {
+      e.preventDefault();
+      const delta = e.deltaY > 0 ? -0.1 : 0.1;
+      setZoom((z) => Math.min(Math.max(z + delta, 0.2), 4));
+    };
+    el.addEventListener("wheel", handleWheel, { passive: false });
+    return () => el.removeEventListener("wheel", handleWheel);
   }, []);
   const onMouseDown = useCallback((e) => {
     isDragging.current = true;
@@ -12438,7 +12444,6 @@ function MermaidRender({ content, className }) {
         "bg-[radial-gradient(circle,_#94a3b8_1px,_transparent_1px)] bg-[size:22px_22px] bg-slate-50",
         className
       ),
-      onWheel,
       onMouseDown,
       onMouseMove,
       onMouseUp,
@@ -12465,11 +12470,10 @@ function MermaidRender({ content, className }) {
 function MermaidDiagram({
   title,
   description,
-  mermaidNotation,
-  state
+  mermaidNotation
 }) {
   const [modalOpen, setModalOpen] = useState(false);
-  if (state !== "output-available") {
+  if (!mermaidNotation) {
     return /* @__PURE__ */ jsx("div", { className: "w-full h-[400px] rounded-lg border bg-muted/30 animate-pulse" });
   }
   return /* @__PURE__ */ jsxs(Fragment, { children: [
@@ -32006,6 +32010,7 @@ var DocyAssistant = ({
   const authTokenRef = useRef("");
   const messageOptionsRef = useRef(null);
   const initialPromptSentRef = useRef(false);
+  const isSendingRef = useRef(false);
   const clientToolMap = useMemo(() => buildClientToolMap(clientTools), [clientTools]);
   const clientToolDefs = useMemo(() => serializeClientTools(clientTools), [clientTools]);
   const clientToolMapRef = useRef(clientToolMap);
@@ -32282,75 +32287,82 @@ var DocyAssistant = ({
   };
   const handleSendMessage = async (e, options3, overrideText) => {
     e?.preventDefault();
+    if (isSendingRef.current) return;
+    if (status === "submitted" || status === "streaming") return;
     const messageText = (overrideText ?? input).trim();
     if (!messageText) return;
+    isSendingRef.current = true;
+    setInput("");
     setShowWelcome(false);
-    messageOptionsRef.current = options3;
-    let currentThreadId = selectedSessionIdRef.current;
-    if (!currentThreadId) {
-      const subject = messageText.substring(0, 100);
-      const newThread = await createThread(apiClient, {
-        subject,
-        body_text: messageText,
-        sender_name: userDisplayName,
-        deploymentId,
-        tenantAiAgentId: activeAgentId,
-        projectId: projectState.projectContext?.id
-      });
-      if (newThread) {
-        currentThreadId = newThread.id;
-        sessionActions.selectSessionId(currentThreadId);
-        selectedSessionIdRef.current = currentThreadId;
-        const session = {
-          id: newThread.id,
-          title: newThread.subject || subject,
-          messages: [],
-          createdAt: new Date(newThread.created_on || /* @__PURE__ */ new Date()),
-          updatedAt: new Date(newThread.last_modified_on || /* @__PURE__ */ new Date())
-        };
-        sessionActions.addSession(session);
-        fetchThreads();
-        if (projectState.projectContext?.id) {
-          fetchProjectThreads2(projectState.projectContext.id);
+    try {
+      messageOptionsRef.current = options3;
+      let currentThreadId = selectedSessionIdRef.current;
+      if (!currentThreadId) {
+        const subject = messageText.substring(0, 100);
+        const newThread = await createThread(apiClient, {
+          subject,
+          body_text: messageText,
+          sender_name: userDisplayName,
+          deploymentId,
+          tenantAiAgentId: activeAgentId,
+          projectId: projectState.projectContext?.id
+        });
+        if (newThread) {
+          currentThreadId = newThread.id;
+          sessionActions.selectSessionId(currentThreadId);
+          selectedSessionIdRef.current = currentThreadId;
+          const session = {
+            id: newThread.id,
+            title: newThread.subject || subject,
+            messages: [],
+            createdAt: new Date(newThread.created_on || /* @__PURE__ */ new Date()),
+            updatedAt: new Date(newThread.last_modified_on || /* @__PURE__ */ new Date())
+          };
+          sessionActions.addSession(session);
+          fetchThreads();
+          if (projectState.projectContext?.id) {
+            fetchProjectThreads2(projectState.projectContext.id);
+          }
+        } else {
+          return;
         }
-      } else {
-        return;
       }
-    }
-    onMessageSend?.(messageText);
-    setInput("");
-    const rawFiles = options3?.files || [];
-    const filePaths = [];
-    if (rawFiles.length > 0 && currentThreadId) {
-      for (const rawFile of rawFiles) {
-        try {
-          const file = dataUrlToFile(rawFile.url, rawFile.filename);
-          const path = await uploadThreadFile(apiClient, currentThreadId, file);
-          if (path) {
-            filePaths.push(path);
+      onMessageSend?.(messageText);
+      const rawFiles = options3?.files || [];
+      const filePaths = [];
+      if (rawFiles.length > 0 && currentThreadId) {
+        for (const rawFile of rawFiles) {
+          try {
+            const file = dataUrlToFile(rawFile.url, rawFile.filename);
+            const path = await uploadThreadFile(apiClient, currentThreadId, file);
+            if (path) {
+              filePaths.push(path);
+            }
+          } catch (error) {
+            console.error("[FILE_UPLOAD] Error:", error);
           }
-        } catch (error) {
-          console.error("[FILE_UPLOAD] Error:", error);
         }
       }
+      messageOptionsRef.current = {
+        ...options3,
+        filePaths: filePaths.length > 0 ? filePaths : void 0
+      };
+      await sendMessage({
+        text: messageText,
+        metadata: {
+          modelId: options3?.modelId,
+          supportMultipleModels: options3?.supportMultipleModels,
+          supportFiles: options3?.supportFiles,
+          supportWebSearch: options3?.supportWebSearch,
+          supportDeepResearch: options3?.supportDeepResearch,
+          supportDocumentSearch: options3?.supportDocumentSearch,
+          supportThinking: options3?.supportThinking,
+          supportWorkCanvas: options3?.supportWorkCanvas
+        }
+      });
+    } finally {
+      isSendingRef.current = false;
     }
-    messageOptionsRef.current = {
-      ...options3,
-      filePaths: filePaths.length > 0 ? filePaths : void 0
-    };
-    await sendMessage({
-      text: messageText,
-      metadata: {
-        modelId: options3?.modelId,
-        supportMultipleModels: options3?.supportMultipleModels,
-        supportFiles: options3?.supportFiles,
-        supportWebSearch: options3?.supportWebSearch,
-        supportDeepResearch: options3?.supportDeepResearch,
-        supportDocumentSearch: options3?.supportDocumentSearch,
-        supportThinking: options3?.supportThinking,
-        supportWorkCanvas: options3?.supportWorkCanvas
-      }
-    });
   };
   useEffect(() => {
     if (!initialPrompt || initialPromptSentRef.current) return;