hide create index action if there are multiple vectors in the collection (#486)

client/src/consts/Milvus.ts

@@ -25,7 +25,7 @@ export enum DataTypeEnum {
   Array = 22,
 }
 
-export const vectorTypes = [
+export const VectorTypes = [
   DataTypeEnum.BinaryVector,
   DataTypeEnum.FloatVector,
   DataTypeEnum.BFloat16Vector,

client/src/i18n/cn/collection.ts

@@ -127,8 +127,8 @@ const collectionTrans = {
     '它确保在同一会话中所有数据写入可以立即在读取中感知。',
   consistencyEventuallyTooltip:
     '没有保证读写的顺序，副本最终会在没有进一步写操作的情况下收敛到相同的状态。',
-  releaseCollectionFirst: '请先释放collection.',
-  noVectorIndexTooltip: '没有向量索引，请先创建一个。',
+  releaseCollectionFirst: '请先释放collection。',
+  noVectorIndexTooltip: '请保证所有向量列都有索引。',
 
   clickToLoad: '点击加载collection。',
   clickToRelease: '点击释放collection。',

client/src/i18n/en/collection.ts

@@ -126,7 +126,7 @@ const collectionTrans = {
   consistencySessionTooltip: `It ensures that all data writes can be immediately perceived in reads during the same session.`,
   consistencyEventuallyTooltip: `There is no guaranteed order of reads and writes, and replicas eventually converge to the same state given that no further write operations are done.`,
   releaseCollectionFirst: `Please release your collection first.`,
-  noVectorIndexTooltip: `No vector index, create one first.`,
+  noVectorIndexTooltip: `Please make sure all vector fields have index.`,
 
   clickToLoad: 'Click to load the collection.',
   clickToRelease: 'Click to release the collection.',

client/src/pages/databases/collections/CreateFields.tsx

@@ -24,7 +24,7 @@ import {
   VECTOR_FIELDS_OPTIONS,
 } from './Constants';
 import { CreateFieldsProps, CreateFieldType, FieldType } from './Types';
-import { DataTypeEnum, vectorTypes } from '@/consts';
+import { DataTypeEnum, VectorTypes } from '@/consts';
 import {
   DEFAULT_ATTU_DIM,
   DEFAULT_ATTU_MAX_CAPACITY,
@@ -243,7 +243,7 @@ const CreateFields: FC<CreateFieldsProps> = ({
     className?: string
   ) => {
     const defaultLabal = collectionTrans(
-      vectorTypes.includes(field.data_type) ? 'vectorFieldName' : 'fieldName'
+      VectorTypes.includes(field.data_type) ? 'vectorFieldName' : 'fieldName'
     );
 
     return getInput({
@@ -464,7 +464,7 @@ const CreateFields: FC<CreateFieldsProps> = ({
 
       // remove dimension, if not vector
       if (
-        !vectorTypes.includes(updatedField.data_type) ||
+        !VectorTypes.includes(updatedField.data_type) ||
         updatedField.data_type === DataTypeEnum.SparseFloatVector
       ) {
         delete updatedField.dimension;

client/src/pages/databases/collections/overview/Create.tsx

@@ -10,7 +10,7 @@ import {
   INDEX_TYPES_ENUM,
   DataTypeEnum,
   DataTypeStringEnum,
-  vectorTypes,
+  VectorTypes,
 } from '@/consts';
 import { useFormValidation } from '@/hooks';
 import { getCreateIndexJSCode } from '@/utils/code/Js';
@@ -115,7 +115,7 @@ const CreateIndex = (props: {
   }, [indexSetting.index_type]);
 
   const metricOptions = useMemo(() => {
-    return vectorTypes.includes(dataType)
+    return VectorTypes.includes(dataType)
       ? getMetricOptions(indexSetting.index_type, dataType)
       : [];
   }, [indexSetting.index_type, fieldType]);
@@ -156,7 +156,7 @@ const CreateIndex = (props: {
   }, [fieldType]);
 
   const checkedForm = useMemo(() => {
-    if (!vectorTypes.includes(dataType)) {
+    if (!VectorTypes.includes(dataType)) {
       return [];
     }
     const paramsForm: any = { metric_type: indexSetting.metric_type };
@@ -171,7 +171,7 @@ const CreateIndex = (props: {
    * create index code mode
    */
   const codeBlockData: CodeViewData[] = useMemo(() => {
-    const isScalarField = !vectorTypes.includes(dataType);
+    const isScalarField = !VectorTypes.includes(dataType);
     const getCodeParams = {
       collectionName,
       fieldName,

client/src/pages/databases/collections/overview/Overview.tsx

@@ -257,6 +257,28 @@ const Overview = () => {
     },
   ];
 
+
+  // only show create index element when there is only one vector field
+  let CreateIndexElement = null;
+  if (
+    collection &&
+    collection.schema &&
+    collection.schema.vectorFields.length === 1
+  ) {
+    CreateIndexElement = (
+      <IndexTypeElement
+        field={
+          (collection.schema && collection.schema.vectorFields[0]) ||
+          ({} as FieldObject)
+        }
+        collectionName={collectionName}
+        cb={async () => {
+          await fetchCollection(collectionName);
+        }}
+      />
+    );
+  }
+
   // get loading state label
   return (
     <section className={classes.wrapper}>
@@ -280,19 +302,7 @@ const Overview = () => {
                 percentage={collection.loadedPercentage}
                 collection={collection}
                 showExtraAction={true}
-                createIndexElement={
-                  <IndexTypeElement
-                    field={
-                      (collection.schema &&
-                        collection.schema.vectorFields[0]) ||
-                      ({} as FieldObject)
-                    }
-                    collectionName={collectionName}
-                    cb={async () => {
-                      await fetchCollection(collectionName);
-                    }}
-                  />
-                }
+                createIndexElement={CreateIndexElement}
               />
             </div>
           </section>

client/src/pages/dialogs/CreateCollectionDialog.tsx

@@ -17,7 +17,7 @@ import {
   DataTypeEnum,
   ConsistencyLevelEnum,
   DEFAULT_ATTU_DIM,
-  vectorTypes,
+  VectorTypes,
 } from '@/consts';
 import CreateFields from '../databases/collections/CreateFields';
 import {
@@ -233,7 +233,7 @@ const CreateCollectionDialog: FC<CollectionCreateProps> = ({ onCreate }) => {
 
         v.is_primary_key && (data.autoID = form.autoID);
 
-        const param = vectorTypes.includes(v.data_type)
+        const param = VectorTypes.includes(v.data_type)
           ? {
               ...data,
               type_params: {

client/src/utils/Format.ts

@@ -2,8 +2,7 @@ import {
   BYTE_UNITS,
   DEFAULT_MILVUS_PORT,
   DEFAULT_PROMETHEUS_PORT,
-  DataTypeEnum,
-  vectorTypes,
+  VectorTypes,
 } from '@/consts';
 import {
   CreateFieldType,
@@ -108,7 +107,7 @@ export const getCreateFieldType = (config: CreateField): CreateFieldType => {
     return 'defaultVector';
   }
 
-  if (vectorTypes.includes(config.data_type)) {
+  if (VectorTypes.includes(config.data_type)) {
     return 'vector';
   }
 

server/src/collections/collections.service.ts

@@ -39,6 +39,7 @@ import {
   DYNAMIC_FIELD,
   SimpleQueue,
   MIN_INT64,
+  VectorTypes,
 } from '../utils';
 import { QueryDto, ImportSampleDto, GetReplicasDto } from './dto';
 import {
@@ -112,10 +113,7 @@ export class CollectionsService {
         ) || -1;
 
       // classify fields
-      if (
-        field.data_type === 'BinaryVector' ||
-        field.data_type === 'FloatVector'
-      ) {
+      if (VectorTypes.includes(field.data_type)) {
         vectorFields.push(field);
       } else {
         scalarFields.push(field);
@@ -127,7 +125,7 @@ export class CollectionsService {
     });
 
     // add extra data to schema
-    res.schema.hasVectorIndex = vectorFields.some(v => v.index);
+    res.schema.hasVectorIndex = vectorFields.every(v => v.index);
     res.schema.scalarFields = scalarFields;
     res.schema.vectorFields = vectorFields;
     res.schema.dynamicFields = res.schema.enable_dynamic_field

server/src/utils/Const.ts

@@ -156,3 +156,11 @@ export enum LOADING_STATE {
 
 export const MIN_INT64 = `-9223372036854775807`; // safe int64 min value
 export const DYNAMIC_FIELD = `$meta`;
+
+export const VectorTypes = [
+  'BinaryVector',
+  'FloatVector',
+  'SpaseFloatVector',
+  'Float16Vector',
+  'BFloat16Vector',
+];