一篇文章如何入库的

一篇文章如何入库的
  1. 从文章里抽实体(人、公司、地点、产品、时间…)
  2. 抽实体之间的关系(A 工作于 B、C 位于 D、E 认识 F…)
  3. 生成节点 + 关系(Node + Relationship)
  4. 批量写入 Neo4j(CREATE / MERGE)

最终在 Neo4j 里形成一张知识图谱,RAG 检索时就可以沿着关系查上下文。

我们用一段超短新闻做演示:

张三在2023年加入了字节跳动,担任算法工程师。 李四是张三的同事,他们都在北京工作。

以上的超短新闻,要想配合入图数据库,首先是需要抽象出以下信息。

实体(Nodes)

  • 张三 (Person)
  • 李四 (Person)
  • 字节跳动 (Company)
  • 北京 (Location)

关系(Relationships)

  • 张三 ——WORKS_AT——> 字节跳动
  • 张三 ——COLLEAGUE_OF——> 李四
  • 张三 ——WORKS_IN——> 北京
  • 李四 ——WORKS_IN——> 北京

属性

  • 张三:age=None, job=算法工程师, join_year=2023

实体,关系,属性信息都确认了之后,就可以通过以下语句入库了。

// 创建节点 MERGE (p1:Person {name:"张三"}) SET p1.job="算法工程师", p1.join_year=2023 MERGE (p2:Person {name:"李四"}) MERGE (c:Company {name:"字节跳动"}) MERGE (l:Location {name:"北京"}) // 创建关系 MERGE (p1)-[:WORKS_AT]->(c) MERGE (p1)-[:COLLEAGUE_OF]->(p2) MERGE (p1)-[:WORKS_IN]->(l) MERGE (p2)-[:WORKS_IN]->(l)

为什么用 MERGE 不用 CREATE?

  • CREATE:重复执行会重复创建节点
  • MERGE:不存在则创建,存在则更新 →Graph RAG 标准写法

那么如何把文章里的信息是如何抽象出的实体和关系呢,直接说过程:文章不是人工转的,是 LLM 自动转的!

下面是真实工程化流程

首先,给大模型的提示词

你是知识图谱抽取专家。请从文本中抽取: 1. 实体(类型:Person, Company, Location, Organization) 2. 关系(只能用:WORKS_AT, COLLEAGUE_OF, WORKS_IN, FOUNDER_OF) 输出格式严格JSON: { "nodes": [{"label":"Person", "name":"张三", "attributes":{"job":"算法工程师"}}], "relations": [{"from":"张三", "to":"字节跳动", "type":"WORKS_AT"}] } 文本: 张三在2023年加入了字节跳动,担任算法工程师。李四是张三的同事,他们都在北京工作。

以上提示词可以放在任何工具里测试,包括龙虾,豆包等。

实体和关系总结出来了,可以看到结果如下:

{ "nodes": [ {"label":"Person","name":"张三","attributes":{"job":"算法工程师","join_year":2023}}, {"label":"Person","name":"李四","attributes":{}}, {"label":"Company","name":"字节跳动","attributes":{}}, {"label":"Location","name":"北京","attributes":{}} ], "relations": [ {"from":"张三","to":"字节跳动","type":"WORKS_AT"}, {"from":"张三","to":"李四","type":"COLLEAGUE_OF"}, {"from":"张三","to":"北京","type":"WORKS_IN"}, {"from":"李四","to":"北京","type":"WORKS_IN"} ] }

注:如果模型是在线API,那么基本上都能按照要求做出指定的输出,如果是本地部署的模型,参数量比较低的话,输出可能会走样。

最后,通过python代码自动把 JSON 转成 Cypher 写入 Neo4j

from neo4j import GraphDatabase # 连接 Neo4j uri = "bolt://localhost:7687" user = "neo4j" password = "your-password" driver = GraphDatabase.driver(uri, auth=(user, password)) # 从LLM拿到的知识图谱数据 data = { ...上面的JSON... } # 写入节点 for node in data["nodes"]: if node["attributes"]: attrs = ", ".join([f"{k}: '{v}'" for k, v in node["attributes"].items()]) cypher = f"MERGE (n:{node['label']} {{name:'{node['name']}', {attrs}}})" else: cypher = f"MERGE (n:{node['label']} {{name:'{node['name']}'}})" driver.session().run(cypher) # 写入关系 for rel in data["relations"]: cypher = f""" MATCH (a {{name:'{rel['from']}'}}), (b {{name:'{rel['to']}'}}) MERGE (a)-[:{rel['type']}]->(b) """ driver.session().run(cypher)

运行完,一篇文章就变成知识图谱了!


最终效果(Neo4j 里看到的图)
你会看到:

  • 张三、李四、字节跳动、北京
  • 它们之间用箭头连起来
  • Graph RAG 检索时就能沿着关系推理

最后再说一种更简单粗暴的方法,跳过大模型生成JSON和python的JSON遍历,直接让大模型生成语句。