使用大语言模型创建 Graph 数据

Neo4j 是开源的 Graph 数据库，Graph 数据通过三元组举行表现，两个顶点+一条边，从语意上可以明确为：主语、谓语和宾语。GraphDB 能够通过图来表达复杂的结构，非常恰当存储知识型数据，本文将通过大语言实现图数据库的创建。
首先，需要安装 LangChain 对应的依靠。

1. langchain==0.3.7
2. python-dotenv==1.0.1
3. langchain-google-genai==2.0.4
4. langserve==0.3.0
5. langchain-community==0.3.5
6. dashscope==1.20.12
7. langgraph==0.2.45
8. tavily-python==0.5.0
9. langchain-experimental==0.3.3

复制代码

初始化模型

本文我们使用 Qwen-Turbo。

1. from langchain_community.chat_models import ChatTongyi
2. llm = ChatTongyi(model="qwen-turbo")

复制代码

使用 LLMGraphTransformer 创建

创建的图关系为英文，这是因为 Langchain 提示词是英文，可以自行修改。

1. ### 创建 Graph
2. from langchain_core.documents import Document
3. from langchain_experimental.graph_transformers import LLMGraphTransformer
4. from langchain_core.output_parsers import StrOutputParser
5. from IPython.display import Markdown, display
6. parser = StrOutputParser()
8. llm_transformer = LLMGraphTransformer(llm=llm)
10. content = """
11. 1 、项目编号：遂政采 [2024]G 049 号
12. 2 、项目名称：无人机设备采购项目
13. 3 、项目预算： 1550967.00元 4 、最高限价： 1550967.00元
15. 1 、本项目投标截止时间和开标时间为 2024 年 9 月 11 日 10 点 00 分 （北京
16. 时间）。投标人必须在投标截止时间前将电子投标文件上传至江西省公共资源
17. 交易网，逾期为无效投标。
18. 2 、开标地点：江西省公共资源交易网不见面开标大厅。
19. """
20. with open("out.md", "r", encoding="utf-8") as file:
21.     markdown_content = file.read()
23. chain = llm | parser
24. # display(Markdown(chain.invoke("转换为 Graph 三元组， 请用中文回答，并只返回 Cypher QL， 。 {} " + content)))
26. documents = [Document(page_content=content)]
27. graph_documents = llm_transformer.convert_to_graph_documents(documents)
28. print(f"Nodes:{graph_documents[0].nodes}")
29. print(f"Relationships:{graph_documents[0].relationships}")
30. graph.add_graph_documents(graph_documents, include_source=False)

复制代码

使用 Cypher 创建

Neo4j 提供了使用 Cypher 举行数据创建、查询等，类似于关系数据库中的 SQL。用模型生成 Cypher 语句。

1. ### 创建 Graph
2. from langchain_core.documents import Document
3. from langchain_experimental.graph_transformers import LLMGraphTransformer
4. from langchain_core.output_parsers import StrOutputParser
5. from IPython.display import Markdown, display
6. parser = StrOutputParser()
8. llm_transformer = LLMGraphTransformer(llm=llm)
10. content = """
11. 1 、项目编号：遂政采 [2024]G 049 号
12. 2 、项目名称：无人机设备采购项目
13. 3 、项目预算： 1550967.00元 4 、最高限价： 1550967.00元
15. 1 、本项目投标截止时间和开标时间为 2024 年 9 月 11 日 10 点 00 分 （北京
16. 时间）。投标人必须在投标截止时间前将电子投标文件上传至江西省公共资源
17. 交易网，逾期为无效投标。
18. 2 、开标地点：江西省公共资源交易网不见面开标大厅。
19. """
20. with open("out.md", "r", encoding="utf-8") as file:
21.     markdown_content = file.read()
23. chain = llm | parser
24. display(Markdown(chain.invoke("转换为 Graph 三元组， 请用中文回答，并只返回 Cypher QL， 。 {} " + content)))

复制代码

总结

本文介绍了怎样使用大语言模型创建图数据，同样的方式也可以创建其他类型的数据源，只需要对提示词举行修改即可。

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。