mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-09 03:18:44 +00:00
fix: make user-agent header check case-insensitive
The setting disable_add_user_agent_to_request_tags was not working when headers used capitalized keys like 'User-Agent' instead of lowercase 'user-agent'. HTTP headers are case-insensitive per RFC 7230, so we now iterate through all header keys and match case-insensitively. Fixes #23553
This commit is contained in:
parent
1df71606a3
commit
095f5e8399
1 changed files with 15 additions and 11 deletions
|
|
@ -5063,18 +5063,22 @@ class StandardLoggingPayloadSetup:
|
|||
user_agent_tags: Optional[List[str]] = None
|
||||
headers = proxy_server_request.get("headers", {})
|
||||
if headers is not None and isinstance(headers, dict):
|
||||
if "user-agent" in headers:
|
||||
user_agent = headers["user-agent"]
|
||||
# Check for case-insensitive "user-agent" header key
|
||||
user_agent = None
|
||||
for key in headers:
|
||||
if key.lower() == "user-agent":
|
||||
user_agent = headers[key]
|
||||
break
|
||||
if user_agent is not None:
|
||||
if user_agent_tags is None:
|
||||
user_agent_tags = []
|
||||
user_agent_part: Optional[str] = None
|
||||
if "/" in user_agent:
|
||||
user_agent_part = user_agent.split("/")[0]
|
||||
if user_agent_part is not None:
|
||||
user_agent_tags.append("User-Agent: " + user_agent_part)
|
||||
if user_agent is not None:
|
||||
if user_agent_tags is None:
|
||||
user_agent_tags = []
|
||||
user_agent_part: Optional[str] = None
|
||||
if "/" in user_agent:
|
||||
user_agent_part = user_agent.split("/")[0]
|
||||
if user_agent_part is not None:
|
||||
user_agent_tags.append("User-Agent: " + user_agent_part)
|
||||
if user_agent is not None:
|
||||
user_agent_tags.append("User-Agent: " + user_agent)
|
||||
user_agent_tags.append("User-Agent: " + user_agent)
|
||||
return user_agent_tags
|
||||
|
||||
@staticmethod
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue