(feat) proxy: set custom headers in metadata

This commit is contained in:
ishaan-jaff 2023-12-04 10:48:59 -08:00
parent ec579d4821
commit e96a608938

View file

@ -923,9 +923,10 @@ async def chat_completion(request: Request, model: Optional[str] = None, user_ap
if "metadata" in data: if "metadata" in data:
data["metadata"]["user_api_key"] = user_api_key_dict["api_key"] data["metadata"]["user_api_key"] = user_api_key_dict["api_key"]
data["metadata"]["headers"] = request.headers
else: else:
data["metadata"] = {"user_api_key": user_api_key_dict["api_key"]} data["metadata"] = {"user_api_key": user_api_key_dict["api_key"]}
data["metadata"]["headers"] = request.headers
global user_temperature, user_request_timeout, user_max_tokens, user_api_base global user_temperature, user_request_timeout, user_max_tokens, user_api_base
# override with user settings, these are params passed via cli # override with user settings, these are params passed via cli
if user_temperature: if user_temperature: