--- tests --- TestWithCancel_BeforeExecutionStarts verifies that a cancel issued before the graph begins executing still produces a CancelError withou

TestWithCancel_CancelImmediate_NestedAgentTool_ResumeFromToolsNode verifies that when a nested ChatModelAgent (wrapped as an AgentTool inside an outer

TestWithCancel_CancelImmediate_StreamAborted verifies that CancelImmediate during model execution surfaces CancelError and completes quickly. Uses blo

TestWithCancel_CancelImmediate_StreamableToolAborted verifies that CancelImmediate during StreamableTool streaming surfaces ErrStreamCanceled on the t

TestWithCancel_DerivedAgentToolCancelContextMarkedDoneAfterRun verifies that an explicitly derived AgentTool child cancel context is owned by the chil

TestWithCancel_GraphInterruptRaceBeforeSet verifies that a CancelImmediate issued before setGraphInterruptFunc is called still results in cancellation

TestWithCancel_ModelError verifies that a model error marks the cancelCtx as done so that a subsequent cancelFn call returns ErrExecutionEnded.

TestWithCancel_MultipleToolsConcurrent verifies that CancelAfterToolCalls waits for ALL concurrent tool calls to complete before cancelling.

TestWithCancel_NoCheckpointStore verifies cancel completes and does not panic when no checkpoint store is configured.

TestWithCancel_PlanExecute_BetweenTransitions verifies that cancel works when fired between agent transitions (e.g., after planner, before executor st

TestWithCancel_PlanExecute_DuringExecution verifies that cancel works during the executor (ChatModelAgent) phase of the PlanExecute agent.

TestWithCancel_Resume tests the workflow of Cancel followed by Resume. To avoid data races, we create new agent and runner instances for the Resume p

TestWithCancel_Resume_CancelAfterChatModel_MessagePreserved tests both the ReAct (with-tools) and noTools paths to ensure that when a CancelAfterChatM

TestWithCancel_Resume_SafePoint covers CancelAfterChatModel and CancelAfterToolCalls on a Resume path.

TestWithCancel_TargetedResume_CancelImmediate cancels an agent via CancelImmediate, extracts InterruptContexts from the resulting CancelError, and use

TestWithCancel_TargetedResume_SafePoint cancels an agent via CancelAfterChatModel (safe-point) and verifies that InterruptContexts are populated on th

TestWithCancel_TimeoutEscalation tests that WithAgentCancelTimeout causes the cancel to escalate to immediate when the safe-point hasn't fired yet, an

Add these test functions to the existing workflow_test.go file Replace the existing TestWorkflowAgentPanicRecovery function

TestWorkflowAgentUnsupportedMode tests unsupported workflow mode error (lines 65-71)

Transform mocks base method.

Transform indicates an expected call of Transform.

TrySend attempts to put an item into the channel. Returns false if the channel is closed, true otherwise.

WithAgentToolRunOptions specifies per-tool run options for the agent.

WithChatTemplateOption is a functional option type for chat template component.

WithDocumentTransformerOption is a functional option type for document transformer component.

WithEagerExecution enables the eager execution mode for the graph. In eager mode, nodes will be executed immediately once they are ready to run, witho

WithEagerExecutionDisabled disables the eager execution mode for the graph. By default, eager execution is enabled for Workflow and Graph with the All

WithExtraMeta attaches extra metadata to the parsed document.

WithHistoryModifier sets a function to modify history during resume. Deprecated: use ResumeWithData and ChatModelAgentResumeData instead.

WithHistoryRewriter sets a rewriter to transform conversation history during agent transfers. NOT RECOMMENDED: Agent transfer with full context shari

WithIndexerOption is a functional option type for indexer component. e.g. indexerOption := compose.WithIndexerOption(indexer.WithSubIndexes([]string

WithMarshalOutput wraps the marshal output option. when you want to marshal the output by yourself, you can use this option.

WithPreempt signals that the current agent turn should be cancelled at the specified safePoint after pushing the new item. The loop cancels the curren

WithPreemptDelay sets a delay duration before resolving a preemptive Push. When used with WithPreempt or WithPreemptTimeout, the pushed item is buffer