uipath-langchain 0.0.76__tar.gz → 0.0.77__tar.gz

{uipath_langchain-0.0.76 → uipath_langchain-0.0.77}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-langchain
-Version: 0.0.76
+Version: 0.0.77
 Summary: UiPath Langchain
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-python
@@ -25,5 +25,5 @@ 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-sdk>=0.0.101
+Requires-Dist: uipath-sdk>=0.0.106
 Provides-Extra: langchain

{uipath_langchain-0.0.76 → uipath_langchain-0.0.77}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "uipath-langchain"
-version = "0.0.76"
+version = "0.0.77"
 description = "UiPath Langchain"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.9"
 dependencies = [
-    "uipath-sdk>=0.0.101",
+    "uipath-sdk>=0.0.106",
     "langgraph>=0.2.70",
     "langchain-core>=0.3.34",
     "langgraph-checkpoint-sqlite>=2.0.3",

{uipath_langchain-0.0.76 → uipath_langchain-0.0.77}/uipath_langchain/_cli/_runtime/_escalation.py

@@ -162,7 +162,7 @@ class Escalation:
         return current
 
     def extract_response_value(self, action_data: Dict[str, Any]) -> Any:
-        if not self.enabled or not self._config:
+        if not self._config:
             return ""
 
         response_template = self._config.get("response")
@@ -208,7 +208,7 @@ class Escalation:
 
                     return extracted_value
 
-        return ""
+        return action_data
 
     async def create(self, value: Any) -> Optional[Action]:
         """

{uipath_langchain-0.0.76 → uipath_langchain-0.0.77}/uipath_langchain/_cli/_runtime/_input.py

@@ -65,7 +65,7 @@ class LangGraphInputProcessor:
             logger.debug(f"Action: {action}")
             if action.data is None:
                 return Command(resume={})
-            if self.escalation:
+            if self.escalation and self.escalation.enabled:
                 extracted_value = self.escalation.extract_response_value(action.data)
                 return Command(resume=extracted_value)
             return Command(resume=action.data)

{uipath_langchain-0.0.76 → uipath_langchain-0.0.77}/uipath_langchain/_cli/_runtime/_output.py

@@ -15,7 +15,7 @@ from uipath_sdk._cli._runtime._contracts import (
     UiPathRuntimeResult,
     UiPathRuntimeStatus,
 )
-from uipath_sdk._models import InvokeProcess
+from uipath_sdk._models import CreateAction, InvokeProcess, WaitAction, WaitJob
 from uipath_sdk._models.actions import Action
 
 from ._context import LangGraphRuntimeContext
@@ -34,10 +34,14 @@ class InterruptInfo:
     @property
     def type(self) -> Optional[UiPathResumeTriggerType]:
         """Returns the type of the interrupt value."""
-        if isinstance(self.value, Action):
+        if isinstance(self.value, CreateAction):
+            return UiPathResumeTriggerType.ACTION
+        if isinstance(self.value, WaitAction):
             return UiPathResumeTriggerType.ACTION
         if isinstance(self.value, InvokeProcess):
             return UiPathResumeTriggerType.JOB
+        if isinstance(self.value, WaitJob):
+            return UiPathResumeTriggerType.JOB
         return None
 
     @property
@@ -218,33 +222,62 @@ class LangGraphOutputProcessor:
 
                 try:
                     default_escalation = Escalation()
-
-                    if default_escalation.enabled:
+                    if default_escalation.enabled and isinstance(
+                        self.interrupt_value, str
+                    ):
                         action = await default_escalation.create(self.interrupt_value)
                         if action:
                             self._resume_trigger = UiPathResumeTrigger(
                                 trigger_type=UiPathResumeTriggerType.ACTION,
                                 item_key=action.key,
                             )
-
-                    if (
-                        isinstance(self.interrupt_info, InterruptInfo)
-                        and self.interrupt_info.type is UiPathResumeTriggerType.JOB
-                    ):
-                        if not isinstance(self.interrupt_value, InvokeProcess):
-                            raise Exception(
-                                "Expected 'InvokeProcess' as interrupt value type"
-                            )
-
+                        return
+                    if isinstance(self.interrupt_info, InterruptInfo):
                         uipath_sdk = UiPathSDK()
-                        job = await uipath_sdk.processes.invoke_async(
-                            name=self.interrupt_value.name,
-                            input_arguments=self.interrupt_value.input_arguments,
-                        )
-                        if job:
-                            self._resume_trigger = UiPathResumeTrigger(
-                                trigger_type=UiPathResumeTriggerType.JOB,
-                                item_key=job.Key,
+                        if self.interrupt_info.type is UiPathResumeTriggerType.JOB:
+                            if isinstance(self.interrupt_value, InvokeProcess):
+                                job = await uipath_sdk.processes.invoke_async(
+                                    name=self.interrupt_value.name,
+                                    input_arguments=self.interrupt_value.input_arguments,
+                                )
+                                if job:
+                                    self._resume_trigger = UiPathResumeTrigger(
+                                        trigger_type=UiPathResumeTriggerType.JOB,
+                                        item_key=job.Key,
+                                    )
+                            elif isinstance(self.interrupt_value, WaitJob):
+                                self._resume_trigger = UiPathResumeTrigger(
+                                    triggerType=UiPathResumeTriggerType.JOB,
+                                    itemKey=self.interrupt_value.job.Key,
+                                )
+                        elif self.interrupt_info.type is UiPathResumeTriggerType.ACTION:
+                            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
+                                    else "",
+                                    app_key=self.interrupt_value.key
+                                    if self.interrupt_value.key
+                                    else "",
+                                    app_version=self.interrupt_value.app_version
+                                    if self.interrupt_value.app_version
+                                    else 1,
+                                    data=self.interrupt_value.data,
+                                )
+                                if action:
+                                    self._resume_trigger = UiPathResumeTrigger(
+                                        trigger_type=UiPathResumeTriggerType.ACTION,
+                                        item_key=action.key,
+                                    )
+                            elif isinstance(self.interrupt_value, WaitAction):
+                                self._resume_trigger = UiPathResumeTrigger(
+                                    triggerType=UiPathResumeTriggerType.ACTION,
+                                    itemKey=self.interrupt_value.action.key,
+                                )
+                        else:
+                            raise Exception(
+                                f"Unknown interrupt value type {type(self.interrupt_value)}"
                             )
 
                 except Exception as e: