uipath-langchain 0.0.88__py3-none-any.whl → 0.0.90__py3-none-any.whl

uipath_langchain/_cli/_runtime/_input.py

@@ -58,10 +58,12 @@ class LangGraphInputProcessor:
         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 @@ class LangGraphInputProcessor:
                 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,7 +108,7 @@ class LangGraphInputProcessor:
                 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
@@ -114,7 +116,7 @@ class LangGraphInputProcessor:
                 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",

uipath_langchain/_cli/_runtime/_output.py

@@ -232,6 +232,9 @@ class LangGraphOutputProcessor:
                             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 @@ class LangGraphOutputProcessor:
                             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,25 @@ class LangGraphOutputProcessor:
                                     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.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,
                                 )
 
                 except Exception as e:
@@ -324,8 +347,14 @@ class LangGraphOutputProcessor:
                 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:

uipath_langchain/_cli/_runtime/_runtime.py

@@ -1,5 +1,6 @@
 import json
 import logging
+import os
 from typing import List, Optional
 
 from langchain_core.callbacks.base import BaseCallbackHandler
@@ -54,6 +55,11 @@ class LangGraphRuntime(UiPathBaseRuntime):
         tracer = None
 
         try:
+            if self.context.resume is False and self.context.job_id is None:
+                # Delete the previous graph state file at debug time
+                if os.path.exists(self.state_file_path):
+                    os.remove(self.state_file_path)
+
             async with AsyncSqliteSaver.from_conn_string(
                 self.state_file_path
             ) as memory:
@@ -88,6 +94,14 @@ class LangGraphRuntime(UiPathBaseRuntime):
                     "callbacks": callbacks,
                 }
 
+                recursion_limit = os.environ.get("LANGCHAIN_RECURSION_LIMIT", None)
+                max_concurrency = os.environ.get("LANGCHAIN_MAX_CONCURRENCY", None)
+
+                if recursion_limit is not None:
+                    graph_config["recursion_limit"] = int(recursion_limit)
+                if max_concurrency is not None:
+                    graph_config["max_concurrency"] = int(max_concurrency)
+
                 # Stream the output at debug time
                 if self.context.job_id is None:
                     # Get final chunk while streaming

{uipath_langchain-0.0.88.dist-info → uipath_langchain-0.0.90.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-langchain
-Version: 0.0.88
+Version: 0.0.90
 Summary: UiPath Langchain
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-langchain-python
@@ -25,7 +25,7 @@ Requires-Dist: pydantic-settings>=2.6.0
 Requires-Dist: python-dotenv>=1.0.1
 Requires-Dist: requests>=2.23.3
 Requires-Dist: types-requests>=2.32.0.20241016
-Requires-Dist: uipath<2.1.0,>=2.0.4
+Requires-Dist: uipath<2.1.0,>=2.0.9
 Provides-Extra: langchain
 Description-Content-Type: text/markdown
 

{uipath_langchain-0.0.88.dist-info → uipath_langchain-0.0.90.dist-info}/RECORD

@@ -6,9 +6,9 @@ uipath_langchain/_cli/cli_run.py,sha256=zMn2P1gB1ogM8_EmZsIhqNpeI9Oc02Z5Gj-oni7e
 uipath_langchain/_cli/_runtime/_context.py,sha256=wr4aNn06ReIXmetEZ6b6AnpAt64p13anQ2trZ5Bzgio,807
 uipath_langchain/_cli/_runtime/_escalation.py,sha256=oA5NvZvCo8ngELFJRyhZNM69DxVHrshhMY6CUk_cukQ,8055
 uipath_langchain/_cli/_runtime/_exception.py,sha256=USKkLYkG-dzjX3fEiMMOHnVUpiXJs_xF0OQXCCOvbYM,546
-uipath_langchain/_cli/_runtime/_input.py,sha256=aaqU7ZiTwVmE1CgaiqgOtArr925bYr9c4CuWt1oC-jk,5418
-uipath_langchain/_cli/_runtime/_output.py,sha256=we6sdXGI-dARRWUgcwblB8CMytBmYncJh6xiMkwcAVA,14238
-uipath_langchain/_cli/_runtime/_runtime.py,sha256=PWzs0EnmpzkEIceoPq_SVLLgFbTC1aU71HbPYvlzVOg,11121
+uipath_langchain/_cli/_runtime/_input.py,sha256=gKzPaGW-EzgeAskWJjbCWnfZRLu_BM7lCXkq0XkVGLU,5614
+uipath_langchain/_cli/_runtime/_output.py,sha256=WQSrsvGaaclZ6GLWEh6Nk1Mz1iGaIB45PgIX3DS3AN4,16130
+uipath_langchain/_cli/_runtime/_runtime.py,sha256=M82qhxU6OVrobiaqoktPYEN9HQyK4NjuW9VgrTAxfWs,11803
 uipath_langchain/_cli/_utils/_graph.py,sha256=WLBSJfPc3_C07SqJhePRe17JIc5wcBvEqLviMcNOdTA,6950
 uipath_langchain/_utils/__init__.py,sha256=Sp2qnEXLAp9ftQ09x7CZMenYnpXIIGFJNv8zNN7vAsw,172
 uipath_langchain/_utils/_request_mixin.py,sha256=t_1HWBxqEl-wsSk9ubmIM-8vs9BlNy4ZVBxtDxktn6U,18489
@@ -33,8 +33,8 @@ uipath_langchain/utils/_request_mixin.py,sha256=WFyTDyAthSci1DRwUwS21I3hLntD7HdV
 uipath_langchain/utils/_settings.py,sha256=MhwEVj4gVRSar0RBf2w2hTjO-5Qm-HpCuufqN3gSWjA,3390
 uipath_langchain/utils/_sleep_policy.py,sha256=e9pHdjmcCj4CVoFM1jMyZFelH11YatsgWfpyrfXzKBQ,1251
 uipath_langchain/vectorstores/context_grounding_vectorstore.py,sha256=eTa5sX43-ydB1pj9VNHUPbB-hC36fZK_CGrNe5U2Nrw,9393
-uipath_langchain-0.0.88.dist-info/METADATA,sha256=rSi1XhzmDNx8D5LB0F4lhuYTCrlNoMMU6bbTJHknIUY,3818
-uipath_langchain-0.0.88.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-uipath_langchain-0.0.88.dist-info/entry_points.txt,sha256=FUtzqGOEntlJKMJIXhQUfT7ZTbQmGhke1iCmDWZaQZI,81
-uipath_langchain-0.0.88.dist-info/licenses/LICENSE,sha256=JDpt-uotAkHFmxpwxi6gwx6HQ25e-lG4U_Gzcvgp7JY,1063
-uipath_langchain-0.0.88.dist-info/RECORD,,
+uipath_langchain-0.0.90.dist-info/METADATA,sha256=BVn0Hp0FpVC3BrXtGBfMNwixP9VapW8dxRyWCJvnutI,3818
+uipath_langchain-0.0.90.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+uipath_langchain-0.0.90.dist-info/entry_points.txt,sha256=FUtzqGOEntlJKMJIXhQUfT7ZTbQmGhke1iCmDWZaQZI,81
+uipath_langchain-0.0.90.dist-info/licenses/LICENSE,sha256=JDpt-uotAkHFmxpwxi6gwx6HQ25e-lG4U_Gzcvgp7JY,1063
+uipath_langchain-0.0.90.dist-info/RECORD,,