import json
|
from datetime import datetime
|
from enum import IntEnum
|
from sqlalchemy import Column, String, Enum as SQLAlchemyEnum, Integer, DateTime, JSON
|
|
from app.models import AgentType, current_time
|
from app.models.base_model import Base
|
|
|
class SessionModel(Base):
|
__tablename__ = "sessions"
|
id = Column(String(255), primary_key=True)
|
name = Column(String(255))
|
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) # 更新时间，默认值为当前时区时间，更新时自动更新
|
tenant_id = Column(Integer) # 创建人
|
message = Column(JSON) # 说明
|
|
# to_dict 方法
|
def to_dict(self):
|
return {
|
'id': self.id,
|
'name': self.name,
|
'agent_type': self.agent_type,
|
'agent_id': self.agent_id,
|
'create_date': self.create_date,
|
'update_date': self.update_date,
|
}
|
|
def log_to_json(self):
|
return {
|
'id': self.id,
|
'name': self.name,
|
'agent_type': self.agent_type,
|
'agent_id': self.agent_id,
|
'create_date': self.create_date,
|
'update_date': self.update_date,
|
'message': self.message
|
}
|
