| | |
| | | import json |
| | | from datetime import datetime |
| | | from enum import IntEnum |
| | | |
| | | import pytz |
| | | from sqlalchemy import Column, String, Enum as SQLAlchemyEnum, Integer, DateTime, JSON, TEXT |
| | | |
| | | from app.models import AgentType, current_time |
| | | from app.models.agent_model import AgentType |
| | | # from app.models import current_time |
| | | from app.models.base_model import Base |
| | | |
| | | def current_time(): |
| | | tz = pytz.timezone('Asia/Shanghai') |
| | | return datetime.now(tz) |
| | | |
| | | class SessionModel(Base): |
| | | __tablename__ = "sessions" |
| | |
| | | agent_id = Column(String(255)) |
| | | agent_type = Column(SQLAlchemyEnum(AgentType), nullable=False) # 目前只存basic的,ragflow和bisheng的调接口获取 |
| | | create_date = Column(DateTime, default=current_time) # 创建时间,默认值为当前时区时间 |
| | | update_date = Column(DateTime, default=current_time, onupdate=current_time) # 更新时间,默认值为当前时区时间,更新时自动更新 |
| | | update_date = Column(DateTime, default=current_time, onupdate=current_time, index=True) # 更新时间,默认值为当前时区时间,更新时自动更新 |
| | | tenant_id = Column(Integer) # 创建人 |
| | | message = Column(TEXT) # 说明 |
| | | conversation_id = Column(String(64)) |