fix VisualizationsToolIT (#229) (#231)

(cherry picked from commit 919f455) Signed-off-by: Jing Zhang <[email protected]> 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>
opensearch-project · Feb 15, 2024 · 04c6616 · 04c6616
1 parent 112d4ec
commit 04c6616
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"