Mark some more endpoints as retryable

they are idempotent (list/stream)

Author: FlorentinD

graphdatascience/graph/graph_entity_ops_runner.py

@@ -343,7 +343,7 @@ def drop(
     def stream(self, G: Graph, relationship_types: list[str] = ["*"], **config: Any) -> TopologyDataFrame:
         self._namespace += ".stream"
         params = CallParameters(graph_name=G.name(), relationship_types=relationship_types, config=config)
-        result = self._query_runner.call_procedure(endpoint=self._namespace, params=params)
+        result = self._query_runner.call_procedure(endpoint=self._namespace, params=params, retryable=True)
 
         return TopologyDataFrame(result)
 
@@ -361,7 +361,9 @@ def stream(self, G: Graph, relationship_types: list[str] = ["*"], **config: Any)
         self._namespace += ".stream"
         params = CallParameters(graph_name=G.name(), relationship_types=relationship_types, config=config)
 
-        return TopologyDataFrame(self._query_runner.call_procedure(endpoint=self._namespace, params=params))
+        return TopologyDataFrame(
+            self._query_runner.call_procedure(endpoint=self._namespace, params=params, retryable=True)
+        )
 
     @property
     @compatible_with("toUndirected", min_inclusive=ServerVersion(2, 3, 0))
@@ -382,7 +384,7 @@ def stream(
         self._namespace += ".stream"
         params = CallParameters(graph_name=G.name(), graph_property=graph_property, config=config)
 
-        return self._query_runner.call_procedure(endpoint=self._namespace, params=params)
+        return self._query_runner.call_procedure(endpoint=self._namespace, params=params, retryable=True)
 
     @compatible_with("drop", min_inclusive=ServerVersion(2, 2, 0))
     @graph_type_check

graphdatascience/utils/direct_util_endpoints.py

@@ -48,8 +48,7 @@ def find_node_id(self, labels: list[str] = [], properties: dict[str, Any] = {})
         else:
             query = "MATCH (n) RETURN id(n) AS id"
 
-        # TODO use execute query
-        node_match = self._query_runner.run_cypher(query, custom_error=False)
+        node_match = self._query_runner.run_retryable_cypher(query, custom_error=False)
 
         if len(node_match) != 1:
             raise ValueError(f"Filter did not match with exactly one node: {node_match}")
@@ -84,7 +83,7 @@ def list(self) -> DataFrame:
             A DataFrame containing all available GDS procedures.
         """
         namespace = self._namespace + ".list"
-        return self._query_runner.call_procedure(endpoint=namespace, custom_error=False)
+        return self._query_runner.call_procedure(endpoint=namespace, custom_error=False, retryable=True)
 
 
 class IndirectUtilAlphaEndpoints(CallerBase):