zhaoqingang
2025-03-06 e26a7859a8900b152e10961d91fa6ad19a8deb9c
app/service/session.py
@@ -1,6 +1,10 @@
import json
from typing import Type
from sqlalchemy.orm import Session
from app.models import AgentType
from Log import logger
from app.models import AgentType, current_time
from app.models.session_model import SessionModel
@@ -8,7 +12,10 @@
    def __init__(self, db: Session):
        self.db = db
    def create_session(self, session_id: str, name: str, agent_id: str, agent_type: AgentType, user_id: int) -> SessionModel:
    def create_session(self, session_id: str, name: str, agent_id: str, agent_type: AgentType, user_id: int, message: dict = None, workflow_type: int = 0) -> Type[
                                                                                                                    SessionModel] | SessionModel:
        """
        创建一个新的会话记录。
@@ -21,26 +28,33 @@
        返回:
            SessionModel: 新创建的会话模型实例,如果会话ID已存在则返回None。
        """
        if not message:
            message = {"role": "user", "content": name}
        existing_session = self.get_session_by_id(session_id)
        if existing_session:
            message=existing_session.message
            message.append({"role": "user", "content": name})
            self.update_session(session_id, message=message)
            print("update success")
            # existing_session.add_message({"role": "user", "content": name})
            existing_session.add_message(message)
            existing_session.update_date = current_time()
            self.db.commit()
            self.db.refresh(existing_session)
            return existing_session
        new_session = SessionModel(
            id=session_id,
            name=name,
            name=name[0:50],
            agent_id=agent_id,
            agent_type=agent_type,
            tenant_id = user_id,
            message=[{"role": "user", "content": name}]
            tenant_id=user_id,
            workflow=workflow_type,
            message = json.dumps([message])
        )
        self.db.add(new_session)
        self.db.commit()
        self.db.refresh(new_session)
        return new_session
    def get_session_by_id(self, session_id: str) -> SessionModel:
    def get_session_by_id(self, session_id: str) -> Type[SessionModel] | None:
        """
        根据会话ID获取会话记录。
@@ -50,9 +64,12 @@
        返回:
            SessionModel: 查找到的会话模型实例,如果未找到则返回None。
        """
        return self.db.query(SessionModel).filter_by(id=session_id).first()
        session = self.db.query(SessionModel).filter_by(id=session_id).first()
        if  session and session.message is None:
            session.message = '[]'
        return session
    def update_session(self, session_id: str, **kwargs) -> SessionModel:
    def update_session(self, session_id: str, **kwargs) -> Type[SessionModel] | None:
        """
        更新会话记录。
@@ -63,16 +80,23 @@
        返回:
            SessionModel: 更新后的会话模型实例。
        """
        logger.error("更新数据---------------------------")
        self.db.commit()
        session = self.get_session_by_id(session_id)
        if session:
            if "message" in kwargs:
                message =  session.message
                message.append(kwargs["message"])
                kwargs["message"] = message
                session.add_message(kwargs["message"])
            # 替换其他字段
            for key, value in kwargs.items():
                setattr(session, key, value)
            self.db.commit()
            self.db.refresh(session)
                if key != "message":
                    setattr(session, key, value)
            session.update_date = current_time()
            try:
                self.db.commit()
                self.db.refresh(session)
            except Exception as e:
                logger.error(e)
                self.db.rollback()
        return session
    def delete_session(self, session_id: str) -> None: