Resolve routing mode only after driver > 5.5 check

its not available for 4.4

Author: FlorentinD

graphdatascience/query_runner/neo4j_query_runner.py

@@ -206,7 +206,7 @@ def run_retryable_cypher(
         params: Optional[dict[str, Any]] = None,
         database: Optional[str] = None,
         custom_error: bool = True,
-        routing: Optional[neo4j.RoutingControl] = None,
+        mode: Optional[QueryMode] = None,
         connectivity_retry_config: Optional[ConnectivityRetriesConfig] = None,
     ) -> DataFrame:
         if not database:
@@ -215,8 +215,10 @@ def run_retryable_cypher(
         if self._NEO4J_DRIVER_VERSION < SemanticVersion(5, 5, 0):
             return self.run_cypher(query, params, database, custom_error, connectivity_retry_config)
 
-        if not routing:
+        if not mode:
             routing = neo4j.RoutingControl.READ
+        else:
+            routing = mode.neo4j_routing()
 
         try:
             return self._driver.execute_query(
@@ -240,9 +242,7 @@ def call_function(self, endpoint: str, params: Optional[CallParameters] = None,
         query = f"RETURN {endpoint}({params.placeholder_str()})"
 
         # we can use retryable cypher as we expect all gds functions to be idempotent
-        return self.run_retryable_cypher(
-            query, params, custom_error=custom_error, routing=neo4j.RoutingControl.READ
-        ).squeeze()
+        return self.run_retryable_cypher(query, params, custom_error=custom_error, mode=QueryMode.READ).squeeze()
 
     def call_procedure(
         self,
@@ -263,7 +263,7 @@ def call_procedure(
 
         def run_cypher_query() -> DataFrame:
             if retryable:
-                return self.run_retryable_cypher(query, params, database, custom_error, routing=mode.neo4j_mode())
+                return self.run_retryable_cypher(query, params, database, custom_error, mode=mode)
             else:
                 return self.run_cypher(query, params, database, custom_error)
 

graphdatascience/query_runner/query_mode.py

@@ -7,7 +7,7 @@ class QueryMode(str, Enum):
     READ = "read"
     WRITE = "write"
 
-    def neo4j_mode(self) -> neo4j.RoutingControl:
+    def neo4j_routing(self) -> neo4j.RoutingControl:
         if self == QueryMode.READ:
             return neo4j.RoutingControl.READ
         elif self == QueryMode.WRITE: