@@ -632,7 +632,7 @@ def to_streamed_response_wrapper(func: Callable[P, R]) -> Callable[P, ResponseCo
632
632
633
633
@functools .wraps (func )
634
634
def wrapped (* args : P .args , ** kwargs : P .kwargs ) -> ResponseContextManager [APIResponse [R ]]:
635
- extra_headers = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
635
+ extra_headers : dict [ str , str ] = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
636
636
extra_headers [RAW_RESPONSE_HEADER ] = "stream"
637
637
638
638
kwargs ["extra_headers" ] = extra_headers
@@ -653,7 +653,7 @@ def async_to_streamed_response_wrapper(
653
653
654
654
@functools .wraps (func )
655
655
def wrapped (* args : P .args , ** kwargs : P .kwargs ) -> AsyncResponseContextManager [AsyncAPIResponse [R ]]:
656
- extra_headers = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
656
+ extra_headers : dict [ str , str ] = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
657
657
extra_headers [RAW_RESPONSE_HEADER ] = "stream"
658
658
659
659
kwargs ["extra_headers" ] = extra_headers
@@ -677,7 +677,7 @@ def to_custom_streamed_response_wrapper(
677
677
678
678
@functools .wraps (func )
679
679
def wrapped (* args : P .args , ** kwargs : P .kwargs ) -> ResponseContextManager [_APIResponseT ]:
680
- extra_headers = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
680
+ extra_headers : dict [ str , Any ] = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
681
681
extra_headers [RAW_RESPONSE_HEADER ] = "stream"
682
682
extra_headers [OVERRIDE_CAST_TO_HEADER ] = response_cls
683
683
@@ -702,7 +702,7 @@ def async_to_custom_streamed_response_wrapper(
702
702
703
703
@functools .wraps (func )
704
704
def wrapped (* args : P .args , ** kwargs : P .kwargs ) -> AsyncResponseContextManager [_AsyncAPIResponseT ]:
705
- extra_headers = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
705
+ extra_headers : dict [ str , Any ] = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
706
706
extra_headers [RAW_RESPONSE_HEADER ] = "stream"
707
707
extra_headers [OVERRIDE_CAST_TO_HEADER ] = response_cls
708
708
@@ -722,7 +722,7 @@ def to_raw_response_wrapper(func: Callable[P, R]) -> Callable[P, APIResponse[R]]
722
722
723
723
@functools .wraps (func )
724
724
def wrapped (* args : P .args , ** kwargs : P .kwargs ) -> APIResponse [R ]:
725
- extra_headers = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
725
+ extra_headers : dict [ str , str ] = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
726
726
extra_headers [RAW_RESPONSE_HEADER ] = "raw"
727
727
728
728
kwargs ["extra_headers" ] = extra_headers
@@ -739,7 +739,7 @@ def async_to_raw_response_wrapper(func: Callable[P, Awaitable[R]]) -> Callable[P
739
739
740
740
@functools .wraps (func )
741
741
async def wrapped (* args : P .args , ** kwargs : P .kwargs ) -> AsyncAPIResponse [R ]:
742
- extra_headers = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
742
+ extra_headers : dict [ str , str ] = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
743
743
extra_headers [RAW_RESPONSE_HEADER ] = "raw"
744
744
745
745
kwargs ["extra_headers" ] = extra_headers
@@ -761,7 +761,7 @@ def to_custom_raw_response_wrapper(
761
761
762
762
@functools .wraps (func )
763
763
def wrapped (* args : P .args , ** kwargs : P .kwargs ) -> _APIResponseT :
764
- extra_headers = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
764
+ extra_headers : dict [ str , Any ] = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
765
765
extra_headers [RAW_RESPONSE_HEADER ] = "raw"
766
766
extra_headers [OVERRIDE_CAST_TO_HEADER ] = response_cls
767
767
@@ -784,7 +784,7 @@ def async_to_custom_raw_response_wrapper(
784
784
785
785
@functools .wraps (func )
786
786
def wrapped (* args : P .args , ** kwargs : P .kwargs ) -> Awaitable [_AsyncAPIResponseT ]:
787
- extra_headers = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
787
+ extra_headers : dict [ str , Any ] = {** (cast (Any , kwargs .get ("extra_headers" )) or {})}
788
788
extra_headers [RAW_RESPONSE_HEADER ] = "raw"
789
789
extra_headers [OVERRIDE_CAST_TO_HEADER ] = response_cls
790
790
0 commit comments