yohua

zhaoqingang

2025-01-06 a01c838f32839349c243f586fb003333899cab81

 app/models/agent_model.py

@@ -9,6 +9,7 @@
    RAGFLOW = 1
    BISHENG = 2
    BASIC = 3
    DIFY = 4


class AgentModel(Base):
@@ -30,7 +31,7 @@


class CanvasModel(Base):
    __tablename__ = 'user_canvas'
    __tablename__ = 'canvas'

    id = Column(String(32), primary_key=True)      # id
    create_date = Column(DateTime, default=datetime.now)
@@ -51,9 +52,7 @@
    def to_json(self):
        return {
            'id': self.id,
            'create_time': self.create_time,
            'create_date': self.create_date,
            'update_time': self.update_time,
            'update_date': self.update_date,
            'avatar': self.avatar,
            'user_id': self.user_id,
@@ -130,5 +129,7 @@

    @staticmethod
    def is_type(record_id, t):
        record = UnifiedAgent.get_by_id(record_id)
        return record and record.type == t
        record = UnifiedAgentModel.get_by_id(record_id)
        return record and record.type == t