error handling with missing input field

Signed-off-by: yuye-aws <[email protected]>
opensearch-project · Jul 16, 2024 · ed36bed · ed36bed
1 parent acc51f1
commit ed36bed
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/main/java/org/opensearch/agent/tools/KnowledgeBaseTool.java b/src/main/java/org/opensearch/agent/tools/KnowledgeBaseTool.java
@@ -58,6 +58,10 @@ public KnowledgeBaseTool(Client client, String modelId) {
     @SuppressWarnings("unchecked")
     public <T> void run(Map<String, String> parameters, ActionListener<T> listener) {
         String text = parameters.get(INPUT_FIELD);
+        if (StringUtils.isBlank(text)) {
+            throw new IllegalArgumentException("[" + INPUT_FIELD + "] is null or empty, can not process it in knowledge base tool.");
+        }
+
         RemoteInferenceInputDataSet inputDataSet = RemoteInferenceInputDataSet
             .builder()
             .parameters(Collections.singletonMap(INPUT_FIELD, text))