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

rename to _response_headers

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-20 17:31:16 -07:00
parent 5e52f50a82
commit 5e4d291244
4 changed files with 35 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/llms/openai.py
View file

@ -960,7 +960,7 @@ class OpenAIChatCompletion(BaseLLM):
return convert_to_model_response_object(
response_object=stringified_response,
model_response_object=model_response,
response_headers=headers,
_response_headers=headers,
)
except Exception as e:
if print_verbose is not None:
@ -1059,7 +1059,7 @@ class OpenAIChatCompletion(BaseLLM):
response_object=stringified_response,
model_response_object=model_response,
hidden_params={"headers": headers},
response_headers=headers,
_response_headers=headers,
)
except Exception as e:
raise e
@ -1110,7 +1110,7 @@ class OpenAIChatCompletion(BaseLLM):
custom_llm_provider="openai",
logging_obj=logging_obj,
stream_options=data.get("stream_options", None),
response_headers=headers,
_response_headers=headers,
)
return streamwrapper
@ -1160,7 +1160,7 @@ class OpenAIChatCompletion(BaseLLM):
custom_llm_provider="openai",
logging_obj=logging_obj,
stream_options=data.get("stream_options", None),
response_headers=headers,
_response_headers=headers,
)
return streamwrapper
except (
@ -1269,7 +1269,7 @@ class OpenAIChatCompletion(BaseLLM):
response_object=stringified_response,
model_response_object=model_response,
response_type="embedding",
response_headers=headers,
_response_headers=headers,
) # type: ignore
except Exception as e:
## LOGGING
@ -1348,7 +1348,7 @@ class OpenAIChatCompletion(BaseLLM):
return convert_to_model_response_object(
response_object=sync_embedding_response.model_dump(),
model_response_object=model_response,
response_headers=headers,
_response_headers=headers,
response_type="embedding",
) # type: ignore
except OpenAIError as e: