Skip to content

Commit 7eb3a5f

Browse files
committed
Send versioned get messages
1 parent faab172 commit 7eb3a5f

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

graphdatascience/query_runner/arrow_query_runner.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -289,8 +289,15 @@ def _run_arrow_property_get(self, graph_name: str, procedure_name: str, configur
289289
"procedure_name": procedure_name,
290290
"configuration": configuration,
291291
}
292-
ticket = flight.Ticket(json.dumps(payload).encode("utf-8"))
293292

293+
if self._arrow_version == ArrowVersion.V1:
294+
payload = {
295+
"name": "GET_MESSAGE",
296+
"version": ArrowVersion.V1.name(),
297+
"body": payload,
298+
}
299+
300+
ticket = flight.Ticket(json.dumps(payload).encode("utf-8"))
294301
get = self._flight_client.do_get(ticket)
295302
arrow_table = get.read_all()
296303

0 commit comments

Comments
 (0)