proper Speckle client

main.py

@@ -57,12 +57,9 @@ def automate_function(
     """
     # the context provides a conveniet way, to receive the triggering version
     try:
-
         projInfo = query_version_info(automate_context)
         if not projInfo.speckle_type.endswith("Revit.ProjectInfo"):
-            automate_context.mark_run_failed(
-                "Not a valid 'Revit.ProjectInfo' provided"
-            )
+            automate_context.mark_run_failed("Not a valid 'Revit.ProjectInfo' provided")
 
         lon = np.rad2deg(projInfo["longitude"])
         lat = np.rad2deg(projInfo["latitude"])
@@ -88,7 +85,7 @@ def automate_function(
             source_data="© OpenStreetMap",
             source_url="https://www.openstreetmap.org/",
         )
-        r'''
+        r"""
         roads_line_layer = Collection(
             elements=roads_lines,
             units="m",
@@ -97,7 +94,7 @@ def automate_function(
             source_data="© OpenStreetMap",
             source_url="https://www.openstreetmap.org/",
         )
-        '''
+        """
         roads_mesh_layer = Collection(
             elements=roads_meshes,
             units="m",

utils/utils_server.py

@@ -1,8 +1,7 @@
-
-
 from specklepy.api.wrapper import StreamWrapper
 from gql import gql
 
+
 def query_version_info(automate_context):
     automation_run_data = automate_context.automation_run_data
     # get branch name
@@ -19,10 +18,7 @@ def query_version_info(automate_context):
         }
     """
     )
-    sw = StreamWrapper(
-        f"{automation_run_data.speckle_server_url}/projects/{automation_run_data.project_id}"
-    )
-    client = sw.get_client()
+    client = automate_context.speckle_client
     params = {
         "project_id": automation_run_data.project_id,
         "model_id": automation_run_data.model_id,