vectordb-bench 0.0.18__py3-none-any.whl → 0.0.20__py3-none-any.whl

vectordb_bench/frontend/components/custom/getCustomConfig.py

@@ -32,8 +32,7 @@ def get_custom_configs():
 
 def save_custom_configs(custom_configs: list[CustomDatasetConfig]):
     with open(config.CUSTOM_CONFIG_DIR, "w") as f:
-        json.dump([custom_config.dict()
-                   for custom_config in custom_configs], f, indent=4)
+        json.dump([custom_config.dict() for custom_config in custom_configs], f, indent=4)
 
 
 def generate_custom_case():

vectordb_bench/frontend/components/custom/initStyle.py

@@ -12,4 +12,4 @@ def initStyle(st):
             */
         </style>""",
         unsafe_allow_html=True,
-    )
+    )

vectordb_bench/frontend/components/get_results/saveAsImage.py

@@ -9,10 +9,12 @@ HTML_2_CANVAS_URL = "https://unpkg.com/html2canvas@1.4.1/dist/html2canvas.js"
 def load_unpkg(src: str) -> str:
     return requests.get(src).text
 
+
 def getResults(container, pageName="vectordb_bench"):
     container.subheader("Get results")
     saveAsImage(container, pageName)
 
+
 def saveAsImage(container, pageName):
     html2canvasJS = load_unpkg(HTML_2_CANVAS_URL)
     container.write()

vectordb_bench/frontend/components/run_test/caseSelector.py

@@ -1,6 +1,4 @@
-
 from vectordb_bench.frontend.config.styles import *
-from vectordb_bench.backend.cases import CaseType
 from vectordb_bench.frontend.config.dbCaseConfigs import *
 from collections import defaultdict
 
@@ -23,8 +21,7 @@ def caseSelector(st, activedDbList: list[DB]):
     dbToCaseConfigs = defaultdict(lambda: defaultdict(dict))
     caseClusters = UI_CASE_CLUSTERS + [get_custom_case_cluter()]
     for caseCluster in caseClusters:
-        activedCaseList += caseClusterExpander(
-            st, caseCluster, dbToCaseClusterConfigs, activedDbList)
+        activedCaseList += caseClusterExpander(st, caseCluster, dbToCaseClusterConfigs, activedDbList)
     for db in dbToCaseClusterConfigs:
         for uiCaseItem in dbToCaseClusterConfigs[db]:
             for case in uiCaseItem.cases:
@@ -40,8 +37,7 @@ def caseClusterExpander(st, caseCluster: UICaseItemCluster, dbToCaseClusterConfi
         if uiCaseItem.isLine:
             addHorizontalLine(expander)
         else:
-            activedCases += caseItemCheckbox(expander,
-                                             dbToCaseClusterConfigs, uiCaseItem, activedDbList)
+            activedCases += caseItemCheckbox(expander, dbToCaseClusterConfigs, uiCaseItem, activedDbList)
     return activedCases
 
 
@@ -53,9 +49,7 @@ def caseItemCheckbox(st, dbToCaseClusterConfigs, uiCaseItem: UICaseItem, actived
     )
 
     if selected:
-        caseConfigSetting(
-            st.container(), dbToCaseClusterConfigs, uiCaseItem, activedDbList
-        )
+        caseConfigSetting(st.container(), dbToCaseClusterConfigs, uiCaseItem, activedDbList)
 
     return uiCaseItem.cases if selected else []
 

vectordb_bench/frontend/components/run_test/dbConfigSetting.py

@@ -42,10 +42,7 @@ def dbConfigSettingItem(st, activeDb: DB):
 
     # db config (unique)
     for key, property in properties.items():
-        if (
-            key not in dbConfigClass.common_short_configs()
-            and key not in dbConfigClass.common_long_configs()
-        ):
+        if key not in dbConfigClass.common_short_configs() and key not in dbConfigClass.common_long_configs():
             column = columns[idx % DB_CONFIG_SETTING_COLUMNS]
             idx += 1
             dbConfig[key] = column.text_input(

vectordb_bench/frontend/components/run_test/dbSelector.py

@@ -22,7 +22,7 @@ def dbSelector(st):
         dbIsActived[db] = column.checkbox(db.name)
         try:
             column.image(DB_TO_ICON.get(db, ""))
-        except MediaFileStorageError as e:
+        except MediaFileStorageError:
             column.warning(f"{db.name} image not available")
             pass
     activedDbList = [db for db in DB_LIST if dbIsActived[db]]

vectordb_bench/frontend/components/run_test/generateTasks.py

@@ -7,13 +7,13 @@ def generate_tasks(activedDbList: list[DB], dbConfigs, activedCaseList: list[Cas
     for db in activedDbList:
         for case in activedCaseList:
             task = TaskConfig(
-                    db=db.value,
-                    db_config=dbConfigs[db],
-                    case_config=case,
-                    db_case_config=db.case_config_cls(
-                        allCaseConfigs[db][case].get(CaseConfigParamType.IndexType, None)
-                    )(**{key.value: value for key, value in allCaseConfigs[db][case].items()}),
-                )
+                db=db.value,
+                db_config=dbConfigs[db],
+                case_config=case,
+                db_case_config=db.case_config_cls(allCaseConfigs[db][case].get(CaseConfigParamType.IndexType, None))(
+                    **{key.value: value for key, value in allCaseConfigs[db][case].items()}
+                ),
+            )
             tasks.append(task)
-    
+
     return tasks

vectordb_bench/frontend/components/run_test/submitTask.py

@@ -1,6 +1,6 @@
 from datetime import datetime
-from vectordb_bench.frontend.config.styles import *
-from vectordb_bench.interface import benchMarkRunner
+from vectordb_bench.frontend.config import styles
+from vectordb_bench.interface import benchmark_runner
 
 
 def submitTask(st, tasks, isAllValid):
@@ -27,10 +27,8 @@ def submitTask(st, tasks, isAllValid):
 
 def taskLabelInput(st):
     defaultTaskLabel = datetime.now().strftime("%Y%m%d%H")
-    columns = st.columns(TASK_LABEL_INPUT_COLUMNS)
-    taskLabel = columns[0].text_input(
-        "task_label", defaultTaskLabel, label_visibility="collapsed"
-    )
+    columns = st.columns(styles.TASK_LABEL_INPUT_COLUMNS)
+    taskLabel = columns[0].text_input("task_label", defaultTaskLabel, label_visibility="collapsed")
     return taskLabel
 
 
@@ -46,10 +44,8 @@ def advancedSettings(st):
     )
 
     container = st.columns([1, 2])
-    k = container[0].number_input("k",min_value=1, value=100, label_visibility="collapsed")
-    container[1].caption(
-        "K value for number of nearest neighbors to search"
-    )
+    k = container[0].number_input("k", min_value=1, value=100, label_visibility="collapsed")
+    container[1].caption("K value for number of nearest neighbors to search")
 
     return index_already_exists, use_aliyun, k
 
@@ -58,20 +54,20 @@ def controlPanel(st, tasks, taskLabel, isAllValid):
     index_already_exists, use_aliyun, k = advancedSettings(st)
 
     def runHandler():
-        benchMarkRunner.set_drop_old(not index_already_exists)
+        benchmark_runner.set_drop_old(not index_already_exists)
         for task in tasks:
             task.case_config.k = k
-        benchMarkRunner.set_download_address(use_aliyun)
-        benchMarkRunner.run(tasks, taskLabel)
+        benchmark_runner.set_download_address(use_aliyun)
+        benchmark_runner.run(tasks, taskLabel)
 
     def stopHandler():
-        benchMarkRunner.stop_running()
+        benchmark_runner.stop_running()
 
-    isRunning = benchMarkRunner.has_running()
+    isRunning = benchmark_runner.has_running()
 
     if isRunning:
-        currentTaskId = benchMarkRunner.get_current_task_id()
-        tasksCount = benchMarkRunner.get_tasks_count()
+        currentTaskId = benchmark_runner.get_current_task_id()
+        tasksCount = benchmark_runner.get_tasks_count()
         text = f":running: Running Task {currentTaskId} / {tasksCount}"
         st.progress(currentTaskId / tasksCount, text=text)
 
@@ -89,7 +85,7 @@ def controlPanel(st, tasks, taskLabel, isAllValid):
         )
 
     else:
-        errorText = benchMarkRunner.latest_error or ""
+        errorText = benchmark_runner.latest_error or ""
         if len(errorText) > 0:
             st.error(errorText)
         disabled = True if len(tasks) == 0 or not isAllValid else False

vectordb_bench/frontend/components/tables/data.py

@@ -1,12 +1,11 @@
 from dataclasses import asdict
-from vectordb_bench.backend.cases import CaseType
-from vectordb_bench.interface import benchMarkRunner
+from vectordb_bench.interface import benchmark_runner
 from vectordb_bench.models import CaseResult, ResultLabel
 import pandas as pd
 
 
 def getNewResults():
-    allResults = benchMarkRunner.get_results()
+    allResults = benchmark_runner.get_results()
     newResults: list[CaseResult] = []
 
     for res in allResults:
@@ -14,7 +13,6 @@ def getNewResults():
         for result in results:
             if result.label == ResultLabel.NORMAL:
                 newResults.append(result)
-                    
 
     df = pd.DataFrame(formatData(newResults))
     return df
@@ -26,7 +24,6 @@ def formatData(caseResults: list[CaseResult]):
         db = caseResult.task_config.db.value
         db_label = caseResult.task_config.db_config.db_label
         case_config = caseResult.task_config.case_config
-        db_case_config = caseResult.task_config.db_case_config
         case = case_config.case_id.case_cls()
         filter_rate = case.filter_rate
         dataset = case.dataset.data.name
@@ -41,4 +38,4 @@ def formatData(caseResults: list[CaseResult]):
                 **metrics,
             }
         )
-    return data
+    return data