From b8d4d27b7a1b407ff44ddb9ddce62dbdb95fca84 Mon Sep 17 00:00:00 2001
From: Ben Batman <bbbatman@pbs.org>
Date: Tue, 19 May 2026 20:00:46 -0400
Subject: [PATCH] Example bedrock script

---
 how-to-write-claude-md/bedrock_example.py | 47 +++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 how-to-write-claude-md/bedrock_example.py

diff --git a/how-to-write-claude-md/bedrock_example.py b/how-to-write-claude-md/bedrock_example.py
new file mode 100644
index 0000000000..8fec446e78
--- /dev/null
+++ b/how-to-write-claude-md/bedrock_example.py
@@ -0,0 +1,47 @@
+"""
+Demonstrate the utility of CLAUDE.md
+"""
+
+import boto3
+from botocore.config import Config
+
+config = Config(
+    connect_timeout=15,
+    read_timeout=3600,
+    retries={"max_attempts": 4},
+)
+
+bedrock_client = boto3.client("bedrock-runtime", config=config)
+
+MODEL_ID = "amazon.nova-premier-v1:0"
+MAX_TOKENS = 100
+TEMPERATURE = 0.1
+
+SYSTEM_PROMPT = """You are a helpful, harmless assistant.
+Your task is to assist customers with any questions they may have.
+"""
+
+
+def query_llm(prompt: str) -> str:
+    system = [
+        {"text": SYSTEM_PROMPT},
+    ]
+
+    messages = [{"role": "user", "content": [{"text": prompt}]}]
+    inf_params = {"maxTokens": MAX_TOKENS, "temperature": TEMPERATURE}
+
+    response = bedrock_client.converse(
+        modelId=MODEL_ID,
+        system=system,
+        messages=messages,
+        inferenceConfig=inf_params,
+    )
+
+    response_text = response["output"]["message"]["content"][0]["text"]
+    return response_text
+
+
+if __name__ == "__main__":
+    query = input("Ask a question!\n>")
+    response = query_llm(query)
+    print(response)