Address Review Comments

Signed-off-by: Naveen Tatikonda <[email protected]>
opensearch-project · Mar 18, 2024 · efd70d2 · efd70d2
1 parent e0cb80d
commit efd70d2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/main/java/org/opensearch/knn/index/mapper/KNNVectorFieldMapper.java b/src/main/java/org/opensearch/knn/index/mapper/KNNVectorFieldMapper.java
@@ -599,6 +599,11 @@ protected boolean isFaissSQfp16(MethodComponentContext methodComponentContext) {
             return false;
         }
 
+        // Validate if the object is of type MethodComponentContext before casting it later
+        if (!(methodComponentParams.get(METHOD_ENCODER_PARAMETER) instanceof MethodComponentContext)) {
+            return false;
+        }
+
         MethodComponentContext encoderMethodComponentContext = (MethodComponentContext) methodComponentParams.get(METHOD_ENCODER_PARAMETER);
 
         // returns true if encoder name is "sq" and type is "fp16"