Merge pull request #15 from UiPath/fix/handle_schema_exceptions

fix: handle schema exceptions

Author: cristipufu

pyproject.toml

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

src/uipath/dev/_demo/mock_runtime.py

@@ -45,7 +45,7 @@ async def new_runtime(
         )
         return MockGreetingRuntime(entrypoint=entrypoint)
 
-    def discover_runtimes(self) -> list[UiPathRuntimeProtocol]:
+    async def discover_runtimes(self) -> list[UiPathRuntimeProtocol]:
         """Return prototype instances for discovery (not really used by the UI)."""
         return [
             MockGreetingRuntime(entrypoint=ENTRYPOINT_GREETING),
@@ -62,3 +62,7 @@ def discover_entrypoints(self) -> list[str]:
             ENTRYPOINT_ANALYZE_NUMBERS,
             ENTRYPOINT_SUPPORT_CHAT,
         ]
+
+    async def dispose(self) -> None:
+        """Dispose of any resources (no-op for mock)."""
+        pass

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

@@ -6,7 +6,7 @@
 from textual.app import ComposeResult
 from textual.containers import Container, Horizontal, Vertical
 from textual.reactive import reactive
-from textual.widgets import Button, Select, TabbedContent, TabPane, TextArea
+from textual.widgets import Button, Select, TabbedContent, TabPane
 from uipath.runtime import UiPathRuntimeFactoryProtocol, UiPathRuntimeProtocol
 
 from uipath.dev.ui.widgets.json_input import JsonInput
@@ -149,7 +149,7 @@ async def on_mount(self) -> None:
 
         select = self.query_one("#entrypoint-select", Select)
 
-        json_input = self.query_one("#json-input", TextArea)
+        json_input = self.query_one("#json-input", JsonInput)
         run_button = self.query_one("#execute-btn", Button)
 
         if not self.entrypoints:
@@ -166,14 +166,17 @@ async def on_mount(self) -> None:
 
         # Use the first entrypoint as default
         self.selected_entrypoint = self.entrypoints[0]
-        select.value = self.selected_entrypoint
 
-        # Lazily fetch schema and populate input
+        # Lazily fetch schema and populate input BEFORE setting select.value
+        # to avoid triggering on_select_changed
         await self._load_schema_and_update_input(self.selected_entrypoint)
 
+        # Set the select value after loading the schema
+        select.value = self.selected_entrypoint
+
     async def _load_schema_and_update_input(self, entrypoint: str) -> None:
         """Ensure schema for entrypoint is loaded, then update JSON input."""
-        json_input = self.query_one("#json-input", TextArea)
+        json_input = self.query_one("#json-input", JsonInput)
 
         if not entrypoint or entrypoint == "no-entrypoints":
             json_input.text = "{}"
@@ -192,33 +195,40 @@ async def _load_schema_and_update_input(self, entrypoint: str) -> None:
                 input_schema = schema_obj.input or {}
                 self.entrypoint_schemas[entrypoint] = input_schema
                 schema = input_schema
-            except Exception:
-                schema = {}
-                self.entrypoint_schemas[entrypoint] = schema
+            except Exception as e:
+                json_input.text = "{}"
+                self.app.notify(
+                    f"Error loading schema for '{entrypoint}': {str(e)}",
+                    severity="error",
+                    timeout=5,
+                )
+                return
             finally:
                 if runtime is not None:
                     await runtime.dispose()
 
-        json_input.text = json.dumps(
-            mock_json_from_schema(schema),
-            indent=2,
-        )
+        # Generate mock JSON from schema
+        mock_data = mock_json_from_schema(schema)
+        json_input.text = json.dumps(mock_data, indent=2)
 
     async def on_select_changed(self, event: Select.Changed) -> None:
         """Update JSON input when user selects an entrypoint."""
-        self.selected_entrypoint = cast(str, event.value) if event.value else ""
+        new_entrypoint = cast(str, event.value) if event.value else ""
 
-        await self._load_schema_and_update_input(self.selected_entrypoint)
+        # Only load schema if the entrypoint actually changed
+        if new_entrypoint != self.selected_entrypoint:
+            self.selected_entrypoint = new_entrypoint
+            await self._load_schema_and_update_input(self.selected_entrypoint)
 
     def get_input_values(self) -> Tuple[str, str, bool]:
         """Get the selected entrypoint and JSON input values."""
-        json_input = self.query_one("#json-input", TextArea)
+        json_input = self.query_one("#json-input", JsonInput)
         return self.selected_entrypoint, json_input.text.strip(), self.conversational
 
     def reset_form(self) -> None:
         """Reset selection and JSON input to defaults."""
         select = self.query_one("#entrypoint-select", Select)
-        json_input = self.query_one("#json-input", TextArea)
+        json_input = self.query_one("#json-input", JsonInput)
 
         if not self.entrypoints:
             self.selected_entrypoint = ""