update embedding type

client/src/consts/Milvus.tsx

@@ -1,3 +1,5 @@
+import { DataTypeEnum } from '../pages/collections/Types';
+
 export enum METRIC_TYPES_VALUES {
   L2 = 'L2',
   IP = 'IP',
@@ -121,11 +123,11 @@ export const m_OPTIONS = [
 ];
 
 export const INDEX_OPTIONS_MAP = {
-  FLOAT_INDEX: Object.keys(FLOAT_INDEX_CONFIG).map(v => ({
+  [DataTypeEnum.FloatVector]: Object.keys(FLOAT_INDEX_CONFIG).map(v => ({
     label: v,
     value: v,
   })),
-  BINARY_INDEX: Object.keys(BINARY_INDEX_CONFIG).map(v => ({
+  [DataTypeEnum.BinaryVector]: Object.keys(BINARY_INDEX_CONFIG).map(v => ({
     label: v,
     value: v,
   })),
@@ -133,13 +135,8 @@ export const INDEX_OPTIONS_MAP = {
 
 export const PRIMARY_KEY_FIELD = 'INT64 (Primary key)';
 
-export enum EmbeddingTypeEnum {
-  float = 'FLOAT_INDEX',
-  binary = 'BINARY_INDEX',
-}
-
 export const METRIC_OPTIONS_MAP = {
-  [EmbeddingTypeEnum.float]: [
+  [DataTypeEnum.FloatVector]: [
     {
       value: METRIC_TYPES_VALUES.L2,
       label: METRIC_TYPES_VALUES.L2,
@@ -149,7 +146,7 @@ export const METRIC_OPTIONS_MAP = {
       label: METRIC_TYPES_VALUES.IP,
     },
   ],
-  [EmbeddingTypeEnum.binary]: [
+  [DataTypeEnum.BinaryVector]: [
     {
       value: METRIC_TYPES_VALUES.SUBSTRUCTURE,
       label: METRIC_TYPES_VALUES.SUBSTRUCTURE,
@@ -178,8 +175,8 @@ export const METRIC_OPTIONS_MAP = {
  * use Hamming as binary default metric type
  */
 export const DEFAULT_METRIC_VALUE_MAP = {
-  [EmbeddingTypeEnum.float]: METRIC_TYPES_VALUES.L2,
-  [EmbeddingTypeEnum.binary]: METRIC_TYPES_VALUES.HAMMING,
+  [DataTypeEnum.FloatVector]: METRIC_TYPES_VALUES.L2,
+  [DataTypeEnum.BinaryVector]: METRIC_TYPES_VALUES.HAMMING,
 };
 
 // search params default value map

client/src/pages/seach/Types.ts

@@ -1,5 +1,5 @@
 import { Option } from '../../components/customSelector/Types';
-import { EmbeddingTypeEnum, searchKeywordsType } from '../../consts/Milvus';
+import { searchKeywordsType } from '../../consts/Milvus';
 import { DataType, DataTypeEnum } from '../collections/Types';
 import { IndexView } from '../schema/Types';
 
@@ -8,7 +8,7 @@ export interface SearchParamsProps {
   // else pass empty string
   metricType: string;
   // used for getting metric type options
-  embeddingType: EmbeddingTypeEnum;
+  embeddingType: DataTypeEnum.FloatVector | DataTypeEnum.BinaryVector;
   // default index type is FLAT
   indexType: string;
   // index extra params, e.g. nlist

client/src/pages/seach/VectorSearch.tsx

@@ -5,10 +5,7 @@ import { ALL_ROUTER_TYPES } from '../../router/Types';
 import CustomSelector from '../../components/customSelector/CustomSelector';
 import { useCallback, useEffect, useMemo, useState } from 'react';
 import SearchParams from './SearchParams';
-import {
-  DEFAULT_METRIC_VALUE_MAP,
-  EmbeddingTypeEnum,
-} from '../../consts/Milvus';
+import { DEFAULT_METRIC_VALUE_MAP } from '../../consts/Milvus';
 import { FieldOption, SearchResultView, VectorSearchParam } from './Types';
 import MilvusGrid from '../../components/grid/Grid';
 import EmptyCard from '../../components/cards/EmptyCard';
@@ -149,7 +146,7 @@ const VectorSearch = () => {
         indexType: '',
         indexParams: [],
         fieldType: 0,
-        embeddingType: EmbeddingTypeEnum.float,
+        embeddingType: DataTypeEnum.FloatVector,
       };
     }, [selectedField, fieldOptions]);
 
@@ -326,7 +323,11 @@ const VectorSearch = () => {
           <SearchParams
             wrapperClass={classes.paramsWrapper}
             metricType={metricType!}
-            embeddingType={embeddingType}
+            embeddingType={
+              embeddingType as
+                | DataTypeEnum.BinaryVector
+                | DataTypeEnum.FloatVector
+            }
             indexType={indexType}
             indexParams={indexParams!}
             searchParamsForm={searchParam}

client/src/utils/search.ts

@@ -1,6 +1,5 @@
 import { Field } from '../components/advancedSearch/Types';
-import { EmbeddingTypeEnum } from '../consts/Milvus';
-import { DataType } from '../pages/collections/Types';
+import { DataType, DataTypeEnum } from '../pages/collections/Types';
 import {
   FieldData,
   IndexType,
@@ -32,11 +31,13 @@ export const transferSearchResult = (
  * @param fieldType only vector type fields: 'BinaryVector' or 'FloatVector'
  * @returns 'FLOAT_INDEX' or 'BINARY_INDEX'
  */
-export const getEmbeddingType = (fieldType: DataType): EmbeddingTypeEnum => {
+export const getEmbeddingType = (
+  fieldType: DataType
+): DataTypeEnum.BinaryVector | DataTypeEnum.FloatVector => {
   const type =
     fieldType === 'BinaryVector'
-      ? EmbeddingTypeEnum.binary
-      : EmbeddingTypeEnum.float;
+      ? DataTypeEnum.BinaryVector
+      : DataTypeEnum.FloatVector;
   return type;
 };
 
@@ -46,11 +47,9 @@ export const getEmbeddingType = (fieldType: DataType): EmbeddingTypeEnum => {
  * @param embeddingType float or binary
  * @returns index type
  */
-export const getDefaultIndexType = (
-  embeddingType: EmbeddingTypeEnum
-): IndexType => {
+export const getDefaultIndexType = (embeddingType: DataTypeEnum): IndexType => {
   const defaultIndexType =
-    embeddingType === EmbeddingTypeEnum.float
+    embeddingType === DataTypeEnum.FloatVector
       ? INDEX_TYPES_ENUM.FLAT
       : INDEX_TYPES_ENUM.BIN_FLAT;
   return defaultIndexType;