AI画组织架构图：同名员工按ID连接，空岗位不补人

{
  "positions": [
    {
      "id": "P0",
      "name": "负责人",
      "employee_id": "E0",
      "employee": "林青",
      "manager": null
    },
    {
      "id": "P1",
      "name": "研发",
      "employee_id": "E1",
      "employee": "李明",
      "manager": "P0"
    },
    {
      "id": "P2",
      "name": "运营",
      "employee_id": "E2",
      "employee": "李明",
      "manager": "P0"
    },
    {
      "id": "P3",
      "name": "顾问兼职",
      "employee_id": "E1",
      "employee": "李明",
      "manager": "P1"
    },
    {
      "id": "P4",
      "name": "设计",
      "employee_id": null,
      "employee": null,
      "manager": "P0"
    }
  ]
}

python3 -m venv .venv
source .venv/bin/activate
python -m pip install -r requirements.txt
python run.py

{
  "model": "gemini-2.5-flash-lite",
  "messages": [
    {
      "role": "system",
      "content": "严格只输出单个JSON对象，不要Markdown代码围栏，不要解释。只输出JSON：nodes数组逐项保留id、name、employee_id、employee；edges数组每项from、to，方向为manager到岗位id。同名不同员工ID不能合并，同一人兼职保留两个岗位，空岗employee仍null。不要直接生成DOT。"
    },
    {
      "role": "user",
      "content": "任务约束（只输出JSON对象，不用代码围栏）：\n只输出JSON：nodes数组逐项保留id、name、employee_id、employee；edges数组每项from、to，方向为manager到岗位id。同名不同员工ID不能合并，同一人兼职保留两个岗位，空岗employee仍null。不要直接生成DOT。\n待处理合成材料：\n{\"positions\": [{\"id\": \"P0\", \"name\": \"负责人\", \"employee_id\": \"E0\", \"employee\": \"林青\", \"manager\": null}, {\"id\": \"P1\", \"name\": \"研发\", \"employee_id\": \"E1\", \"employee\": \"李明\", \"manager\": \"P0\"}, {\"id\": \"P2\", \"name\": \"运营\", \"employee_id\": \"E2\", \"employee\": \"李明\", \"manager\": \"P0\"}, {\"id\": \"P3\", \"name\": \"顾问兼职\", \"employee_id\": \"E1\", \"employee\": \"李明\", \"manager\": \"P1\"}, {\"id\": \"P4\", \"name\": \"设计\", \"employee_id\": null, \"employee\": null, \"manager\": \"P0\"}]}"
    }
  ],
  "temperature": 0,
  "max_tokens": 1536,
  "provider": {
    "sort": "success_rate"
  },
  "response_format": {
    "type": "json_object"
  }
}

nodes = output.get('nodes', [])
edges = output.get('edges', [])
source = {r['id']: r for r in original['positions']}
by_id = {r.get('id'): r for r in nodes}
expected_edges = {(r['manager'], r['id']) for r in source.values() if r['manager']}
got_edges = {(r.get('from'), r.get('to')) for r in edges}
check('岗位ID不合并，同名员工各有身份', set(by_id) == set(source) and all((by_id[k].get('employee_id') == v['employee_id'] and by_id[k].get('employee') == v['employee'] for k, v in source.items())))
check('空岗未补造员工', by_id.get('P4', {}).get('employee') is None and by_id.get('P4', {}).get('employee_id') is None)
check('经理到岗位关系完整且无新边', got_edges == expected_edges)
lines = ['digraph Organization {', 'graph [rankdir=TB, bgcolor="#f8f4ed"];', 'node [shape=box, style=rounded, fontname="sans-serif"];']
for key in sorted(by_id):
    r = by_id[key]
    person = r.get('employee') or '待聘'
    lab = f"{key} · {r.get('name', '')}\n{person} / {r.get('employee_id') or 'VACANT'}"
    lines.append(json.dumps(key) + ' [label=' + json.dumps(lab, ensure_ascii=False) + '];')
for a, z in sorted(got_edges):
    lines.append(json.dumps(a) + ' -> ' + json.dumps(z) + ';')
lines.append('}')
artifact('organization.dot').write_text('\n'.join(lines))
script = "import fs from 'node:fs';import path from 'node:path';import {pathToFileURL} from 'node:url';\nconst modules=process.env.BATCH100O_NODE_MODULES || path.resolve('web-deps/node_modules');\nconst {Graphviz}=await import(pathToFileURL(path.join(modules,'@hpcc-js/wasm-graphviz/dist/index.js')).href);\nconst engine=await Graphviz.load();const dot=fs.readFileSync(process.argv[2],'utf8');fs.writeFileSync(process.argv[3],engine.layout(dot,'svg','dot'));\n"
artifact('render-graph.mjs').write_text(script)
node = shutil.which('node')
if not node:
    raise RuntimeError('Node.js is required for actual Graphviz WASM layout')
subprocess.run([node, str((folder / 'render-graph.mjs').resolve()), str((folder / 'organization.dot').resolve()), str((folder / 'organization.svg').resolve())], check=True, capture_output=True, text=True, timeout=40)
artifacts.append('organization.svg')
tree = ET.parse(folder / 'organization.svg')
ns = {'s': 'http://www.w3.org/2000/svg'}
graph_nodes = tree.findall('.//s:g[@class="node"]', ns)
check('官方Graphviz引擎真实布局5个岗位', len(graph_nodes) == 5)
result = {'nodes': nodes, 'edges': edges, 'renderEngine': '@hpcc-js/wasm-graphviz 1.30.0; real Graphviz WASM', 'partTimeEmployee': 'E1 appears in P1 and P3'}
dump(artifact('organization-map.json'), result)

{
  "nodes": [
    {
      "id": "P0",
      "name": "负责人",
      "employee_id": "E0",
      "employee": "林青"
    },
    {
      "id": "P1",
      "name": "研发",
      "employee_id": "E1",
      "employee": "李明"
    },
    {
      "id": "P2",
      "name": "运营",
      "employee_id": "E2",
      "employee": "李明"
    },
    {
      "id": "P3",
      "name": "顾问兼职",
      "employee_id": "E1",
      "employee": "李明"
    },
    {
      "id": "P4",
      "name": "设计",
      "employee_id": null,
      "employee": null
    }
  ],
  "edges": [
    {
      "from": "P0",
      "to": "P1"
    },
    {
      "from": "P0",
      "to": "P2"
    },
    {
      "from": "P1",
      "to": "P3"
    },
    {
      "from": "P0",
      "to": "P4"
    }
  ],
  "renderEngine": "@hpcc-js/wasm-graphviz 1.30.0; real Graphviz WASM",
  "partTimeEmployee": "E1 appears in P1 and P3"
}