Types

RawSearchResult

class neo4j_graphrag.types.RawSearchResult(*, records, metadata=None)

Represents the raw result returned from the retriever get_search_result method. It needs to be formatted further before being returned as a RetrieverResult.

Parameters:

• records (list[Record])

• metadata (dict[str, Any] | None)

records

A list of records from neo4j.

Type:

list[neo4j.Record]

metadata

Record-related metadata, such as score.

Type:

dict[str, Any] | None

RetrieverResult

class neo4j_graphrag.types.RetrieverResult(*, items, metadata=None)

Represents a result returned from a retriever.

Parameters:

• items (list[RetrieverResultItem])

• metadata (dict[str, Any] | None)

items

A list of retrieved items.

Type:

list[RetrieverResultItem]

metadata

Context-related metadata such as generated Cypher query in the Text2CypherRetriever.

Type:

dict[str, Any] | None

RetrieverResultItem

class neo4j_graphrag.types.RetrieverResultItem(*, content, metadata=None)

A single record returned from a retriever.

Parameters:

• content (Any)

• metadata (dict[str, Any] | None)

content

The context as will be provided to the LLM

Type:

str

metadata

Any metadata that can be included together with the text, related to that record (e.g. another node property)

Type:

Optional[dict]

LLMResponse

class neo4j_graphrag.llm.types.LLMResponse(*, content)

Parameters:

content (str)

RagResultModel

class neo4j_graphrag.generation.types.RagResultModel(*, answer, retriever_result=None)

Parameters:

• answer (str)

• retriever_result (RetrieverResult | None)

TextChunk

class neo4j_graphrag.experimental.components.types.TextChunk(*, text, index, metadata=None)

A chunk of text split from a document by a text splitter.

Parameters:

• text (str)

• index (int)

• metadata (dict[str, Any] | None)

text

The raw chunk text.

Type:

str

index

The position of this chunk in the original document.

Type:

int

metadata

Metadata associated with this chunk such as the id of the next chunk in the original document.

Type:

Optional[dict[str, Any]]

TextChunks

class neo4j_graphrag.experimental.components.types.TextChunks(*, chunks)

A collection of text chunks returned from a text splitter.

Parameters:

chunks (list[TextChunk])

chunks

A list of text chunks.

Type:

list[TextChunk]

Neo4jNode

class neo4j_graphrag.experimental.components.types.Neo4jNode(*, id, label, properties={}, embedding_properties=None)

Represents a Neo4j node.

Parameters:

• id (str)

• label (str)

• properties (dict[str, Any])

• embedding_properties (dict[str, list[float]] | None)

id

The element ID of the node.

Type:

str

label

The label of the node.

Type:

str

properties

A dictionary of properties attached to the node.

Type:

dict[str, Any]

embedding_properties

A list of embedding properties attached to the node.

Type:

Optional[dict[str, list[float]]]

Neo4jRelationship

class neo4j_graphrag.experimental.components.types.Neo4jRelationship(*, start_node_id, end_node_id, type, properties={}, embedding_properties=None)

Represents a Neo4j relationship.

Parameters:

• start_node_id (str)

• end_node_id (str)

• type (str)

• properties (dict[str, Any])

• embedding_properties (dict[str, list[float]] | None)

start_node_id

The ID of the start node.

Type:

str

end_node_id

The ID of the end node.

Type:

str

type

The relationship type.

Type:

str

properties

A dictionary of properties attached to the relationship.

Type:

dict[str, Any]

embedding_properties

A list of embedding properties attached to the relationship.

Type:

Optional[dict[str, list[float]]]

Neo4jGraph

class neo4j_graphrag.experimental.components.types.Neo4jGraph(*, nodes=[], relationships=[])

Represents a Neo4j graph.

Parameters:

• nodes (list[Neo4jNode])

• relationships (list[Neo4jRelationship])

nodes

A list of nodes in the graph.

Type:

list[Neo4jNode]

relationships

A list of relationships in the graph.

Type:

list[Neo4jRelationship]

KGWriterModel

class neo4j_graphrag.experimental.components.kg_writer.KGWriterModel(*, status, metadata=None)

Data model for the output of the Knowledge Graph writer.

Parameters:

• status (Literal['SUCCESS', 'FAILURE'])

• metadata (dict[str, Any] | None)

status

Whether the write operation was successful.

Type:

Literal[“SUCCESS”, “FAILURE”]

SchemaProperty

class neo4j_graphrag.experimental.components.schema.SchemaProperty(*, name, type, description='')

Represents a property on a node or relationship in the graph.

Parameters:

• name (str)

• type (Literal['BOOLEAN', 'DATE', 'DURATION', 'FLOAT', 'INTEGER', 'LIST', 'LOCAL_DATETIME', 'LOCAL_TIME', 'POINT', 'STRING', 'ZONED_DATETIME', 'ZONED_TIME'])

• description (str)

SchemaEntity

class neo4j_graphrag.experimental.components.schema.SchemaEntity(*, label, description='', properties=[])

Represents a possible node in the graph.

Parameters:

• label (str)

• description (str)

• properties (List[SchemaProperty])

SchemaRelation

class neo4j_graphrag.experimental.components.schema.SchemaRelation(*, label, description='', properties=[])

Represents a possible relationship between nodes in the graph.

Parameters:

• label (str)

• description (str)

• properties (List[SchemaProperty])

SchemaConfig

class neo4j_graphrag.experimental.components.schema.SchemaConfig(*, entities, relations, potential_schema)

Represents possible relationships between entities and relations in the graph.

Parameters:

• entities (Dict[str, Dict[str, Any]])

• relations (Dict[str, Dict[str, Any]] | None)

• potential_schema (List[Tuple[str, str, str]] | None)