fix: imports

Author: nazarfil

openhexa/sdk/client/__init__.py

@@ -1,16 +1,7 @@
 """OpenHexa Client."""
-
 from ..utils import OpenHexaClient
 
-_openhexa = None
-
-
-def get_openhexa():
-    """Get the OpenHexa client instance."""
-    global _openhexa
-    if _openhexa is None:
-        _openhexa = OpenHexaClient()
-    return _openhexa
+openhexa = OpenHexaClient()
 
 
-__all__ = ["get_openhexa"]
+__all__ = ["openhexa"]

openhexa/sdk/pipelines/run.py

@@ -6,11 +6,12 @@
 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 ..utils import OpenHexaClient
+
 
 class CurrentRun:
     """Represents the current run of a pipeline.
@@ -33,15 +34,20 @@ def _is_connected(self):
             )
 
     @property
-    def pipeline_run_id(self) -> Any:
+    def id(self) -> Any:
         """Get the current pipeline run ID."""
-        self._is_connected()
-        return OpenHexaClient().current_pipeline_run().id
+        if "HEXA_RUN_ID" in os.environ:
+            return os.environ["HEXA_RUN_ID"]
+        else:
+            return OpenHexaClient().current_pipeline_run().id
 
-    def get_pipline(self) -> CurrentPipelineRunCurrentPipelineRunPipeline:
+    def get_pipline_info(self) -> CurrentPipelineRunCurrentPipelineRunPipeline:
         """Get the current pipeline."""
         self._is_connected()
-        return OpenHexaClient().current_pipeline_run().pipeline
+        if "HEXA_RUN_ID" in os.environ:
+            os.environ["HEXA_RUN_ID"]
+        else:
+            return OpenHexaClient().current_pipeline_run().pipeline
 
     def add_file_output(self, path: str):
         """Record a run output for a file creation operation.