return azure response headers

This commit is contained in:
Ishaan Jaff 2024-07-01 17:09:06 -07:00
parent 48946f7528
commit 140f7fe254
2 changed files with 46 additions and 4 deletions

View file

@ -658,6 +658,11 @@ class OpenAIChatCompletion(BaseLLM):
data: dict,
timeout: Union[float, httpx.Timeout],
):
"""
Helper to:
- call chat.completions.create.with_raw_response when litellm.return_response_headers is True
- call chat.completions.create by default
"""
try:
if litellm.return_response_headers is True:
raw_response = (