Add example for Nova Canvas

Author: DennisTraub

.doc_gen/metadata/bedrock-runtime_metadata.yaml

@@ -100,6 +100,14 @@ bedrock-runtime_Converse_AmazonNovaText:
             - description: Send a text message to Amazon Nova, using Bedrock's Converse API.
               snippet_tags:
                 - bedrock-runtime.java2.Converse_AmazonNovaText
+    Kotlin:
+      versions:
+        - sdk_version: 1
+          github: kotlin/services/bedrock-runtime
+          excerpts:
+            - description: Send a text message to Amazon Nova, using Bedrock's Converse API.
+              snippet_tags:
+                - bedrock-runtime.kotlin.Converse_AmazonNovaText
     .NET:
       versions:
         - sdk_version: 3
@@ -423,6 +431,14 @@ bedrock-runtime_ConverseStream_AmazonNovaText:
             - description: Send a text message to Amazon Nova using Bedrock's Converse API and process the response stream in real-time.
               snippet_tags:
                 - bedrock-runtime.java2.ConverseStream_AmazonNovaText
+    Kotlin:
+      versions:
+        - sdk_version: 1
+          github: kotlin/services/bedrock-runtime
+          excerpts:
+            - description: Send a text message to Amazon Nova using Bedrock's Converse API and process the response stream in real-time.
+              snippet_tags:
+                - bedrock-runtime.kotlin.ConverseStream_AmazonNovaText
     .NET:
       versions:
         - sdk_version: 3
@@ -736,7 +752,7 @@ bedrock-runtime_InvokeModel_TitanText:
         - sdk_version: 1
           github: kotlin/services/bedrock-runtime
           excerpts:
-            - description: Use the Invoke Model API to generate a short story.
+            - description: Use the Invoke Model API to send a text message.
               snippet_tags:
                 - bedrock-runtime.kotlin.InvokeModel_AmazonTitanText
     .NET:
@@ -1235,6 +1251,14 @@ bedrock-runtime_InvokeModel_AmazonNovaImageGeneration:
             - description: Create an image with Amazon Nova Canvas.
               snippet_tags:
                 - bedrock-runtime.java2.InvokeModel_AmazonNovaImageGeneration
+    Kotlin:
+      versions:
+        - sdk_version: 1
+          github: kotlin/services/bedrock-runtime
+          excerpts:
+            - description: Create an image with Amazon Nova Canvas.
+              snippet_tags:
+                - bedrock-runtime.kotlin.InvokeModel_AmazonNovaImageGeneration
     .NET:
       versions:
         - sdk_version: 3

.gitignore

@@ -36,3 +36,4 @@ kotlin/services/**/build/
 kotlin/services/**/gradle/
 kotlin/services/**/gradlew
 kotlin/services/**/gradlew.bat
+kotlin/services/**/.kotlin/

kotlin/services/bedrock-runtime/build.gradle.kts

@@ -21,6 +21,7 @@ dependencies {
 
     testImplementation("org.junit.jupiter:junit-jupiter:$junitVersion")
     testImplementation("org.jetbrains.kotlinx:kotlinx-coroutines-core:1.10.1")
+    testImplementation("org.jetbrains.kotlin:kotlin-reflect")
     testRuntimeOnly("org.junit.platform:junit-platform-launcher")
 }
 

kotlin/services/bedrock-runtime/src/main/kotlin/com/example/bedrockruntime/libs/ImageTools.kt

@@ -0,0 +1,27 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+package com.example.bedrockruntime.libs
+
+import java.io.ByteArrayInputStream
+import java.io.IOException
+import javax.imageio.ImageIO
+import javax.swing.ImageIcon
+import javax.swing.JFrame
+import javax.swing.JLabel
+
+object ImageTools {
+    fun displayImage(imageData: ByteArray) {
+        try {
+            val image = ImageIO.read(ByteArrayInputStream(imageData))
+            JFrame("Image").apply {
+                defaultCloseOperation = JFrame.EXIT_ON_CLOSE
+                contentPane.add(JLabel(ImageIcon(image)))
+                pack()
+                isVisible = true
+            }
+        } catch (e: IOException) {
+            throw RuntimeException(e)
+        }
+    }
+}

kotlin/services/bedrock-runtime/src/main/kotlin/com/example/bedrockruntime/models/amazon/nova/canvas/InvokeModel.kt

@@ -0,0 +1,93 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+package com.example.bedrockruntime.models.amazon.nova.canvas
+
+// snippet-start:[bedrock-runtime.kotlin.InvokeModel_AmazonNovaImageGeneration]
+
+import aws.sdk.kotlin.services.bedrockruntime.BedrockRuntimeClient
+import aws.sdk.kotlin.services.bedrockruntime.model.InvokeModelRequest
+import com.example.bedrockruntime.libs.ImageTools.displayImage
+import kotlinx.serialization.Serializable
+import kotlinx.serialization.json.Json
+import java.util.*
+
+/**
+ * This example demonstrates how to use Amazon Nova Canvas to generate images.
+ * It shows how to:
+ * - Set up the Amazon Bedrock runtime client
+ * - Configure the image generation parameters
+ * - Send a request to generate an image
+ * - Process the response and display the generated image
+ */
+suspend fun main() {
+    println("Generating image. This may take a few seconds...")
+    val imageData = invokeModel()
+    displayImage(imageData)
+}
+
+// Data class for parsing the model's response
+@Serializable
+private data class Response(val images: List<String>)
+
+// Configure JSON parser to ignore unknown fields in the response
+private val json = Json { ignoreUnknownKeys = true }
+
+suspend fun invokeModel(): ByteArray {
+    // Create and configure the Bedrock runtime client
+    BedrockRuntimeClient { region = "us-east-1" }.use { client ->
+
+        // Specify the model ID. For the latest available models, see:
+        // https://docs.aws.amazon.com/bedrock/latest/userguide/models-supported.html
+        val modelId = "amazon.nova-canvas-v1:0"
+
+        // Configure the generation parameters and create the request
+        // First, set the main parameters:
+        // - prompt: Text description of the image to generate
+        // - seed: Random number for reproducible generation (0 to 858,993,459)
+        val prompt = "A stylized picture of a cute old steampunk robot"
+        val seed = (0..858_993_459).random()
+
+        // Then, create the request using a template with the following structure:
+        // - taskType: TEXT_IMAGE (specifies text-to-image generation)
+        // - textToImageParams: Contains the text prompt
+        // - imageGenerationConfig: Contains optional generation settings (seed, quality, etc.)
+        // For a list of available request parameters, see:
+        // https://docs.aws.amazon.com/nova/latest/userguide/image-gen-req-resp-structure.html
+        val request = """
+            {
+                "taskType": "TEXT_IMAGE",
+                "textToImageParams": {
+                    "text": "$prompt"
+                },
+                "imageGenerationConfig": {
+                    "seed": $seed,
+                    "quality": "standard"
+                }
+            }
+            """.trimIndent()
+
+        // Send the request and process the model's response
+        runCatching {
+            // Send the request to the model
+            val response = client.invokeModel(InvokeModelRequest {
+                this.modelId = modelId
+                body = request.toByteArray()
+            })
+
+            // Parse the response and extract the generated image
+            val jsonResponse = response.body.toString(Charsets.UTF_8)
+            val parsedResponse = json.decodeFromString<Response>(jsonResponse)
+
+            // Extract the generated image and return it as a byte array for better handling
+            val base64Image = parsedResponse.images.first()
+            return Base64.getDecoder().decode(base64Image)
+
+        }.getOrElse { error ->
+            System.err.println("ERROR: Can't invoke '$modelId'. Reason: ${error.message}")
+            throw RuntimeException("Failed to generate image with model $modelId", error)
+        }
+    }
+}
+
+// snippet-end:[bedrock-runtime.kotlin.InvokeModel_AmazonNovaImageGeneration]

kotlin/services/bedrock-runtime/src/main/kotlin/com/example/bedrockruntime/models/amazon/nova/text/Converse.kt

@@ -27,7 +27,7 @@ suspend fun converse(): String {
     // Create and configure the Bedrock runtime client
     BedrockRuntimeClient { region = "us-east-1" }.use { client ->
 
-        // For the latest available models, see:
+        // Specify the model ID. For the latest available models, see:
         // https://docs.aws.amazon.com/bedrock/latest/userguide/models-supported.html
         val modelId = "amazon.nova-lite-v1:0"
 

kotlin/services/bedrock-runtime/src/main/kotlin/com/example/bedrockruntime/models/amazon/nova/text/ConverseStream.kt

@@ -28,7 +28,7 @@ suspend fun converseStream(): String {
     // Create and configure the Bedrock runtime client
     BedrockRuntimeClient { region = "us-east-1" }.use { client ->
 
-        // For the latest available models, see:
+        // Specify the model ID. For the latest available models, see:
         // https://docs.aws.amazon.com/bedrock/latest/userguide/models-supported.html
         val modelId = "amazon.nova-lite-v1:0"
 

kotlin/services/bedrock-runtime/src/main/kotlin/com/example/bedrockruntime/models/amazon/titan/text/InvokeModel.kt

@@ -22,14 +22,24 @@ suspend fun main() {
     invokeModel().also { println(it) }
 }
 
+// Data class for parsing the model's response
+@Serializable
+private data class BedrockResponse(val results: List<Result>) {
+    @Serializable
+    data class Result(
+        val outputText: String
+    )
+}
+
+
 // Initialize JSON parser with relaxed configuration
 private val json = Json { ignoreUnknownKeys = true }
 
 suspend fun invokeModel(): String {
     // Create and configure the Bedrock runtime client
     BedrockRuntimeClient { region = "us-east-1" }.use { client ->
 
-        // For the latest available models, see:
+        // Specify the model ID. For the latest available models, see:
         // https://docs.aws.amazon.com/bedrock/latest/userguide/models-supported.html
         val modelId = "amazon.titan-text-lite-v1"
 
@@ -45,7 +55,7 @@ suspend fun invokeModel(): String {
                     "temperature": 0.5
                 }
             }
-        """.trimIndent()
+            """.trimIndent()
 
         // Send the request and process the model's response
         runCatching {
@@ -73,10 +83,4 @@ suspend fun invokeModel(): String {
     }
 }
 
-@Serializable
-private data class BedrockResponse(val results: List<Result>)
-
-@Serializable
-private data class Result(val outputText: String)
-
 // snippet-end:[bedrock-runtime.kotlin.InvokeModel_AmazonTitanText]

kotlin/services/bedrock-runtime/src/test/kotlin/models/AbstractModelTest.kt

@@ -81,6 +81,6 @@ abstract class AbstractModelTest {
      */
     data class ModelTest(
         val name: String,
-        val function: suspend () -> String
+        val function: suspend () -> Any
     )
 }

kotlin/services/bedrock-runtime/src/test/kotlin/models/TestConverse.kt

@@ -3,8 +3,6 @@
 
 package models
 
-import com.example.bedrockruntime.models.amazon.nova.text.converse
-
 import java.util.stream.Stream
 
 /**
@@ -18,7 +16,7 @@ class TestConverse : AbstractModelTest() {
      */
     override fun modelProvider(): Stream<ModelTest> {
         return listOf(
-            ModelTest("Amazon Nova", ::converse)
+            ModelTest("Amazon Nova") { com.example.bedrockruntime.models.amazon.nova.text.converse() }
         ).stream()
     }
 }