注册问题优化

zhaoqingang

2024-12-31 c8da96955f57e439231f8a748f0bfe9a98283c1a

 app/models/agent_model.py

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


class AgentModel(Base):
@@ -128,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