fix: added retry mehanism to integration tools

Author: gauravsingh-rajput

src/uipath_langchain/agent/tools/integration_tool.py

@@ -6,12 +6,18 @@
 from langchain.tools import BaseTool
 from langchain_core.messages import ToolCall
 from langchain_core.tools import StructuredTool
+from tenacity import (
+    retry,
+    retry_if_exception,
+    stop_after_attempt,
+    wait_exponential_jitter,
+)
 from uipath.agent.models.agent import AgentIntegrationToolResourceConfig
 from uipath.eval.mocks import mockable
 from uipath.platform import UiPath
 from uipath.platform.connections import ActivityMetadata, ActivityParameterLocationInfo
+from uipath.platform.errors import EnrichedException
 from uipath.runtime.errors import UiPathErrorCategory
-
 from uipath_langchain.agent.exceptions import AgentStartupError, AgentStartupErrorCode
 from uipath_langchain.agent.react.jsonschema_pydantic_converter import create_model
 from uipath_langchain.agent.react.types import AgentGraphState
@@ -165,6 +171,18 @@ def create_integration_tool(
 
     sdk = UiPath()
 
+    def _is_429(exc: BaseException) -> bool:
+        return (
+            isinstance(exc, EnrichedException)
+            and getattr(exc, "status_code", None) == 429
+        )
+
+    @retry(
+        retry=retry_if_exception(_is_429),
+        wait=wait_exponential_jitter(initial=10, max=120, jitter=5),
+        stop=stop_after_attempt(5),
+        reraise=True,
+    )
     @mockable(
         name=resource.name,
         description=resource.description,
@@ -173,15 +191,11 @@ def create_integration_tool(
         example_calls=resource.properties.example_calls,
     )
     async def integration_tool_fn(**kwargs: Any):
-        try:
-            result = await sdk.connections.invoke_activity_async(
-                activity_metadata=activity_metadata,
-                connection_id=connection_id,
-                activity_input=sanitize_dict_for_serialization(kwargs),
-            )
-        except Exception:
-            raise
-
+        result = await sdk.connections.invoke_activity_async(
+            activity_metadata=activity_metadata,
+            connection_id=connection_id,
+            activity_input=sanitize_dict_for_serialization(kwargs),
+        )
         return result
 
     async def integration_tool_wrapper(