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

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

@ -15,27 +15,28 @@ 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:
openai_headers = {} return {}
if "x-ratelimit-limit-requests" in _response_headers:
openai_headers["x-ratelimit-limit-requests"] = _response_headers[ openai_headers = {}
"x-ratelimit-limit-requests" if "x-ratelimit-limit-requests" in _response_headers:
] openai_headers["x-ratelimit-limit-requests"] = _response_headers[
if "x-ratelimit-remaining-requests" in _response_headers: "x-ratelimit-limit-requests"
openai_headers["x-ratelimit-remaining-requests"] = _response_headers[ ]
"x-ratelimit-remaining-requests" if "x-ratelimit-remaining-requests" in _response_headers:
] openai_headers["x-ratelimit-remaining-requests"] = _response_headers[
if "x-ratelimit-limit-tokens" in _response_headers: "x-ratelimit-remaining-requests"
openai_headers["x-ratelimit-limit-tokens"] = _response_headers[ ]
"x-ratelimit-limit-tokens" if "x-ratelimit-limit-tokens" in _response_headers:
] openai_headers["x-ratelimit-limit-tokens"] = _response_headers[
if "x-ratelimit-remaining-tokens" in _response_headers: "x-ratelimit-limit-tokens"
openai_headers["x-ratelimit-remaining-tokens"] = _response_headers[ ]
"x-ratelimit-remaining-tokens" if "x-ratelimit-remaining-tokens" in _response_headers:
] openai_headers["x-ratelimit-remaining-tokens"] = _response_headers[
llm_provider_headers = _get_llm_provider_headers(_response_headers) "x-ratelimit-remaining-tokens"
return {**llm_provider_headers, **openai_headers} ]
return {} llm_provider_headers = _get_llm_provider_headers(_response_headers)
return {**llm_provider_headers, **openai_headers}
def _get_llm_provider_headers(response_headers: dict) -> dict: def _get_llm_provider_headers(response_headers: dict) -> dict: