diff --git a/litellm/types/utils.py b/litellm/types/utils.py index 4747a9a87..55e5335f3 100644 --- a/litellm/types/utils.py +++ b/litellm/types/utils.py @@ -536,6 +536,8 @@ class ModelResponse(OpenAIObject): _hidden_params: dict = {} + response_headers: Optional[dict] = None + def __init__( self, id=None, @@ -549,6 +551,7 @@ class ModelResponse(OpenAIObject): stream_options=None, response_ms=None, hidden_params=None, + response_headers=None, **params, ) -> None: if stream is not None and stream is True: @@ -598,6 +601,9 @@ class ModelResponse(OpenAIObject): if hidden_params: self._hidden_params = hidden_params + if response_headers: + self.response_headers = response_headers + init_values = { "id": id, "choices": choices, @@ -668,6 +674,8 @@ class EmbeddingResponse(OpenAIObject): _hidden_params: dict = {} + response_headers: Optional[dict] = None + def __init__( self, model=None, @@ -675,6 +683,8 @@ class EmbeddingResponse(OpenAIObject): stream=False, response_ms=None, data=None, + hidden_params=None, + response_headers=None, **params, ): object = "list" @@ -974,6 +984,7 @@ class TranscriptionResponse(OpenAIObject): text: Optional[str] = None _hidden_params: dict = {} + response_headers: Optional[dict] = None def __init__(self, text=None): super().__init__(text=text)