From 04c66162eb32fe66d559879f4198bcc1ab35f933 Mon Sep 17 00:00:00 2001
From: "opensearch-trigger-bot[bot]"
 <98922864+opensearch-trigger-bot[bot]@users.noreply.github.com>
Date: Thu, 15 Feb 2024 15:15:04 -0800
Subject: [PATCH] fix VisualizationsToolIT (#229) (#231)

(cherry picked from commit 919f45525f1adfc9439db29afcdd3035731c8933)

Signed-off-by: Jing Zhang <jngz@amazon.com>
Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
---
 src/test/java/org/opensearch/integTest/PromptHandler.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/test/java/org/opensearch/integTest/PromptHandler.java b/src/test/java/org/opensearch/integTest/PromptHandler.java
index a3f9314d..0ef03501 100644
--- a/src/test/java/org/opensearch/integTest/PromptHandler.java
+++ b/src/test/java/org/opensearch/integTest/PromptHandler.java
@@ -23,7 +23,7 @@ LLMThought llmThought() {
     }
 
     String response(String prompt) {
-        if (prompt.contains("TOOL RESPONSE: ")) {
+        if (prompt.contains("Human: TOOL RESPONSE ")) {
             return "```json{\n"
                 + "    \"thought\": \"Thought: Now I know the final answer\",\n"
                 + "    \"final_answer\": \"final answer\"\n"