support bf16 and f16 search (#491)

Signed-off-by: shanghaikid <jiangruiyi@gmail.com>

client/src/pages/search/VectorSearch.tsx

@@ -100,7 +100,13 @@ const VectorSearch = () => {
     const fields = (s.schema && s.schema.fields) || [];
 
     // vector field can't be output fields
-    const invalidTypes = ['BinaryVector', 'FloatVector', 'SparseFloatVector'];
+    const invalidTypes = [
+      'BinaryVector',
+      'FloatVector',
+      'Float16Vector',
+      'BFloat16Vector',
+      'SparseFloatVector',
+    ];
     const nonVectorFields = fields.filter(
       field => !invalidTypes.includes(field.data_type)
     );

server/src/collections/dto.ts

@@ -21,6 +21,8 @@ enum VectorTypes {
   Binary = DataType.BinaryVector,
   Float = DataType.FloatVector,
   SparseFloatVector = DataType.SparseFloatVector,
+  Float16Vector = DataType.Float16Vector,
+  BFloat16Vector = DataType.BFloat16Vector,
 }
 
 export class CreateCollectionDto {
@@ -99,7 +101,7 @@ export class VectorSearchDto {
   @IsEnum(VectorTypes, {
     message: ({ value }) => `Wrong vector type, ${value}`,
   })
-  vector_type: DataType.BinaryVector | DataType.FloatVector;
+  vector_type: VectorTypes;
 }
 
 export class CreateAliasDto {