flowyml 1.3.0__py3-none-any.whl → 1.5.0__py3-none-any.whl

flowyml/storage/metadata.py

@@ -1,11 +1,6 @@
 """Metadata storage backends for flowyml."""
 
-import json
-import sqlite3
 from abc import ABC, abstractmethod
-from pathlib import Path
-import contextlib
-import builtins
 
 
 class MetadataStore(ABC):
@@ -52,1041 +47,7 @@ class MetadataStore(ABC):
         pass
 
 
-class SQLiteMetadataStore(MetadataStore):
-    """SQLite-based metadata storage."""
+from flowyml.storage.sql import SQLMetadataStore  # noqa: E402
 
-    def __init__(self, db_path: str = ".flowyml/metadata.db"):
-        """Initialize SQLite metadata store.
-
-        Args:
-            db_path: Path to SQLite database file
-        """
-        self.db_path = Path(db_path)
-        self.db_path.parent.mkdir(parents=True, exist_ok=True)
-        self._init_db()
-
-    def _init_db(self) -> None:
-        """Initialize database schema."""
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        # Runs table
-        cursor.execute(
-            """
-            CREATE TABLE IF NOT EXISTS runs (
-                run_id TEXT PRIMARY KEY,
-                pipeline_name TEXT,
-                status TEXT,
-                start_time TEXT,
-                end_time TEXT,
-                duration REAL,
-                metadata TEXT,
-                project TEXT,
-                created_at TEXT DEFAULT CURRENT_TIMESTAMP
-            )
-        """,
-        )
-
-        # Migration: Add project column if it doesn't exist
-        # Migration: Add project column if it doesn't exist
-        with contextlib.suppress(sqlite3.OperationalError):
-            cursor.execute("ALTER TABLE runs ADD COLUMN project TEXT")
-
-        # Artifacts table
-        cursor.execute(
-            """
-            CREATE TABLE IF NOT EXISTS artifacts (
-                artifact_id TEXT PRIMARY KEY,
-                name TEXT,
-                type TEXT,
-                run_id TEXT,
-                path TEXT,
-                metadata TEXT,
-                project TEXT,
-                created_at TEXT DEFAULT CURRENT_TIMESTAMP,
-                FOREIGN KEY (run_id) REFERENCES runs(run_id)
-            )
-        """,
-        )
-
-        # Migration: Add project column to artifacts if it doesn't exist
-        # Migration: Add project column to artifacts if it doesn't exist
-        with contextlib.suppress(sqlite3.OperationalError):
-            cursor.execute("ALTER TABLE artifacts ADD COLUMN project TEXT")
-
-        # Metrics table
-        cursor.execute(
-            """
-            CREATE TABLE IF NOT EXISTS metrics (
-                id INTEGER PRIMARY KEY AUTOINCREMENT,
-                run_id TEXT,
-                name TEXT,
-                value REAL,
-                step INTEGER,
-                timestamp TEXT DEFAULT CURRENT_TIMESTAMP,
-                FOREIGN KEY (run_id) REFERENCES runs(run_id)
-            )
-        """,
-        )
-
-        # Model metrics table
-        cursor.execute(
-            """
-            CREATE TABLE IF NOT EXISTS model_metrics (
-                id INTEGER PRIMARY KEY AUTOINCREMENT,
-                project TEXT,
-                model_name TEXT,
-                run_id TEXT,
-                metric_name TEXT,
-                metric_value REAL,
-                environment TEXT,
-                tags TEXT,
-                created_at TEXT DEFAULT CURRENT_TIMESTAMP
-            )
-        """,
-        )
-
-        # Parameters table
-        cursor.execute(
-            """
-            CREATE TABLE IF NOT EXISTS parameters (
-                id INTEGER PRIMARY KEY AUTOINCREMENT,
-                run_id TEXT,
-                name TEXT,
-                value TEXT,
-                FOREIGN KEY (run_id) REFERENCES runs(run_id)
-            )
-        """,
-        )
-
-        # Experiments table
-        cursor.execute(
-            """
-            CREATE TABLE IF NOT EXISTS experiments (
-                experiment_id TEXT PRIMARY KEY,
-                name TEXT,
-                description TEXT,
-                tags TEXT,
-                project TEXT,
-                created_at TEXT DEFAULT CURRENT_TIMESTAMP
-            )
-        """,
-        )
-
-        # Migration: Add project column to experiments if it doesn't exist
-        # Migration: Add project column to experiments if it doesn't exist
-        with contextlib.suppress(sqlite3.OperationalError):
-            cursor.execute("ALTER TABLE experiments ADD COLUMN project TEXT")
-
-        # Experiment Runs link table
-        cursor.execute(
-            """
-            CREATE TABLE IF NOT EXISTS experiment_runs (
-                experiment_id TEXT,
-                run_id TEXT,
-                metrics TEXT,
-                parameters TEXT,
-                timestamp TEXT DEFAULT CURRENT_TIMESTAMP,
-                PRIMARY KEY (experiment_id, run_id),
-                FOREIGN KEY (experiment_id) REFERENCES experiments(experiment_id),
-                FOREIGN KEY (run_id) REFERENCES runs(run_id)
-            )
-        """,
-        )
-
-        # Traces table for GenAI monitoring
-        cursor.execute(
-            """
-            CREATE TABLE IF NOT EXISTS traces (
-                event_id TEXT PRIMARY KEY,
-                trace_id TEXT,
-                parent_id TEXT,
-                event_type TEXT,
-                name TEXT,
-                inputs TEXT,
-                outputs TEXT,
-                start_time REAL,
-                end_time REAL,
-                duration REAL,
-                status TEXT,
-                error TEXT,
-                metadata TEXT,
-                prompt_tokens INTEGER,
-                completion_tokens INTEGER,
-                total_tokens INTEGER,
-                cost REAL,
-                model TEXT,
-                project TEXT,
-                created_at TEXT DEFAULT CURRENT_TIMESTAMP
-            )
-        """,
-        )
-
-        # Migration: Add project column to traces if it doesn't exist
-        # Migration: Add project column to traces if it doesn't exist
-        with contextlib.suppress(sqlite3.OperationalError):
-            cursor.execute("ALTER TABLE traces ADD COLUMN project TEXT")
-
-        # Create indexes for better query performance
-        cursor.execute("CREATE INDEX IF NOT EXISTS idx_runs_pipeline ON runs(pipeline_name)")
-        cursor.execute("CREATE INDEX IF NOT EXISTS idx_runs_status ON runs(status)")
-        cursor.execute("CREATE INDEX IF NOT EXISTS idx_runs_project ON runs(project)")
-        cursor.execute("CREATE INDEX IF NOT EXISTS idx_artifacts_run ON artifacts(run_id)")
-        cursor.execute("CREATE INDEX IF NOT EXISTS idx_artifacts_project ON artifacts(project)")
-        cursor.execute("CREATE INDEX IF NOT EXISTS idx_metrics_run ON metrics(run_id)")
-        cursor.execute("CREATE INDEX IF NOT EXISTS idx_parameters_run ON parameters(run_id)")
-        cursor.execute("CREATE INDEX IF NOT EXISTS idx_experiments_name ON experiments(name)")
-        cursor.execute("CREATE INDEX IF NOT EXISTS idx_experiments_project ON experiments(project)")
-        cursor.execute("CREATE INDEX IF NOT EXISTS idx_traces_trace_id ON traces(trace_id)")
-        cursor.execute("CREATE INDEX IF NOT EXISTS idx_traces_type ON traces(event_type)")
-        cursor.execute("CREATE INDEX IF NOT EXISTS idx_traces_project ON traces(project)")
-
-        # Pipeline definitions for scheduling
-        cursor.execute(
-            """
-            CREATE TABLE IF NOT EXISTS pipeline_definitions (
-                pipeline_name TEXT PRIMARY KEY,
-                definition TEXT NOT NULL,
-                created_at TEXT NOT NULL,
-                updated_at TEXT NOT NULL
-            )
-            """,
-        )
-
-        conn.commit()
-        conn.close()
-
-    def save_run(self, run_id: str, metadata: dict) -> None:
-        """Save run metadata to database.
-
-        Args:
-            run_id: Unique run identifier
-            metadata: Run metadata dictionary
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        cursor.execute(
-            """
-            INSERT OR REPLACE INTO runs
-            (run_id, pipeline_name, status, start_time, end_time, duration, metadata, project)
-            VALUES (?, ?, ?, ?, ?, ?, ?, ?)
-        """,
-            (
-                run_id,
-                metadata.get("pipeline_name"),
-                metadata.get("status"),
-                metadata.get("start_time"),
-                metadata.get("end_time"),
-                metadata.get("duration"),
-                json.dumps(metadata),
-                metadata.get("project"),
-            ),
-        )
-
-        # Save parameters
-        if "parameters" in metadata:
-            cursor.execute("DELETE FROM parameters WHERE run_id = ?", (run_id,))
-            for name, value in metadata["parameters"].items():
-                cursor.execute(
-                    "INSERT INTO parameters (run_id, name, value) VALUES (?, ?, ?)",
-                    (run_id, name, json.dumps(value)),
-                )
-
-        # Save metrics
-        if "metrics" in metadata:
-            cursor.execute("DELETE FROM metrics WHERE run_id = ?", (run_id,))
-            for name, value in metadata["metrics"].items():
-                cursor.execute(
-                    "INSERT INTO metrics (run_id, name, value, step) VALUES (?, ?, ?, ?)",
-                    (run_id, name, value, 0),
-                )
-
-        conn.commit()
-        conn.close()
-
-    def load_run(self, run_id: str) -> dict | None:
-        """Load run metadata from database.
-
-        Args:
-            run_id: Unique run identifier
-
-        Returns:
-            Run metadata dictionary or None if not found
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        cursor.execute("SELECT metadata FROM runs WHERE run_id = ?", (run_id,))
-        row = cursor.fetchone()
-
-        conn.close()
-
-        if row:
-            data = json.loads(row[0])
-            # Ensure project is in metadata if it's in the column but not the JSON blob
-            # (This might happen if we update the column directly)
-            # Actually, let's just return what's in the blob for now,
-            # but we should probably sync them.
-            return data
-        return None
-
-    def update_run_project(self, run_id: str, project_name: str) -> None:
-        """Update the project for a run.
-
-        Args:
-            run_id: Run identifier
-            project_name: Name of the project
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        # 1. Update the column
-        cursor.execute("UPDATE runs SET project = ? WHERE run_id = ?", (project_name, run_id))
-
-        # 2. Update the JSON blob
-        cursor.execute("SELECT metadata FROM runs WHERE run_id = ?", (run_id,))
-        row = cursor.fetchone()
-        if row:
-            metadata = json.loads(row[0])
-            metadata["project"] = project_name
-            cursor.execute(
-                "UPDATE runs SET metadata = ? WHERE run_id = ?",
-                (json.dumps(metadata), run_id),
-            )
-
-        conn.commit()
-        conn.close()
-
-    def list_runs(self, limit: int | None = None) -> list[dict]:
-        """List all runs from database.
-
-        Args:
-            limit: Optional limit on number of results
-
-        Returns:
-            List of run metadata dictionaries
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        query = "SELECT metadata FROM runs ORDER BY created_at DESC"
-        if limit:
-            query += f" LIMIT {limit}"
-
-        cursor.execute(query)
-        rows = cursor.fetchall()
-
-        conn.close()
-
-        return [json.loads(row[0]) for row in rows]
-
-    def list_pipelines(self, project: str = None) -> list[str]:
-        """List all unique pipeline names.
-
-        Args:
-            project: Optional project name to filter by
-
-        Returns:
-            List of pipeline names
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        if project:
-            cursor.execute(
-                "SELECT DISTINCT pipeline_name FROM runs WHERE project = ? ORDER BY pipeline_name",
-                (project,),
-            )
-        else:
-            cursor.execute("SELECT DISTINCT pipeline_name FROM runs ORDER BY pipeline_name")
-
-        rows = cursor.fetchall()
-
-        conn.close()
-
-        return [row[0] for row in rows if row[0]]
-
-    def save_artifact(self, artifact_id: str, metadata: dict) -> None:
-        """Save artifact metadata to database.
-
-        Args:
-            artifact_id: Unique artifact identifier
-            metadata: Artifact metadata dictionary
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        cursor.execute(
-            """
-            INSERT OR REPLACE INTO artifacts
-            (artifact_id, name, type, run_id, path, metadata, project)
-            VALUES (?, ?, ?, ?, ?, ?, ?)
-        """,
-            (
-                artifact_id,
-                metadata.get("name"),
-                metadata.get("type"),
-                metadata.get("run_id"),
-                metadata.get("path"),
-                json.dumps(metadata),
-                metadata.get("project"),
-            ),
-        )
-
-        conn.commit()
-        conn.close()
-
-    def load_artifact(self, artifact_id: str) -> dict | None:
-        """Load artifact metadata from database.
-
-        Args:
-            artifact_id: Unique artifact identifier
-
-        Returns:
-            Artifact metadata dictionary or None if not found
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        cursor.execute("SELECT metadata FROM artifacts WHERE artifact_id = ?", (artifact_id,))
-        row = cursor.fetchone()
-
-        conn.close()
-
-        if row:
-            return json.loads(row[0])
-        return None
-
-    def list_assets(self, limit: int | None = None, **filters) -> list[dict]:
-        """List assets from database with optional filters.
-
-        Args:
-            limit: Optional limit on number of results
-            **filters: Filter criteria (type, run_id, etc.)
-
-        Returns:
-            List of artifact metadata dictionaries
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        conditions = []
-        params = []
-
-        for key, value in filters.items():
-            if value is not None:
-                conditions.append(f"{key} = ?")
-                params.append(value)
-
-        query = "SELECT metadata FROM artifacts"
-        if conditions:
-            query += " WHERE " + " AND ".join(conditions)
-
-        query += " ORDER BY created_at DESC"
-
-        if limit:
-            query += f" LIMIT {limit}"
-
-        cursor.execute(query, params)
-        rows = cursor.fetchall()
-
-        conn.close()
-
-        return [json.loads(row[0]) for row in rows]
-
-    def query(self, **filters) -> list[dict]:
-        """Query runs with filters.
-
-        Args:
-            **filters: Filter criteria (pipeline_name, status, etc.)
-
-        Returns:
-            List of matching run metadata dictionaries
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        where_clauses = []
-        params = []
-
-        for key, value in filters.items():
-            where_clauses.append(f"{key} = ?")
-            params.append(value)
-
-        query = "SELECT metadata FROM runs"
-        if where_clauses:
-            query += " WHERE " + " AND ".join(where_clauses)
-        query += " ORDER BY created_at DESC"
-
-        cursor.execute(query, params)
-        rows = cursor.fetchall()
-
-        conn.close()
-
-        return [json.loads(row[0]) for row in rows]
-
-    def save_metric(self, run_id: str, name: str, value: float, step: int = 0) -> None:
-        """Save a single metric value.
-
-        Args:
-            run_id: Run identifier
-            name: Metric name
-            value: Metric value
-            step: Training step/iteration
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        cursor.execute(
-            "INSERT INTO metrics (run_id, name, value, step) VALUES (?, ?, ?, ?)",
-            (run_id, name, value, step),
-        )
-
-        conn.commit()
-        conn.close()
-
-    def get_metrics(self, run_id: str, name: str | None = None) -> list[dict]:
-        """Get metrics for a run.
-
-        Args:
-            run_id: Run identifier
-            name: Optional metric name filter
-
-        Returns:
-            List of metric dictionaries
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        if name:
-            cursor.execute(
-                "SELECT name, value, step, timestamp FROM metrics WHERE run_id = ? AND name = ? ORDER BY step",
-                (run_id, name),
-            )
-        else:
-            cursor.execute(
-                "SELECT name, value, step, timestamp FROM metrics WHERE run_id = ? ORDER BY step",
-                (run_id,),
-            )
-
-        rows = cursor.fetchall()
-        conn.close()
-
-        return [{"name": row[0], "value": row[1], "step": row[2], "timestamp": row[3]} for row in rows]
-
-    def log_model_metrics(
-        self,
-        project: str,
-        model_name: str,
-        metrics: dict[str, float],
-        run_id: str | None = None,
-        environment: str | None = None,
-        tags: dict | None = None,
-    ) -> None:
-        """Log production model metrics independent of pipeline runs."""
-        if not metrics:
-            return
-
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-        tags_json = json.dumps(tags or {})
-
-        for metric_name, value in metrics.items():
-            try:
-                metric_value = float(value)
-            except (TypeError, ValueError):
-                continue
-
-            cursor.execute(
-                """
-                INSERT INTO model_metrics
-                (project, model_name, run_id, metric_name, metric_value, environment, tags)
-                VALUES (?, ?, ?, ?, ?, ?, ?)
-                """,
-                (project, model_name, run_id, metric_name, metric_value, environment, tags_json),
-            )
-
-        conn.commit()
-        conn.close()
-
-    def list_model_metrics(
-        self,
-        project: str | None = None,
-        model_name: str | None = None,
-        limit: int = 100,
-    ) -> list[dict]:
-        """List logged model metrics."""
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        query = """
-            SELECT project, model_name, run_id, metric_name, metric_value, environment, tags, created_at
-            FROM model_metrics
-        """
-        params: list = []
-        clauses = []
-
-        if project:
-            clauses.append("project = ?")
-            params.append(project)
-        if model_name:
-            clauses.append("model_name = ?")
-            params.append(model_name)
-
-        if clauses:
-            query += " WHERE " + " AND ".join(clauses)
-
-        query += " ORDER BY created_at DESC LIMIT ?"
-        params.append(limit)
-
-        cursor.execute(query, params)
-        rows = cursor.fetchall()
-        conn.close()
-
-        results = []
-        for row in rows:
-            results.append(
-                {
-                    "project": row[0],
-                    "model_name": row[1],
-                    "run_id": row[2],
-                    "metric_name": row[3],
-                    "metric_value": row[4],
-                    "environment": row[5],
-                    "tags": json.loads(row[6]) if row[6] else {},
-                    "created_at": row[7],
-                },
-            )
-        return results
-
-    def save_experiment(self, experiment_id: str, name: str, description: str = "", tags: dict = None) -> None:
-        """Save experiment metadata.
-
-        Args:
-            experiment_id: Unique experiment identifier
-            name: Experiment name
-            description: Experiment description
-            tags: Experiment tags
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        cursor.execute(
-            """
-            INSERT OR REPLACE INTO experiments
-            (experiment_id, name, description, tags)
-            VALUES (?, ?, ?, ?)
-        """,
-            (
-                experiment_id,
-                name,
-                description,
-                json.dumps(tags or {}),
-            ),
-        )
-
-        conn.commit()
-        conn.close()
-
-    def log_experiment_run(
-        self,
-        experiment_id: str,
-        run_id: str,
-        metrics: dict = None,
-        parameters: dict = None,
-    ) -> None:
-        """Log a run to an experiment.
-
-        Args:
-            experiment_id: Experiment identifier
-            run_id: Run identifier
-            metrics: Metrics from the run
-            parameters: Parameters used in the run
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        cursor.execute(
-            """
-            INSERT OR REPLACE INTO experiment_runs
-            (experiment_id, run_id, metrics, parameters)
-            VALUES (?, ?, ?, ?)
-        """,
-            (
-                experiment_id,
-                run_id,
-                json.dumps(metrics or {}),
-                json.dumps(parameters or {}),
-            ),
-        )
-
-        conn.commit()
-        conn.close()
-
-    def list_experiments(self) -> list[dict]:
-        """List all experiments.
-
-        Returns:
-            List of experiment dictionaries
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        cursor.execute(
-            "SELECT experiment_id, name, description, tags, created_at, project FROM experiments ORDER BY created_at DESC",
-        )
-        rows = cursor.fetchall()
-
-        experiments = []
-        for row in rows:
-            # Count runs for each experiment
-            cursor.execute("SELECT COUNT(*) FROM experiment_runs WHERE experiment_id = ?", (row[0],))
-            run_count = cursor.fetchone()[0]
-
-            experiments.append(
-                {
-                    "experiment_id": row[0],
-                    "name": row[1],
-                    "description": row[2],
-                    "tags": json.loads(row[3]),
-                    "created_at": row[4],
-                    "project": row[5],
-                    "run_count": run_count,
-                },
-            )
-        conn.close()
-        return experiments
-
-    def update_experiment_project(self, experiment_name: str, project_name: str) -> None:
-        """Update the project for an experiment.
-
-        Args:
-            experiment_name: Name of the experiment
-            project_name: New project name
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        try:
-            cursor.execute(
-                "UPDATE experiments SET project = ? WHERE name = ?",
-                (project_name, experiment_name),
-            )
-            conn.commit()
-        finally:
-            conn.close()
-
-    def get_experiment(self, experiment_id: str) -> dict | None:
-        """Get experiment details.
-
-        Args:
-            experiment_id: Experiment identifier
-
-        Returns:
-            Experiment dictionary or None
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        cursor.execute(
-            "SELECT experiment_id, name, description, tags, created_at FROM experiments WHERE experiment_id = ?",
-            (experiment_id,),
-        )
-        row = cursor.fetchone()
-
-        if not row:
-            conn.close()
-            return None
-
-        experiment = {
-            "experiment_id": row[0],
-            "name": row[1],
-            "description": row[2],
-            "tags": json.loads(row[3]),
-            "created_at": row[4],
-        }
-
-        # Get runs
-        cursor.execute(
-            """
-            SELECT er.run_id, er.metrics, er.parameters, er.timestamp, r.status, r.duration
-            FROM experiment_runs er
-            LEFT JOIN runs r ON er.run_id = r.run_id
-            WHERE er.experiment_id = ?
-            ORDER BY er.timestamp DESC
-        """,
-            (experiment_id,),
-        )
-
-        runs = []
-        for r in cursor.fetchall():
-            runs.append(
-                {
-                    "run_id": r[0],
-                    "metrics": json.loads(r[1]),
-                    "parameters": json.loads(r[2]),
-                    "timestamp": r[3],
-                    "status": r[4],
-                    "duration": r[5],
-                },
-            )
-
-        experiment["runs"] = runs
-
-        conn.close()
-        return experiment
-
-    def get_statistics(self) -> dict:
-        """Get database statistics.
-
-        Returns:
-            Dictionary with statistics
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        stats = {}
-
-        cursor.execute("SELECT COUNT(*) FROM runs")
-        stats["total_runs"] = cursor.fetchone()[0]
-
-        cursor.execute("SELECT COUNT(*) FROM artifacts")
-        stats["total_artifacts"] = cursor.fetchone()[0]
-
-        cursor.execute("SELECT COUNT(*) FROM metrics")
-        stats["total_metrics"] = cursor.fetchone()[0]
-
-        cursor.execute("SELECT COUNT(DISTINCT pipeline_name) FROM runs")
-        stats["total_pipelines"] = cursor.fetchone()[0]
-
-        cursor.execute("SELECT COUNT(*) FROM experiments")
-        stats["total_experiments"] = cursor.fetchone()[0]
-
-        conn.close()
-
-        return stats
-
-    def save_trace_event(self, event: dict) -> None:
-        """Save a trace event.
-
-        Args:
-            event: Trace event dictionary
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        cursor.execute(
-            """
-            INSERT OR REPLACE INTO traces
-            (event_id, trace_id, parent_id, event_type, name, inputs, outputs,
-             start_time, end_time, duration, status, error, metadata,
-             prompt_tokens, completion_tokens, total_tokens, cost, model)
-            VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-        """,
-            (
-                event["event_id"],
-                event["trace_id"],
-                event["parent_id"],
-                event["event_type"],
-                event["name"],
-                json.dumps(event.get("inputs", {})),
-                json.dumps(event.get("outputs", {})),
-                event.get("start_time"),
-                event.get("end_time"),
-                event.get("duration"),
-                event.get("status"),
-                event.get("error"),
-                json.dumps(event.get("metadata", {})),
-                event.get("prompt_tokens", 0),
-                event.get("completion_tokens", 0),
-                event.get("total_tokens", 0),
-                event.get("cost", 0.0),
-                event.get("model"),
-            ),
-        )
-
-        conn.commit()
-        conn.close()
-
-    def get_trace(self, trace_id: str) -> list[dict]:
-        """Get all events for a trace.
-
-        Args:
-            trace_id: Trace identifier
-
-        Returns:
-            List of event dictionaries
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        cursor.execute(
-            """
-            SELECT * FROM traces WHERE trace_id = ? ORDER BY start_time
-        """,
-            (trace_id,),
-        )
-
-        columns = [description[0] for description in cursor.description]
-        rows = cursor.fetchall()
-
-        events = []
-        for row in rows:
-            event = dict(zip(columns, row, strict=False))
-            # Parse JSON fields
-            for field in ["inputs", "outputs", "metadata"]:
-                if event[field]:
-                    with contextlib.suppress(builtins.BaseException):
-                        event[field] = json.loads(event[field])
-            events.append(event)
-
-        conn.close()
-        return events
-
-    def save_pipeline_definition(self, pipeline_name: str, definition: dict) -> None:
-        """Save pipeline definition for scheduling."""
-        from datetime import datetime
-
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-        now = datetime.now().isoformat()
-
-        # Check if definition already exists
-        cursor.execute(
-            "SELECT pipeline_name FROM pipeline_definitions WHERE pipeline_name = ?",
-            (pipeline_name,),
-        )
-        exists = cursor.fetchone()
-
-        if exists:
-            # Update existing
-            cursor.execute(
-                """
-                UPDATE pipeline_definitions
-                SET definition = ?, updated_at = ?
-                WHERE pipeline_name = ?
-                """,
-                (json.dumps(definition), now, pipeline_name),
-            )
-        else:
-            # Insert new
-            cursor.execute(
-                """
-                INSERT INTO pipeline_definitions (pipeline_name, definition, created_at, updated_at)
-                VALUES (?, ?, ?, ?)
-                """,
-                (pipeline_name, json.dumps(definition), now, now),
-            )
-
-        conn.commit()
-        conn.close()
-
-    def update_pipeline_project(self, pipeline_name: str, project_name: str) -> None:
-        """Update the project for all runs of a pipeline.
-
-        Args:
-            pipeline_name: Name of the pipeline
-            project_name: New project name
-        """
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-
-        try:
-            # 1. Update the project column for all runs
-            cursor.execute(
-                "UPDATE runs SET project = ? WHERE pipeline_name = ?",
-                (project_name, pipeline_name),
-            )
-
-            # 2. Update the JSON metadata blob for each run
-            cursor.execute(
-                "SELECT run_id, metadata FROM runs WHERE pipeline_name = ?",
-                (pipeline_name,),
-            )
-            rows = cursor.fetchall()
-            for run_id, metadata_json in rows:
-                metadata = json.loads(metadata_json)
-                metadata["project"] = project_name
-                cursor.execute(
-                    "UPDATE runs SET metadata = ? WHERE run_id = ?",
-                    (json.dumps(metadata), run_id),
-                )
-
-            # 3. Update artifacts table
-            cursor.execute("PRAGMA table_info(artifacts)")
-            columns = [info[1] for info in cursor.fetchall()]
-            if "project" in columns:
-                cursor.execute(
-                    """
-                    UPDATE artifacts
-                    SET project = ?
-                    WHERE run_id IN (SELECT run_id FROM runs WHERE pipeline_name = ?)
-                    """,
-                    (project_name, pipeline_name),
-                )
-
-            # 4. Update traces table
-            cursor.execute("PRAGMA table_info(traces)")
-            columns = [info[1] for info in cursor.fetchall()]
-            if "project" in columns:
-                # Update traces linked to runs of this pipeline
-                # Note: This assumes we can link traces to runs via metadata or some other way
-                # For now, let's assume traces might have run_id in metadata or we just update by project if we had it
-                # But here we are moving a pipeline to a project.
-                # If traces have a 'project' column, we should update it for traces belonging to these runs.
-                # Since traces don't explicitly have run_id column in schema (it's in metadata),
-                # we might need a more complex query or just skip if not easily linkable.
-                # However, if we assume traces are logged with project context, we might not need to update them
-                # if they were already correct. But if we are MOVING, we need to.
-                # Let's try to update traces that have run_id in their metadata matching these runs.
-                # This is expensive in SQLite with JSON.
-                # Alternative: If traces are associated with the pipeline name directly?
-                # For now, let's skip complex JSON matching for traces to avoid performance issues
-                # unless we add a run_id column to traces.
-                pass
-
-            # 5. Update model_metrics table
-            cursor.execute("PRAGMA table_info(model_metrics)")
-            columns = [info[1] for info in cursor.fetchall()]
-            if "project" in columns:
-                cursor.execute(
-                    """
-                    UPDATE model_metrics
-                    SET project = ?
-                    WHERE run_id IN (SELECT run_id FROM runs WHERE pipeline_name = ?)
-                    """,
-                    (project_name, pipeline_name),
-                )
-
-            # 6. Update experiments table
-            # If an experiment contains runs from this pipeline, should the experiment be moved?
-            # Maybe not automatically, as an experiment might contain runs from multiple pipelines.
-            # But if the user wants "recursive", let's at least update the experiment_runs link
-            # (which doesn't have project) - wait, experiments have project.
-            # Let's find experiments that ONLY contain runs from this pipeline and move them?
-            # Or just leave experiments as is?
-            # The user said "same for all related objects, experiments etc".
-            # Let's be safe and NOT move experiments automatically as they are higher level grouping.
-            # BUT, we should ensure that the runs inside the experiment are consistent.
-            # The runs are already updated in step 1.
-            # So, we are good on experiments.
-
-            conn.commit()
-        finally:
-            conn.close()
-
-    def get_pipeline_definition(self, pipeline_name: str) -> dict | None:
-        """Retrieve pipeline definition."""
-        conn = sqlite3.connect(self.db_path)
-        cursor = conn.cursor()
-        cursor.execute(
-            "SELECT definition FROM pipeline_definitions WHERE pipeline_name = ?",
-            (pipeline_name,),
-        )
-        row = cursor.fetchone()
-        conn.close()
-
-        if row:
-            return json.loads(row[0])
-        return None
+# Alias for backward compatibility
+SQLiteMetadataStore = SQLMetadataStore