Použití extrahovaných entit
Sekce Dramatis Personæ popisuje postavy v divadelní hře a lze ji považovat za kompletní zdroj pravdy. Obsahuje také bohaté informace o osobních vztazích mezi postavami.
Dramatis Personæ
ESCALUS, Prince of Verona.
MERCUTIO, kinsman to the Prince, and friend to Romeo.
PARIS, a young Nobleman, kinsman to the Prince.
Page to Paris.
Doplň kód tak, aby z textu personae extrahoval postavy pomocí character_chain, a pak na základě seznamu postav sestav pomocí relationship_chain seznam vztahů z textu.
Funkce visualize_families() zobrazí výsledné vztahy jako graf.
Toto cvičení je součástí kurzu
Graph RAG s LangChain a Neo4j
Pokyny k cvičení
- Zavolej
character_chaina předej mupersonae, aby extrahoval entity postav. - Zavolej
relationship_chainse jménem postavy,personaea extrahovanými postavami ve formátu JSON (character_json), aby extrahoval vztahy mezi třemi uvedenými postavami a postavami vpersonae.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
# Extract characters from the personae variable
output = character_chain.____({"personae": ____})
character_json = dumps([dict(character) for character in output.characters])
# Extract the relationships
for character_name in ["Romeo", "Juliet", "Friar Lawrence"]:
relationship_output = relationship_chain.invoke({
"character_name": character_name,
"personae": ____,
"characters": ____
})
for relationship in relationship_output.relationships:
all_relationships.append(relationship)
visualize_families(all_relationships, output)