Fixed import mistakes from merge conflicts

neo4j · Mar 18, 2024 · 8615b53 · 8615b53
1 parent afbfd14
commit 8615b53
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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):