Merge pull request #14 from UiPath/fix/mock_schame

fix: update uipath-runtime, dispose

Author: cristipufu

pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "uipath-dev"
-version = "0.0.4"
+version = "0.0.5"
 description = "UiPath Developer Console"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
 dependencies = [
-  "uipath-runtime>=0.0.11, <0.1.0",
+  "uipath-runtime>=0.0.15, <0.1.0",
   "textual>=6.6.0, <7.0.0",
   "pyperclip>=1.11.0, <2.0.0",
 ]

src/uipath/dev/_demo/mock_runtime.py

@@ -26,7 +26,9 @@
 class MockRuntimeFactory:
     """Runtime factory compatible with UiPathRuntimeFactoryProtocol."""
 
-    async def new_runtime(self, entrypoint: str) -> UiPathRuntimeProtocol:
+    async def new_runtime(
+        self, entrypoint: str, runtime_id: str
+    ) -> UiPathRuntimeProtocol:
         """Create a new runtime instance for the given entrypoint."""
         if entrypoint == ENTRYPOINT_GREETING:
             return MockGreetingRuntime(entrypoint=entrypoint)

src/uipath/dev/services/run_service.py

@@ -77,6 +77,7 @@ async def execute(self, run: ExecutionRun) -> None:
 
         This is the extracted version of the old `_execute_runtime` method.
         """
+        new_runtime: UiPathRuntimeProtocol | None = None
         try:
             execution_input: dict[str, Any] | None = {}
             execution_options: UiPathExecuteOptions = UiPathExecuteOptions()
@@ -100,7 +101,8 @@ async def execute(self, run: ExecutionRun) -> None:
             )
 
             new_runtime = await self.runtime_factory.new_runtime(
-                entrypoint=run.entrypoint
+                entrypoint=run.entrypoint,
+                runtime_id=run.id,
             )
 
             runtime: UiPathRuntimeProtocol
@@ -177,6 +179,9 @@ async def execute(self, run: ExecutionRun) -> None:
                 title=str(e),
                 detail=traceback.format_exc(),
             )
+        finally:
+            if new_runtime is not None:
+                await new_runtime.dispose()
 
         self.runs[run.id] = run
         self._emit_run_updated(run)

src/uipath/dev/ui/panels/new_run_panel.py

@@ -7,7 +7,7 @@
 from textual.containers import Container, Horizontal, Vertical
 from textual.reactive import reactive
 from textual.widgets import Button, Select, TabbedContent, TabPane, TextArea
-from uipath.runtime import UiPathRuntimeFactoryProtocol
+from uipath.runtime import UiPathRuntimeFactoryProtocol, UiPathRuntimeProtocol
 
 from uipath.dev.ui.widgets.json_input import JsonInput
 
@@ -182,18 +182,22 @@ async def _load_schema_and_update_input(self, entrypoint: str) -> None:
         schema = self.entrypoint_schemas.get(entrypoint)
 
         if schema is None:
+            runtime: UiPathRuntimeProtocol | None = None
             try:
-                runtime = await self._runtime_factory.new_runtime(entrypoint)
+                runtime = await self._runtime_factory.new_runtime(
+                    entrypoint, runtime_id="default"
+                )
                 schema_obj = await runtime.get_schema()
 
                 input_schema = schema_obj.input or {}
                 self.entrypoint_schemas[entrypoint] = input_schema
                 schema = input_schema
-
-                await runtime.dispose()
             except Exception:
                 schema = {}
                 self.entrypoint_schemas[entrypoint] = schema
+            finally:
+                if runtime is not None:
+                    await runtime.dispose()
 
         json_input.text = json.dumps(
             mock_json_from_schema(schema),