from sqlalchemy import Column, Integer, String, Text
from backend.app.db import Base


class Experiment(Base):
    __tablename__ = "experiments"

    id = Column(String, primary_key=True, index=True)
    dataset_id = Column(String, nullable=False, index=True)
    algorithm = Column(String, nullable=False)
    status = Column(String, nullable=False)
    config_json = Column(Text, nullable=True)
    metrics_json = Column(Text, nullable=True)
    summary_json = Column(Text, nullable=True)
    runtime_ms = Column(Integer, nullable=True)
    error_message = Column(Text, nullable=True)