add pgdiskann load config in frontend.

zilliztech · Oct 25, 2024 · 69ce9f4 · 69ce9f4
1 parent d081997
commit 69ce9f4
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/vectordb_bench/frontend/config/dbCaseConfigs.py b/vectordb_bench/frontend/config/dbCaseConfigs.py
@@ -950,6 +950,12 @@ class CaseConfigInput(BaseModel):
     CaseConfigParamInput_query_search_list_size,
 ]
 
+PgDiskANNLoadConfig = [
+    CaseConfigParamInput_IndexType_PgDiskANN,
+    CaseConfigParamInput_max_neighbors,
+    CaseConfigParamInput_l_value_ib,
+]
+
 PgDiskANNPerformanceConfig = [
     CaseConfigParamInput_IndexType_PgDiskANN,
     CaseConfigParamInput_max_neighbors,
@@ -986,6 +992,7 @@ class CaseConfigInput(BaseModel):
         CaseLabel.Performance: PgVectorScalePerformanceConfig,
     },
     DB.PgDiskANN: {
+        CaseLabel.Load: PgDiskANNLoadConfig,
         CaseLabel.Performance: PgDiskANNPerformanceConfig,
     },
 }