Skip to content

fix : integration tool retry mechanism #645

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
gauravsingh-rajput wants to merge 2 commits into
base: main
Choose a base branch
from
Draft

fix : integration tool retry mechanism #645

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
8863e33
fix: added retry mehanism to integration tools
gauravsingh-rajput Feb 26, 2026
51a6110
fix: fixed lint error
gauravsingh-rajput Feb 27, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
33 changes: 24 additions & 9 deletions src/uipath_langchain/agent/tools/integration_tool.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,10 +6,17 @@
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
Expand Down Expand Up @@ -165,6 +172,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,
Expand All @@ -173,15 +192,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(
Expand Down