fix: add union of return types for streaming or non streaming call

fix: add union of return types for streaming or non streaming call #337