Doménové grafy

Graph RAG s LangChain a Neo4j

Adam Cowley

Manager, Developer Education at Neo4j

Doménové grafy

diagram znalostního grafu zobrazující lexikální graf vlevo s knihou, kapitolami a stránkami a vpravo osoby a společnosti z knihy spolu s odkazy z jiných zdrojů včetně Wikipedie

Od lexikálních grafů k doménovým

Text písně Don't Stop Believing od skupiny Journey

Od lexikálních grafů k doménovým

Text písně Don't Stop Believing od skupiny Journey se zvýrazněním „Small Town Boy" jako osoby

Od lexikálních grafů k doménovým

Text písně Don't Stop Believing od skupiny Journey se zvýrazněním „South Detroit" jako místa

Od lexikálních grafů k doménovým

Text písně Don't Stop Believing od skupiny Journey se zvýrazněním „Born" jako vztahu mezi uzlem osoby „Small Town Boy" a uzlem místa „South Detroit"

Od lexikálních grafů k doménovým

Text písně Don't Stop Believing od skupiny Journey se zvýrazněním „Raised" jako vztahu mezi uzlem osoby „Small Town Boy" a uzlem místa „South Detroit"

Ukázkový doménový graf

Ukázkový doménový graf: osoba pracuje ve firmě na dané adrese, vlastní kreditní kartu, přes kterou provedla objednávku u firmy ve stejné lokalitě

Strukturované výstupy

from pydantic import BaseModel, Field

from typing import Optional


class Character(BaseModel):
    """A character from the play."""

    id: str = Field(description="A unique identifier for the character in slug-case",

                    examples=["montague-romeo", "tybalt"])

    name: str = Field(description="The full name of the character", 
                    examples=["Romeo", "Tybalt"])

    family: Optional[str] = Field(description="The name of the family they belong to", 
                    examples=["Montague", "Capulet"])


class CharacterOutput(BaseModel):
    characters: list[Character] = Field(description="A list of characters")

Použití strukturovaných výstupů

from langchain.chat_models import init_chat_model
llm = init_chat_model("gpt-4o-mini", model_provider="openai", api_key="...")


# With a structured output
character_model = llm.with_structured_output(CharacterOutput)


# Use within a chain
chain = extraction_prompt | character_model


# Handle the response
output = chain.invoke(...)

CharacterOutput(characters=[Character, Character, Character...])

Ukládání entit

nodes = [

  Node(

      type="Character",

      id=character.id,

      properties={"name": character.name, "family": character.family}

  )

  for character in output.characters

]



# Build a GraphDocument
graph_document = GraphDocument(nodes=nodes, relationships=[])


# Save to the graph
graph.add_graph_documents([graph_document])

Pojďme cvičit!

Graph RAG s LangChain a Neo4j