(PPLTool) update claude and openai prompts (#184)

Signed-off-by: Joshua Li <[email protected]>
opensearch-project · Feb 7, 2024 · e001987 · e001987
1 parent df90be3
commit e001987
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/src/main/java/org/opensearch/agent/tools/PPLTool.java b/src/main/java/org/opensearch/agent/tools/PPLTool.java
@@ -130,7 +130,8 @@ public class PPLTool implements Tool {
 
     public enum PPLModelType {
         CLAUDE,
-        FINETUNE;
+        FINETUNE,
+        OPENAI;
 
         public static PPLModelType from(String value) {
             if (value.isEmpty()) {
@@ -139,7 +140,7 @@ public static PPLModelType from(String value) {
             try {
                 return PPLModelType.valueOf(value.toUpperCase(Locale.ROOT));
             } catch (Exception e) {
-                log.error("Wrong PPL Model type, should be CLAUDE or FINETUNE");
+                log.error("Wrong PPL Model type, should be CLAUDE, FINETUNE, or OPENAI");
                 return PPLModelType.CLAUDE;
             }
         }