Skip to content
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.

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

feat(agents-api): Add retry policies to temporal workflows/activities #551

Merged
merged 12 commits into from
Oct 5, 2024
Prev Previous commit
Next Next commit
Bug fix: Remove test/debug situations
  • Loading branch information
HamadaSalhab committed Oct 3, 2024
commit 2d945d3934c154487211cc8d50d5e8f532987e01
7 changes: 2 additions & 5 deletions agents-api/agents_api/common/retry_policies.py
Original file line number Diff line number Diff line change
@@ -1,15 +1,12 @@
from datetime import timedelta

from temporalio.common import RetryPolicy
from ..env import debug, testing

DEFAULT_RETRY_POLICY = RetryPolicy(
initial_interval=timedelta(seconds=1),
backoff_coefficient=2,
maximum_attempts=2 if debug or testing else 25,
maximum_interval=timedelta(seconds=10)
if debug or testing
else timedelta(seconds=300),
maximum_attempts=25,
maximum_interval=timedelta(seconds=300),
non_retryable_error_types=[
# Temporal-specific errors
"WorkflowExecutionAlreadyStarted",
Expand Down
3 changes: 2 additions & 1 deletion agents-api/agents_api/workflows/task_execution/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@
from temporalio import workflow
from temporalio.exceptions import ApplicationError

from ...common.retry_policies import DEFAULT_RETRY_POLICY

# Import necessary modules and types
with workflow.unsafe.imports_passed_through():
Expand Down Expand Up @@ -55,6 +54,8 @@
execute_switch_branch,
)
from .transition import transition
from ...common.retry_policies import DEFAULT_RETRY_POLICY


# Supported steps
# ---------------
Expand Down
Loading