add smithy traits dependency

smithy-lang · May 14, 2024 · 8b4ea40 · 8b4ea40
1 parent 446c0d9
commit 8b4ea40
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/smithy-typescript-codegen/build.gradle.kts b/smithy-typescript-codegen/build.gradle.kts
@@ -39,6 +39,7 @@ dependencies {
     // Smithy generic dependencies
     api("software.amazon.smithy:smithy-codegen-core:$smithyVersion")
     api("software.amazon.smithy:smithy-model:$smithyVersion")
+    api("software.amazon.smithy:smithy-protocol-traits:$smithyVersion")
     api("software.amazon.smithy:smithy-protocol-test-traits:$smithyVersion")
     api("software.amazon.smithy:smithy-rules-engine:$smithyVersion")
     api("software.amazon.smithy:smithy-waiters:$smithyVersion")

diff --git a/...c/main/java/software/amazon/smithy/typescript/codegen/protocols/cbor/SmithyRpcV2Cbor.java b/...c/main/java/software/amazon/smithy/typescript/codegen/protocols/cbor/SmithyRpcV2Cbor.java
@@ -10,10 +10,10 @@
 import software.amazon.smithy.model.shapes.Shape;
 import software.amazon.smithy.model.shapes.ShapeId;
 import software.amazon.smithy.model.shapes.StructureShape;
+import software.amazon.smithy.protocol.traits.Rpcv2CborTrait;
 import software.amazon.smithy.typescript.codegen.integration.HttpRpcProtocolGenerator;
 import software.amazon.smithy.utils.SmithyInternalApi;
 
-
 /**
  * Generator for Smithy RPCv2 CBOR.
  */
@@ -26,7 +26,7 @@ public SmithyRpcV2Cbor() {
 
     @Override
     protected String getDocumentContentType() {
-        return "";
+        return "application/cbor";
     }
 
     @Override
@@ -65,7 +65,7 @@ protected void deserializeOutputDocument(GenerationContext generationContext,
 
     @Override
     public ShapeId getProtocol() {
-        return null;
+        return Rpcv2CborTrait.ID;
     }
 
     @Override