| from sqlalchemy import JSON, Column, DateTime, Integer, String, Text, UniqueConstraint |
|
|
| from .db import Base |
| from .time_utils import now_ist |
|
|
|
|
| class ResearchRunRecord(Base): |
| __tablename__ = "research_runs" |
|
|
| id = Column(String(64), primary_key=True) |
| status = Column(String(32), nullable=False, index=True) |
| input_type = Column(String(32), nullable=False, index=True) |
| input_payload = Column(JSON, nullable=False) |
| report_payload = Column(JSON, nullable=True) |
| error_message = Column(Text, nullable=True) |
| created_at = Column(DateTime(timezone=True), nullable=False, default=now_ist) |
| updated_at = Column(DateTime(timezone=True), nullable=False, default=now_ist, onupdate=now_ist) |
| completed_at = Column(DateTime(timezone=True), nullable=True) |
|
|
|
|
| class ArtifactCacheRecord(Base): |
| __tablename__ = "artifact_cache" |
| __table_args__ = (UniqueConstraint("source", "cache_key", name="uq_source_cache_key"),) |
|
|
| id = Column(Integer, primary_key=True, autoincrement=True) |
| source = Column(String(64), nullable=False, index=True) |
| cache_key = Column(String(255), nullable=False, index=True) |
| payload = Column(JSON, nullable=False) |
| created_at = Column(DateTime(timezone=True), nullable=False, default=now_ist) |
| updated_at = Column(DateTime(timezone=True), nullable=False, default=now_ist, onupdate=now_ist) |
|
|
