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

Refactor: apply early return (#6369)

Browse source
This commit is contained in:
Hakan Taşköprü 2024-10-22 14:11:54 +03:00 • committed by GitHub
parent a5fe87d6a5
commit a0c5fee61d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 22 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/litellm_core_utils/llm_response_utils/get_headers.py
View file

@ -15,7 +15,9 @@ def get_response_headers(_response_headers: Optional[dict] = None) -> dict:
dict: _response_headers with OpenAI headers and llm_provider-{header} dict: _response_headers with OpenAI headers and llm_provider-{header}
""" """
if _response_headers is not None: if _response_headers is None:
return {}
openai_headers = {} openai_headers = {}
if "x-ratelimit-limit-requests" in _response_headers: if "x-ratelimit-limit-requests" in _response_headers:
openai_headers["x-ratelimit-limit-requests"] = _response_headers[ openai_headers["x-ratelimit-limit-requests"] = _response_headers[
@ -35,7 +37,6 @@ def get_response_headers(_response_headers: Optional[dict] = None) -> dict:
] ]
llm_provider_headers = _get_llm_provider_headers(_response_headers) llm_provider_headers = _get_llm_provider_headers(_response_headers)
return {**llm_provider_headers, **openai_headers} return {**llm_provider_headers, **openai_headers}
return {}
def _get_llm_provider_headers(response_headers: dict) -> dict: def _get_llm_provider_headers(response_headers: dict) -> dict: