uPjrdZddlmZddlZddlmZGddeZd dedzdefd Zd d ed e dzde fd Z y)z. Common utilities for Volcengine LLM provider )FinalN)BaseLLMExceptioncLeZdZdZddededejdzffd ZxZ S)VolcEngineErrorz@ Custom exception class for Volcengine provider errors. N status_codemessageheadersc||_||_|xstj|_t |||t|jy)N)rrr )rrhttpxHeadersr super__init__dict)selfrrr __class__s i/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/volcengine/common_utils.pyrzVolcEngineError.__init__sD& 1%--/  ['4PTP\P\K]^N) __name__ __module__ __qualname____doc__intstrr r r __classcell__)rs@rrr s6_C_#_ PT@T__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)sY F _& _ /cDj /C /Ct tr