diff --git a/llms/openai/internal/openaiclient/openaiclient.go b/llms/openai/internal/openaiclient/openaiclient.go index e0a09d1fb..a6dd25f18 100644 --- a/llms/openai/internal/openaiclient/openaiclient.go +++ b/llms/openai/internal/openaiclient/openaiclient.go @@ -225,7 +225,7 @@ func sanitizeHTTPError(err error) error { // Check for context cancellation if errors.Is(err, context.Canceled) { - return errors.New("request cancelled") + return context.Canceled } // Check for network timeout errors diff --git a/llms/openai/internal/openaiclient/openaiclient_test.go b/llms/openai/internal/openaiclient/openaiclient_test.go index a9d0f4140..be7513da5 100644 --- a/llms/openai/internal/openaiclient/openaiclient_test.go +++ b/llms/openai/internal/openaiclient/openaiclient_test.go @@ -374,8 +374,7 @@ func TestSanitizeHTTPError(t *testing.T) { t.Run("context cancelled", func(t *testing.T) { err := context.Canceled sanitized := sanitizeHTTPError(err) - assert.Error(t, sanitized) - assert.Equal(t, "request cancelled", sanitized.Error()) + assert.ErrorIs(t, sanitized, context.Canceled) }) t.Run("network timeout", func(t *testing.T) {