feat(status_interrupt): determine status based on err info

Author: giuliastf

pyproject.toml

@@ -1,35 +1,35 @@
 [project]
 name = "uipath-langchain"
-version = "0.0.136"
+version = "0.0.137"
 description = "UiPath Langchain"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
 dependencies = [
-    "uipath>=2.1.65, <2.2.0",
-    "langgraph>=0.5.0, <0.7.0",
-    "langchain-core>=0.3.34",
-    "langgraph-checkpoint-sqlite>=2.0.3",
-    "langchain-community>=0.3.21",
-    "langchain-openai>=0.3.3",
-    "langchain>=0.3.4",
-    "pydantic-settings>=2.6.0",
-    "python-dotenv>=1.0.1",
-    "httpx>=0.27.0",
-    "openai>=1.65.5",
-    "openinference-instrumentation-langchain>=0.1.50",
-    "jsonschema-pydantic>=0.6",
+  "uipath>=2.1.65, <2.2.0",
+  "langgraph>=0.5.0, <0.7.0",
+  "langchain-core>=0.3.34",
+  "langgraph-checkpoint-sqlite>=2.0.3",
+  "langchain-community>=0.3.21",
+  "langchain-openai>=0.3.3",
+  "langchain>=0.3.4",
+  "pydantic-settings>=2.6.0",
+  "python-dotenv>=1.0.1",
+  "httpx>=0.27.0",
+  "openai>=1.65.5",
+  "openinference-instrumentation-langchain>=0.1.50",
+  "jsonschema-pydantic>=0.6",
 ]
 classifiers = [
-    "Development Status :: 3 - Alpha",
-    "Intended Audience :: Developers",
-    "Topic :: Software Development :: Build Tools",
-    "Programming Language :: Python :: 3.10",
-    "Programming Language :: Python :: 3.11",
-    "Programming Language :: Python :: 3.12",
+  "Development Status :: 3 - Alpha",
+  "Intended Audience :: Developers",
+  "Topic :: Software Development :: Build Tools",
+  "Programming Language :: Python :: 3.10",
+  "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
 ]
 maintainers = [
-    { name = "Marius Cosareanu", email = "marius.cosareanu@uipath.com" },
-    { name = "Cristian Pufu", email = "cristian.pufu@uipath.com" }
+  { name = "Marius Cosareanu", email = "marius.cosareanu@uipath.com" },
+  { name = "Cristian Pufu", email = "cristian.pufu@uipath.com" },
 ]
 
 [project.entry-points."uipath.middlewares"]
@@ -46,21 +46,19 @@ build-backend = "hatchling.build"
 
 [dependency-groups]
 dev = [
-    "mypy>=1.14.1",
-    "ruff>=0.9.4",
-    "pytest>=7.4.0",
-    "pytest-cov>=4.1.0",
-    "pytest-mock>=3.11.1",
-    "pytest-asyncio>=1.0.0",
-    "pre-commit>=4.1.0",
-    "numpy>=1.24.0",
-    "pytest_httpx>=0.35.0"
+  "mypy>=1.14.1",
+  "ruff>=0.9.4",
+  "pytest>=7.4.0",
+  "pytest-cov>=4.1.0",
+  "pytest-mock>=3.11.1",
+  "pytest-asyncio>=1.0.0",
+  "pre-commit>=4.1.0",
+  "numpy>=1.24.0",
+  "pytest_httpx>=0.35.0",
 ]
 
 [project.optional-dependencies]
-langchain = [
-  "uipath-langchain>=0.0.2"
-]
+langchain = ["uipath-langchain>=0.0.2"]
 
 [tool.hatch.build.targets.wheel]
 packages = ["src/uipath_langchain"]
@@ -83,13 +81,8 @@ skip-magic-trailing-comma = false
 line-ending = "auto"
 
 [tool.mypy]
-plugins = [
-  "pydantic.mypy"
-]
-exclude = [
-    "samples/.*",
-    "testcases/.*"
-]
+plugins = ["pydantic.mypy"]
+exclude = ["samples/.*", "testcases/.*"]
 
 follow_imports = "silent"
 warn_redundant_casts = true

src/uipath_langchain/_tracing/_oteladapter.py

@@ -1,10 +1,8 @@
 import json
 import logging
-from typing import Any, Dict, List
+from typing import Any, Dict, List, Optional
 
-from opentelemetry.sdk.trace.export import (
-    SpanExportResult,
-)
+from opentelemetry.sdk.trace.export import SpanExportResult
 from uipath.tracing import LlmOpsHttpExporter
 
 logger = logging.getLogger(__name__)
@@ -71,6 +69,11 @@ class LangChainExporter(LlmOpsHttpExporter):
         # Add more mappings as needed
     }
 
+    class Status:
+        SUCCESS = 1
+        ERROR = 2
+        INTERRUPTED = 3
+
     def __init__(self, *args: Any, **kwargs: Any) -> None:
         super().__init__(*args, **kwargs)
 
@@ -161,6 +164,15 @@ def _map_tool_call_attributes(self, attributes: Dict[str, Any]) -> Dict[str, Any
 
         return result
 
+    def _determine_status(self, error: Optional[str]) -> int:
+        if error:
+            logger.debug(f"Determining status from error: {error}")
+            if error and error.startswith("GraphInterrupt("):
+                logger.debug(f"Status determined: {self.Status.INTERRUPTED}")
+                return self.Status.INTERRUPTED
+            return self.Status.ERROR
+        return self.Status.SUCCESS
+
     def _process_span_attributes(self, span_data: Dict[str, Any]) -> Dict[str, Any]:
         """Extracts, transforms, and maps attributes for a span."""
         if "Attributes" not in span_data:
@@ -206,6 +218,13 @@ def _process_span_attributes(self, span_data: Dict[str, Any]) -> Dict[str, Any]:
 
         span_data["Attributes"] = json.dumps(processed_attributes)
 
+        # Determine status based on error information
+        error = attributes.get("error") or attributes.get("exception.message")
+        logger.debug(f"Span error info: {error}")
+        status = self._determine_status(error)
+        logger.debug(f"Determined status: {status}")
+        span_data["Status"] = status
+
         logger.debug(f"Transformed span: {span_data}")
         return span_data