diff --git a/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/endpointsV2/AddDefaultEndpointRuleSet.java b/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/endpointsV2/AddDefaultEndpointRuleSet.java index 8a4575e3d44..092b9baf41f 100644 --- a/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/endpointsV2/AddDefaultEndpointRuleSet.java +++ b/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/endpointsV2/AddDefaultEndpointRuleSet.java @@ -15,6 +15,7 @@ import software.amazon.smithy.codegen.core.SymbolProvider; import software.amazon.smithy.model.Model; import software.amazon.smithy.model.node.Node; +import software.amazon.smithy.model.shapes.ServiceShape; import software.amazon.smithy.rulesengine.traits.EndpointRuleSetTrait; import software.amazon.smithy.typescript.codegen.CodegenUtils; import software.amazon.smithy.typescript.codegen.LanguageTarget; @@ -109,15 +110,14 @@ public List getClientPlugins() { public Model preprocessModel(Model model, TypeScriptSettings settings) { Model.Builder modelBuilder = model.toBuilder(); - model.getServiceShapes().forEach(serviceShape -> { - if (!serviceShape.hasTrait(EndpointRuleSetTrait.class)) { - usesDefaultEndpointRuleset = true; - modelBuilder.removeShape(serviceShape.toShapeId()); - modelBuilder.addShape(serviceShape.toBuilder() - .addTrait(DEFAULT_RULESET) - .build()); - } - }); + ServiceShape serviceShape = settings.getService(model); + if (!serviceShape.hasTrait(EndpointRuleSetTrait.class)) { + usesDefaultEndpointRuleset = true; + modelBuilder.removeShape(serviceShape.toShapeId()); + modelBuilder.addShape(serviceShape.toBuilder() + .addTrait(DEFAULT_RULESET) + .build()); + } return modelBuilder.build(); } diff --git a/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/schema/SchemaGenerator.java b/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/schema/SchemaGenerator.java index ee7a6f86aab..551c1bff1db 100644 --- a/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/schema/SchemaGenerator.java +++ b/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/schema/SchemaGenerator.java @@ -259,7 +259,7 @@ private void writeStructureSchema(StructureShape shape) { private void writeBaseError() { String serviceName = CodegenUtils.getServiceName(settings, model, symbolProvider); String serviceExceptionName = CodegenUtils.getServiceExceptionName(serviceName); - String namespace = model.getServiceShapes().stream().findFirst().get().getId().getNamespace(); + String namespace = settings.getService(model).getId().getNamespace(); String exceptionCtorSymbolName = "__" + serviceExceptionName; writer.addImportSubmodule("error", "error", TypeScriptDependency.SMITHY_CORE, "/schema");