update

formkiq · May 21, 2023 · 2a14663 · 2a14663
1 parent 884e5bb
commit 2a14663
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/lambda-s3/src/main/java/com/formkiq/stacks/lambda/s3/DocumentTaggingAction.java b/lambda-s3/src/main/java/com/formkiq/stacks/lambda/s3/DocumentTaggingAction.java
@@ -101,8 +101,7 @@ private String createChatGptPrompt(final String siteId, final String documentId,
       throw new IOException("missing 'tags' parameter");
     }
 
-    DocumentItem item =
-        this.serviceCache.getExtension(DocumentService.class).findDocument(siteId, documentId);
+    DocumentItem item = this.documentService.findDocument(siteId, documentId);
 
     DocumentContentFunction docContentFucn = new DocumentContentFunction(this.serviceCache);
     List<String> contentUrls = docContentFucn.getContentUrls(siteId, item);
@@ -157,6 +156,7 @@ private void parseChatGptResponse(final String siteId, final String documentId,
 
     if (!tags.isEmpty()) {
       this.documentService.addTags(siteId, documentId, tags, null);
+      this.documentService.deleteDocumentTag(siteId, documentId, "untagged");
     }
   }
 

diff --git a/lambda-s3/src/test/java/com/formkiq/stacks/lambda/s3/DocumentActionsProcessorTest.java b/lambda-s3/src/test/java/com/formkiq/stacks/lambda/s3/DocumentActionsProcessorTest.java
@@ -277,6 +277,8 @@ public void testDocumentTaggingAction02() throws Exception {
           "text/plain");
 
       documentService.saveDocument(siteId, item, null);
+      documentService.addTags(siteId, documentId, Arrays.asList(new DocumentTag(documentId,
+          "untagged", "", new Date(), "joe", DocumentTagType.SYSTEMDEFINED)), null);
 
       List<Action> actions =
           Arrays.asList(new Action().type(ActionType.DOCUMENTTAGGING).parameters(Map.of("engine",