| | |
|---|
| | | 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) -> Type[ |
|---|
| | | |
|---|
| | | 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: |
|---|
| | | |
|---|
| | | """ |
|---|
| | | 创建一个新的会话记录。 |
|---|
| | | |
|---|
| | |
|---|
| | | 返回: |
|---|
| | | SessionModel: 新创建的会话模型实例，如果会话ID已存在则返回None。 |
|---|
| | | """ |
|---|
| | | if not message: |
|---|
| | | 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}) |
|---|
| | | print("update success") |
|---|
| | | # existing_session.add_message({"role": "user", "content": name}) |
|---|
| | | existing_session.add_message(message) |
|---|
| | | existing_session.update_date = current_time() |
|---|
| | | self.db.commit() |
|---|
| | | self.db.refresh(existing_session) |
|---|
| | |
|---|
| | | agent_id=agent_id, |
|---|
| | | agent_type=agent_type, |
|---|
| | | tenant_id=user_id, |
|---|
| | | message=json.dumps([{"role": "user", "content": name}]) |
|---|
| | | workflow=workflow_type, |
|---|
| | | message = json.dumps([message]) |
|---|
| | | ) |
|---|
| | | self.db.add(new_session) |
|---|
| | | self.db.commit() |
|---|
| | |
|---|
| | | self.db.commit() |
|---|
| | | self.db.refresh(session) |
|---|
| | | except Exception as e: |
|---|
| | | logger.error(e) |
|---|
| | | self.db.rollback() |
|---|
| | | return session |
|---|
| | | |
|---|
