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
