Search for Operation Context Params trait on OperationShape

Author: trivikr

smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/CommandGenerator.java

@@ -300,7 +300,7 @@ private void generateEndpointParameterInstructionProvider() {
                 paramNames.add(name);
             });
 
-            parameterFinder.getOperationContextParamValues(operationInput).forEach((name, jmesPathForInputInJs) -> {
+            parameterFinder.getOperationContextParamValues(operation).forEach((name, jmesPathForInputInJs) -> {
                 writer.write(
                     """
                     $L: { type: \"operationContextParams\", name: $L },

smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/endpointsV2/RuleSetParameterFinder.java

@@ -156,10 +156,10 @@ public Map<String, String> getContextParams(Shape operationInput) {
     /**
      * Get map of params to JavaScript equivalent of provided JMESPath expressions.
      */
-    public Map<String, String> getOperationContextParamValues(Shape operationInput) {
+    public Map<String, String> getOperationContextParamValues(OperationShape operation) {
         Map<String, String> map = new HashMap<>();
 
-        Optional<OperationContextParamsTrait> trait = operationInput.getTrait(OperationContextParamsTrait.class);
+        Optional<OperationContextParamsTrait> trait = operation.getTrait(OperationContextParamsTrait.class);
         if (trait.isPresent()) {
             OperationContextParamsTrait operationContextParamsTrait = trait.get();
             operationContextParamsTrait.getParameters().forEach((name, definition) -> {