Preserve parent context in agent loop timeout

Use the caller-provided context when creating the agent loop inactivity timeout instead of oc.backgroundContext(ctx), so parent cancellations propagate to the derived context. Add TestWithAgentLoopInactivityTimeoutPreservesParentCancellation to verify the derived context cancels and reports context.Canceled when the parent is cancelled. Changes in bridges/ai/agent_loop_runtime.go and bridges/ai/agent_loop_test.go.

Author: batuhan

bridges/ai/agent_loop_runtime.go

@@ -82,7 +82,7 @@ func agentLoopInactivityCause(ctx context.Context) error {
 }
 
 func (oc *AIClient) withAgentLoopInactivityTimeout(ctx context.Context) (context.Context, context.CancelFunc) {
-	runCtx, touch, cancel := withActivityTimeout(oc.backgroundContext(ctx), agentLoopInactivityTimeout, errAgentLoopInactivityTimeout)
+	runCtx, touch, cancel := withActivityTimeout(ctx, agentLoopInactivityTimeout, errAgentLoopInactivityTimeout)
 	return withAgentLoopActivity(runCtx, touch), cancel
 }
 

bridges/ai/agent_loop_test.go

@@ -157,3 +157,23 @@ func TestWithActivityTimeoutResetsOnTouchAndCancelsOnIdle(t *testing.T) {
 		t.Fatalf("expected inactivity timeout cause, got %v", context.Cause(ctx))
 	}
 }
+
+func TestWithAgentLoopInactivityTimeoutPreservesParentCancellation(t *testing.T) {
+	parent, parentCancel := context.WithCancel(context.Background())
+	client := &AIClient{}
+
+	ctx, cancel := client.withAgentLoopInactivityTimeout(parent)
+	defer cancel()
+
+	parentCancel()
+
+	select {
+	case <-ctx.Done():
+	case <-time.After(100 * time.Millisecond):
+		t.Fatal("expected derived context to cancel when parent is cancelled")
+	}
+
+	if !errors.Is(context.Cause(ctx), context.Canceled) {
+		t.Fatalf("expected cancelled cause from parent, got %v", context.Cause(ctx))
+	}
+}