Bedrock managed prompts examples (#7418)

* work on python hello bedrock example

* fixed metadata

* reformatted string

* updated readme

* updated readme

* updated readme

* updated readme

* ran writeme script to update bedrock readme

* removed region declaration

* updated requirements.txt

* updated requirements

* added prompts

* workon readme

* changed scenario to match docs

* updated JSON support to match flow docs example. Other celan up

* made invoke_prompt function generic

* clean up

* added integ test

* ongoing work

* added metadata, other updates

* Update bedrock-agent_metadata.yaml

* Update bedrock-agent_metadata.yaml

* Update bedrock-agent_metadata.yaml

* Update bedrock-agent_metadata.yaml

* Update bedrock-agent_metadata.yaml

* Update bedrock-agent_metadata.yaml

* Update bedrock-agent_metadata.yaml

* updates for review feedback

---------

Co-authored-by: Christopher Rees <[email protected]>
Co-authored-by: Rachel Hagerman <[email protected]>
Co-authored-by: Brian Murray <[email protected]>

Author: 4 people

.doc_gen/metadata/bedrock-agent_metadata.yaml

@@ -186,7 +186,6 @@ bedrock-agent_PrepareAgent:
   services:
     bedrock-agent: {PrepareAgent}
 
-
 bedrock-agent_GettingStartedWithBedrockAgents:
   title: An end-to-end example showing how to create and invoke &BRA; using an &AWS; SDK
   title_abbrev: Create and invoke an agent
@@ -211,8 +210,6 @@ bedrock-agent_GettingStartedWithBedrockAgents:
   services:
     bedrock-agent: {CreateAgent, CreateAgentActionGroup, CreateAgentAlias, DeleteAgent, DeleteAgentAlias, GetAgent, ListAgentActionGroups,
                     ListAgents, ListAgentKnowledgeBases, PrepareAgent}
-    
-
 
 bedrock-agent_GettingStartedWithBedrockFlows:
   title: An end-to-end example showing how to create and invoke an Amazon Bedrock flow using an &AWS; SDK
@@ -242,7 +239,6 @@ bedrock-agent_GettingStartedWithBedrockFlows:
                     GetFlowVersion, PrepareFlow}
     bedrock-agent-runtime: {InvokeFlow}
 
-
 bedrock-agent_CreateFlow:
   languages:
     Python:
@@ -433,3 +429,95 @@ bedrock-agent_ListFlowVersions:
 
   services:
     bedrock-agent: {ListFlowVersions}
+
+bedrock-agent_CreatePrompt:
+  languages:
+    Python:
+      versions:
+        - sdk_version: 3
+          github: python/example_code/bedrock-agent
+          excerpts:
+            - description: Create an Amazon Bedrock managed prompt.
+              snippet_tags:
+                - python.example_code.bedrock.create_prompt
+
+  services:
+    bedrock-agent: {CreatePrompt}
+
+bedrock-agent_CreatePromptVersion:
+  languages:
+    Python:
+      versions:
+        - sdk_version: 3
+          github: python/example_code/bedrock-agent
+          excerpts:
+            - description: Create a version of an Amazon Bedrock managed prompt.
+              snippet_tags:
+                - python.example_code.bedrock.create_prompt_version
+
+  services:
+    bedrock-agent: {CreatePromptVersion}
+
+bedrock-agent_GetPrompt:
+  languages:
+    Python:
+      versions:
+        - sdk_version: 3
+          github: python/example_code/bedrock-agent
+          excerpts:
+            - description: Get an Amazon Bedrock managed prompt.
+              snippet_tags:
+                - python.example_code.bedrock.get_prompt
+
+  services:
+    bedrock-agent: {GetPrompt}
+
+bedrock-agent_DeletePrompt:
+  languages:
+    Python:
+      versions:
+        - sdk_version: 3
+          github: python/example_code/bedrock-agent
+          excerpts:
+            - description: Delete an Amazon Bedrock managed prompt.
+              snippet_tags:
+                - python.example_code.bedrock.delete_prompt
+
+  services:
+    bedrock-agent: {DeletePrompt}
+
+bedrock-agent_ListPrompts:
+  languages:
+    Python:
+      versions:
+        - sdk_version: 3
+          github: python/example_code/bedrock-agent
+          excerpts:
+            - description: List Amazon Bedrock managed prompts.
+              snippet_tags:
+                - python.example_code.bedrock.list_prompts
+
+  services:
+    bedrock-agent: {ListPrompts}
+    
+bedrock-agent_GettingStartedWithBedrockPrompts:
+  title: An end-to-end example showing how to create and invoke Amazon Bedrock managed prompts using an &AWS; SDK
+  title_abbrev: Create and invoke a managed prompt
+  synopsis_list:
+    - Create a managed prompt.
+    - Create a version of the prompt.
+    - Invoke the prompt using the version.
+    - Clean up resources (optional).
+  category: Scenarios
+  languages:
+    Python:
+      versions:
+        - sdk_version: 3
+          github: python/example_code/bedrock-agent
+          excerpts:
+            - description: Create and invoke a managed prompt.
+              snippet_tags:
+                - python.example_code.bedrock-agent-runtime.Scenario_GettingStartedBedrockManagedPrompts
+  services:
+    bedrock-agent: {CreatePrompt, CreatePromptVersion, DeletePrompt}
+    bedrock-runtime: {Converse}

python/example_code/bedrock-agent/README.md

@@ -2,13 +2,15 @@
 
 ## Overview
 
-Shows how to use the AWS SDK for Python (Boto3) to work with Amazon Bedrock Agents.
+Shows how to use the AWS SDK for Python (Boto3) to work with Amazon Bedrock agents, flows, and managed prompts.
 
 <!--custom.overview.start-->
 <!--custom.overview.end-->
 
 _Amazon Bedrock Agents offer you the ability to build and configure autonomous agents in your application._
 
+
+
 ## ⚠ Important
 
 * Running this code might result in charges to your AWS account. For more details, see [AWS Pricing](https://aws.amazon.com/pricing/) and [Free Tier](https://aws.amazon.com/free/).
@@ -63,6 +65,13 @@ Code excerpts that show you how to call individual service functions.
 - [UpdateFlowAlias](flows/flow_alias.py#L55) 
 - [DeleteFlowAlias](flows/flow_alias.py#L98) 
 - [ListFlowAliases](flows/flow_alias.py#L132) 
+- [FlowConversation](flows/flow-conversation.py)
+- [CreatePrompt](prompts/prompt.py#L32)
+- [CreatePromptVersion](prompts/prompt.py#L61)
+- [GetPrompt](prompts/prompt.py#L71)
+- [DeletePrompt](prompts/prompt.py#L154)
+- [ListPrompts](prompts/prompt.py#L187)
+
 
 
 
@@ -73,6 +82,7 @@ functions within the same service.
 
 - [Create and invoke an agent](scenario_get_started_with_agents.py)
 - [Create and invoke a flow](flows/playlist_flow.py)
+- [Create and invoke a managed prompt](prompts/scenario_get_started_with_prompts.py)
 
 
 <!--custom.examples.start-->
@@ -144,6 +154,49 @@ python flows/list_flows.py
 The example first lists the flows in the current AWS Region. It
 then prompts for a flow ID, which you can get from the list of flows. Finally, the example lists the flow versions and flow aliases for the flow ID that you entered.
 
+#### Create and invoke a managed prompt
+
+This example shows you how to do the following:
+
+- Create a managed prompt
+- Create a version of the prompt
+- Invoke the prompt using the version
+- Update the prompt
+- Create a new version
+- Invoke the updated prompt
+- Clean up resources (optional)
+
+Start the example by running the following at a command prompt:
+
+```
+python prompts/scenario_get_started_with_prompts.py
+```
+
+By default, the example will clean up all resources it creates. If you want to keep the resources for further exploration, use the `--no-cleanup` flag:
+
+```
+python prompts/scenario_get_started_with_prompts.py --no-cleanup
+```
+
+You can also specify a different AWS region or model ID:
+
+```
+python prompts/scenario_get_started_with_prompts.py --region us-west-2 --model-id anthropic.claude-3-sonnet-20240229-v1:0
+```
+
+#### List prompts
+
+Shows how to list Amazon Bedrock managed prompts and versions of a prompt.
+
+Start the example by running the following at a command prompt:
+
+```
+python prompts/list_prompts.py
+```
+
+The example first lists the prompts in the current AWS Region. 
+
+
 ### Tests
 
 ⚠ Running tests might result in charges to your AWS account.
@@ -170,4 +223,4 @@ in the `python` folder.
 
 Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 
-SPDX-License-Identifier: Apache-2.0
+SPDX-License-Identifier: Apache-2.0

python/example_code/bedrock-agent/prompts/list_prompts.py

@@ -0,0 +1,61 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+# SPDX-License-Identifier: Apache-2.0
+"""
+Shows how to use the AWS SDK for Python (Boto3) with Amazon Bedrock
+to list Amazon Bedrock managed prompts.
+"""
+
+import argparse
+import boto3
+import logging
+
+from prompt import list_prompts
+
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(levelname)s: %(message)s'
+)
+logger = logging.getLogger(__name__)
+
+def main():
+    """
+    Lists Amazon Bedrock managed prompts.
+    """
+    parser = argparse.ArgumentParser(
+        description="Lists Amazon Bedrock managed prompts."
+    )
+    parser.add_argument(
+        '--region',
+        default='us-east-1',
+        help="The AWS Region to use."
+    )
+    args = parser.parse_args()
+
+    # Use bedrock-agent client for prompt management
+    bedrock_client = boto3.client('bedrock-agent', region_name=args.region)
+    
+    try:
+
+        # List all prompts
+        print("\n=== Amazon Bedrock Managed Prompts ===")
+        prompts = list_prompts(bedrock_client)
+        
+        if not prompts:
+            print("No prompts found in this region.")
+            return
+            
+        for prompt in prompts:
+            print(f"ID: {prompt['id']}")
+            print(f"Name: {prompt['name']}")
+            print(f"Description: {prompt.get('description', 'No description')}")
+            print(f"ARN: {prompt.get('arn', 'No ARN')}")
+            print(f"Created: {prompt.get('createdAt', 'Unknown')}")
+            print(f"Updated: {prompt.get('updatedAt', 'Unknown')}")
+            print(f"Version: {prompt.get('version', 'Unknown')}")
+            print("-" * 40)
+        
+    except Exception as e:
+        logger.exception("Error listing prompts: %s", str(e))
+        
+if __name__ == "__main__":
+    main()