"""
文档向量化模型
"""
from sqlalchemy import Column, BigInteger, String, DateTime, Index, Text
from sqlalchemy.sql import func
from app.core.database import Base


class DocumentVector(Base):
    """文档向量表模型"""

    __tablename__ = "document_vector"

    id = Column(BigInteger, primary_key=True, autoincrement=True, comment="向量ID")
    project_id = Column(BigInteger, nullable=False, index=True, comment="项目ID")
    file_path = Column(String(500), nullable=False, comment="文件相对路径")
    content_hash = Column(String(64), comment="内容哈希值，用于判断文件是否变更")
    zvec_id = Column(String(256), comment="ZVec返回的向量ID")
    zvec_response = Column(Text, comment="ZVec完整响应JSON")
    status = Column(String(32), nullable=False, default="success", comment="向量化状态：success/failed/pending")
    error_message = Column(String(500), comment="错误信息")
    created_at = Column(DateTime, server_default=func.now(), comment="创建时间")
    updated_at = Column(DateTime, server_default=func.now(), onupdate=func.now(), comment="更新时间")

    __table_args__ = (
        Index("idx_project_file", "project_id", "file_path"),
    )

    def __repr__(self):
        return f"<DocumentVector(id={self.id}, project_id={self.project_id}, file_path='{self.file_path}', status='{self.status}')>"