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
refactor: Lint agents-api (CI)
  • Loading branch information
creatorrr authored and github-actions[bot] committed Oct 5, 2024
commit 2cbc451e9680fc539e4a3436945c4bb2324e402c
3 changes: 1 addition & 2 deletions 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


# Import necessary modules and types
with workflow.unsafe.imports_passed_through():
from ...activities import task_steps
Expand Down Expand Up @@ -48,6 +47,7 @@
StepContext,
StepOutcome,
)
from ...common.retry_policies import DEFAULT_RETRY_POLICY
from ...env import debug, testing
from .helpers import (
continue_as_child,
Expand All @@ -58,7 +58,6 @@
execute_switch_branch,
)
from .transition import transition
from ...common.retry_policies import DEFAULT_RETRY_POLICY


# Supported steps
Expand Down
1 change: 0 additions & 1 deletion agents-api/tests/test_activities.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,6 @@
from agents_api.clients import temporal
from agents_api.env import temporal_task_queue
from agents_api.workflows.demo import DemoWorkflow

from agents_api.workflows.task_execution.helpers import DEFAULT_RETRY_POLICY

from .fixtures import (
Expand Down