31 lines
1.3 KiB
Python
31 lines
1.3 KiB
Python
|
"""
|
|||
|
|
文档向量化模型
|
|||
|
|
"""
|
|||
|
|
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}')>"
|