Params

Author: gmorgachev

src/compressa/perf/cli/__main__.py

@@ -37,7 +37,8 @@ def report_experiment_args(args):
 def list_experiments_args(args):
     list_experiments(
         db=args.db,
-        show_parameters=args.show_parameters
+        show_parameters=args.show_parameters,
+        show_metrics=args.show_metrics
     )
 
 
@@ -176,6 +177,11 @@ def main():
         action="store_true",
         help="Show all parameters for each experiment"
     )
+    parser_list.add_argument(
+        "--show-metrics",
+        action="store_true",
+        help="Show metrics for each experiment"
+    )
     parser_list.set_defaults(func=list_experiments_args)
 
     parser_yaml = subparsers.add_parser(

src/compressa/perf/cli/tools.py

@@ -178,6 +178,7 @@ def report_experiment(
 def list_experiments(
     db: str = DEFAULT_DB_PATH,
     show_parameters: bool = False,
+    show_metrics: bool = False,
 ):
     with sqlite3.connect(db) as conn:
         ensure_db_initialized(conn)
@@ -194,7 +195,7 @@ def list_experiments(
         if show_parameters:
             headers.extend(["Parameters"])
 
-        desciptiont_length = 20 if show_parameters else 50
+        desciptiont_length = 20 if show_parameters or show_metrics else 50
         for exp in experiments:
             row = [
                 exp.id,
@@ -205,9 +206,19 @@ def list_experiments(
 
             if show_parameters:
                 parameters = fetch_parameters_by_experiment(conn, exp.id)
-                param_str = "\n".join([f"{p.key}: {format_value(p.value, precision=2)}" for p in parameters])
+                param_str = "\n".join([
+                    f"{p.key}: {format_value(p.value, precision=2)[:10] + '...' 
+                    if len(format_value(p.value, precision=2)) > 10 
+                    else format_value(p.value, precision=2)}" 
+                    for p in parameters
+                ])
                 row.append(param_str)
 
+            if show_metrics:
+                metrics = fetch_metrics_by_experiment(conn, exp.id)
+                metrics_str = "\n".join([f"{m.metric_name.value}: {format_value(m.metric_value)}" for m in metrics])
+                row.append(metrics_str)
+
             table_data.append(row)
 
         print("\nList of Experiments:")

src/compressa/perf/experiment/inference.py

@@ -134,6 +134,12 @@ def store_experiment_parameters(
                 key="max_tokens",
                 value=str(max_tokens),
             ),
+            Parameter(
+                id=None,
+                experiment_id=experiment_id,
+                key="model_name",
+                value=self.model_name,
+            ),
         ]
         for param in parameters:
             insert_parameter(self.conn, param)