File tree Expand file tree Collapse file tree
model-bundle/model-bundle-api/src/main/java/software/amazon/smithy/modelbundle/api Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -53,17 +53,9 @@ private static Model getModel(SmithyBundle bundle) {
5353 }
5454 var b = model .toBuilder ();
5555
56- var serviceShapes = model .getServiceShapes ();
57-
58- if (serviceShapes .size () != 1 ) {
59- throw new IllegalStateException ("Expected exactly one service shape but got "
60- + serviceShapes .stream ().map (ServiceShape ::getId ).toList ());
61- }
62-
63- var service = serviceShapes .iterator ().next ();
56+ var serviceShape = model .expectShape (ShapeId .from (bundle .getServiceName ()), ServiceShape .class );
6457
6558 // mix in the generic arg members
66- var serviceShape = service .asServiceShape ().get ();
6759 var serviceBuilder = serviceShape .toBuilder ();
6860 for (var opId : serviceShape .getAllOperations ()) {
6961 var op = model .expectShape (opId , OperationShape .class );
You can’t perform that action at this time.
0 commit comments