work on list flows calling code

Author: AWSChris

.doc_gen/metadata/bedrock-agent_metadata.yaml

@@ -240,7 +240,7 @@ bedrock-agent_GettingStartedWithBedrockFlows:
 
   services:
     bedrock-agent: {CreateFlow, CreateFlowAlias, CreateFlowVersion, DeleteFlow, DeleteFlowVersion, DeleteFlowAlias, GetFlow, GetFlowAlias,
-                    GetFlowVersion, ListFlows, ListFlowVersions, ListFLowAliases, PrepareFlow}
+                    GetFlowVersion, PrepareFlow}
     bedrock-agent-runtime: {InvokeFlow}
 
 

python/example_code/bedrock-agent/flows/flow.py

@@ -236,7 +236,7 @@ def list_flows(client):
         flow_id (str): The identifier of the flow.
 
     Returns:
-        dict: The response from ListFlowVersions.
+        Nothing.
     """
     try:
         finished = False
@@ -262,8 +262,8 @@ def list_flows(client):
             else:
                 finished = True
 
-        logging.info("Successfully listed flows")
-        return response
+        logging.info("Successfully listed flows.")
+
 
     except ClientError as e:
         logging.exception("Client error listing flow versions: %s", str(e))

python/example_code/bedrock-agent/flows/list_flows.py

@@ -0,0 +1,57 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+# SPDX-License-Identifier: Apache-2.0
+
+"""
+Amazon Bedrock Flow lists
+
+Shows how to list Amazon Bedrock flows, flow versions in a flow,
+and flow aliases in a flow.
+
+"""
+
+import logging
+import boto3
+
+from botocore.exceptions import ClientError
+
+
+from flow import list_flows
+from flow_version import list_flow_versions
+from flow_alias import  list_flow_aliases
+
+
+def main():
+    """
+    List the Amazon Bedrock flows, flow versions in a flow,
+    and flow aliases in a flow. The call to List_flows shows
+    the IDs that you can enter for the flow ID.
+
+    Note:
+        Requires valid AWS credentials in the default profile.
+    """
+
+    try:
+
+        session = boto3.Session(profile_name='default')
+        bedrock_agent_client = session.client('bedrock-agent')
+
+        print("Listing flows")
+        list_flows(bedrock_agent_client)
+        flow_id = input("Enter Flow ID: ")
+        print(f"Listing flow versions for flow {flow_id}")
+        list_flow_versions(bedrock_agent_client, flow_id)
+        print(f"Listing flow aliases for flow {flow_id}")
+        list_flow_aliases(bedrock_agent_client, flow_id)
+
+    except ClientError as e:
+        logging.exception("Client error running example: %s", str(e))
+   
+    except Exception as e:
+        print(f"Fatal error: {str(e)}")
+    
+    finally:
+        print ("Done")
+ 
+if __name__ == "__main__":
+    main()
+

python/example_code/bedrock-agent/test/test_flow.py

@@ -264,7 +264,7 @@ def test_list_flows(make_stubber, error_code):
         call_response = flow.list_flows(
             bedrock_agent_client)
 
-        assert call_response is not None
+        assert call_response is None
 
     else:
         with pytest.raises(ClientError) as exc_info:

python/example_code/bedrock-agent/test/test_playlist_flow.py

@@ -13,8 +13,7 @@
 @pytest.mark.parametrize("file", files_under_test)
 def test_playlist_flow(file):
     # Simulate user input - each string represents one input() call
-    # If you're using the docs at https://docs.aws.amazon.com/bedrock/latest/userguide/flows-multi-turn-invocation.html,
-    # "Create a playlist\n 3\n pop, castles\n" should work with Antropic Haiku.
+    # Creates a playlist of 3 pop songs.
     test_input = "pop\n3\ny"  
 
     result = subprocess.run(