Claude PDF Support on Amazon Bedrock

import json
import boto3
import base64

# Read and encode the PDF
with open("sample_technical_report.pdf", "rb") as pdf_file:
    encoded_pdf = base64.b64encode(pdf_file.read()).decode("utf-8")

# Prepare the request with prompt caching enabled
request_body = {
    "anthropic_version": "bedrock-2023-05-31",
    "messages": [
        {
            "role": "user",
            "content": [
                {
                    "type": "document",
                    "source": {
                        "type": "base64",
                        "media_type": "application/pdf",
                        "data": encoded_pdf
                    },
                    "title": "Technical Report",
                    "citations": {"enabled": True},
                    "cache_control": {"type": "ephemeral"}  # Enable prompt caching!
                },
                {
                    "type": "text",
                    "text": "What charts or graphs are shown in this document? Describe what you see in any visual elements."
                }
            ]
        }
    ],
    "max_tokens": 500
}

# Make the API call
response = bedrock_client.invoke_model(
    modelId="us.anthropic.claude-sonnet-4-20250514-v1:0",
    contentType='application/json',
    accept='application/json',
    body=json.dumps(request_body)
)

# Parse and display the response
response_body = json.loads(response['body'].read())
print("Claude's Analysis:")

for block in response_body.get('content', []):
    if block.get('type') == 'text':
        print(block.get('text', ''))

with open("sample_technical_report.pdf", "rb") as pdf_file:
    pdf_bytes = pdf_file.read()

# Prepare the message
messages = [
    {
        "role": "user",
        "content": [
            {
                "document": {
                    "name": "Technical Report",
                    "format": "pdf",
                    "source": {
                        "bytes": pdf_bytes  # Raw bytes, not base64!
                    },
                    "citations": {"enabled": True}  # REQUIRED for Claude's PDF support in Converse!
                }
            },
            {
                "text": "What visual elements (charts, graphs, diagrams) are present in this document? Describe what they show."
            }
        ]
    }
]

# Make the converse call
response = bedrock_client.converse(
    modelId="us.anthropic.claude-sonnet-4-20250514-v1:0",
    messages=messages,
    inferenceConfig={
        "maxTokens": 500,
        "temperature": 0.5
    }
)

# Display the response
print("Claude's Analysis:")
output = response.get('output', {})

if 'message' in output:
    content = output['message'].get('content', [])
    for item in content:
        if 'text' in item:
            print(item['text'], end="")
        elif 'citationsContent' in item:
            print(f"'{item['citationsContent']['content'][0]['text']}'", end="")