Work around license check in the AuraDS case

Co-Authored-By: Brian Shi <[email protected]>

Author: adamnsch

changelog/1.2.0.md

@@ -11,6 +11,7 @@
 
 * Fixed a bug where the `separate_property_columns=True` option of `gds.graph.streamNodeProperties` did not handle list node properties correctly.
 * Fixed a bug where an irrelevant warning was shown when creating a `GraphDataScience` object targeting an AuraDS instance with GDS server version >= 2.1.0.
+* Fixed a bug where calling `gds.alpha.graph.construct` targeting an AuraDS instance would raise an exception.
 
 
 ## Improvements

graphdatascience/query_runner/cypher_graph_constructor.py

@@ -19,7 +19,7 @@ def __init__(
         self._graph_name = graph_name
 
     def run(self, node_dfs: List[DataFrame], relationship_dfs: List[DataFrame]) -> None:
-        if self._is_enterprise():
+        if self._should_warn_about_arrow_missing():
             warnings.warn(
                 "GDS Enterprise users can use Apache Arrow for fast graph construction; please see the documentation "
                 "for instructions on how to enable it. Without Arrow enabled, this installation will use community "
@@ -55,12 +55,23 @@ def run(self, node_dfs: List[DataFrame], relationship_dfs: List[DataFrame]) -> N
             },
         )
 
-    def _is_enterprise(self) -> bool:
-        license: str = self._query_runner.run_query(
-            "CALL gds.debug.sysInfo() YIELD key, value WHERE key = 'gdsEdition' RETURN value"
-        ).squeeze()
-
-        return license == "Licensed"
+    def _should_warn_about_arrow_missing(self) -> bool:
+        try:
+            license: str = self._query_runner.run_query(
+                "CALL gds.debug.sysInfo() YIELD key, value WHERE key = 'gdsEdition' RETURN value"
+            ).squeeze()
+            should_warn = license == "Licensed"
+        except Exception as e:
+            # It's not a user's concern whether Arrow is set up or not in AuraDS.
+            if (
+                "There is no procedure with the name `gds.debug.sysInfo` "
+                "registered for this database instance." in str(e)
+            ):
+                should_warn = False
+            else:
+                raise e
+
+        return should_warn
 
     def _node_query(self, node_df: DataFrame) -> Tuple[str, List[List[Any]]]:
         node_list = node_df.values.tolist()

graphdatascience/system/system_endpoints.py

@@ -42,8 +42,18 @@ def debug(self) -> DebugProcRunner:
 
     @client_only_endpoint("gds")
     def is_licensed(self) -> bool:
-        license: str = self._query_runner.run_query(
-            "CALL gds.debug.sysInfo() YIELD key, value WHERE key = 'gdsEdition' RETURN value"
-        ).squeeze()
+        try:
+            license: str = self._query_runner.run_query(
+                "CALL gds.debug.sysInfo() YIELD key, value WHERE key = 'gdsEdition' RETURN value"
+            ).squeeze()
+        except Exception as e:
+            # AuraDS does not have `gds.debug.sysInfo`, but is always GDS EE.
+            if (
+                "There is no procedure with the name `gds.debug.sysInfo` "
+                "registered for this database instance." in str(e)
+            ):
+                license = "Licensed"
+            else:
+                raise e
 
         return license == "Licensed"