fix: delay openehxa client instantiation

Author: nazarfil

openhexa/sdk/client/__init__.py

@@ -2,7 +2,12 @@
 
 from ..utils import OpenHexaClient
 
-openhexa = OpenHexaClient()
+_openhexa = None
 
+def get_openhexa():
+    global _openhexa
+    if _openhexa is None:
+        _openhexa = OpenHexaClient()
+    return _openhexa
 
-__all__ = ["openhexa"]
+__all__ = ["get_openhexa"]

openhexa/sdk/pipelines/run.py

@@ -5,12 +5,15 @@
 import os
 from typing import Any
 
-from openhexa.graphql import CurrentPipelineRunCurrentPipelineRunPipeline
-from openhexa.sdk.client import OpenHexaClient
+
 from openhexa.sdk.pipelines.log_level import LogLevel
 from openhexa.sdk.utils import Environment, get_environment, graphql
 from openhexa.sdk.workspaces import workspace
 
+from openhexa.sdk.client import get_openhexa
+
+from openhexa.graphql import CurrentPipelineRunCurrentPipelineRunPipeline
+
 
 class CurrentRun:
     """Represents the current run of a pipeline.
@@ -27,18 +30,18 @@ def _connected(self):
     def pipeline_run_id(self) -> Any:
         """Get the current pipeline run ID."""
         if self._connected:
-            # Use the OpenHexaClient to get the current pipeline run ID
-            return OpenHexaClient().current_pipeline_run()["id"]
+            return get_openhexa().current_pipeline_run()["id"]
         else:
             raise RuntimeError(
                 "CurrentRun is not connected to the OpenHEXA server. "
                 "Make sure to set the HEXA_SERVER_URL environment variable."
             )
 
+
     def get_pipline(self) -> CurrentPipelineRunCurrentPipelineRunPipeline:
         """Get the current pipeline."""
         if self._connected:
-            return OpenHexaClient().current_pipeline_run()["pipeline"]
+            return get_openhexa().current_pipeline_run()["pipeline"]
         else:
             raise RuntimeError(
                 "CurrentRun is not connected to the OpenHEXA server. "