Skip to content

Commit 388a5f3

Browse files
committed
chore(codegen): enable websocket for Bedrock Runtime
1 parent ca426fe commit 388a5f3

File tree

1 file changed

+3
-1
lines changed
  • codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen

1 file changed

+3
-1
lines changed

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ public Map<String, Consumer<TypeScriptWriter>> getRuntimeConfigWriters(
126126
}
127127

128128
private static boolean isWebsocketSupported(ServiceShape service) {
129-
Set<String> websocketServices = SetUtils.of("Transcribe Streaming", "RekognitionStreaming");
129+
Set<String> websocketServices = SetUtils.of("Transcribe Streaming", "RekognitionStreaming", "Bedrock Runtime");
130130
String serviceId = service.getTrait(ServiceTrait.class).map(ServiceTrait::getSdkId).orElse("");
131131
return websocketServices.contains(serviceId);
132132
}
@@ -141,6 +141,8 @@ private static Map<String, Object> getPluginFunctionParams(
141141
return MapUtils.of("headerPrefix", "x-amzn-transcribe-");
142142
} else if (serviceId.equals("RekognitionStreaming")) {
143143
return MapUtils.of("headerPrefix", "x-amz-rekognition-streaming-liveness-");
144+
} else if (serviceId.equals("Bedrock Runtime")) {
145+
return MapUtils.of("headerPrefix", "x-amz-bedrock-");
144146
} else {
145147
throw new CodegenException("Missing endpoint prefix for Websocket plugin of service " + serviceId);
146148
}

0 commit comments

Comments
 (0)