Skip to content

Commit 460cb4a

Browse files
srchasesrchase
committed
chore(codegen): update codegen to use @smithy dependencies
1 parent 3e8ab5f commit 460cb4a

File tree

56 files changed

+426
-233
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

56 files changed

+426
-233
lines changed

‎codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/AddAwsAuthPlugin.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ public void addConfigInterfaceFields(
9191
}
9292

9393
if (!areAllOptionalAuthOperations(model, service)) {
94-
writer.addImport("Credentials", "__Credentials", TypeScriptDependency.AWS_SDK_TYPES.packageName);
94+
writer.addImport("Credentials", "__Credentials", TypeScriptDependency.AWS_SDK_TYPES);
9595
writer.writeDocs("Default credentials provider; Not available in browser runtime.\n"
9696
+ "@internal");
9797
writer.write("credentialDefaultProvider?: (input: any) => __Provider<__Credentials>;\n");

‎codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/AddBodyChecksumGeneratorDependency.java

+2-1
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
import software.amazon.smithy.model.Model;
2525
import software.amazon.smithy.model.shapes.ServiceShape;
2626
import software.amazon.smithy.typescript.codegen.LanguageTarget;
27+
import software.amazon.smithy.typescript.codegen.TypeScriptDependency;
2728
import software.amazon.smithy.typescript.codegen.TypeScriptSettings;
2829
import software.amazon.smithy.typescript.codegen.TypeScriptWriter;
2930
import software.amazon.smithy.typescript.codegen.integration.TypeScriptIntegration;
@@ -48,7 +49,7 @@ public void addConfigInterfaceFields(
4849
if (!needsBodyChecksumGeneratorDep(settings.getService(model))) {
4950
return;
5051
}
51-
writer.addImport("HttpRequest", "__HttpRequest", "@aws-sdk/types");
52+
writer.addImport("HttpRequest", "__HttpRequest", TypeScriptDependency.SMITHY_TYPES);
5253
writer.writeDocs("Function that returns body checksums.\n"
5354
+ "@internal");
5455
writer.write("bodyChecksumGenerator?: (request: __HttpRequest, "

‎codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/AddEndpointDiscoveryPlugin.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ public void addConfigInterfaceFields(
6363
if (hasClientEndpointDiscovery(service)) {
6464
// Add import for endpoint discovery command here, as getClientPlugins doesn't have access to writer.
6565
addEndpointDiscoveryCommandImport(model, symbolProvider, service, writer);
66-
writer.addImport("Provider", "__Provider", TypeScriptDependency.SMITHY_TYPES.packageName);
66+
writer.addImport("Provider", "__Provider", TypeScriptDependency.SMITHY_TYPES);
6767
writer.writeDocs("The provider which populates default for endpointDiscoveryEnabled configuration,"
6868
+ " if it's\nnot passed during client creation.\n@internal")
6969
.write("endpointDiscoveryEnabledProvider?: __Provider<boolean | undefined>;\n");

‎codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/AddHttpChecksumDependency.java

+7-6
Original file line numberDiff line numberDiff line change
@@ -58,17 +58,17 @@ public void addConfigInterfaceFields(
5858
}
5959

6060
writer.addImport("Readable", "Readable", "stream");
61-
writer.addImport("StreamHasher", "__StreamHasher", "@aws-sdk/types");
61+
writer.addImport("StreamHasher", "__StreamHasher", TypeScriptDependency.SMITHY_TYPES);
6262
writer.writeDocs("A function that, given a hash constructor and a stream, calculates the \n"
6363
+ "hash of the streamed value.\n"
6464
+ "@internal");
6565
writer.write("streamHasher?: __StreamHasher<Readable> | __StreamHasher<Blob>;\n");
6666

67-
writer.addImport("Hash", "__Hash", "@aws-sdk/types");
68-
writer.addImport("HashConstructor", "__HashConstructor", "@aws-sdk/types");
67+
writer.addImport("Hash", "__Hash", TypeScriptDependency.SMITHY_TYPES);
68+
writer.addImport("HashConstructor", "__HashConstructor", TypeScriptDependency.SMITHY_TYPES);
6969

70-
writer.addImport("Checksum", "__Checksum", "@aws-sdk/types");
71-
writer.addImport("ChecksumConstructor", "__ChecksumConstructor", "@aws-sdk/types");
70+
writer.addImport("Checksum", "__Checksum", TypeScriptDependency.SMITHY_TYPES);
71+
writer.addImport("ChecksumConstructor", "__ChecksumConstructor", TypeScriptDependency.SMITHY_TYPES);
7272

7373
writer.writeDocs("A constructor for a class implementing the {@link __Checksum} interface \n"
7474
+ "that computes MD5 hashes.\n"
@@ -80,7 +80,8 @@ public void addConfigInterfaceFields(
8080
+ "@internal");
8181
writer.write("sha1?: __ChecksumConstructor | __HashConstructor;\n");
8282

83-
writer.addImport("GetAwsChunkedEncodingStream", "GetAwsChunkedEncodingStream", "@aws-sdk/types");
83+
writer.addImport("GetAwsChunkedEncodingStream", "GetAwsChunkedEncodingStream",
84+
TypeScriptDependency.AWS_SDK_TYPES);
8485
writer.writeDocs("A function that returns Readable Stream which follows aws-chunked encoding stream.\n"
8586
+ "@internal");
8687
writer.write("getAwsChunkedEncodingStream?: GetAwsChunkedEncodingStream;\n");

‎codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/AddS3Config.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ public void addConfigInterfaceFields(
122122
writer.writeDocs(
123123
"Whether to override the request region with the region inferred from requested resource's ARN."
124124
+ " Defaults to false.")
125-
.addImport("Provider", "Provider", TypeScriptDependency.AWS_SDK_TYPES.packageName)
125+
.addImport("Provider", "Provider", TypeScriptDependency.SMITHY_TYPES)
126126
.write("useArnRegion?: boolean | Provider<boolean>;");
127127
}
128128

‎codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/AddSqsDependency.java

+7-7
Original file line numberDiff line numberDiff line change
@@ -73,10 +73,10 @@ public void addConfigInterfaceFields(
7373
return;
7474
}
7575

76-
writer.addImport("Hash", "__Hash", "@aws-sdk/types");
77-
writer.addImport("HashConstructor", "__HashConstructor", "@aws-sdk/types");
78-
writer.addImport("Checksum", "__Checksum", "@aws-sdk/types");
79-
writer.addImport("ChecksumConstructor", "__ChecksumConstructor", "@aws-sdk/types");
76+
writer.addImport("Hash", "__Hash", TypeScriptDependency.SMITHY_TYPES);
77+
writer.addImport("HashConstructor", "__HashConstructor", TypeScriptDependency.SMITHY_TYPES);
78+
writer.addImport("Checksum", "__Checksum", TypeScriptDependency.SMITHY_TYPES);
79+
writer.addImport("ChecksumConstructor", "__ChecksumConstructor", TypeScriptDependency.SMITHY_TYPES);
8080
writer.writeDocs("A constructor for a class implementing the {@link __Checksum} interface \n"
8181
+ "that computes MD5 hashes.\n"
8282
+ "@internal");
@@ -99,15 +99,15 @@ public Map<String, Consumer<TypeScriptWriter>> getRuntimeConfigWriters(
9999
return MapUtils.of("md5", writer -> {
100100
writer.addDependency(TypeScriptDependency.AWS_SDK_TYPES);
101101
writer.addImport("HashConstructor", "__HashConstructor",
102-
TypeScriptDependency.AWS_SDK_TYPES.packageName);
102+
TypeScriptDependency.SMITHY_TYPES);
103103
writer.addImport("ChecksumConstructor", "__ChecksumConstructor",
104-
TypeScriptDependency.AWS_SDK_TYPES.packageName);
104+
TypeScriptDependency.SMITHY_TYPES);
105105
writer.write("Hash.bind(null, \"md5\")");
106106
});
107107
case BROWSER:
108108
return MapUtils.of("md5", writer -> {
109109
writer.addDependency(TypeScriptDependency.MD5_BROWSER);
110-
writer.addImport("Md5", "Md5", TypeScriptDependency.MD5_BROWSER.packageName);
110+
writer.addImport("Md5", "Md5", TypeScriptDependency.MD5_BROWSER);
111111
writer.write("Md5");
112112
});
113113
default:

‎codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/AddUserAgentDependency.java

+2-2
Original file line numberDiff line numberDiff line change
@@ -52,8 +52,8 @@ public void addConfigInterfaceFields(
5252
SymbolProvider symbolProvider,
5353
TypeScriptWriter writer
5454
) {
55-
writer.addImport("Provider", "Provider", TypeScriptDependency.AWS_SDK_TYPES.packageName);
56-
writer.addImport("UserAgent", "__UserAgent", TypeScriptDependency.AWS_SDK_TYPES.packageName);
55+
writer.addImport("Provider", "Provider", TypeScriptDependency.SMITHY_TYPES);
56+
writer.addImport("UserAgent", "__UserAgent", TypeScriptDependency.SMITHY_TYPES);
5757
writer.writeDocs("The provider populating default tracking information to be sent with `user-agent`, "
5858
+ "`x-amz-user-agent` header\n@internal");
5959
writer.write("defaultUserAgentProvider?: Provider<__UserAgent>;\n");

‎codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/AwsEndpointGeneratorIntegration.java

+2-4
Original file line numberDiff line numberDiff line change
@@ -96,11 +96,9 @@ public void addConfigInterfaceFields(
9696
Paths.get(".", CodegenUtils.SOURCE_FOLDER, "endpoint/EndpointParameters").toString()
9797
));
9898

99-
writer.addImport("EndpointV2", "__EndpointV2", TypeScriptDependency.AWS_SDK_TYPES.packageName);
99+
writer.addImport("EndpointV2", "__EndpointV2", TypeScriptDependency.SMITHY_TYPES);
100100
} else {
101-
writer.addImport(
102-
"RegionInfoProvider", null, TypeScriptDependency.AWS_SDK_TYPES.packageName
103-
);
101+
writer.addImport("RegionInfoProvider", null, TypeScriptDependency.SMITHY_TYPES);
104102
writer.writeDocs("Fetch related hostname, signing name or signing region with given region.\n"
105103
+ "@internal");
106104
writer.write("regionInfoProvider?: RegionInfoProvider;\n");

‎codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/AwsProtocolUtils.java

+4-3
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ static void generateJsonParseBody(GenerationContext context) {
123123

124124
static void generateJsonParseBodyWithQueryHeader(GenerationContext context) {
125125
TypeScriptWriter writer = context.getWriter();
126-
writer.addImport("HeaderBag", "__HeaderBag", "@aws-sdk/types");
126+
writer.addImport("HeaderBag", "__HeaderBag", TypeScriptDependency.SMITHY_TYPES);
127127
writer.write(IoUtils.readUtf8Resource(
128128
AwsProtocolUtils.class, "populate-body-with-query-compatibility-code-stub.ts"));
129129
}
@@ -160,7 +160,7 @@ static void generateXmlParseBody(GenerationContext context) {
160160

161161
// Include an XML body parser used to deserialize documents from HTTP responses.
162162
writer.addImport("SerdeContext", "__SerdeContext", TypeScriptDependency.SMITHY_TYPES);
163-
writer.addImport("getValueFromTextNode", "__getValueFromTextNode", "@aws-sdk/smithy-client");
163+
writer.addImport("getValueFromTextNode", "__getValueFromTextNode", TypeScriptDependency.AWS_SMITHY_CLIENT);
164164
writer.addDependency(AwsDependency.XML_PARSER);
165165
writer.addImport("XMLParser", null, "fast-xml-parser");
166166
writer.openBlock("const parseBody = (streamBody: any, context: __SerdeContext): "
@@ -223,7 +223,8 @@ static void generateBuildFormUrlencodedString(GenerationContext context) {
223223
TypeScriptWriter writer = context.getWriter();
224224

225225
// Write a single function to handle combining a map in to a valid query string.
226-
writer.addImport("extendedEncodeURIComponent", "__extendedEncodeURIComponent", "@aws-sdk/smithy-client");
226+
writer.addImport("extendedEncodeURIComponent", "__extendedEncodeURIComponent",
227+
TypeScriptDependency.AWS_SMITHY_CLIENT);
227228
writer.openBlock("const buildFormUrlencodedString = (formEntries: Record<string, string>): "
228229
+ "string => Object.entries(formEntries).map(", ").join(\"&\");",
229230
() -> writer.write("([key, value]) => __extendedEncodeURIComponent(key) + '=' + "

‎codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/DocumentBareBonesClientGenerator.java

+2-1
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
import software.amazon.smithy.model.shapes.OperationShape;
2828
import software.amazon.smithy.model.shapes.ServiceShape;
2929
import software.amazon.smithy.typescript.codegen.ApplicationProtocol;
30+
import software.amazon.smithy.typescript.codegen.TypeScriptDependency;
3031
import software.amazon.smithy.typescript.codegen.TypeScriptSettings;
3132
import software.amazon.smithy.typescript.codegen.TypeScriptWriter;
3233
import software.amazon.smithy.utils.SmithyInternalApi;
@@ -70,7 +71,7 @@ public void run() {
7071
// Add required imports.
7172
writer.addImport(serviceName, serviceName, "@aws-sdk/client-dynamodb");
7273
writer.addImport(configType, configType, "@aws-sdk/client-dynamodb");
73-
writer.addImport("Client", "__Client", "@aws-sdk/smithy-client");
74+
writer.addImport("Client", "__Client", TypeScriptDependency.AWS_SMITHY_CLIENT);
7475
writer.writeDocs("@public");
7576
writer.write("export { __Client };");
7677
generateInputOutputImports(serviceInputTypes, serviceOutputTypes);

‎codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/DocumentClientCommandGenerator.java

+4-3
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434
import software.amazon.smithy.model.shapes.StructureShape;
3535
import software.amazon.smithy.model.traits.IdempotencyTokenTrait;
3636
import software.amazon.smithy.typescript.codegen.ApplicationProtocol;
37+
import software.amazon.smithy.typescript.codegen.TypeScriptDependency;
3738
import software.amazon.smithy.typescript.codegen.TypeScriptSettings;
3839
import software.amazon.smithy.typescript.codegen.TypeScriptWriter;
3940
import software.amazon.smithy.utils.SmithyInternalApi;
@@ -107,7 +108,7 @@ public void run() {
107108
"DynamoDBDocumentClientCommand",
108109
"./baseCommand/DynamoDBDocumentClientCommand"
109110
);
110-
writer.addImport("Command", "$Command", "@aws-sdk/smithy-client");
111+
writer.addImport("Command", "$Command", TypeScriptDependency.AWS_SMITHY_CLIENT);
111112

112113
writer.writeDocs("@public");
113114
writer.write("export { DynamoDBDocumentClientCommand, $$Command };");
@@ -182,8 +183,8 @@ private void generateCommandMiddlewareResolver(String configType) {
182183
String servicePath = Paths.get(".", DocumentClientUtils.CLIENT_NAME).toString();
183184
writer.addImport(serviceInputTypes, serviceInputTypes, servicePath);
184185
writer.addImport(serviceOutputTypes, serviceOutputTypes, servicePath);
185-
writer.addImport(handler, handler, "@aws-sdk/types");
186-
writer.addImport(middlewareStack, middlewareStack, "@aws-sdk/types");
186+
writer.addImport(handler, handler, TypeScriptDependency.SMITHY_TYPES);
187+
writer.addImport(middlewareStack, middlewareStack, TypeScriptDependency.SMITHY_TYPES);
187188

188189
writer.write("resolveMiddleware(")
189190
.indent()

‎codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/DocumentClientPaginationGenerator.java

+3-2
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
import software.amazon.smithy.model.knowledge.PaginationInfo;
2626
import software.amazon.smithy.model.shapes.OperationShape;
2727
import software.amazon.smithy.model.shapes.ServiceShape;
28+
import software.amazon.smithy.typescript.codegen.TypeScriptDependency;
2829
import software.amazon.smithy.typescript.codegen.TypeScriptWriter;
2930
import software.amazon.smithy.utils.SmithyInternalApi;
3031

@@ -91,7 +92,7 @@ public void run() {
9192
Paths.get(".", DocumentClientUtils.CLIENT_FULL_NAME).toString());
9293

9394
// Import Pagination types
94-
writer.addImport("Paginator", "Paginator", "@aws-sdk/types");
95+
writer.addImport("Paginator", "Paginator", TypeScriptDependency.SMITHY_TYPES);
9596
writer.addImport(paginationType, paginationType,
9697
Paths.get(".", getInterfaceFilelocation().replace(".ts", "")).toString());
9798

@@ -115,7 +116,7 @@ static String getInterfaceFilelocation() {
115116
}
116117

117118
static void generateServicePaginationInterfaces(TypeScriptWriter writer) {
118-
writer.addImport("PaginationConfiguration", "PaginationConfiguration", "@aws-sdk/types");
119+
writer.addImport("PaginationConfiguration", "PaginationConfiguration", TypeScriptDependency.SMITHY_TYPES);
119120

120121
writer.addImport(
121122
DocumentClientUtils.CLIENT_NAME,

‎codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/JsonMemberDeserVisitor.java

+2-1
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
import software.amazon.smithy.model.shapes.MemberShape;
2222
import software.amazon.smithy.model.shapes.Shape;
2323
import software.amazon.smithy.model.traits.TimestampFormatTrait.Format;
24+
import software.amazon.smithy.typescript.codegen.TypeScriptDependency;
2425
import software.amazon.smithy.typescript.codegen.integration.DocumentMemberDeserVisitor;
2526
import software.amazon.smithy.typescript.codegen.integration.ProtocolGenerator.GenerationContext;
2627
import software.amazon.smithy.utils.SmithyInternalApi;
@@ -43,7 +44,7 @@ final class JsonMemberDeserVisitor extends DocumentMemberDeserVisitor {
4344
Format defaultTimestampFormat) {
4445
super(context, dataSource, defaultTimestampFormat);
4546
this.memberShape = memberShape;
46-
context.getWriter().addImport("_json", null, "@aws-sdk/smithy-client");
47+
context.getWriter().addImport("_json", null, TypeScriptDependency.AWS_SMITHY_CLIENT);
4748
this.serdeElisionEnabled = !context.getSettings().generateServerSdk();
4849
}
4950

‎codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/JsonMemberSerVisitor.java

+2-1
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
import software.amazon.smithy.model.shapes.BigIntegerShape;
2121
import software.amazon.smithy.model.shapes.Shape;
2222
import software.amazon.smithy.model.traits.TimestampFormatTrait.Format;
23+
import software.amazon.smithy.typescript.codegen.TypeScriptDependency;
2324
import software.amazon.smithy.typescript.codegen.integration.DocumentMemberSerVisitor;
2425
import software.amazon.smithy.typescript.codegen.integration.ProtocolGenerator.GenerationContext;
2526
import software.amazon.smithy.utils.SmithyInternalApi;
@@ -38,7 +39,7 @@ final class JsonMemberSerVisitor extends DocumentMemberSerVisitor {
3839
*/
3940
JsonMemberSerVisitor(GenerationContext context, String dataSource, Format defaultTimestampFormat) {
4041
super(context, dataSource, defaultTimestampFormat);
41-
context.getWriter().addImport("_json", null, "@aws-sdk/smithy-client");
42+
context.getWriter().addImport("_json", null, TypeScriptDependency.AWS_SMITHY_CLIENT);
4243
this.serdeElisionEnabled = !context.getSettings().generateServerSdk();
4344
}
4445

‎codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/JsonRpcProtocolGenerator.java

+2-1
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
import software.amazon.smithy.model.shapes.Shape;
2323
import software.amazon.smithy.model.shapes.StructureShape;
2424
import software.amazon.smithy.model.traits.TimestampFormatTrait.Format;
25+
import software.amazon.smithy.typescript.codegen.TypeScriptDependency;
2526
import software.amazon.smithy.typescript.codegen.TypeScriptWriter;
2627
import software.amazon.smithy.typescript.codegen.integration.DocumentMemberDeserVisitor;
2728
import software.amazon.smithy.typescript.codegen.integration.DocumentMemberSerVisitor;
@@ -112,7 +113,7 @@ protected void writeRequestHeaders(GenerationContext context, OperationShape ope
112113
protected void writeSharedRequestHeaders(GenerationContext context) {
113114
ServiceShape serviceShape = context.getService();
114115
TypeScriptWriter writer = context.getWriter();
115-
writer.addImport("HeaderBag", "__HeaderBag", "@aws-sdk/types");
116+
writer.addImport("HeaderBag", "__HeaderBag", TypeScriptDependency.SMITHY_TYPES);
116117
String targetHeader = serviceShape.getId().getName(serviceShape) + ".${operation}";
117118
writer.openBlock("function sharedHeaders(operation: string): __HeaderBag { return {", "}};",
118119
() -> {

‎codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/JsonShapeDeserVisitor.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,7 @@ protected void deserializeStructure(GenerationContext context, StructureShape sh
181181
// Prepare the document contents structure.
182182
// Use a TreeMap to sort the members.
183183
Map<String, MemberShape> members = new TreeMap<>(shape.getAllMembers());
184-
writer.addImport("take", null, "@aws-sdk/smithy-client");
184+
writer.addImport("take", null, TypeScriptDependency.AWS_SMITHY_CLIENT);
185185
writer.openBlock("return take(output, {", "}) as any;", () -> {
186186
// Set all the members to undefined to meet type constraints.
187187
members.forEach((memberName, memberShape) -> {

‎codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/JsonShapeSerVisitor.java

+2-1
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434
import software.amazon.smithy.model.traits.SparseTrait;
3535
import software.amazon.smithy.model.traits.TimestampFormatTrait;
3636
import software.amazon.smithy.model.traits.TimestampFormatTrait.Format;
37+
import software.amazon.smithy.typescript.codegen.TypeScriptDependency;
3738
import software.amazon.smithy.typescript.codegen.TypeScriptWriter;
3839
import software.amazon.smithy.typescript.codegen.integration.DocumentMemberSerVisitor;
3940
import software.amazon.smithy.typescript.codegen.integration.DocumentShapeSerVisitor;
@@ -143,7 +144,7 @@ public void serializeMap(GenerationContext context, MapShape shape) {
143144
@Override
144145
public void serializeStructure(GenerationContext context, StructureShape shape) {
145146
TypeScriptWriter writer = context.getWriter();
146-
writer.addImport("take", null, "@aws-sdk/smithy-client");
147+
writer.addImport("take", null, TypeScriptDependency.AWS_SMITHY_CLIENT);
147148
writer.openBlock("return take(input, {", "});", () -> {
148149
// Use a TreeMap to sort the members.
149150
Map<String, MemberShape> members = new TreeMap<>(shape.getAllMembers());

0 commit comments

Comments
 (0)