Fix new ingestion status of enriched

SciPhi-AI · Nov 5, 2024 · 746b05b · 746b05b
1 parent 77f9af4
commit 746b05b
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 6 deletions.
diff --git a/src/components/ChatDemo/DocumentsTable.tsx b/src/components/ChatDemo/DocumentsTable.tsx
@@ -57,7 +57,7 @@ const DocumentsTable: React.FC<DocumentsTableProps> = ({
     order: 'asc' | 'desc';
   }>({ key: 'title', order: 'asc' });
   const [filters, setFilters] = useState<Record<string, any>>({
-    ingestion_status: ['success', 'failed', 'pending'],
+    ingestion_status: ['success', 'failed', 'pending', 'enriched'],
     kg_extraction_status: ['success', 'failed', 'pending'],
   });
   const [currentPage, setCurrentPage] = useState(1);
@@ -73,6 +73,9 @@ const DocumentsTable: React.FC<DocumentsTableProps> = ({
     if (lowerStatus === 'failed') {
       return IngestionStatus.FAILED;
     }
+    if (lowerStatus === 'enriched') {
+      return IngestionStatus.ENRICHED;
+    }
     return IngestionStatus.PENDING;
   };
 
@@ -121,13 +124,16 @@ const DocumentsTable: React.FC<DocumentsTableProps> = ({
       label: 'Ingestion',
       filterable: true,
       filterType: 'multiselect',
-      filterOptions: ['success', 'failed', 'pending'],
+      filterOptions: ['success', 'failed', 'pending', 'enriched'],
       renderCell: (doc) => {
-        let variant: 'success' | 'destructive' | 'pending' = 'pending';
+        let variant: 'success' | 'destructive' | 'pending' | 'enriched' = 'pending';
         switch (doc.ingestion_status) {
           case IngestionStatus.SUCCESS:
             variant = 'success';
             break;
+          case IngestionStatus.ENRICHED:
+            variant = 'success';
+            break;
           case IngestionStatus.FAILED:
             variant = 'destructive';
             break;
@@ -203,7 +209,10 @@ const DocumentsTable: React.FC<DocumentsTableProps> = ({
             setIsDocumentInfoDialogOpen(true);
           }}
           color="filled"
-          disabled={doc.ingestion_status !== IngestionStatus.SUCCESS}
+          disabled={
+            doc.ingestion_status !== IngestionStatus.SUCCESS &&
+            doc.ingestion_status !== IngestionStatus.ENRICHED
+          }
           shape="slim"
           tooltip="View Document Info"
         >

diff --git a/src/pages/collection/[collection_id].tsx b/src/pages/collection/[collection_id].tsx
@@ -143,6 +143,7 @@ const CollectionIdPage: React.FC = () => {
             .filter(
               (doc: DocumentInfoType) =>
                 doc.ingestion_status !== IngestionStatus.SUCCESS &&
+                doc.ingestion_status !== IngestionStatus.ENRICHED &&
                 doc.ingestion_status !== IngestionStatus.FAILED
             )
             .map((doc: DocumentInfoType) => doc.id)
@@ -193,6 +194,7 @@ const CollectionIdPage: React.FC = () => {
               (doc: DocumentInfoType) =>
                 doc.id === id &&
                 doc.ingestion_status !== IngestionStatus.SUCCESS &&
+                doc.ingestion_status !== IngestionStatus.ENRICHED &&
                 doc.ingestion_status !== IngestionStatus.FAILED
             )
           )
@@ -348,11 +350,11 @@ const CollectionIdPage: React.FC = () => {
       label: 'Ingestion',
       filterable: true,
       filterType: 'multiselect',
-      filterOptions: ['success', 'failed', 'pending'],
+      filterOptions: ['success', 'failed', 'pending', 'enriched'],
       renderCell: (doc) => (
         <Badge
           variant={
-            doc.ingestion_status === IngestionStatus.SUCCESS
+            doc.ingestion_status === IngestionStatus.SUCCESS || IngestionStatus.EMBEDDING
               ? 'success'
               : doc.ingestion_status === IngestionStatus.FAILED
                 ? 'destructive'

diff --git a/src/pages/documents.tsx b/src/pages/documents.tsx
@@ -80,6 +80,7 @@ const Index: React.FC = () => {
       .filter(
         (doc) =>
           doc.ingestion_status !== IngestionStatus.SUCCESS &&
+          doc.ingestion_status !== IngestionStatus.ENRICHED &&
           doc.ingestion_status !== IngestionStatus.FAILED
       )
       .map((doc) => doc.id);

diff --git a/src/types.ts b/src/types.ts
@@ -138,6 +138,7 @@ export enum IngestionStatus {
   STORING = 'storing',
   FAILED = 'failed',
   SUCCESS = 'success',
+  ENRICHED = 'enriched',
 }
 
 export enum KGExtractionStatus {