From 5262a5b56475c7b05ee897925401d1f4e0ad2ae2 Mon Sep 17 00:00:00 2001 From: Diwank Singh Tomer Date: Wed, 16 Oct 2024 15:21:17 -0400 Subject: [PATCH] hotfix: raise_complete_async bad beartype annotation Signed-off-by: Diwank Singh Tomer --- .../activities/task_steps/raise_complete_async.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/agents-api/agents_api/activities/task_steps/raise_complete_async.py b/agents-api/agents_api/activities/task_steps/raise_complete_async.py index 904b298d..adae15b7 100644 --- a/agents-api/agents_api/activities/task_steps/raise_complete_async.py +++ b/agents-api/agents_api/activities/task_steps/raise_complete_async.py @@ -1,13 +1,11 @@ import base64 +from typing import Any from beartype import beartype from temporalio import activity from ...autogen.openapi_model import CreateTransitionRequest -from ...common.protocol.tasks import ( - StepContext, - StepOutcome, -) +from ...common.protocol.tasks import StepContext from ...common.storage_handler import auto_blob_store from .transition_step import original_transition_step @@ -15,7 +13,7 @@ @activity.defn @auto_blob_store @beartype -async def raise_complete_async(context: StepContext, output: StepOutcome) -> None: +async def raise_complete_async(context: StepContext, output: Any) -> None: activity_info = activity.info() captured_token = base64.b64encode(activity_info.task_token).decode("ascii")