update

formkiq · May 20, 2023 · 884e5bb · 884e5bb
1 parent 49f0608
commit 884e5bb
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 7 deletions.
diff --git a/lambda-s3/src/main/java/com/formkiq/stacks/lambda/s3/DocumentAction.java b/lambda-s3/src/main/java/com/formkiq/stacks/lambda/s3/DocumentAction.java
@@ -24,6 +24,7 @@
 package com.formkiq.stacks.lambda.s3;
 
 import java.io.IOException;
+import com.amazonaws.services.lambda.runtime.LambdaLogger;
 import com.formkiq.module.actions.Action;
 
 /**
@@ -36,10 +37,11 @@ public interface DocumentAction {
   /**
    * Run Action.
    * 
+   * @param logger {@link LambdaLogger}
    * @param siteId {@link String}
    * @param documentId {@link String}
    * @param action {@link Action}
    * @throws IOException IOException
    */
-  void run(String siteId, String documentId, Action action) throws IOException;
+  void run(LambdaLogger logger, String siteId, String documentId, Action action) throws IOException;
 }
diff --git a/lambda-s3/src/main/java/com/formkiq/stacks/lambda/s3/DocumentActionsProcessor.java b/lambda-s3/src/main/java/com/formkiq/stacks/lambda/s3/DocumentActionsProcessor.java
@@ -396,7 +396,7 @@ private void processAction(final LambdaLogger logger, final String siteId,
     if (ActionType.DOCUMENTTAGGING.equals(action.type())) {
 
       DocumentTaggingAction dtAction = new DocumentTaggingAction(this.serviceCache);
-      dtAction.run(siteId, documentId, action);
+      dtAction.run(logger, siteId, documentId, action);
 
       List<Action> updatedActions = this.actionsService.updateActionStatus(siteId, documentId,
           action.type(), ActionStatus.COMPLETE);

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
@@ -35,6 +35,7 @@
 import java.util.Map;
 import java.util.Map.Entry;
 import java.util.Optional;
+import com.amazonaws.services.lambda.runtime.LambdaLogger;
 import com.formkiq.aws.dynamodb.DynamicObject;
 import com.formkiq.aws.dynamodb.model.DocumentItem;
 import com.formkiq.aws.dynamodb.model.DocumentTag;
@@ -160,12 +161,12 @@ private void parseChatGptResponse(final String siteId, final String documentId,
   }
 
   @Override
-  public void run(final String siteId, final String documentId, final Action action)
-      throws IOException {
+  public void run(final LambdaLogger logger, final String siteId, final String documentId,
+      final Action action) throws IOException {
 
     String engine = action.parameters().get("engine");
     if (engine != null && "chatgpt".equals(engine.toLowerCase())) {
-      runChatGpt(siteId, documentId, action);
+      runChatGpt(logger, siteId, documentId, action);
     } else {
       throw new IOException("Unknown engine: " + engine);
     }
@@ -174,13 +175,14 @@ public void run(final String siteId, final String documentId, final Action actio
   /**
    * Run ChatGpt Document Tagging.
    * 
+   * @param logger {@link LambdaLogger}
    * @param siteId {@link String}
    * @param documentId {@link String}
    * @param action {@link Action}
    * @throws IOException IOException
    */
-  private void runChatGpt(final String siteId, final String documentId, final Action action)
-      throws IOException {
+  private void runChatGpt(final LambdaLogger logger, final String siteId, final String documentId,
+      final Action action) throws IOException {
     DynamicObject configs = this.configsService.get(siteId);
     String chatGptApiKey = configs.getString(CHATGPT_API_KEY);
 
@@ -203,6 +205,8 @@ private void runChatGpt(final String siteId, final String documentId, final Acti
 
     String url = this.serviceCache.environment("CHATGPT_API_COMPLETIONS_URL");
     HttpResponse<String> response = this.http.post(url, headers, this.gson.toJson(payload));
+    logger.log(String.format("{\"engine\":\"%s\",\"statusCode\":\"%s\",\"body\":\"%s\"}", "chatgpt",
+        String.valueOf(response.statusCode()), response.body()));
 
     if (is2XX(response)) {
       parseChatGptResponse(siteId, documentId, response);