Unable to run neo4j create constraint cypher query from Databricks using pyspark connector

Question

I am using databricks notebook and neo4j spark connector to run cypher query to create constraints. While executing its given an error.. I tried multiple ways to change the databricks runtime version and spark connector version, but no luck.

Error msg: IllegalArgumentException: Please provide a valid WRITE query.

I tried to test using multiple version of databricks runtime along with spark connector and did not get any luck. Here is the code.

from pyspark.sql import SparkSession

# Initialize Spark session
spark = SparkSession.builder \
 .appName("Neo4j Integration") \
 .getOrCreate()

# Neo4j connection config
neo4j_config = {
 "url": "bolt://localhost:7687",
 "authentication.type": "basic",
 "authentication.basic.username": "<username>",
 "authentication.basic.password": "password"
}

# Sample data
data = [
{"id": "1", "name": "Alice"}
]
 
# Create DataFrame
df = spark.createDataFrame(data)
 
 
Constraints_Query = """
CREATE CONSTRAINT unique_person IF NOT EXISTS FOR (p:Person) REQUIRE p.name IS UNIQUE
"""
# Execute Cypher query to create constraint
df.write.format("org.neo4j.spark.DataSource") \
    .options(**neo4j_config) \
    .option("query", Constraints_Query) \
    .option("database", "neo4j")\
    .mode('Overwrite')\
    .save()

fbiville · Accepted Answer · 2025-07-03 13:35:36Z

1

If you want to initialize constraints with the Spark Connector for Neo4j, you have two choices:

configure the right schema optimization strategy, constraints will be inferred from your dataframe schema
use the script option (where the value would be CREATE CONSTRAINT unique_person IF NOT EXISTS FOR (p:Person) REQUIRE p.name IS UNIQUE)

In any case, your query call must be for data imports, not schema initialization.
The official documentation hopefully helps in that regards.

answered Jul 3 at 13:35

fbiville

9,0657 gold badges58 silver badges83 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Arbind Chandra Jul 3 at 15:12

Thank you @fbiville for your quick help. I am able to re-write with schema optimization and it worked.

fbiville Jul 4 at 7:49

Glad to hear that! Can you please mark the answer as accepted if it worked for you?

Collectives™ on Stack Overflow

Unable to run neo4j create constraint cypher query from Databricks using pyspark connector

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related