zhaoqingang
2025-02-13 25bf0a51088fa0d0acdc39c6096cc45d699f63a1
app/service/session.py
@@ -12,8 +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,
                       message: dict = None, workflow_type: int = 0) -> Type[SessionModel] | 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:
        """
        创建一个新的会话记录。
@@ -30,6 +32,7 @@
            message = {"role": "user", "content": name}
        existing_session = self.get_session_by_id(session_id)
        if existing_session:
            # existing_session.add_message({"role": "user", "content": name})
            existing_session.add_message(message)
            existing_session.update_date = current_time()
            self.db.commit()
@@ -43,7 +46,7 @@
            agent_type=agent_type,
            tenant_id=user_id,
            workflow=workflow_type,
            message=json.dumps([message])
            message = json.dumps([message])
        )
        self.db.add(new_session)
        self.db.commit()