zhaoqingang
2024-12-06 9e51ec752c93d850db15cd61a7a3463d8fe94344
app/models/agent_model.py
@@ -9,6 +9,7 @@
    RAGFLOW = 1
    BISHENG = 2
    BASIC = 3
    DIFY = 4
class AgentModel(Base):
@@ -129,4 +130,6 @@
    @staticmethod
    def is_type(record_id, t):
        record = UnifiedAgentModel.get_by_id(record_id)
        return record and record.type == t
        return record and record.type == t