增加会话列表类型

zhaoqingang

2025-01-06 f14a9e9a8a8636da2da7c217b0e5057951238347

增加会话列表类型

 app/api/chat.py

@@ -577,7 +577,7 @@
                                                    outputs = answer.get("outputs", {})
                                                    if outputs:
                                                        message = outputs.get("output", "")
                                                        download_url = outputs.get("download_url", "")
                                                        download_url = outputs.get("download_url", "").rstrip("/")
                                                    else:
                                                        message = answer.get("error", "")
                                                    if download_url:

 app/models/session_model.py

@@ -8,7 +8,7 @@


class SessionModel(Base):
    __tablename__ = "sessions"
    __tablename__ = "sessions_log"
    id = Column(String(255), primary_key=True)
    name = Column(String(255))
    agent_id = Column(String(255))
@@ -18,7 +18,7 @@
    tenant_id = Column(Integer)  # 创建人
    message = Column(TEXT)  # 说明
    conversation_id = Column(String(64))
    # workflow = Column(Integer, default=0)
    workflow = Column(Integer, default=0)

    # to_dict 方法
    def to_dict(self):
@@ -27,7 +27,7 @@
            'name': self.name,
            'agent_type': self.agent_type,
            'agent_id': self.agent_id,
            # 'workflow': self.workflow,
            'workflow': self.workflow if self.workflow else 0,
            'create_date': self.create_date.strftime("%Y-%m-%d %H:%M:%S"),
            'update_date': self.update_date.strftime("%Y-%m-%d %H:%M:%S"),
        }

 app/service/session.py

@@ -42,7 +42,7 @@
            agent_id=agent_id,
            agent_type=agent_type,
            tenant_id=user_id,
            # workflow=workflow_type,
            workflow=workflow_type,
            message=json.dumps([message])
        )
        self.db.add(new_session)