WITH clause fails to parse, despite being mentioned in DeepWiki docs

[prrao87]

Summary

Running a query that uses a WITH clause causes a parse error, even though the docs say it’s supported.

Example

import pyarrow as pa
from lance_graph import GraphConfig, CypherQuery

people = pa.table({"id": [1, 2]})
follows = pa.table({"src": [1], "dst": [2]})
cfg = (
    GraphConfig.builder()
    .with_node_label("Person", "id")
    .with_relationship("FOLLOWS", "src", "dst")
    .build()
)
datasets = {"Person": people, "FOLLOWS": follows}

query = """
MATCH (a:Person)-[:FOLLOWS]->(b:Person)
WITH b, count(a.id) AS n
RETURN b.id, n
"""
CypherQuery(query).with_config(cfg).execute(datasets)

Expected

Query executes, returning counts per b.

Actual

• ValueError: Cypher parse error ... input: WITH b, count(...)

Proposal

Rewriting without WITH works, but loses some expressiveness. But WITH does help a lot of Cypher beginners write queries as they're learning, so it's a useful add.

Environment

The following environment was used to test this:

lance-graph 0.4.0
Python 3.13
macOS Tahoe 26.2

[prrao87]

Related to #86, hopefully can test this on the real datasets once a new release of lance-graph is made.

[aheev]

I'm getting the following error when running in local

>>> CypherQuery(query).with_config(cfg).explain(datasets)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: Query planning error: Failed to build aggregate: Schema error: No field named b. Valid fields are a__id, follows_1__src, follows_1__dst, b__id, a__id, follows_1__src, follows_1__dst, b__id.

[aheev]

@prrao87 Got it. You can't alias whole nodes or relationships. For ex below query works

MATCH (a:Person)-[:FOLLOWS]->(b:Person)
WITH a.id as c
RETURN c

[aheev]

raised new issue: #109

[prrao87]

Ah yes, but this is the pattern I need to use. And combining WITH with DISTINCT doesn't seem to work.

MATCH (p:Person)<-[:commentHasCreator]-(c:Comment)<-[:likeComment]-(p2:Person)
WHERE p.firstName = "Rafael" AND p.lastName = "Alonso"
WITH DISTINCT p2.ID AS id
RETURN count(id) AS num_persons;

Gives:

Traceback (most recent call last):
  File "/Users/prrao/code/graph-benchmark-ldbc-snb/lance_graph/query.py", line 490, in <module>
    main(selected_queries)
    ~~~~^^^^^^^^^^^^^^^^^^
  File "/Users/prrao/code/graph-benchmark-ldbc-snb/lance_graph/query.py", line 483, in main
    func(cfg, datasets)
    ~~~~^^^^^^^^^^^^^^^
  File "/Users/prrao/code/graph-benchmark-ldbc-snb/lance_graph/query.py", line 252, in run_query13
    return _execute(cfg, datasets, 13, query)
  File "/Users/prrao/code/graph-benchmark-ldbc-snb/lance_graph/query.py", line 119, in _execute
    result = execute_query(query, cfg, datasets)
  File "/Users/prrao/code/graph-benchmark-ldbc-snb/lance_graph/query.py", line 107, in execute_query
    cypher = CypherQuery(query)
ValueError: Cypher parse error at position 0: Failed to parse Cypher query: Parsing Error: Error { input: "WITH DISTINCT p2.ID AS id\nRETURN count(id) AS num_persons;\n    ", code: Tag }

[aheev]

Ah yes, but this is the pattern I need to use. And combining WITH with DISTINCT doesn't seem to work.

MATCH (p:Person)<-[:commentHasCreator]-(c:Comment)<-[:likeComment]-(p2:Person)
WHERE p.firstName = "Rafael" AND p.lastName = "Alonso"
WITH DISTINCT p2.ID AS id
RETURN count(id) AS num_persons;
Gives:

Traceback (most recent call last):
  File "/Users/prrao/code/graph-benchmark-ldbc-snb/lance_graph/query.py", line 490, in <module>
    main(selected_queries)
    ~~~~^^^^^^^^^^^^^^^^^^
  File "/Users/prrao/code/graph-benchmark-ldbc-snb/lance_graph/query.py", line 483, in main
    func(cfg, datasets)
    ~~~~^^^^^^^^^^^^^^^
  File "/Users/prrao/code/graph-benchmark-ldbc-snb/lance_graph/query.py", line 252, in run_query13
    return _execute(cfg, datasets, 13, query)
  File "/Users/prrao/code/graph-benchmark-ldbc-snb/lance_graph/query.py", line 119, in _execute
    result = execute_query(query, cfg, datasets)
  File "/Users/prrao/code/graph-benchmark-ldbc-snb/lance_graph/query.py", line 107, in execute_query
    cypher = CypherQuery(query)
ValueError: Cypher parse error at position 0: Failed to parse Cypher query: Parsing Error: Error { input: "WITH DISTINCT p2.ID AS id\nRETURN count(id) AS num_persons;\n    ", code: Tag }

It's already listed here

[prrao87]

Fixed in 0.5.0 release.