add array data type for milvus vector store collection create

Author: Rohit Gupta

pymilvus/orm/types.py

@@ -70,6 +70,11 @@ def is_float_datatype(data_type: DataType):
 def is_numeric_datatype(data_type: DataType):
     return is_float_datatype(data_type) or is_integer_datatype(data_type)
 
+def is_varchar_datatype(data_type: DataType):
+    return data_type in (DataType.VARCHAR,)
+
+def is_bool_datatype(data_type: DataType):
+    return data_type in (DataType.BOOL,)
 
 # pylint: disable=too-many-return-statements
 def infer_dtype_by_scalar_data(data: Any):
@@ -105,7 +110,7 @@ def infer_dtype_by_scalar_data(data: Any):
     return DataType.UNKNOWN
 
 
-def infer_dtype_bydata(data: Any):
+def infer_dtype_bydata(data: Any, **kargs):
     d_type = DataType.UNKNOWN
     if is_scalar(data):
         return infer_dtype_by_scalar_data(data)
@@ -121,7 +126,17 @@ def infer_dtype_bydata(data: Any):
             failed = True
         if not failed:
             d_type = dtype_str_map.get(type_str, DataType.UNKNOWN)
-            return DataType.FLOAT_VECTOR if is_numeric_datatype(d_type) else DataType.ARRAY
+            if is_varchar_datatype(d_type) or is_bool_datatype(d_type):
+                return DataType.ARRAY
+            if kargs is None or len(kargs) == 0:
+                return DataType.FLOAT_VECTOR if \
+                    is_numeric_datatype(d_type) else DataType.UNKNOWN
+            else:
+                if kargs["type"] is not None and kargs["type"] == "vector":
+                    return DataType.FLOAT_VECTOR \
+                        if is_numeric_datatype(d_type) else DataType.UNKNOWN
+                else:
+                    return DataType.ARRAY
 
     if d_type == DataType.UNKNOWN:
         try: