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,10 @@ 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={})
@@ -106,7 +106,7 @@ 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

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,17 @@ 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_path=self.interrupt_value.action.app_folder_path if self.interrupt_value.app_folder_path else None,
+                                    folder_key=self.interrupt_value.action.app_folder_key if self.interrupt_value.app_folder_key else None
                                 )
 
                 except Exception as e:
@@ -324,8 +339,8 @@ 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: