Add Python examples for Amazon Nova and Nova Canvas

Author: DennisTraub

.doc_gen/metadata/bedrock-runtime_metadata.yaml

@@ -108,6 +108,14 @@ bedrock-runtime_Converse_AmazonNovaText:
             - description: Send a text message to Amazon Nova, using Bedrock's Converse API.
               snippet_tags:
                 - BedrockRuntime.dotnetv3.Converse_AmazonNovaText
+    Python:
+      versions:
+        - sdk_version: 3
+          github: python/example_code/bedrock-runtime
+          excerpts:
+            - description: Send a text message to Amazon Nova, using Bedrock's Converse API.
+              snippet_tags:
+                - python.example_code.bedrock-runtime.Converse_AmazonNovaText
   services:
     bedrock-runtime: {Converse}
 
@@ -365,6 +373,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:
                 - BedrockRuntime.dotnetv3.ConverseStream_AmazonNovaText
+    Python:
+      versions:
+        - sdk_version: 3
+          github: python/example_code/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:
+                - python.example_code.bedrock-runtime.ConverseStream_AmazonNovaText
   services:
     bedrock-runtime: {ConverseStream}
 
@@ -1161,6 +1177,14 @@ bedrock-runtime_InvokeModel_AmazonNovaImageGeneration:
             - description: Create an image with Amazon Nova Canvas.
               snippet_tags:
                 - BedrockRuntime.dotnetv3.InvokeModel_AmazonNovaImageGeneration
+    Python:
+      versions:
+        - sdk_version: 3
+          github: python/example_code/bedrock-runtime
+          excerpts:
+            - description: Create an image with the Amazon Nova Canvas.
+              snippet_tags:
+                - python.example_code.bedrock-runtime.InvokeModel_AmazonNovaImageGeneration
   services:
     bedrock-runtime: {InvokeModel}
 

python/example_code/bedrock-runtime/README.md

@@ -48,6 +48,15 @@ python -m pip install -r requirements.txt
 - [Converse](models/ai21_labs_jurassic2/converse.py#L4)
 - [InvokeModel](models/ai21_labs_jurassic2/invoke_model.py#L4)
 
+### Amazon Nova
+
+- [Converse](models/amazon_nova/amazon_nova_text/converse.py#L4)
+- [ConverseStream](models/amazon_nova/amazon_nova_text/converse_stream.py#L4)
+
+### Amazon Nova Canvas
+
+- [InvokeModel](models/amazon_nova/amazon_nova_canvas/invoke_model.py#L4)
+
 ### Amazon Titan Image Generator
 
 - [InvokeModel](models/amazon_titan_image_generator/invoke_model.py#L4)

python/example_code/bedrock-runtime/models/amazon_nova/amazon_nova_canvas/invoke_model.py

@@ -0,0 +1,66 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+# SPDX-License-Identifier: Apache-2.0
+
+# snippet-start:[python.example_code.bedrock-runtime.InvokeModel_AmazonNovaImageGeneration]
+# Use the native inference API to create an image with Amazon Nova Canvas
+
+import base64
+import json
+import os
+import random
+
+import boto3
+
+# Create a Bedrock Runtime client in the AWS Region of your choice.
+client = boto3.client("bedrock-runtime", region_name="us-east-1")
+
+# Set the model ID.
+model_id = "amazon.nova-canvas-v1:0"
+
+# Define the image generation prompt for the model.
+prompt = "A stylized picture of a cute old steampunk robot."
+
+# Generate a random seed between 0 and 858,993,459
+seed = random.randint(0, 858993460)
+
+# Format the request payload using the model's native structure.
+native_request = {
+    "taskType": "TEXT_IMAGE",
+    "textToImageParams": {"text": prompt},
+    "imageGenerationConfig": {
+        "seed": seed,
+        "quality": "standard",
+        "height": 512,
+        "width": 512,
+        "numberOfImages": 1,
+    },
+}
+
+# Convert the native request to JSON.
+request = json.dumps(native_request)
+
+# Invoke the model with the request.
+response = client.invoke_model(modelId=model_id, body=request)
+
+# Decode the response body.
+model_response = json.loads(response["body"].read())
+
+# Extract the image data.
+base64_image_data = model_response["images"][0]
+
+# Save the generated image to a local folder.
+i, output_dir = 1, "output"
+if not os.path.exists(output_dir):
+    os.makedirs(output_dir)
+while os.path.exists(os.path.join(output_dir, f"nova_canvas_{i}.png")):
+    i += 1
+
+image_data = base64.b64decode(base64_image_data)
+
+image_path = os.path.join(output_dir, f"nova_canvas_{i}.png")
+with open(image_path, "wb") as file:
+    file.write(image_data)
+
+print(f"The generated image has been saved to {image_path}")
+
+# snippet-end:[python.example_code.bedrock-runtime.InvokeModel_AmazonNovaImageGeneration]

python/example_code/bedrock-runtime/models/amazon_nova/amazon_nova_text/converse.py

@@ -0,0 +1,41 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+# SPDX-License-Identifier: Apache-2.0
+
+# snippet-start:[python.example_code.bedrock-runtime.Converse_AmazonNovaText]
+# Use the Conversation API to send a text message to Amazon Nova.
+
+import boto3
+from botocore.exceptions import ClientError
+
+# Create a Bedrock Runtime client in the AWS Region you want to use.
+client = boto3.client("bedrock-runtime", region_name="us-east-1")
+
+# Set the model ID, e.g., Amazon Nova Lite.
+model_id = "amazon.nova-lite-v1:0"
+
+# Start a conversation with the user message.
+user_message = "Describe the purpose of a 'hello world' program in one line."
+conversation = [
+    {
+        "role": "user",
+        "content": [{"text": user_message}],
+    }
+]
+
+try:
+    # Send the message to the model, using a basic inference configuration.
+    response = client.converse(
+        modelId=model_id,
+        messages=conversation,
+        inferenceConfig={"maxTokens": 512, "temperature": 0.5, "topP": 0.9},
+    )
+
+    # Extract and print the response text.
+    response_text = response["output"]["message"]["content"][0]["text"]
+    print(response_text)
+
+except (ClientError, Exception) as e:
+    print(f"ERROR: Can't invoke '{model_id}'. Reason: {e}")
+    exit(1)
+
+# snippet-end:[python.example_code.bedrock-runtime.Converse_AmazonNovaText]

python/example_code/bedrock-runtime/models/amazon_nova/amazon_nova_text/converse_stream.py

@@ -0,0 +1,44 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+# SPDX-License-Identifier: Apache-2.0
+
+# snippet-start:[python.example_code.bedrock-runtime.ConverseStream_AmazonNovaText]
+# Use the Conversation API to send a text message to Amazon Nova Text
+# and print the response stream.
+
+import boto3
+from botocore.exceptions import ClientError
+
+# Create a Bedrock Runtime client in the AWS Region you want to use.
+client = boto3.client("bedrock-runtime", region_name="us-east-1")
+
+# Set the model ID, e.g., Amazon Nova Lite.
+model_id = "amazon.nova-lite-v1:0"
+
+# Start a conversation with the user message.
+user_message = "Describe the purpose of a 'hello world' program in one line."
+conversation = [
+    {
+        "role": "user",
+        "content": [{"text": user_message}],
+    }
+]
+
+try:
+    # Send the message to the model, using a basic inference configuration.
+    streaming_response = client.converse_stream(
+        modelId=model_id,
+        messages=conversation,
+        inferenceConfig={"maxTokens": 512, "temperature": 0.5, "topP": 0.9},
+    )
+
+    # Extract and print the streamed response text in real-time.
+    for chunk in streaming_response["stream"]:
+        if "contentBlockDelta" in chunk:
+            text = chunk["contentBlockDelta"]["delta"]["text"]
+            print(text, end="")
+
+except (ClientError, Exception) as e:
+    print(f"ERROR: Can't invoke '{model_id}'. Reason: {e}")
+    exit(1)
+
+# snippet-end:[python.example_code.bedrock-runtime.ConverseStream_AmazonNovaText]

python/example_code/bedrock-runtime/requirements.txt

@@ -1,35 +1,12 @@
-beautifulsoup4==4.12.3
-boto3==1.35.28
-botocore==1.35.28
-certifi==2024.8.30
-charset-normalizer==3.3.2
+boto3==1.36.22
+botocore==1.36.22
 colorama==0.4.6
-contourpy==1.3.0
-cycler==0.12.1
-fonttools==4.54.1
-geojson==3.1.0
-idna==3.10
 iniconfig==2.0.0
 jmespath==1.0.1
-kiwisolver==1.4.7
-lxml==5.3.0
-matplotlib==3.9.2
-numpy==2.1.1
-packaging==24.1
-pandas==2.2.3
-pillow==10.4.0
-pip-review==1.3.0
+packaging==24.2
 pluggy==1.5.0
-pyparsing==3.1.4
-pytest==8.3.3
-pytest-asyncio==0.24.0
+pytest==8.3.4
 python-dateutil==2.9.0.post0
-pytz==2024.2
-requests==2.32.3
-s3transfer==0.10.2
-six==1.16.0
-soupsieve==2.6
-tzdata==2024.2
-ujson==5.10.0
-urllib3==2.2.3
-xarray==2024.9.0
+s3transfer==0.11.2
+six==1.17.0
+urllib3==2.3.0

python/example_code/bedrock-runtime/test/test_converse.py

@@ -1,12 +1,14 @@
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 # SPDX-License-Identifier: Apache-2.0
 
-import pytest
 import subprocess
 import sys
 
+import pytest
+
 files_under_test = [
     "models/ai21_labs_jurassic2/converse.py",
+    "models/amazon_nova/amazon_nova_text/converse.py",
     "models/amazon_titan_text/converse.py",
     "models/anthropic_claude/converse.py",
     "models/cohere_command/converse.py",

python/example_code/bedrock-runtime/test/test_invoke_model.py

@@ -1,13 +1,15 @@
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 # SPDX-License-Identifier: Apache-2.0
 
-import pytest
 import subprocess
 import sys
 
+import pytest
+
 files_under_test = [
     # Text models
     "models/ai21_labs_jurassic2/invoke_model.py",
+    "models/amazon_nova/amazon_nova_canvas/invoke_model.py",
     "models/amazon_titan_text/invoke_model.py",
     "models/anthropic_claude/invoke_model.py",
     "models/cohere_command/command_invoke_model.py",