feat: include folder info for actions

Author: radu-mocanu

pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "uipath-langchain"
-version = "0.0.88"
+version = "0.0.89"
 description = "UiPath Langchain"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
 dependencies = [
-    "uipath>=2.0.4, <2.1.0",
+    "uipath>=2.0.8, <2.1.0",
     "langgraph>=0.2.70",
     "langchain-core>=0.3.34",
     "langgraph-checkpoint-sqlite>=2.0.3",

src/uipath_langchain/_cli/_runtime/_input.py

@@ -58,10 +58,12 @@ async def process(self) -> Any:
         if not trigger:
             return Command(resume=self.context.input_json)
 
-        type, key = trigger
+        type, key, folder_path, folder_key, payload = trigger
         logger.debug(f"ResumeTrigger: {type} {key}")
         if type == UiPathResumeTriggerType.ACTION.value and key:
-            action = self.uipath.actions.retrieve(key)
+            action = await self.uipath.actions.retrieve_async(
+                key, app_folder_key=folder_key, app_folder_path=folder_path
+            )
             logger.debug(f"Action: {action}")
             if action.data is None:
                 return Command(resume={})
@@ -95,7 +97,7 @@ async def process(self) -> Any:
                 return Command(resume=try_convert_to_json_format(job.output_arguments))
         return Command(resume=self.context.input_json)
 
-    async def _get_latest_trigger(self) -> Optional[tuple[str, str]]:
+    async def _get_latest_trigger(self) -> Optional[tuple[str, str, str, str, str]]:
         """Fetch the most recent trigger from the database."""
         if self.context.memory is None:
             return None
@@ -106,15 +108,15 @@ async def _get_latest_trigger(self) -> Optional[tuple[str, str]]:
                 self.context.memory.conn.cursor() as cur,
             ):
                 await cur.execute(f"""
-                    SELECT type, key
+                    SELECT type, key, folder_path, folder_key, payload
                     FROM {self.context.resume_triggers_table}
                     ORDER BY timestamp DESC
                     LIMIT 1
                 """)
                 result = await cur.fetchone()
                 if result is None:
                     return None
-                return cast(tuple[str, str], tuple(result))
+                return cast(tuple[str, str, str, str, str], tuple(result))
         except Exception as e:
             raise LangGraphRuntimeError(
                 "DB_QUERY_FAILED",

src/uipath_langchain/_cli/_runtime/_output.py

@@ -232,6 +232,9 @@ async def _save_resume_trigger(self) -> None:
                             id INTEGER PRIMARY KEY AUTOINCREMENT,
                             type TEXT NOT NULL,
                             key TEXT,
+                            folder_key TEXT,
+                            folder_path TEXT,
+                            payload TEXT,
                             timestamp DATETIME DEFAULT (strftime('%Y-%m-%d %H:%M:%S', 'now', 'utc'))
                         )
                     """)
@@ -277,11 +280,17 @@ async def _save_resume_trigger(self) -> None:
                             if isinstance(self.interrupt_value, CreateAction):
                                 action = uipath_sdk.actions.create(
                                     title=self.interrupt_value.title,
-                                    app_name=self.interrupt_value.name
-                                    if self.interrupt_value.name
+                                    app_name=self.interrupt_value.app_name
+                                    if self.interrupt_value.app_name
                                     else "",
-                                    app_key=self.interrupt_value.key
-                                    if self.interrupt_value.key
+                                    app_folder_path=self.interrupt_value.app_folder_path
+                                    if self.interrupt_value.app_folder_path
+                                    else "",
+                                    app_folder_key=self.interrupt_value.app_folder_key
+                                    if self.interrupt_value.app_folder_key
+                                    else "",
+                                    app_key=self.interrupt_value.app_key
+                                    if self.interrupt_value.app_key
                                     else "",
                                     app_version=self.interrupt_value.app_version
                                     if self.interrupt_value.app_version
@@ -295,11 +304,22 @@ async def _save_resume_trigger(self) -> None:
                                     self._resume_trigger = UiPathResumeTrigger(
                                         trigger_type=UiPathResumeTriggerType.ACTION,
                                         item_key=action.key,
+                                        payload=self.interrupt_value.model_dump_json(),
+                                        folder_path=self.interrupt_value.app_folder_path
+                                        if self.interrupt_value.app_folder_path
+                                        else None,
+                                        folder_key=self.interrupt_value.app_folder_key
+                                        if self.interrupt_value.app_folder_key
+                                        else None,
                                     )
                             elif isinstance(self.interrupt_value, WaitAction):
                                 self._resume_trigger = UiPathResumeTrigger(
                                     triggerType=UiPathResumeTriggerType.ACTION,
                                     itemKey=self.interrupt_value.action.key,
+                                    payload=self.interrupt_value.model_dump_json(),
+                                    folder_key=self.interrupt_value.action.organization_unit_id
+                                    if self.interrupt_value.action.organization_unit_id
+                                    else None,
                                 )
 
                 except Exception as e:
@@ -324,8 +344,14 @@ async def _save_resume_trigger(self) -> None:
                 try:
                     logger.debug(f"ResumeTrigger: {trigger_type} {trigger_key}")
                     await cur.execute(
-                        f"INSERT INTO {self.context.resume_triggers_table} (type, key) VALUES (?, ?)",
-                        (trigger_type, trigger_key),
+                        f"INSERT INTO {self.context.resume_triggers_table} (type, key, payload, folder_path, folder_key) VALUES (?, ?, ?, ?, ?)",
+                        (
+                            trigger_type,
+                            trigger_key,
+                            self.resume_trigger.payload,
+                            self.resume_trigger.folder_path,
+                            self.resume_trigger.folder_key,
+                        ),
                     )
                     await self.context.memory.conn.commit()
                 except Exception as e: