reduce benchmark time by reducing default context num and adding a co…

…nfig with a lot of contexts

chatsky/utils/db_benchmark/basic_config.py

@@ -91,7 +91,7 @@ class BasicBenchmarkConfig(BenchmarkConfig, frozen=True):
     Dialog length is configured using `from_dialog_len`, `to_dialog_len`, `step_dialog_len`.
     """
 
-    context_num: int = 30
+    context_num: int = 5
     """
     Number of times the contexts will be benchmarked.
     Increasing this number decreases standard error of the mean for benchmarked data.
@@ -188,25 +188,25 @@ def context_updater(self, context: Context) -> Optional[Context]:
         misc_dimensions=(0, 0),
     ),
     "default": BasicBenchmarkConfig(),
-    "large-misc--long-dialog": BasicBenchmarkConfig(
+    "many-contexts": BasicBenchmarkConfig(
+        context_num=50,
+    ),
+    "large-misc-long-dialog": BasicBenchmarkConfig(
         from_dialog_len=500,
         to_dialog_len=550,
         message_dimensions=(3, 5, 6, 5, 3),
         misc_dimensions=(2, 4, 3, 8, 100),
     ),
     "very-long-dialog-len": BasicBenchmarkConfig(
-        context_num=10,
         from_dialog_len=10000,
         to_dialog_len=10050,
     ),
     "very-long-message-len": BasicBenchmarkConfig(
-        context_num=10,
         from_dialog_len=1,
         to_dialog_len=3,
         message_dimensions=(10000, 1),
     ),
     "very-long-misc-len": BasicBenchmarkConfig(
-        context_num=10,
         from_dialog_len=1,
         to_dialog_len=3,
         misc_dimensions=(10000, 1),