Merge pull request #54 from UiPath/fix/wrap-output-primitives

fix: wrap output str primitive

Author: radu-mocanu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-runtime"
-version = "0.2.8"
+version = "0.2.9"
 description = "Runtime abstractions and interfaces for building agents and automation scripts in the UiPath ecosystem"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

src/uipath/runtime/result.py

@@ -37,6 +37,8 @@ def to_dict(self) -> dict[str, Any]:
             output_data = {}
         elif isinstance(self.output, BaseModel):
             output_data = self.output.model_dump()
+        elif isinstance(self.output, str):
+            output_data = {"output": self.output}
         else:
             output_data = self.output
 

tests/test_context.py

@@ -234,3 +234,16 @@ def test_from_config_loads_runtime_and_fps_properties(tmp_path: Path) -> None:
     assert ctx.message_id == "msg-ghi"
     assert ctx.mcp_server_id == "mcp-server-456"
     assert ctx.mcp_server_slug == "test-server-slug"
+
+
+def test_string_output_wrapped_in_dict() -> None:
+    """Test that string output is wrapped in a dict with key 'output'."""
+    result = UiPathRuntimeResult(
+        status=UiPathRuntimeStatus.SUCCESSFUL,
+        output="primitive str",
+    )
+
+    result_dict = result.to_dict()
+
+    assert result_dict["output"] == {"output": "primitive str"}
+    assert result_dict["status"] == UiPathRuntimeStatus.SUCCESSFUL