diff --git a/src/neo4j_genai/types.py b/src/neo4j_genai/types.py
index 3182a49c3..4fa90594b 100644
--- a/src/neo4j_genai/types.py
+++ b/src/neo4j_genai/types.py
@@ -2,9 +2,8 @@
 from pydantic import BaseModel, PositiveInt, model_validator
 
 
-class Neo4jRecord(BaseModel):
-    node: Any
-    score: float
+Neo4jRecord = dict[str, Any]
+"""Type alias for data items returned from Neo4j queries"""
 
 
 class EmbeddingVector(BaseModel):