[Go & Cpp] Fix vector type check

Signed-off-by: sunby <sunbingyi1992@gmail.com>
milvus-io · Oct 8, 2023 · 2838023 · 2838023
1 parent c58990f
commit 2838023
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/cpp/src/storage/options.cpp b/cpp/src/storage/options.cpp
@@ -1,4 +1,5 @@
 #include "storage/options.h"
+#include <arrow/type_fwd.h>
 #include "arrow/type.h"
 #include "common/status.h"
 
@@ -30,7 +31,7 @@ Status SchemaOptions::Validate(const arrow::Schema* schema) {
     auto vector_field = schema->GetFieldByName(vector_column);
     if (!vector_field) {
       return Status::InvalidArgument("vector column is not exist");
-    } else if (vector_field->type()->id() != arrow::Type::FIXED_SIZE_BINARY) {
+    } else if (vector_field->type()->id() != arrow::Type::FIXED_SIZE_BINARY && vector_field->type()->id() != arrow::Type::FIXED_SIZE_LIST) {
       return Status::InvalidArgument("vector column is not fixed size binary");
     }
   } else {

diff --git a/go/storage/schema/schema_option.go b/go/storage/schema/schema_option.go
@@ -69,7 +69,7 @@ func (o *SchemaOptions) Validate(schema *arrow.Schema) error {
 		vectorField, b := schema.FieldsByName(o.VectorColumn)
 		if !b {
 			return ErrVectorColumnNotFound
-		} else if vectorField[0].Type.ID() != arrow.FIXED_SIZE_BINARY {
+		} else if vectorField[0].Type.ID() != arrow.FIXED_SIZE_BINARY && vectorField[0].Type.ID() != arrow.FIXED_SIZE_LIST {
 			return ErrVectorColumnType
 		}
 	} else {