fix: Add support for numpy ndarrays in Array fields (#3069) (#3070)

This commit fixes issue #3069 where numpy ndarrays were being rejected
for Array fields with an error: "Expected list, got ndarray".


Fixes #3069

Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>

Author: zhuwenxing

pymilvus/bulk_writer/constants.py

@@ -62,7 +62,7 @@
     DataType.BFLOAT16_VECTOR.name: lambda x, dim: float16_vector_validator(x, dim, True),
     DataType.SPARSE_FLOAT_VECTOR.name: lambda x: sparse_vector_validator(x),
     DataType.INT8_VECTOR.name: lambda x, dim: int8_vector_validator(x, dim),
-    DataType.ARRAY.name: lambda x, cap: isinstance(x, list) and len(x) <= cap,
+    DataType.ARRAY.name: lambda x, cap: (isinstance(x, (list, np.ndarray)) and len(x) <= cap),
 }
 
 NUMPY_TYPE_CREATOR = {

pymilvus/client/entity_helper.py

@@ -246,6 +246,10 @@ def convert_to_array_arr(objs: List[Any], field_info: Any):
 
 
 def convert_to_array(obj: List[Any], field_info: Any):
+    # Convert numpy ndarray to list if needed
+    if isinstance(obj, np.ndarray):
+        obj = obj.tolist()
+
     field_data = schema_types.ScalarField()
     element_type = field_info.get("element_type", None)
     if element_type == DataType.BOOL:

pymilvus/client/prepare.py

@@ -543,6 +543,10 @@ def _process_struct_field(
             struct_sub_field_info: Two-level dict [struct_name][field_name] -> field info
             struct_sub_fields_data: Two-level dict [struct_name][field_name] -> FieldData
         """
+        # Convert numpy ndarray to list if needed
+        if isinstance(values, np.ndarray):
+            values = values.tolist()
+
         if not isinstance(values, list):
             msg = f"Field '{field_name}': Expected list, got {type(values).__name__}"
             raise TypeError(msg)