Merge pull request #107 from SciPhi-AI/Nolan/KGExtraction

Use KG Extraction
SciPhi-AI · Oct 2, 2024 · 1c8c4cf · 1c8c4cf
2 parents 24f6aed + 747162c
commit 1c8c4cf
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 14 deletions.
diff --git a/src/components/ChatDemo/DocumentsTable.tsx b/src/components/ChatDemo/DocumentsTable.tsx
@@ -71,7 +71,7 @@ const DocumentsTable: React.FC<DocumentsTableProps> = ({
     return documents.map((doc) => ({
       ...doc,
       ingestion_status: mapIngestionStatus(doc.ingestion_status),
-      kg_creation_status: mapIngestionStatus(doc.kg_creation_status),
+      kg_extraction_status: mapIngestionStatus(doc.kg_extraction_status),
     }));
   }, [documents]);
 
@@ -119,14 +119,14 @@ const DocumentsTable: React.FC<DocumentsTableProps> = ({
       },
     },
     {
-      key: 'kg_creation_status',
-      label: 'KG Creation',
+      key: 'kg_extraction_status',
+      label: 'KG Extraction',
       filterable: true,
       filterType: 'multiselect',
       filterOptions: ['success', 'failure', 'pending'],
       renderCell: (doc) => {
         let variant: 'success' | 'destructive' | 'pending' = 'pending';
-        switch (doc.kg_creation_status) {
+        switch (doc.kg_extraction_status) {
           case IngestionStatus.SUCCESS:
             variant = 'success';
             break;
@@ -137,7 +137,7 @@ const DocumentsTable: React.FC<DocumentsTableProps> = ({
             variant = 'pending';
             break;
         }
-        return <Badge variant={variant}>{doc.kg_creation_status}</Badge>;
+        return <Badge variant={variant}>{doc.kg_extraction_status}</Badge>;
       },
       selected: false,
     },

diff --git a/src/components/ChatDemo/utils/documentDialogInfo.tsx b/src/components/ChatDemo/utils/documentDialogInfo.tsx
@@ -22,7 +22,7 @@ interface DocumentOverview {
   id?: string;
   ingestion_status?: string;
   metadata?: { title?: string; version?: string };
-  kg_creation_status?: string;
+  kg_extraction_status?: string;
   title?: string;
   type?: string;
   updated_at?: string;
@@ -191,8 +191,8 @@ const DocumentInfoDialog: React.FC<DocumentInfoDialogProps> = ({
                           value: documentOverview.ingestion_status,
                         },
                         {
-                          label: 'KG Creation',
-                          value: documentOverview.kg_creation_status,
+                          label: 'KG Extraction',
+                          value: documentOverview.kg_extraction_status,
                         },
                       ]}
                     />

diff --git a/src/pages/collection/[collection_id].tsx b/src/pages/collection/[collection_id].tsx
@@ -229,22 +229,22 @@ const CollectionIdPage: React.FC = () => {
       ),
     },
     {
-      key: 'kg_creation_status',
-      label: 'KG Creation',
+      key: 'kg_extraction_status',
+      label: 'KG Extraction',
       filterable: true,
       filterType: 'multiselect',
       filterOptions: ['success', 'failure', 'pending'],
       renderCell: (doc) => (
         <Badge
           variant={
-            doc.kg_creation_status === 'success'
+            doc.kg_extraction_status === 'success'
               ? 'success'
-              : doc.kg_creation_status === 'failure'
+              : doc.kg_extraction_status === 'failure'
                 ? 'destructive'
                 : 'pending'
           }
         >
-          {doc.kg_creation_status}
+          {doc.kg_extraction_status}
         </Badge>
       ),
       selected: false,

diff --git a/src/types.ts b/src/types.ts
@@ -110,7 +110,7 @@ export interface DocumentInfoType {
   version: string;
   size_in_bytes: number;
   ingestion_status: IngestionStatus;
-  kg_creation_status: string;
+  kg_extraction_status: string;
   created_at: string;
   updated_at: string;
 }