【langchain学习】GraphQAChain通过抽取三元组实现与图的对话

GraphQAChain通过抽取三元组实现与图的对话

from langchain.indexes import GraphIndexCreator
from langchain.chains import GraphQAChain
from config import llm

index_creator = GraphIndexCreator(llm=llm)

text = "温格高于1996年出生在丹麦沿海一个只有370人的小渔村。童年的温格高喜欢足球和手球，是英超利物浦队球迷。"
graph = index_creator.from_text(text)

print(graph.get_triples())
# [('Wenger', 'a small fishing village with only 370 people on the coast of Denmark in 1996', 'was born in'), ('Wenger', 'football and handball in childhood', 'liked'), ('Wenger', 'the Premier League team Liverpool', 'is a fan of')]

chain = GraphQAChain.from_llm(llm=llm, graph=graph)
print(chain.run("温格高是哪队的球迷"))
# Answer to the question: 温格高是利物浦队的球迷。