feat: support TemplateConfigration

aws-cloudformation · Oct 31, 2023 · 1b4f388 · 1b4f388
1 parent b003bf0
commit 1b4f388
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/aws-kendra-datasource/aws-kendra-datasource.json b/aws-kendra-datasource/aws-kendra-datasource.json
@@ -1388,7 +1388,7 @@
       "type": "object",
       "properties": {
         "Template": {
-          "type": "string"
+          "type": "object"
         }
       },
       "additionalProperties": false,

diff --git a/...datasource/src/main/java/software/amazon/kendra/datasource/convert/TemplateConverter.java b/...datasource/src/main/java/software/amazon/kendra/datasource/convert/TemplateConverter.java
@@ -1,5 +1,9 @@
 package software.amazon.kendra.datasource.convert;
 
+import java.util.LinkedHashMap;
+import java.util.Map;
+import java.util.stream.Collectors;
+
 import com.google.gson.Gson;
 import com.google.gson.GsonBuilder;
 import software.amazon.awssdk.core.document.Document;
@@ -18,7 +22,7 @@ public static software.amazon.awssdk.services.kendra.model.TemplateConfiguration
             return null;
         }
         return software.amazon.awssdk.services.kendra.model.TemplateConfiguration.builder()
-                .template(builder.fromJson(model.getTemplate(), Document.class))
+                .template(builder.fromJson(builder.toJson(model.getTemplate()), Document.class))
                 .build();
     }
 
@@ -29,12 +33,13 @@ public static DataSourceConfiguration toModelDataSourceConfiguration(
                 .build();
     }
 
+    @SuppressWarnings("unchecked") 
     private static TemplateConfiguration toModel(software.amazon.awssdk.services.kendra.model.TemplateConfiguration sdk) {
         if (sdk == null) {
             return null;
         }
         return TemplateConfiguration.builder()
-                .template(builder.toJson(sdk.template(), Document.class))
+                .template(builder.fromJson(builder.toJson(sdk.template()), Map.class))
                 .build();
     }
 }