Use Nova Lite in AgentCore Strands sample

Author: lennessyy

bedrock_agentcore/strands_agent/README.md

@@ -31,10 +31,8 @@ workflows kick off.
 - AWS permissions for the AgentCore CLI (S3, IAM, CloudFormation): see [Use the
   AgentCore
   CLI](https://docs.aws.amazon.com/bedrock-agentcore/latest/devguide/runtime-permissions.html#runtime-permissions-cli)
-- Access to an [Amazon
-  Bedrock](https://docs.aws.amazon.com/bedrock/latest/userguide/model-access-modify.html)
-  foundation model in your region -- the agent uses the plugin's default
-  `BedrockModel()`
+- Access to the Amazon Nova Lite model in your target Region. The sample uses
+  the model ID `amazon.nova-lite-v1:0`.
 
 
 Docker is not needed. With the CodeZip build there is no image: the CLI uploads a
@@ -75,8 +73,8 @@ This sample opts to utilize the new AgentCore CLI for creating and updating both
 ### 1. Choose the AWS account and region
 
 Edit `agentcore/aws-targets.json` with the account ID and region to deploy into. AgentCore is only available in
-[certain regions](https://docs.aws.amazon.com/bedrock-agentcore/latest/devguide/agentcore-regions.html), and the region
-needs the Bedrock model enabled.
+[certain regions](https://docs.aws.amazon.com/bedrock-agentcore/latest/devguide/agentcore-regions.html), and the Region
+must support Amazon Nova Lite.
 
 ### 2. Configure the Temporal connection
 

bedrock_agentcore/strands_agent/agentcore_worker.py

@@ -23,6 +23,7 @@
 from datetime import timedelta
 
 from bedrock_agentcore.runtime import BedrockAgentCoreApp
+from strands.models import BedrockModel
 from temporalio.client import Client
 from temporalio.common import VersioningBehavior
 from temporalio.contrib.strands import StrandsPlugin
@@ -111,7 +112,15 @@ async def run_worker() -> None:
         namespace=os.environ.get("TEMPORAL_NAMESPACE", "default"),
         api_key=api_key,
         tls=bool(api_key),
-        plugins=[StrandsPlugin()],
+        plugins=[
+            StrandsPlugin(
+                models={
+                    workflows.MODEL_NAME: lambda: BedrockModel(
+                        model_id=workflows.MODEL_ID
+                    )
+                }
+            )
+        ],
     )
 
     tracker = ActivityTracker()

bedrock_agentcore/strands_agent/starter.py

@@ -1,11 +1,12 @@
 import asyncio
 import sys
 
+from strands.models import BedrockModel
 from temporalio.client import Client
 from temporalio.contrib.strands import StrandsPlugin
 from temporalio.envconfig import ClientConfig
 
-from workflows import TASK_QUEUE, StrandsAgentWorkflow
+from workflows import MODEL_ID, MODEL_NAME, TASK_QUEUE, StrandsAgentWorkflow
 
 DEFAULT_PROMPT = (
     "What is the 30th Fibonacci number, and is it divisible by 7? "
@@ -21,7 +22,12 @@ async def main() -> None:
     # automatically alongside an API key.
     config = ClientConfig.load_client_connect_config()
     config.setdefault("target_host", "localhost:7233")
-    client = await Client.connect(**config, plugins=[StrandsPlugin()])
+    client = await Client.connect(
+        **config,
+        plugins=[
+            StrandsPlugin(models={MODEL_NAME: lambda: BedrockModel(model_id=MODEL_ID)})
+        ],
+    )
     print("Connected to Temporal Service")
 
     result = await client.execute_workflow(

bedrock_agentcore/strands_agent/workflows.py

@@ -8,6 +8,10 @@
 DEPLOYMENT_NAME = "agentcore-strands-agent-python"
 BUILD_ID = "1.0.0"
 
+# @@@SNIPSTART python-agentcore-strands-workflow
+MODEL_NAME = "nova-lite"
+MODEL_ID = "amazon.nova-lite-v1:0"
+
 with workflow.unsafe.imports_passed_through():
     from activities import execute_code
 
@@ -18,13 +22,12 @@
 the code you ran and its output."""
 
 
-# @@@SNIPSTART python-agentcore-strands-workflow
 @workflow.defn
 class StrandsAgentWorkflow:
     def __init__(self) -> None:
-        # Configure with the plugin's default BedrockModel(), custom system
-        # prompt and code interpreter tool.
+        # Configure the model, custom system prompt and code interpreter tool.
         self.agent = TemporalAgent(
+            model=MODEL_NAME,
             start_to_close_timeout=timedelta(seconds=60),
             system_prompt=SYSTEM_PROMPT,
             tools=[