add consistency_level on search page (#366)

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

client/src/consts/Milvus.ts

@@ -297,6 +297,25 @@ export enum ConsistencyLevelEnum {
   Customized = 'Customized', // Users pass their own `guarantee_timestamp`.
 }
 
+export const CONSISTENCY_LEVEL_OPTIONS = [
+  {
+    value: ConsistencyLevelEnum.Bounded,
+    label: ConsistencyLevelEnum.Bounded,
+  },
+  {
+    value: ConsistencyLevelEnum.Strong,
+    label: ConsistencyLevelEnum.Strong,
+  },
+  {
+    value: ConsistencyLevelEnum.Session,
+    label: ConsistencyLevelEnum.Session,
+  },
+  {
+    value: ConsistencyLevelEnum.Eventually,
+    label: ConsistencyLevelEnum.Eventually,
+  },
+];
+
 export enum DataTypeStringEnum {
   Bool = 'Bool',
   Int8 = 'Int8',

client/src/pages/search/SearchParams.tsx

@@ -11,6 +11,7 @@ import {
   INDEX_CONFIG,
   METRIC_OPTIONS_MAP,
   searchKeywordsType,
+  CONSISTENCY_LEVEL_OPTIONS,
 } from '@/consts';
 import { rootContext } from '@/context';
 import { useFormValidation } from '@/hooks';
@@ -40,13 +41,16 @@ const SearchParams: FC<SearchParamsProps> = ({
   searchParamsForm,
   handleFormChange,
   handleMetricTypeChange,
+  handleConsistencyChange,
   embeddingType,
   metricType,
+  consistency_level,
   topK,
   setParamsDisabled,
   wrapperClass = '',
 }) => {
   const { t: indexTrans } = useTranslation('index');
+  const { t: collectionTrans } = useTranslation('collection');
   const { t: warningTrans } = useTranslation('warning');
   const classes = getStyles();
 
@@ -331,6 +335,18 @@ const SearchParams: FC<SearchParamsProps> = ({
           handleMetricTypeChange(metricType);
         }}
       />
+      {/* consistency level */}
+      <CustomSelector
+        options={CONSISTENCY_LEVEL_OPTIONS}
+        value={consistency_level}
+        label={collectionTrans('consistencyLevel')}
+        wrapperClass={classes.selector}
+        variant="filled"
+        onChange={(e: { target: { value: unknown } }) => {
+          const consistency = e.target.value as string;
+          handleConsistencyChange(consistency);
+        }}
+      />
       <div className={classes.inlineInputWrapper}>
         {/* dynamic params, now every type only has one param except metric type */}
         {searchParams.map(param => (

client/src/pages/search/Types.ts

@@ -19,8 +19,10 @@ export interface SearchParamsProps {
   topK: number;
   handleFormChange: (form: { [key in string]: number }) => void;
   handleMetricTypeChange: (type: string) => void;
+  handleConsistencyChange: (type: string) => void;
   wrapperClass?: string;
   setParamsDisabled: (isDisabled: boolean) => void;
+  consistency_level: string;
 }
 
 export interface SearchResultView {

client/src/pages/search/VectorSearch.tsx

@@ -162,6 +162,8 @@ const VectorSearch = () => {
   }, [searchResult, primaryKeyField, orderArray]);
 
   const [selectedMetricType, setSelectedMetricType] = useState<string>('');
+  const [selectedConsistencyLevel, setSelectedConsistencyLevel] =
+    useState<string>('');
 
   const {
     indexType,
@@ -293,6 +295,9 @@ const VectorSearch = () => {
     if (selectedCollection !== '') {
       fetchFieldsWithIndex(selectedCollection, collections);
     }
+    const level = collections.find(c => c.collectionName == selectedCollection)
+      ?.consistency_level!;
+    setSelectedConsistencyLevel(level);
   }, [selectedCollection, collections, fetchFieldsWithIndex]);
 
   // set default collection value if is from overview page
@@ -351,9 +356,10 @@ const VectorSearch = () => {
       search_params: searchParamPairs,
       vectors: [parseValue(vectors)],
       vector_type: fieldType,
-      consistency_level: collections.find(
-        c => c.collectionName == selectedCollection
-      )?.consistency_level!,
+      consistency_level:
+        selectedConsistencyLevel ||
+        collections.find(c => c.collectionName == selectedCollection)
+          ?.consistency_level!,
     };
 
     setTableLoading(true);
@@ -472,6 +478,10 @@ const VectorSearch = () => {
                 | DataTypeEnum.BinaryVector
                 | DataTypeEnum.FloatVector
             }
+            consistency_level={selectedConsistencyLevel}
+            handleConsistencyChange={(level: string) => {
+              setSelectedConsistencyLevel(level);
+            }}
             indexType={indexType}
             indexParams={indexParams!}
             searchParamsForm={searchParam}