From 0f3736d78114b41f73eb8d060db97c3ab158a9e4 Mon Sep 17 00:00:00 2001
From: Daniel Widdis <widdis@gmail.com>
Date: Thu, 28 Mar 2024 16:00:13 -0700
Subject: [PATCH] Add session_token

Signed-off-by: Daniel Widdis <widdis@gmail.com>
---
 sample-templates/query-assist-agent-claude.json | 5 +++--
 sample-templates/query-assist-agent-claude.yml  | 7 ++++---
 2 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/sample-templates/query-assist-agent-claude.json b/sample-templates/query-assist-agent-claude.json
index 91cad3ef1..259b22175 100644
--- a/sample-templates/query-assist-agent-claude.json
+++ b/sample-templates/query-assist-agent-claude.json
@@ -32,8 +32,9 @@
               }
             ],
             "credential": {
-                "access_key": "<put your access key here>",
-                "secret_key": "<put your secret key here>"
+                "access_key": "<YOUR_ACCESS_KEY>",
+                "secret_key": "<YOUR_SECRET_KEY>",
+                "session_token": "<YOUR_SESSION_TOKEN>"
             },
             "parameters": {
               "region": "us-west-2",
diff --git a/sample-templates/query-assist-agent-claude.yml b/sample-templates/query-assist-agent-claude.yml
index 884853d83..51adff023 100644
--- a/sample-templates/query-assist-agent-claude.yml
+++ b/sample-templates/query-assist-agent-claude.yml
@@ -25,7 +25,7 @@ workflows:
     nodes:
     # This node (workflow step) creates the connector to the Claude model on Bedrock.
     # To use a different model, you can update this node.
-    # Be sure to enter your keys in the credentials section, and change the region and endpoint if applicable.
+    # Be sure to enter your keys/token in the credentials section, and change the region and endpoint if applicable.
     - id: create_claude_connector
       type: create_connector
       previous_node_inputs: {}
@@ -45,8 +45,9 @@ workflows:
           action_type: predict
           url: https://bedrock-runtime.us-west-2.amazonaws.com/model/anthropic.claude-instant-v1/invoke
         credential:
-          access_key: "<put your access key here>"
-          secret_key: "<put your secret key here>"
+          access_key: "<YOUR_ACCESS_KEY>"
+          secret_key: "<YOUR_SECRET_KEY>"
+          session_token: "<YOUR_SESSION_TOKEN>"
         parameters:
           region: us-west-2
           endpoint: bedrock-runtime.us-west-2.amazonaws.com