自研agent会话保存和查询

zhaoqingang

2024-11-21 ae30d9a75407c912649f11c4f44ff15c869a4f98

 app/service/session.py

@@ -8,7 +8,7 @@
    def __init__(self, db: Session):
        self.db = db

    def create_session(self, session_id: str, name: str, agent_id: str, agent_type: AgentType) -> SessionModel:
    def create_session(self, session_id: str, name: str, agent_id: str, agent_type: AgentType, user_id: int) -> SessionModel:
        """
        创建一个新的会话记录。

@@ -23,13 +23,17 @@
        """
        existing_session = self.get_session_by_id(session_id)
        if existing_session:
            return None  # 如果会话ID已存在，不进行任何操作
            message=existing_session.message
            message.append({"role": "user", "content": name})
            self.update_session(session_id, message=message)

        new_session = SessionModel(
            id=session_id,
            name=name,
            agent_id=agent_id,
            agent_type=agent_type
            agent_type=agent_type,
            tenant_id = user_id,
            message=[{"role": "user", "content": name}]
        )
        self.db.add(new_session)
        self.db.commit()
@@ -61,6 +65,10 @@
        """
        session = self.get_session_by_id(session_id)
        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)
            self.db.commit()