12-16

zhaoqingang

2024-12-16 2f84e3cc4bff981590ab37bd95da64bd72c1f2e0

 app/models/session_model.py

@@ -1,11 +1,17 @@
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
import pytz
from sqlalchemy import Column, String, Enum as SQLAlchemyEnum, Integer, DateTime, JSON, TEXT

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"
@@ -16,7 +22,8 @@
    create_date = Column(DateTime, default=current_time)  # 创建时间，默认值为当前时区时间
    update_date = Column(DateTime, default=current_time, onupdate=current_time)  # 更新时间，默认值为当前时区时间，更新时自动更新
    tenant_id = Column(Integer)  # 创建人
    message = Column(JSON)  # 说明
    message = Column(TEXT)  # 说明
    conversation_id = Column(String(64))

    # to_dict 方法
    def to_dict(self):
@@ -37,5 +44,15 @@
            'agent_id': self.agent_id,
            'create_date': self.create_date.strftime("%Y-%m-%d %H:%M:%S"),
            'update_date': self.update_date.strftime("%Y-%m-%d %H:%M:%S"),
            'message': self.message
            'message': json.loads(self.message)
        }

    def add_message(self, message: dict):
        if self.message is None:
            self.message = '[]'
        try:
            msg = json.loads(self.message)
            msg.append(message)
        except Exception as e:
            return
        self.message = json.dumps(msg)