知识组织

import json
from typing import Dict, List

class SimpleKnowledgeBase:
    def __init__(self):
        self.entities = {}
        self.relations = []
    
    def add_entity(self, entity_id: str, attributes: Dict):
        self.entities[entity_id] = attributes
    
    def add_relation(self, source: str, target: str, relation_type: str):
        self.relations.append({
            'source': source,
            'target': target,
            'type': relation_type
        })
    
    def query_entity(self, entity_id: str) -> Dict:
        return self.entities.get(entity_id, {})
    
    def query_relations(self, entity_id: str) -> List[Dict]:
        return [r for r in self.relations if r['source'] == entity_id or r['target'] == entity_id]

# 使用示例
kb = SimpleKnowledgeBase()
kb.add_entity('AI', {'type': 'concept', 'description': '人工智能'})
kb.add_entity('ML', {'type': 'concept', 'description': '机器学习'})
kb.add_relation('AI', 'ML', 'includes')

print(kb.query_entity('AI'))
print(kb.query_relations('AI'))

Python实现简单知识抽取与存储

{
  "entity": {
    "id": "AI",
    "type": "concept",
    "attributes": {
      "description": "人工智能",
      "related_fields": [
        "机器学习",
        "深度学习",
        "自然语言处理"
      ]
    }
  },
  "relations": [
    {
      "source": "AI",
      "target": "ML",
      "type": "includes",
      "confidence": 0.95
    }
  ],
  "metadata": {
    "query_time": "2024-01-15T10:30:00Z",
    "source": "knowledge_base",
    "version": "1.0"
  }
}

典型知识查询返回格式