feat: fix namespace for AttributeValue (#294)

smithy-lang · Oct 2, 2023 · 308bb5a · 308bb5a
1 parent 6083aee
commit 308bb5a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/...afny-codegen/src/main/java/software/amazon/polymorph/smithydotnet/DotNetNameResolver.java b/...afny-codegen/src/main/java/software/amazon/polymorph/smithydotnet/DotNetNameResolver.java
@@ -386,7 +386,12 @@ public String baseTypeForShape(final ShapeId shapeId) {
             // therefore it MAY have specific naming requirements.
             // These requirements SHOULD be handled in the `AwsSdkDotNetNameResolver`.
             final ServiceShape awsSdkService = AwsSdkNameResolverHelpers.getAwsServiceShape(model, shapeId);
-            return new AwsSdkDotNetNameResolver(model, awsSdkService).baseTypeSwitch(shape);
+            String value = new AwsSdkDotNetNameResolver(model, awsSdkService).baseTypeSwitch(shape);
+            if (value.equals("Com.Amazonaws.Dynamodb.AttributeValue")) {
+                return "Amazon.DynamoDBv2.Model.AttributeValue";
+            } else {
+                return value;
+            }
         }
     }