Set citations field for Bedrock PDFs (#2547)

Author: imalsogreg

.github/workflows/release.yml

@@ -19,6 +19,7 @@ on:
     branches:
       - gofix2
       - cursor/implement-collector-clear-and-update-docs-7c67
+      - greg/bedrock-pdf-citations
 concurrency:
   # No suffix specified - in reusable workflows we need a statically-defined suffix
   # to prevent workflow_call workflows from triggering a concurrency deadlock

engine/Cargo.lock

@@ -135,7 +135,7 @@ aws-config = { version = "=1.5.3", default-features = false, features = [] }
 # https://github.com/awslabs/aws-sdk-rust/issues/1263
 # same goes for aws-config.
 # Potential migration https://docs.aws.amazon.com/sdk-for-rust/latest/dg/http.html#tlsProviders
-aws-sdk-bedrockruntime = { version = "=1.76.0", default-features = false, features = [
+aws-sdk-bedrockruntime = { version = "=1.106.0", default-features = false, features = [
 
 ] }
 colored = { workspace = true, default-features = false, features = [
@@ -170,7 +170,7 @@ aws-config = "=1.5.3"
 # https://github.com/awslabs/aws-sdk-rust/issues/1263
 # same goes for aws-config.
 # Potential migration https://docs.aws.amazon.com/sdk-for-rust/latest/dg/http.html#tlsProviders
-aws-sdk-bedrockruntime = { version = "=1.76.0", default-features = false, features = [
+aws-sdk-bedrockruntime = { version = "=1.106.0", default-features = false, features = [
 ] }
 axum = "0.7.5"
 axum-extra = { version = "0.9.3", features = ["erased-json", "typed-header"] }

engine/baml-runtime/Cargo.toml

@@ -16,6 +16,7 @@ use aws_sdk_bedrockruntime::{
     self as bedrock,
     config::{Intercept, StalledStreamProtectionConfig},
     operation::converse::ConverseOutput,
+    types::CitationsConfig,
     Client as BedrockRuntimeClient,
 };
 use aws_smithy_json::serialize::JsonObjectWriter;
@@ -1083,6 +1084,9 @@ impl AwsClient {
                                 .set_source(Some(bedrock::types::DocumentSource::Bytes(Blob::new(
                                     url_media.url.as_bytes().to_vec(),
                                 ))))
+                                .set_citations(Some(
+                                    CitationsConfig::builder().set_enabled(Some(true)).build()?,
+                                ))
                                 .build()
                                 .context("Failed to build Pdf document block")?,
                         ))