Add neo4j base label config (#2675)

2025-05-20 03:22:20 +02:00
parent 12a268da30
commit 1786d907f7
2 changed files with 56 additions and 35 deletions
--- a/mem0/graphs/configs.py
+++ b/mem0/graphs/configs.py
@@ -10,6 +10,7 @@ class Neo4jConfig(BaseModel):
    username: Optional[str] = Field(None, description="Username for the graph database")
    password: Optional[str] = Field(None, description="Password for the graph database")
    database: Optional[str] = Field(None, description="Database for the graph database")
+    base_label: Optional[bool] = Field(None, description="Whether to use base node label __Entity__ for all entities")

    @model_validator(mode="before")
    def check_host_port_or_path(cls, values):