app/models/agent_model.py
@@ -9,6 +9,8 @@ RAGFLOW = 1 BISHENG = 2 BASIC = 3 DIFY = 4 OTHER = 5 class AgentModel(Base): @@ -128,5 +130,5 @@ @staticmethod def is_type(record_id, t): record = UnifiedAgent.get_by_id(record_id) record = UnifiedAgentModel.get_by_id(record_id) return record and record.type == t