44822a561a
Changes: - Fix datetime serialization with UTC 'Z' suffix for correct timezone display - Add PDF upload support with extension fallback for MIME detection - Fix LOT add/remove by creating new list for SQLAlchemy JSON change detection - Add file message components (FileMessage, ImageLightbox, UploadPreview) - Add multi-file upload support with progress tracking - Link uploaded files to chat messages via message_id - Include file attachments in AI report generation - Update specs for file-storage, realtime-messaging, and ai-report-generation 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
103 lines
2.9 KiB
Python
103 lines
2.9 KiB
Python
"""SQLAlchemy models for report generation
|
|
|
|
Tables:
|
|
- tr_generated_reports: Stores report metadata and generation status
|
|
|
|
Note: All tables use 'tr_' prefix to avoid conflicts in shared database.
|
|
"""
|
|
from sqlalchemy import Column, String, Text, DateTime, Integer, ForeignKey, Index, JSON
|
|
from sqlalchemy.orm import relationship
|
|
from datetime import datetime
|
|
import uuid
|
|
import enum
|
|
from app.core.database import Base
|
|
|
|
|
|
class ReportStatus(str, enum.Enum):
|
|
"""Report generation status"""
|
|
PENDING = "pending"
|
|
COLLECTING_DATA = "collecting_data"
|
|
GENERATING_CONTENT = "generating_content"
|
|
ASSEMBLING_DOCUMENT = "assembling_document"
|
|
COMPLETED = "completed"
|
|
FAILED = "failed"
|
|
|
|
|
|
class GeneratedReport(Base):
|
|
"""Generated report model for incident reports"""
|
|
|
|
__tablename__ = "tr_generated_reports"
|
|
|
|
report_id = Column(
|
|
String(36), primary_key=True, default=lambda: str(uuid.uuid4()),
|
|
comment="Unique report identifier (UUID)"
|
|
)
|
|
room_id = Column(
|
|
String(36), ForeignKey("tr_incident_rooms.room_id", ondelete="CASCADE"),
|
|
nullable=False, comment="Reference to incident room"
|
|
)
|
|
|
|
# Generation metadata
|
|
generated_by = Column(
|
|
String(255), nullable=False,
|
|
comment="User email who triggered report generation"
|
|
)
|
|
generated_at = Column(
|
|
DateTime, default=datetime.utcnow, nullable=False,
|
|
comment="Report generation timestamp"
|
|
)
|
|
|
|
# Status tracking
|
|
status = Column(
|
|
String(30), default=ReportStatus.PENDING.value, nullable=False,
|
|
comment="Current generation status"
|
|
)
|
|
error_message = Column(
|
|
Text, nullable=True,
|
|
comment="User-friendly error message if generation failed"
|
|
)
|
|
|
|
# DIFY AI metadata
|
|
dify_message_id = Column(
|
|
String(100), nullable=True,
|
|
comment="DIFY API message ID for tracking"
|
|
)
|
|
dify_conversation_id = Column(
|
|
String(100), nullable=True,
|
|
comment="DIFY conversation ID"
|
|
)
|
|
prompt_tokens = Column(
|
|
Integer, nullable=True,
|
|
comment="Number of prompt tokens used"
|
|
)
|
|
completion_tokens = Column(
|
|
Integer, nullable=True,
|
|
comment="Number of completion tokens used"
|
|
)
|
|
|
|
# Report content
|
|
report_title = Column(
|
|
String(255), nullable=True,
|
|
comment="Generated report title"
|
|
)
|
|
report_json = Column(
|
|
JSON, nullable=True,
|
|
comment="Parsed AI output as JSON"
|
|
)
|
|
docx_storage_path = Column(
|
|
String(500), nullable=True,
|
|
comment="Path to generated .docx file in MinIO or local storage"
|
|
)
|
|
|
|
# Relationship
|
|
room = relationship("IncidentRoom", back_populates="reports")
|
|
|
|
# Indexes
|
|
__table_args__ = (
|
|
Index("ix_tr_generated_reports_room_date", "room_id", "generated_at"),
|
|
Index("ix_tr_generated_reports_status", "status"),
|
|
)
|
|
|
|
def __repr__(self):
|
|
return f"<GeneratedReport {self.report_id} status={self.status}>"
|