difyq 接入

zhaoqingang

2024-11-25 dc478b065693dd24e4cae719186d6aafb2d24f6d

 app/service/session.py

@@ -1,3 +1,4 @@
import json
from typing import Type

from sqlalchemy.orm import Session
@@ -39,7 +40,7 @@
            agent_id=agent_id,
            agent_type=agent_type,
            tenant_id=user_id,
            message=[{"role": "user", "content": name}]
            message=json.dumps([{"role": "user", "content": name}])
        )
        self.db.add(new_session)
        self.db.commit()
@@ -57,7 +58,7 @@
            SessionModel: 查找到的会话模型实例，如果未找到则返回None。
        """
        session = self.db.query(SessionModel).filter_by(id=session_id).first()
        if session.message is None:
        if  session and session.message is None:
            session.message = '[]'
        return session

@@ -87,6 +88,7 @@
                self.db.commit()
                self.db.refresh(session)
            except Exception as e:
                logger.error(e)
                self.db.rollback()
        return session