mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-05 08:07:05 +00:00
54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import Iterable
|
|
from typing import Final
|
|
|
|
HOP_BY_HOP_HEADERS: Final[frozenset[str]] = frozenset(
|
|
{
|
|
"connection",
|
|
"keep-alive",
|
|
"proxy-authenticate",
|
|
"proxy-authorization",
|
|
"te",
|
|
"trailer",
|
|
"trailers",
|
|
"transfer-encoding",
|
|
"upgrade",
|
|
}
|
|
)
|
|
|
|
REQUEST_DROPPED_HEADERS: Final[frozenset[str]] = HOP_BY_HOP_HEADERS | {
|
|
"host",
|
|
"content-length",
|
|
"accept-encoding",
|
|
}
|
|
|
|
RESPONSE_DROPPED_HEADERS: Final[frozenset[str]] = HOP_BY_HOP_HEADERS | {
|
|
"content-encoding",
|
|
"content-length",
|
|
"set-cookie",
|
|
}
|
|
|
|
|
|
def connection_header_names(headers: Iterable[tuple[str, str]]) -> frozenset[str]:
|
|
return frozenset(
|
|
token.strip().lower()
|
|
for name, value in headers
|
|
if name.lower() == "connection"
|
|
for token in value.split(",")
|
|
if token.strip()
|
|
)
|
|
|
|
|
|
def dropped_request_headers(headers: Iterable[tuple[str, str]]) -> frozenset[str]:
|
|
materialized: Final = tuple(headers)
|
|
return REQUEST_DROPPED_HEADERS | connection_header_names(materialized)
|
|
|
|
|
|
def dropped_response_headers(headers: Iterable[tuple[str, str]]) -> frozenset[str]:
|
|
materialized: Final = tuple(headers)
|
|
return RESPONSE_DROPPED_HEADERS | connection_header_names(materialized)
|
|
|
|
|
|
def is_streaming_response(content_type: str) -> bool:
|
|
return "text/event-stream" in content_type.lower()
|