@@ -634,7 +634,7 @@ def to_streamed_response_wrapper(func: Callable[P, R]) -> Callable[P, ResponseCo
634
634
635
635
@functools .wraps (func )
636
636
def wrapped (* args : P .args , ** kwargs : P .kwargs ) -> ResponseContextManager [APIResponse [R ]]:
637
- extra_headers = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
637
+ extra_headers : dict [ str , str ] = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
638
638
extra_headers [RAW_RESPONSE_HEADER ] = "stream"
639
639
640
640
kwargs ["extra_headers" ] = extra_headers
@@ -655,7 +655,7 @@ def async_to_streamed_response_wrapper(
655
655
656
656
@functools .wraps (func )
657
657
def wrapped (* args : P .args , ** kwargs : P .kwargs ) -> AsyncResponseContextManager [AsyncAPIResponse [R ]]:
658
- extra_headers = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
658
+ extra_headers : dict [ str , str ] = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
659
659
extra_headers [RAW_RESPONSE_HEADER ] = "stream"
660
660
661
661
kwargs ["extra_headers" ] = extra_headers
@@ -679,7 +679,7 @@ def to_custom_streamed_response_wrapper(
679
679
680
680
@functools .wraps (func )
681
681
def wrapped (* args : P .args , ** kwargs : P .kwargs ) -> ResponseContextManager [_APIResponseT ]:
682
- extra_headers = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
682
+ extra_headers : dict [ str , Any ] = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
683
683
extra_headers [RAW_RESPONSE_HEADER ] = "stream"
684
684
extra_headers [OVERRIDE_CAST_TO_HEADER ] = response_cls
685
685
@@ -704,7 +704,7 @@ def async_to_custom_streamed_response_wrapper(
704
704
705
705
@functools .wraps (func )
706
706
def wrapped (* args : P .args , ** kwargs : P .kwargs ) -> AsyncResponseContextManager [_AsyncAPIResponseT ]:
707
- extra_headers = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
707
+ extra_headers : dict [ str , Any ] = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
708
708
extra_headers [RAW_RESPONSE_HEADER ] = "stream"
709
709
extra_headers [OVERRIDE_CAST_TO_HEADER ] = response_cls
710
710
@@ -724,7 +724,7 @@ def to_raw_response_wrapper(func: Callable[P, R]) -> Callable[P, APIResponse[R]]
724
724
725
725
@functools .wraps (func )
726
726
def wrapped (* args : P .args , ** kwargs : P .kwargs ) -> APIResponse [R ]:
727
- extra_headers = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
727
+ extra_headers : dict [ str , str ] = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
728
728
extra_headers [RAW_RESPONSE_HEADER ] = "raw"
729
729
730
730
kwargs ["extra_headers" ] = extra_headers
@@ -741,7 +741,7 @@ def async_to_raw_response_wrapper(func: Callable[P, Awaitable[R]]) -> Callable[P
741
741
742
742
@functools .wraps (func )
743
743
async def wrapped (* args : P .args , ** kwargs : P .kwargs ) -> AsyncAPIResponse [R ]:
744
- extra_headers = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
744
+ extra_headers : dict [ str , str ] = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
745
745
extra_headers [RAW_RESPONSE_HEADER ] = "raw"
746
746
747
747
kwargs ["extra_headers" ] = extra_headers
@@ -763,7 +763,7 @@ def to_custom_raw_response_wrapper(
763
763
764
764
@functools .wraps (func )
765
765
def wrapped (* args : P .args , ** kwargs : P .kwargs ) -> _APIResponseT :
766
- extra_headers = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
766
+ extra_headers : dict [ str , Any ] = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
767
767
extra_headers [RAW_RESPONSE_HEADER ] = "raw"
768
768
extra_headers [OVERRIDE_CAST_TO_HEADER ] = response_cls
769
769
@@ -786,7 +786,7 @@ def async_to_custom_raw_response_wrapper(
786
786
787
787
@functools .wraps (func )
788
788
def wrapped (* args : P .args , ** kwargs : P .kwargs ) -> Awaitable [_AsyncAPIResponseT ]:
789
- extra_headers = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
789
+ extra_headers : dict [ str , Any ] = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
790
790
extra_headers [RAW_RESPONSE_HEADER ] = "raw"
791
791
extra_headers [OVERRIDE_CAST_TO_HEADER ] = response_cls
792
792
0 commit comments