Merge pull request #5 from UiPath/fix/run_workflow

Basic uipath run implementation

Author: cristipufu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-llamaindex"
-version = "0.0.2"
+version = "0.0.3"
 description = "UiPath LlamaIndex SDK"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"

src/uipath_llamaindex/_cli/_runtime/_runtime.py

@@ -34,12 +34,18 @@ async def execute(self) -> Optional[UiPathRuntimeResult]:
             Dictionary with execution results
 
         Raises:
-            UiPathLlamaRuntimeError: If execution fails
+            UiPathLlamaIndexRuntimeError: If execution fails
         """
         await self.validate()
 
         try:
-            a = 2
+            output = await self.context.workflow.run(self.context.input_json)
+
+            self.context.result = UiPathRuntimeResult(
+                output=self._serialize_object(output)
+            )
+
+            return self.context.result
 
         except Exception as e:
             if isinstance(e, UiPathLlamaIndexRuntimeError):
@@ -98,17 +104,17 @@ async def validate(self) -> None:
                 UiPathErrorCategory.DEPLOYMENT,
             )
 
-        # Get the specified graph
-        self.graph_config = self.context.config.get_workflow(self.context.entrypoint)
-        if not self.graph_config:
+        # Get the specified workflow configuration
+        self.workflow_config = self.context.config.get_workflow(self.context.entrypoint)
+        if not self.workflow_config:
             raise UiPathLlamaIndexRuntimeError(
                 "WORKFLOW_NOT_FOUND",
                 "Workflow not found",
                 f"Workflow '{self.context.entrypoint}' not found.",
                 UiPathErrorCategory.DEPLOYMENT,
             )
         try:
-            self.context.workflow = await self.graph_config.load_workflow()
+            self.context.workflow = await self.workflow_config.load_workflow()
         except ImportError as e:
             raise UiPathLlamaIndexRuntimeError(
                 "WORKFLOW_IMPORT_ERROR",
@@ -120,7 +126,7 @@ async def validate(self) -> None:
             raise UiPathLlamaIndexRuntimeError(
                 "WORKFLOW_TYPE_ERROR",
                 "Invalid workflow type",
-                f"Workflow '{self.context.entrypoint}' is not a valid Workflow: {str(e)}",
+                f"Workflow '{self.context.entrypoint}' is not a valid `Workflow`: {str(e)}",
                 UiPathErrorCategory.USER,
             ) from e
         except ValueError as e:
@@ -141,3 +147,28 @@ async def validate(self) -> None:
     async def cleanup(self) -> None:
         """Clean up all resources."""
         pass
+
+    def _serialize_object(self, obj):
+        """Recursively serializes an object and all its nested components."""
+        # Handle Pydantic models
+        if hasattr(obj, "dict"):
+            return self._serialize_object(obj.dict())
+        elif hasattr(obj, "model_dump"):
+            return self._serialize_object(obj.model_dump(by_alias=True))
+        elif hasattr(obj, "to_dict"):
+            return self._serialize_object(obj.to_dict())
+        # Handle dictionaries
+        elif isinstance(obj, dict):
+            return {k: self._serialize_object(v) for k, v in obj.items()}
+        # Handle lists
+        elif isinstance(obj, list):
+            return [self._serialize_object(item) for item in obj]
+        # Handle other iterable objects (convert to dict first)
+        elif hasattr(obj, "__iter__") and not isinstance(obj, (str, bytes)):
+            try:
+                return self._serialize_object(dict(obj))
+            except (TypeError, ValueError):
+                return obj
+        # Return primitive types as is
+        else:
+            return obj

src/uipath_llamaindex/_cli/cli_init.py

@@ -115,7 +115,7 @@ def generate_schema_from_workflow(workflow: Workflow) -> Dict[str, Any]:
     # For output schema, check if it's the base StopEvent or a custom subclass
     if stop_event_class is StopEvent:
         # For base StopEvent, just use Any type for the result
-        schema["output"] = {"type": "object", "properties": {}}
+        schema["output"] = {"type": "object", "properties": {}, "required": []}
     else:
         # For custom StopEvent subclasses, extract their Pydantic schema
         try:

src/uipath_llamaindex/_cli/cli_run.py

@@ -25,7 +25,7 @@ def llamaindex_run_middleware(
     if not config.exists:
         return MiddlewareResult(
             should_continue=True
-        )  # Continue with normal flow if no llama.json
+        )  # Continue with normal flow if no llama_index.json
 
     try:
 
@@ -54,7 +54,7 @@ async def execute():
             )
 
             env["UIPATH_REQUESTING_PRODUCT"] = "uipath-python-sdk"
-            env["UIPATH_REQUESTING_FEATURE"] = "llama"
+            env["UIPATH_REQUESTING_FEATURE"] = "llamaindex"
 
             async with UiPathLlamaIndexRuntime.from_context(context) as runtime:
                 await runtime.execute()