GRPC client_interceptor does not allow the user to cancel streamed responses

[bkrmalick]

Environment
Python 3.9
opentelemetry-sdk 1.20.0
opentelemetry-instrumentation 0.41b0
opentelemetry-instrumentation-grpc 0.41b0
grpcio = "==1.48.2 || ==1.49.1"

Steps to reproduce

import grpc
from opentelemetry.instrumentation.grpc import client_interceptor

chan = grpc.insecure_channel(MY_GRPC_SERVICE_ADDRESS)
chan = intercept_channel(chan, client_interceptor()))

response = chan.MyStreamRPC(....)  # Server streamed RPC where the client sends a request to the server and gets a stream to read a sequence of messages back.

for idx, item in enumerate(response):
      print(item)
      if idx == 4:
           response.cancel() # this works with standard grpc but throws an attribute error when the otel client_interceptor is added above
      
chan.close()

What is the expected behavior?
Being able to cancel the streamed response client side part-way

What is the actual behavior?
Throws an attribute error

Additional context
The interceptor seems to be modifying the return type for streamed rpc responses to be of type of standard generator

[samuelchan94]

thank you @aidandj for picking it up! is there a timeline on when this will be rolled out? this bug is pretty critical for adopting grpc instrumentation on my end.

[aidandj]

This what I'm doing for now

grpc_client_instrumentor = GrpcInstrumentorClient(
    filter_=lambda call_info: not isinstance(call_info, _StreamClientInfo)
    or not call_info.is_server_stream
)
grpc_client_instrumentor.instrument()