Fix alert summary prompt (#958)

* Fix alert summary prompt Signed-off-by: Heng Qian <qianheng@amazon.com> * Update CHANGELOG Signed-off-by: Heng Qian <qianheng@amazon.com> * Revert "Update CHANGELOG" Signed-off-by: Heng Qian <qianheng@amazon.com> * Add prefix and postfix in requestbody Signed-off-by: Heng Qian <qianheng@amazon.com> --------- Signed-off-by: Heng Qian <qianheng@amazon.com>
opensearch-project · Nov 15, 2024 · 70b5721 · 70b5721
1 parent 52f7870
commit 70b5721
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sample-templates/alert-summary-agent-claude-tested.json b/sample-templates/alert-summary-agent-claude-tested.json
@@ -26,7 +26,7 @@
                   "content-type": "application/json"
                 },
                 "method": "POST",
-                "request_body": "{\"prompt\":\"${parameters.prompt}\", \"max_tokens_to_sample\":${parameters.max_tokens_to_sample}, \"temperature\":${parameters.temperature},  \"anthropic_version\":\"${parameters.anthropic_version}\" }",
+                "request_body": "{\"prompt\":\"\\n\\nHuman: ${parameters.prompt}\\n\\nAssistant:\", \"max_tokens_to_sample\":${parameters.max_tokens_to_sample}, \"temperature\":${parameters.temperature},  \"anthropic_version\":\"${parameters.anthropic_version}\" }",
                 "action_type": "predict",
                 "url": "https://bedrock-runtime.us-west-2.amazonaws.com/model/anthropic.claude-instant-v1/invoke"
               }

diff --git a/sample-templates/alert-summary-agent-claude-tested.yml b/sample-templates/alert-summary-agent-claude-tested.yml
@@ -24,7 +24,7 @@ workflows:
             x-amz-content-sha256: required
             content-type: application/json
           method: POST
-          request_body: '{"prompt":"${parameters.prompt}", "max_tokens_to_sample":${parameters.max_tokens_to_sample},
+          request_body: '{"prompt":"\n\nHuman: ${parameters.prompt}\n\nAssistant:", "max_tokens_to_sample":${parameters.max_tokens_to_sample},
             "temperature":${parameters.temperature},  "anthropic_version":"${parameters.anthropic_version}"
             }'
           action_type: predict