Preserve transport error identities

2026-08-11 23:28:00 +00:00
parent 350b0e76d9
commit c281f721bc
5 changed files with 171 additions and 20 deletions
--- a/engine_test.go
+++ b/engine_test.go
@@ -737,6 +737,9 @@ func TestEngineRunPropagatesCallerCancellation(t *testing.T) {
 		if !errors.Is(err, promptkit.ErrLLMGenerate) {
 			t.Fatalf("expected ErrLLMGenerate after caller cancellation, got %v", err)
 		}
+		if !errors.Is(err, context.Canceled) {
+			t.Fatalf("expected context cancellation identity, got %v", err)
+		}
 	case <-watchdog.C:
 		t.Fatal("timed out waiting for Engine.Run to return after cancellation")
 	}