phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

return response_headers in response

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-20 14:58:14 -07:00
parent 64dbe07593
commit ca8012090c
1 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/types/utils.py
View file

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