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