fix: send started/completed runtime state events

Author: cristipufu

pyproject.toml

@@ -1,12 +1,12 @@
 [project]
 name = "uipath-langchain"
-version = "0.5.62"
+version = "0.5.63"
 description = "Python SDK that enables developers to build and deploy LangGraph agents to the UiPath Cloud Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
 dependencies = [
     "uipath>=2.8.31, <2.9.0",
-    "uipath-runtime>=0.8.4, <0.9.0",
+    "uipath-runtime>=0.8.6, <0.9.0",
     "langgraph>=1.0.0, <2.0.0",
     "langchain-core>=1.2.11, <2.0.0",
     "langgraph-checkpoint-sqlite>=3.0.3, <4.0.0",

src/uipath_langchain/runtime/runtime.py

@@ -21,6 +21,7 @@
     UiPathRuntimeEvent,
     UiPathRuntimeMessageEvent,
     UiPathRuntimeStateEvent,
+    UiPathRuntimeStatePhase,
 )
 from uipath.runtime.schema import UiPathRuntimeSchema
 
@@ -131,7 +132,7 @@ async def stream(
                 graph_input,
                 graph_config,
                 interrupt_before=options.breakpoints if options else None,
-                stream_mode=["messages", "updates"],
+                stream_mode=["messages", "updates", "tasks"],
                 subgraphs=True,
             ):
                 namespace, chunk_type, data = stream_chunk
@@ -178,6 +179,32 @@ async def stream(
                                 ),
                             )
                             yield state_event
+                elif chunk_type == "tasks":
+                    if isinstance(data, dict):
+                        task_name = data.get("name", "")
+
+                        if "input" in data:
+                            phase = UiPathRuntimeStatePhase.STARTED
+                        elif "result" in data:
+                            phase = (
+                                UiPathRuntimeStatePhase.FAULTED
+                                if data.get("error")
+                                else UiPathRuntimeStatePhase.COMPLETED
+                            )
+                        else:
+                            phase = None
+
+                        if phase is not None:
+                            state_event = UiPathRuntimeStateEvent(
+                                payload=serialize_output(data),
+                                node_name=task_name,
+                                qualified_node_name=self._build_node_name(
+                                    namespace,
+                                    task_name,
+                                ),
+                                phase=phase,
+                            )
+                            yield state_event
 
             # Extract output from final chunk
             graph_output = self._extract_graph_result(final_chunk)