Fix run selection issue (#396)

Co-authored-by: gradio-pr-bot <gradio-pr-bot@users.noreply.github.com>

Author: abidlabs

.changeset/silver-baths-exist.md

@@ -0,0 +1,5 @@
+---
+"trackio": patch
+---
+
+feat:Fix run selection issue

tests/unit/test_sqlite_storage.py

@@ -232,3 +232,34 @@ def test_old_database_without_configs_table(temp_dir):
 
     all_configs = SQLiteStorage.get_all_run_configs("test")
     assert all_configs == {}
+
+
+def test_get_runs_returns_chronological_order(temp_dir):
+    db_path = SQLiteStorage.get_project_db_path("proj")
+    db_path.parent.mkdir(parents=True, exist_ok=True)
+
+    with sqlite3.connect(db_path) as conn:
+        conn.execute("""
+            CREATE TABLE metrics (
+                id INTEGER PRIMARY KEY,
+                timestamp TEXT,
+                run_name TEXT,
+                step INTEGER,
+                metrics TEXT
+            )
+        """)
+        conn.execute(
+            "INSERT INTO metrics (timestamp, run_name, step, metrics) VALUES (?, ?, ?, ?)",
+            ("2024-01-01", "run-z", 0, orjson.dumps({"loss": 0.5})),
+        )
+        conn.execute(
+            "INSERT INTO metrics (timestamp, run_name, step, metrics) VALUES (?, ?, ?, ?)",
+            ("2024-01-02", "run-a", 0, orjson.dumps({"loss": 0.5})),
+        )
+        conn.execute(
+            "INSERT INTO metrics (timestamp, run_name, step, metrics) VALUES (?, ?, ?, ?)",
+            ("2024-01-03", "run-m", 0, orjson.dumps({"loss": 0.5})),
+        )
+
+    runs = SQLiteStorage.get_runs("proj")
+    assert runs == ["run-z", "run-a", "run-m"]

trackio/sqlite_storage.py

@@ -734,15 +734,20 @@ def get_projects() -> list[str]:
 
     @staticmethod
     def get_runs(project: str) -> list[str]:
-        """Get list of all runs for a project."""
+        """Get list of all runs for a project, ordered by creation time."""
         db_path = SQLiteStorage.get_project_db_path(project)
         if not db_path.exists():
             return []
 
         with SQLiteStorage._get_connection(db_path) as conn:
             cursor = conn.cursor()
             cursor.execute(
-                "SELECT DISTINCT run_name FROM metrics",
+                """
+                SELECT run_name
+                FROM metrics
+                GROUP BY run_name
+                ORDER BY MIN(timestamp) ASC
+                """,
             )
             return [row[0] for row in cursor.fetchall()]