fix: additionally start knowledge polling when source is pending

obot-platform · Dec 20, 2024 · 5abb422 · 5abb422
1 parent fa7372d
commit 5abb422
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 4 deletions.
diff --git a/ui/admin/app/components/knowledge/AddSourceModal.tsx b/ui/admin/app/components/knowledge/AddSourceModal.tsx
@@ -58,7 +58,10 @@ export const AddSourceModal: FC<AddSourceModalProps> = ({
 
     return (
         <Dialog open={isOpen} onOpenChange={onOpenChange}>
-            <DialogContent aria-describedby={undefined} className="max-w-2xl">
+            <DialogContent
+                aria-describedby="add-source-modal"
+                className="max-w-2xl"
+            >
                 <DialogTitle className="flex flex-row items-center text-xl font-semibold mb-4 justify-between">
                     <div className="flex flex-row items-center">
                         <KnowledgeSourceAvatar

diff --git a/ui/admin/app/components/knowledge/CronDialog.tsx b/ui/admin/app/components/knowledge/CronDialog.tsx
@@ -24,7 +24,7 @@ const CronDialog: FC<CronDialogProps> = ({
 
     return (
         <Dialog open={isOpen} onOpenChange={onOpenChange}>
-            <DialogContent aria-describedby={undefined} className="max-w-md">
+            <DialogContent aria-describedby="cron-dialog" className="max-w-md">
                 <DialogTitle className="text-xl font-semibold mb-4">
                     Enter Cron Expression
                 </DialogTitle>

diff --git a/ui/admin/app/components/knowledge/KnowledgeSourceDetail.tsx b/ui/admin/app/components/knowledge/KnowledgeSourceDetail.tsx
@@ -20,7 +20,12 @@ import KnowledgeSourceAvatar from "~/components/knowledge/KnowledgeSourceAvatar"
 import OauthSignDialog from "~/components/knowledge/OAuthSignDialog";
 import { LoadingSpinner } from "~/components/ui/LoadingSpinner";
 import { Button } from "~/components/ui/button";
-import { Dialog, DialogContent, DialogTitle } from "~/components/ui/dialog";
+import {
+    Dialog,
+    DialogContent,
+    DialogDescription,
+    DialogTitle,
+} from "~/components/ui/dialog";
 import {
     DropdownMenu,
     DropdownMenuCheckboxItem,
@@ -122,6 +127,10 @@ export const KnowledgeSourceDetail: FC<KnowledgeSourceDetailProps> = ({
                 onScroll={handleScroll}
                 ref={tableContainerRef}
             >
+                <DialogDescription hidden>
+                    Knowledge Source Detail
+                </DialogDescription>
+
                 <DialogTitle className="flex justify-between items-center">
                     <div className="flex flex-row items-center">
                         <KnowledgeSourceAvatar

diff --git a/ui/admin/app/hooks/knowledge/useKnowledgeSourceFiles.ts b/ui/admin/app/hooks/knowledge/useKnowledgeSourceFiles.ts
@@ -20,7 +20,10 @@ export function useKnowledgeSourceFiles(
         if (blockPollingFiles) setBlockPollingFiles(false);
     };
 
-    if (knowledgeSource.state === KnowledgeSourceStatus.Syncing) {
+    if (
+        knowledgeSource.state === KnowledgeSourceStatus.Syncing ||
+        knowledgeSource.state === KnowledgeSourceStatus.Pending
+    ) {
         startPolling();
     }