jrdZddlmZddlZddlmZGddeZd deedefd Zd d ed ee de fd Z y)z. Common utilities for Volcengine LLM provider )OptionalN)BaseLLMExceptioncNeZdZdZ ddededeejffd Z xZ S)VolcEngineErrorz@ Custom exception class for Volcengine provider errors. status_codemessageheadersc||_||_|xstj|_t |||t|jy)N)rrr )rrhttpxHeadersr super__init__dict)selfrrr __class__s }/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/litellm/llms/volcengine/common_utils.pyrzVolcEngineError.__init__sH' 1%--/  #Wd4<<>P  N) __name__ __module__ __qualname____doc__intstrrr r r __classcell__)rs@rrr s: RV  ), 7? 7N  rrapi_basereturnc |r|Sy)z Get the base URL for Volcengine API calls. Args: api_base: Optional custom API base URL Returns: The base URL to use for API calls z!https://ark.cn-beijing.volces.com)rs rget_volcengine_base_urlr s .rapi_key extra_headersc<dd|d}|r|j||S)z Get headers for Volcengine API calls. Args: api_key: The API key for authentication extra_headers: Optional additional headers Returns: Dictionary of headers zapplication/jsonzBearer )z Content-Type Authorization)update)r!r"r s rget_volcengine_headersr&+s.+"7),G }% Nrr) rtypingrr )litellm.llms.base_llm.chat.transformationrrrr rr&rrrr)sX F  &  /hsm /s /CRVr