feat: add invoke agent in bedrock-agent-runtime example

Author: mattiapitossi

rustv1/examples/Cargo.toml

@@ -9,7 +9,7 @@ members = [
     "aurora",
     "auto-scaling",
     "autoscalingplans",
-    "batch",
+    "batch", "bedrock-agent-runtime",
     "bedrock-runtime",
     "cloudformation",
     "cloudwatch",

rustv1/examples/bedrock-agent-runtime/Cargo.toml

@@ -0,0 +1,10 @@
+[package]
+name = "bedrock-agent-runtime"
+version = "0.1.0"
+edition = "2021"
+
+[dependencies]
+anyhow = "1.0.98"
+aws-config = "1.6.3"
+aws-sdk-bedrockagentruntime = "1.98.0"
+tokio = { version = "1.45.1", features = ["full"] }

rustv1/examples/bedrock-agent-runtime/src/main.rs

@@ -0,0 +1,51 @@
+use aws_config::{BehaviorVersion, SdkConfig};
+use aws_sdk_bedrockagentruntime::{self as bedrockagentruntime, types::ResponseStream};
+
+const BEDROCK_AGENT_ID: &str = "AJBHXXILZN";
+const BEDROCK_AGENT_ALIAS_ID: &str = "AVKP1ITZAA";
+
+#[::tokio::main]
+async fn main() -> Result<(), bedrockagentruntime::Error> {
+    let result = invoke_bedrock_agent("a prompt".to_string(), "random session id".to_string()).await?;
+    Ok(println!("{}",result))
+}
+
+async fn invoke_bedrock_agent(prompt: String, session_id: String) -> Result<String, bedrockagentruntime::Error> {
+    let aws_config: SdkConfig = aws_config::load_from_env().await;
+    let bedrock_client = bedrockagentruntime::Client::new(&aws_config);
+
+    let command_builder = bedrock_client
+        .invoke_agent()
+        .agent_id(BEDROCK_AGENT_ID)
+        .agent_alias_id(BEDROCK_AGENT_ALIAS_ID)
+        .session_id(session_id)
+        .input_text(prompt);
+
+    let response = command_builder.send().await?;
+
+    let mut response_stream = response.completion;
+    let mut full_agent_text_response = String::new();
+
+    println!("Processing Bedrock agent response stream...");
+    while let Some(event_result) = response_stream.recv().await.unwrap() {
+        match event_result {
+            ResponseStream::Chunk(chunk) => {
+                if let Some(bytes) = chunk.bytes {
+                    match String::from_utf8(bytes.into_inner()) {
+                        Ok(text_chunk) => {
+                            full_agent_text_response.push_str(&text_chunk);
+                        }
+                        Err(e) => {
+                            eprintln!("UTF-8 decoding error for chunk: {}", e);
+                        }
+                    }
+                }
+            }
+            _ => {
+                println!("Received an unhandled event type from Bedrock stream.");
+            }
+        }
+    }
+
+    Ok(full_agent_text_response)
+}

rustv1/examples/bedrock-runtime/src/bin/tool-use.rs

@@ -24,8 +24,8 @@ use aws_smithy_types::Document;
 use tracing::debug;
 
 // Set the model ID, e.g., Claude 3 Haiku.
-const MODEL_ID: &str = "anthropic.claude-3-haiku-20240307-v1:0";
-const CLAUDE_REGION: &str = "us-east-1";
+const MODEL_ID: &str = "anthropic.claude-3-7-sonnet-20250219-v1:0";
+const CLAUDE_REGION: &str = "eu-west-1";
 
 const SYSTEM_PROMPT: &str = "You are a weather assistant that provides current weather data for user-specified locations using only
 the Weather_Tool, which expects latitude and longitude. Infer the coordinates from the location yourself.