Merge pull request #457 from UiPath/fix/runtime-async-func

fix: core runtime await async functions

Author: radu-mocanu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath"
-version = "2.0.79"
+version = "2.0.80"
 description = "Python SDK and CLI for UiPath Platform, enabling programmatic interaction with automation services, process management, and deployment tools."
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"

src/uipath/_cli/_runtime/_runtime.py

@@ -50,7 +50,7 @@ async def execute(self) -> Optional[UiPathRuntimeResult]:
             if self.context.entrypoint is None:
                 return None
 
-            script_result = self._execute_python_script(
+            script_result = await self._execute_python_script(
                 self.context.entrypoint, self.context.input_json
             )
 
@@ -111,7 +111,7 @@ async def cleanup(self) -> None:
         """Cleanup runtime resources."""
         pass
 
-    def _execute_python_script(self, script_path: str, input_data: Any) -> Any:
+    async def _execute_python_script(self, script_path: str, input_data: Any) -> Any:
         """Execute the Python script with the given input."""
         spec = importlib.util.spec_from_file_location("dynamic_module", script_path)
         if not spec or not spec.loader:
@@ -139,10 +139,13 @@ def _execute_python_script(self, script_path: str, input_data: Any) -> Any:
                 sig = inspect.signature(main_func)
                 params = list(sig.parameters.values())
 
+                # Check if the function is asynchronous
+                is_async = inspect.iscoroutinefunction(main_func)
+
                 # Case 1: No parameters
                 if not params:
                     try:
-                        result = main_func()
+                        result = await main_func() if is_async else main_func()
                         return (
                             self._convert_from_class(result)
                             if result is not None
@@ -166,7 +169,11 @@ def _execute_python_script(self, script_path: str, input_data: Any) -> Any:
                     try:
                         valid_type = cast(Type[Any], input_type)
                         typed_input = self._convert_to_class(input_data, valid_type)
-                        result = main_func(typed_input)
+                        result = (
+                            await main_func(typed_input)
+                            if is_async
+                            else main_func(typed_input)
+                        )
                         return (
                             self._convert_from_class(result)
                             if result is not None
@@ -183,7 +190,11 @@ def _execute_python_script(self, script_path: str, input_data: Any) -> Any:
                 # Case 3: Dict parameter
                 else:
                     try:
-                        result = main_func(input_data)
+                        result = (
+                            await main_func(input_data)
+                            if is_async
+                            else main_func(input_data)
+                        )
                         return (
                             self._convert_from_class(result)
                             if result is not None