| | |
| | | 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: |
| | | |
| | | """ |
| | | 创建一个新的会话记录。 |
| | | |
| | |
| | | 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() |
| | |
| | | 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() |