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

test: fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-27 10:46:57 -07:00
parent cd7dd2a511
commit 18b67a455e
3 changed files with 21 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

14
litellm/llms/openai.py
View file

@ -770,7 +770,7 @@ class OpenAIChatCompletion(BaseLLM):
openai_aclient: AsyncOpenAI,
data: dict,
timeout: Union[float, httpx.Timeout],
):
) -> Tuple[dict, BaseModel]:
"""
Helper to:
- call chat.completions.create.with_raw_response when litellm.return_response_headers is True
@ -783,7 +783,10 @@ class OpenAIChatCompletion(BaseLLM):
)
)
headers = dict(raw_response.headers)
if hasattr(raw_response, "headers"):
headers = dict(raw_response.headers)
else:
headers = {}
response = raw_response.parse()
return headers, response
except OpenAIError as e:
@ -800,7 +803,7 @@ class OpenAIChatCompletion(BaseLLM):
openai_client: OpenAI,
data: dict,
timeout: Union[float, httpx.Timeout],
):
) -> Tuple[dict, BaseModel]:
"""
Helper to:
- call chat.completions.create.with_raw_response when litellm.return_response_headers is True
@ -811,7 +814,10 @@ class OpenAIChatCompletion(BaseLLM):
**data, timeout=timeout
)
headers = dict(raw_response.headers)
if hasattr(raw_response, "headers"):
headers = dict(raw_response.headers)
else:
headers = {}
response = raw_response.parse()
return headers, response
except OpenAIError as e: