Method get_statistics

rag_modules/graph_indexing.py:359–374

获取键值对存储统计信息

(self)

    
    def get_statistics(self) -> Dict[str, Any]:
        """获取键值对存储统计信息"""
        return {
            "total_entities": len(self.entity_kv_store),
            "total_relations": len(self.relation_kv_store),
            "total_entity_keys": sum(len(kv.index_keys) for kv in self.entity_kv_store.values()),
            "total_relation_keys": sum(len(kv.index_keys) for kv in self.relation_kv_store.values()),
            "entity_types": {
                "Disease": len([kv for kv in self.entity_kv_store.values() if kv.entity_type == "Disease"]),
                "Symptom": len([kv for kv in self.entity_kv_store.values() if kv.entity_type == "Symptom"]),
                "Treatment": len([kv for kv in self.entity_kv_store.values() if kv.entity_type == "Treatment"]),
                "Medication": len([kv for kv in self.entity_kv_store.values() if kv.entity_type == "Medication"]),
                "RiskFactor": len([kv for kv in self.entity_kv_store.values() if kv.entity_type == "RiskFactor"]),
                "CareTip": len([kv for kv in self.entity_kv_store.values() if kv.entity_type == "CareTip"])
            }
        }