ソースを参照

Merge pull request #240 from zilliztech/diskann

support diskann
ryjiang 1 年間 前
コミット
c6c771c4a7
2 ファイル変更18 行追加9 行削除
  1. 7 9
      client/src/consts/Milvus.tsx
  2. 11 0
      client/src/pages/search/SearchParams.tsx

+ 7 - 9
client/src/consts/Milvus.tsx

@@ -59,7 +59,8 @@ export type searchKeywordsType =
   | 'search_k'
   | 'search_length'
   | 'round_decimal'
-  | 'level';
+  | 'level'
+  | 'search_list';
 
 export type indexConfigType = {
   [x: string]: {
@@ -82,10 +83,6 @@ export const FLOAT_INDEX_CONFIG: indexConfigType = {
     create: ['nlist'],
     search: ['nprobe'],
   },
-  // IVF_SQ8_HYBRID: {
-  //   create: ['nlist'],
-  //   search: ['nprobe'],
-  // },
   FLAT: {
     create: [],
     search: ['nprobe'],
@@ -102,10 +99,10 @@ export const FLOAT_INDEX_CONFIG: indexConfigType = {
     create: [],
     search: ['level'],
   },
-  // RNSG: {
-  //   create: ['out_degree', 'candidate_pool_size', 'search_length', 'knng'],
-  //   search: ['search_length'],
-  // },}
+  DISKANN: {
+    create: [],
+    search: ['search_list'],
+  },
 };
 
 export const BINARY_INDEX_CONFIG: indexConfigType = {
@@ -210,6 +207,7 @@ export const DEFAULT_SEARCH_PARAM_VALUE_MAP: {
   search_length: 10,
   round_decimal: -1,
   level: 1,
+  search_list: 20,
 };
 
 export const DEFAULT_NLIST_VALUE = 1024;

+ 11 - 0
client/src/pages/search/SearchParams.tsx

@@ -236,6 +236,17 @@ const SearchParams: FC<SearchParamsProps> = ({
             handleInputChange('search_length', Number(value));
           },
         },
+        search_list: {
+          label: 'search_list',
+          key: 'search_list',
+          value: searchParamsForm['search_list'] || '',
+          min: 20,
+          max: 65535,
+          isInt: true,
+          handleChange: value => {
+            handleInputChange('search_list', Number(value));
+          },
+        },
       };
 
       const param = configParamMap[paramKey];