┌─────────────────────────────────────────────────┐
│               Claude Desktop / Codex            │
│                      │                           │
│              MCP Protocol (stdio JSON-RPC)       │
│                      ▼                           │
│  ┌─────────────────────────────────────────┐    │
│  │          MCP-KG-Server                   │    │
│  │  ┌─────────────────────────────────┐    │    │
│  │  │      Tool Registry (6 tools)    │    │    │
│  │  │  search_drug / search_tcm       │    │    │
│  │  │  check_compatibility            │    │    │
│  │  │  query_by_symptom (RAG)         │    │    │
│  │  │  get_similar_drugs / get_kg_stats│   │    │
│  │  └──────────────┬──────────────────┘    │    │
│  │                 ▼                        │    │
│  │  ┌─────────────────────────────────┐    │    │
│  │  │    Knowledge Graph (NetworkX)   │    │    │
│  │  │    2,556 nodes / 5,306 edges    │    │    │
│  │  │    289 drugs + 204 TCM herbs    │    │    │
│  │  └──────────────┬──────────────────┘    │    │
│  │                 ▼                        │    │
│  │  ┌─────────────────────────────────┐    │    │
│  │  │    RAG Retriever                │    │    │
│  │  │    Embedding + Keyword Fallback  │    │    │
│  │  └─────────────────────────────────┘    │    │
│  └─────────────────────────────────────────┘    │
└─────────────────────────────────────────────────┘

git clone https://github.com/chwsvveuiygbfiw/mcp-kg-server
cd mcp-kg-server
pip install -e .

streamlit run src/playground.py
# 浏览器打开 http://localhost:8501
# 可测试所有 6 个 MCP 工具

{
  "mcpServers": {
    "medical-kg": {
      "command": "python",
      "args": ["-m", "mcp_kg_server.server"],
      "env": {
        "KG_CONFIG": "/absolute/path/to/medical_kg.yaml"
      }
    }
  }
}

# .codex.yaml
mcp_servers:
  medical-kg:
    command: python
    args: ["-m", "mcp_kg_server.server"]

// Request
{"tool": "search_drug", "arguments": {"query": "布洛芬"}}

// Response
{
  "name": "布洛芬",
  "category": "NSAID",
  "indications": ["发热", "头痛", "牙痛", "关节痛"],
  "contraindications": ["胃溃疡", "孕妇晚期"],
  "side_effects": ["胃痛", "恶心"],
  "dosage": "200-400mg tid",
  "match_type": "exact"
}

// Request
{"tool": "search_tcm", "arguments": {"name": "人参"}}

// Response
{
  "name": "人参",
  "nature": "微温", "taste": ["甘", "微苦"],
  "meridians": ["脾", "肺", "心", "肾"],
  "effects": ["大补元气", "复脉固脱", "补脾益肺"],
  "synergies": [
    {"partner": "黄芪", "effect": "相须为用，补气固表"},
    {"partner": "白术", "effect": "补气健脾"}
  ],
  "incompatibilities": [
    {"partner": "藜芦", "reason": "十八反: 诸参辛芍叛藜芦"},
    {"partner": "五灵脂", "reason": "十九畏: 人参最怕五灵脂"}
  ]
}

// Request
{"tool": "check_compatibility", "arguments": {"drug_a": "人参", "drug_b": "五灵脂"}}

// Response
{
  "compatible": false,
  "reason": "十九畏: 人参最怕五灵脂",
  "danger_level": "禁止联用"
}

// Request
{"tool": "query_by_symptom", "arguments": {"symptom": "牙疼", "top_k": 5}}

// Response
{
  "matched_symptoms": [
    {"symptom": "牙痛", "score": 0.85},
    {"symptom": "口腔溃疡疼痛", "score": 0.52}
  ],
  "inferred_diseases": ["牙髓炎", "牙周炎", "龋齿"],
  "recommended_drugs": [
    {"name": "甲硝唑", "category": "硝基咪唑类"},
    {"name": "布洛芬", "category": "NSAID"}
  ],
  "reasoning_chain": "牙疼→牙痛+口腔溃疡疼痛→牙髓炎+牙周炎+龋齿→甲硝唑+布洛芬"
}

// Request
{"tool": "get_similar_drugs", "arguments": {"name": "布洛芬"}}

// Response
{
  "drug": "布洛芬",
  "alternatives": [
    {"name": "对乙酰氨基酚", "category": "解热镇痛药", "relation": "同类药"},
    {"name": "双氯芬酸", "category": "NSAID", "relation": "同类药/替代药"},
    {"name": "塞来昔布", "category": "COX-2选择性NSAID", "relation": "同类药/替代药"}
  ]
}

// Request
{"tool": "get_kg_stats", "arguments": {}}

// Response
{
  "num_nodes": 2556,
  "num_edges": 5306,
  "num_drugs": 289,
  "num_diseases": 38,
  "num_tcm": 204
}

streamlit run src/playground.py

# Clone
git clone https://github.com/chwsvveuiygbfiw/mcp-kg-server
cd mcp-kg-server

# Install dev dependencies
pip install -e ".[dev]"

# Run tests
python -m pytest tests/ -v

# Run playground
streamlit run src/playground.py