写入记录问题

zhaoqingang

2024-11-21 4eb8d55ec0a664678895132256c97cc63a80be2b

 app/service/session.py

@@ -1,5 +1,6 @@
from sqlalchemy.orm import Session

from Log import logger
from app.models import AgentType
from app.models.session_model import SessionModel

@@ -29,7 +30,7 @@

        new_session = SessionModel(
            id=session_id,
            name=name,
            name=name[0:200],
            agent_id=agent_id,
            agent_type=agent_type,
            tenant_id = user_id,
@@ -63,14 +64,14 @@
        返回:
            SessionModel: 更新后的会话模型实例。
        """
        session = self.get_session_by_id(session_id)
        logger.error("更新数据---------------------------")
        session = self.db.query(SessionModel).filter_by(id=session_id).first()
        if session:
            if "message" in kwargs:

                message =  session.message
                message.append(kwargs["message"])
                kwargs["message"] = message
            for key, value in kwargs.items():
                setattr(session, key, value)
                session = message
            self.db.commit()
            self.db.refresh(session)
        return session