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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-langchain
-Version: 0.0.76
+Version: 0.0.78
 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.78}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "uipath-langchain"
-version = "0.0.76"
+version = "0.0.78"
 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.78}/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.78}/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.78}/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:

{uipath_langchain-0.0.76 → uipath_langchain-0.0.78}/uipath_langchain/tracers/AsyncUiPathTracer.py

@@ -13,13 +13,15 @@ from langchain_core.tracers.base import AsyncBaseTracer
 from langchain_core.tracers.schemas import Run
 from pydantic import PydanticDeprecationWarning
 
+from ._utils import _simple_serialize_defaults
+
 logger = logging.getLogger(__name__)
 
 
 class Status:
     SUCCESS = 1
     ERROR = 2
-    INTERRUPTED = 1  # intentional equal to SUCCESS
+    INTERRUPTED = 3
 
 
 class AsyncUiPathTracer(AsyncBaseTracer):
@@ -235,7 +237,7 @@ class AsyncUiPathTracer(AsyncBaseTracer):
 
     def _safe_json_dump(self, obj) -> str:
         try:
-            json_str = json.dumps(obj, default=str)
+            json_str = json.dumps(obj, default=_simple_serialize_defaults)
             return json_str
         except Exception as e:
             logger.warning(e)

{uipath_langchain-0.0.76 → uipath_langchain-0.0.78}/uipath_langchain/tracers/UiPathTracer.py

@@ -11,6 +11,8 @@ from langchain_core.tracers.base import BaseTracer
 from langchain_core.tracers.schemas import Run
 from pydantic import PydanticDeprecationWarning
 
+from ._utils import _simple_serialize_defaults
+
 logger = logging.getLogger(__name__)
 
 
@@ -142,7 +144,7 @@ class UiPathTracer(BaseTracer):
 
     def _safe_json_dump(self, obj) -> str:
         try:
-            json_str = json.dumps(obj, default=str)
+            json_str = json.dumps(obj, default=_simple_serialize_defaults)
             return json_str
         except Exception as e:
             logger.warning(e)

uipath_langchain-0.0.78/uipath_langchain/tracers/_utils.py

@@ -0,0 +1,24 @@
+import datetime
+from zoneinfo import ZoneInfo
+
+
+def _simple_serialize_defaults(obj):
+    if hasattr(obj, "model_dump"):
+        return obj.model_dump(exclude_none=True, mode="json")
+    if hasattr(obj, "dict"):
+        return obj.dict()
+    if hasattr(obj, "to_dict"):
+        return obj.to_dict()
+
+    if isinstance(obj, (set, tuple)):
+        if hasattr(obj, "_asdict") and callable(obj._asdict):
+            return obj._asdict()
+        return list(obj)
+
+    if isinstance(obj, datetime.datetime):
+        return obj.isoformat()
+
+    if isinstance(obj, (datetime.timezone, ZoneInfo)):
+        return obj.tzname(None)
+
+    return str(obj)