Fix for api change in auto entry extraction status

the-deep · Dec 21, 2023 · 3235ac2 · 3235ac2
1 parent f6412ec
commit 3235ac2
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/app/views/EntryEdit/LeftPane/AutoEntriesModal/index.tsx b/app/views/EntryEdit/LeftPane/AutoEntriesModal/index.tsx
@@ -149,10 +149,10 @@ const AUTO_DRAFT_ENTRIES_STATUS = gql`
     ) {
         project(id: $projectId) {
             id
-            assistedTagging {
-                extractionStatusByLead(leadId: $leadId) {
-                    autoEntryExtractionStatus
-                }
+            lead(
+                id: $leadId,
+            ) {
+                autoEntryExtractionStatus
             }
         }
     }
@@ -481,7 +481,7 @@ function AutoEntriesModal(props: Props) {
             notifyOnNetworkStatusChange: true,
             onCompleted: (response) => {
                 const status = response?.project
-                    ?.assistedTagging?.extractionStatusByLead?.autoEntryExtractionStatus;
+                    ?.lead?.autoEntryExtractionStatus;
                 if (status === 'SUCCESS') {
                     setDraftEntriesLoading(false);
                 }
@@ -490,11 +490,11 @@ function AutoEntriesModal(props: Props) {
     );
 
     const extractionStatus = autoEntryExtractionStatus?.project
-        ?.assistedTagging?.extractionStatusByLead?.autoEntryExtractionStatus;
+        ?.lead?.autoEntryExtractionStatus;
 
     useEffect(() => {
         const extractionStatusInternal = autoEntryExtractionStatus?.project
-            ?.assistedTagging?.extractionStatusByLead?.autoEntryExtractionStatus;
+            ?.lead?.autoEntryExtractionStatus;
 
         const shouldPoll = extractionStatusInternal === 'PENDING' || extractionStatusInternal === 'STARTED';
         if (shouldPoll) {