From e6805488d0cc110088ed30b1ee5536f79bce02f9 Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" Date: Thu, 18 Apr 2024 08:43:23 +0000 Subject: [PATCH] remove logic about replace quota for finetuning model (#289) * remove logic about replace quota for finetuning model Signed-off-by: xinyual * change logic to omit replace on finetune Signed-off-by: xinyual * change equal logic Signed-off-by: xinyual * fix logic error Signed-off-by: xinyual --------- Signed-off-by: xinyual (cherry picked from commit 44b7adcfeae1981ffb0c8bccd589ef8937e1435d) Signed-off-by: github-actions[bot] --- src/main/java/org/opensearch/agent/tools/PPLTool.java | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/main/java/org/opensearch/agent/tools/PPLTool.java b/src/main/java/org/opensearch/agent/tools/PPLTool.java index 2108aa37..b1154b1c 100644 --- a/src/main/java/org/opensearch/agent/tools/PPLTool.java +++ b/src/main/java/org/opensearch/agent/tools/PPLTool.java @@ -524,7 +524,9 @@ private String parseOutput(String llmOutput, String indexName) { throw new IllegalArgumentException("The returned PPL: " + llmOutput + " has wrong format"); } } - ppl = ppl.replace("`", ""); + if (this.pplModelType != PPLModelType.FINETUNE) { + ppl = ppl.replace("`", ""); + } ppl = ppl.replaceAll("\\bSPAN\\(", "span("); if (this.head > 0) { String[] lists = llmOutput.split("\\|");