diff --git a/src/agentex/lib/core/temporal/workflows/workflow.py b/src/agentex/lib/core/temporal/workflows/workflow.py index 727f3ac85..7ef0c7a55 100644 --- a/src/agentex/lib/core/temporal/workflows/workflow.py +++ b/src/agentex/lib/core/temporal/workflows/workflow.py @@ -16,6 +16,22 @@ def __init__( ): self.display_name = display_name + @workflow.query(name="get_current_state") + def get_current_state(self) -> str: + """Query handler for the current workflow state. + + Returns "unknown" by default. Subclasses should override this + to return their actual state, enabling external callers to + detect turn completion. + + Example override for StateMachine-based agents: + + @workflow.query(name="get_current_state") + def get_current_state(self) -> str: + return self.state_machine.get_current_state() + """ + return "unknown" + @abstractmethod @workflow.signal(name=SignalName.RECEIVE_EVENT) async def on_task_event_send(self, params: SendEventParams) -> None: