o ưi@sldZddlmZddlZddlmZGdddeZddeedefd d Zdd ed ee de fd dZ dS)z. Common utilities for Volcengine LLM provider )OptionalN)BaseLLMExceptioncs8eZdZdZ ddededeejffdd Z Z S) VolcEngineErrorz@ Custom exception class for Volcengine provider errors. N status_codemessageheaderscs6||_||_|p t|_tj||t|jddS)N)rrr)rrhttpxHeadersrsuper__init__dict)selfrrr __class__[/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/volcengine/common_utils.pyr s   zVolcEngineError.__init__N) __name__ __module__ __qualname____doc__intstrrrr r __classcell__rrrrr srapi_basereturncCs |r|SdS)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.comr)rrrrget_volcengine_base_urls rapi_key extra_headerscCs"dd|d}|r|||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)rrrrrrget_volcengine_headers+s   r!r) rtypingrrZ)litellm.llms.base_llm.chat.transformationrrrrr r!rrrrs