@docyrus/ui-pro-ai-assistant 0.2.6 → 0.2.8

package/dist/index.js

@@ -2557,48 +2557,77 @@ function toAgentTriggerData(id, data) {
 }
 function useAgentsData(agentIds) {
   const apiClient = useApiClient();
+  const apiClientRef = useRef(apiClient);
+  apiClientRef.current = apiClient;
   const [agents, setAgents] = useState([]);
   const [isLoading, setIsLoading] = useState(false);
   const [error, setError] = useState(null);
+  const [retryCount, setRetryCount] = useState(0);
   const idsKey = agentIds.join(",");
-  const fetchAgents = useCallback(async () => {
-    const ids = idsKey.split(",").filter(Boolean);
-    if (ids.length === 0) {
-      setAgents([]);
-      return;
-    }
-    setIsLoading(true);
-    setError(null);
-    try {
-      const results = await Promise.allSettled(
-        ids.map((id) => apiClient.get(`/ai/agent-deployments/base/${id}`))
-      );
-      const resolved = [];
-      for (let i = 0; i < ids.length; i++) {
-        const result = results[i];
-        if (result.status === "fulfilled" && result.value.success && result.value.data) {
-          resolved.push(
-            toAgentTriggerData(ids[i], result.value.data)
+  useEffect(() => {
+    let cancelled = false;
+    const run = async () => {
+      const ids = idsKey.split(",").filter(Boolean);
+      if (ids.length === 0) {
+        setAgents([]);
+        return;
+      }
+      setIsLoading(true);
+      setError(null);
+      try {
+        const BATCH_SIZE = 3;
+        const BATCH_DELAY = 200;
+        const resolved = [];
+        const client = apiClientRef.current;
+        for (let start = 0; start < ids.length; start += BATCH_SIZE) {
+          if (cancelled) return;
+          if (start > 0) {
+            await new Promise((resolve) => {
+              const timer = setTimeout(resolve, BATCH_DELAY);
+              if (cancelled) clearTimeout(timer);
+            });
+          }
+          const batch = ids.slice(start, start + BATCH_SIZE);
+          const results = await Promise.allSettled(
+            batch.map((id) => client.get(`/ai/agent-deployments/base/${id}`))
+          );
+          for (let i = 0; i < batch.length; i++) {
+            const result = results[i];
+            if (result.status === "fulfilled" && result.value.success && result.value.data) {
+              resolved.push(
+                toAgentTriggerData(batch[i], result.value.data)
+              );
+            }
+          }
+        }
+        if (!cancelled) {
+          setAgents(resolved);
+        }
+      } catch (err) {
+        if (!cancelled) {
+          setError(
+            err instanceof Error ? err : new Error("Failed to fetch agents")
           );
         }
+      } finally {
+        if (!cancelled) {
+          setIsLoading(false);
+        }
       }
-      setAgents(resolved);
-    } catch (err) {
-      setError(
-        err instanceof Error ? err : new Error("Failed to fetch agents")
-      );
-    } finally {
-      setIsLoading(false);
-    }
-  }, [apiClient, idsKey]);
-  useEffect(() => {
-    void fetchAgents();
-  }, [fetchAgents]);
+    };
+    void run();
+    return () => {
+      cancelled = true;
+    };
+  }, [idsKey, retryCount]);
+  const retry = useCallback(() => {
+    setRetryCount((c) => c + 1);
+  }, []);
   return {
     agents,
     isLoading,
     error,
-    retry: fetchAgents
+    retry
   };
 }
 function useAssistantApi({
@@ -24920,11 +24949,11 @@ var DocyAssistant = ({
     uiActions.setActiveTab(0);
     setMessages([]);
   };
-  const handleSendMessage = async (e, options3) => {
+  const handleSendMessage = async (e, options3, overrideText) => {
     e?.preventDefault();
-    if (!input.trim()) return;
+    const messageText = (overrideText ?? input).trim();
+    if (!messageText) return;
     messageOptionsRef.current = options3;
-    const messageText = input.trim();
     let currentThreadId = selectedSessionIdRef.current;
     if (!currentThreadId) {
       const subject = messageText.substring(0, 100);
@@ -24995,9 +25024,7 @@ var DocyAssistant = ({
     if (!initialPrompt || initialPromptSentRef.current) return;
     initialPromptSentRef.current = true;
     setInput(initialPrompt);
-    setTimeout(() => {
-      handleSendMessage();
-    }, 300);
+    void handleSendMessage(void 0, void 0, initialPrompt);
   }, [initialPrompt]);
   const handleInputChange = useCallback((e) => {
     setInput(e.target.value);