Skip to content

Commit 242cda6

Browse files
committed
feat: add invoke agent in bedrock-agent-runtime example
1 parent 5e37073 commit 242cda6

File tree

1 file changed

+4
-3
lines changed
  • rustv1/examples/bedrock-agent-runtime/src

1 file changed

+4
-3
lines changed

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

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,14 @@ const BEDROCK_AGENT_ID: &str = "AJBHXXILZN";
55
const BEDROCK_AGENT_ALIAS_ID: &str = "AVKP1ITZAA";
66

77
#[::tokio::main]
8-
async fn main() -> Result<(), bedrockagentruntime::Error> {
8+
async fn main() -> Result<(), Box<bedrockagentruntime::Error>> {
99
let result = invoke_bedrock_agent("a prompt".to_string(), "random session id".to_string()).await?;
10-
Ok(println!("{}",result))
10+
println!("{}", result);
11+
Ok(())
1112
}
1213

1314
async fn invoke_bedrock_agent(prompt: String, session_id: String) -> Result<String, bedrockagentruntime::Error> {
14-
let aws_config: SdkConfig = aws_config::load_from_env().await;
15+
let aws_config: SdkConfig = aws_config::load_defaults(BehaviorVersion::latest()).await;
1516
let bedrock_client = bedrockagentruntime::Client::new(&aws_config);
1617

1718
let command_builder = bedrock_client

0 commit comments

Comments
 (0)