uPj#UdZddlZddlmZddlZddlmZddlmZ ddl m Z ddl m Z ddlmZmZdd lmZd d lmZd Zeed <Gdde ZGdde Zy)z GigaChat Embedding Transformation Transforms OpenAI /v1/embeddings format to GigaChat format. API Documentation: https://developers.sber.ru/docs/ru/gigachat/api/reference/rest/post-embeddings N)Final) LlmProviders)Logging)BaseLLMException)BaseEmbeddingConfig)AllEmbeddingInputValuesAllMessageValues)EmbeddingResponseget_access_tokenz+https://gigachat.devices.sberbank.ru/api/v1GIGACHAT_BASE_URLceZdZdZy)GigaChatEmbeddingErrorzGigaChat Embedding API error.N)__name__ __module__ __qualname____doc__s/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/gigachat/embedding/transformation.pyrrs'rrceZdZdZddZedZdedeefdZ de d e ded e de f d Z d edzd edzde eedzedzffdZ d d edzd edzded e de de dzdefdZdeded e de de f dZdedej&deded edzde d e de defdZ d!de dedeed e de d edzd edzde fdZdedede ej4zdefdZy)"GigaChatEmbeddingConfigzu Configuration class for GigaChat Embeddings API. GigaChat embeddings endpoint: POST /api/v1/embeddings returnNcyNr)selfs r__init__z GigaChatEmbeddingConfig.__init__%s rc |jjDcic]O\}}|jds9t|tj tj ttfs|||Qc}}Scc}}w)N__) __dict__items startswith isinstancetypes FunctionTypeBuiltinFunctionType classmethod staticmethod)clskvs r get_configz"GigaChatEmbeddingConfig.get_config(st **, 1<<%&&--   qD   sAA6modelcgS)z8GigaChat embeddings don't support additional parameters.r)rr.s rget_supported_openai_paramsz3GigaChatEmbeddingConfig.get_supported_openai_params:s rnon_default_paramsoptional_params drop_paramsc|S)zAMap OpenAI params to GigaChat format (no special mapping needed).r)rr1r2r.r3s rmap_openai_paramsz)GigaChatEmbeddingConfig.map_openai_params>s rapi_baseapi_keycP|xst}tjj||fS)z Returns provider info for GigaChat. Returns: Tuple of (custom_llm_provider, api_base, dynamic_api_key) )rrGIGACHATvalue)rr6r7s r$_get_openai_compatible_provider_infozoW&E CL$AALc>G  %*(" g !1=11rmessagesc4t|}dd|d} i| |S)z? Set up headers with OAuth token for GigaChat. zapplication/jsonzBearer )z Content-Type Authorizationr ) rrBr.r[r2r<r7r6 access_tokendefault_headerss rvalidate_environmentz,GigaChatEmbeddingConfig.validate_environments6/w7 /&|n5" ./-W--r error_message status_codect||S)z%Return GigaChat-specific error class.)rbmessage)r)rrarbrBs rget_error_classz'GigaChatEmbeddingConfig.get_error_classs%#!  r)rNr)NN)rrrrrr(r-rDrEr0dictboolr5tupler;r@rrHhttpxResponser LiteLLMLoggingObjrZr r`intHeadersrrerrrrrs6    "c       >* >t > sC$Jd * + >*# $* $t $ $  $  $t  $  $  '       @/2/2nn/2* /2 ' /2 t /2/2/2/2 /2p##...'( .  .  .t.*. ., S s TTYTaTaMa fv rr)rr%typingrrilitellmr*litellm.litellm_core_utils.litellm_loggingrrk)litellm.llms.base_llm.chat.transformationr.litellm.llms.base_llm.embedding.transformationrlitellm.types.llms.openairr litellm.types.utilsr authenticatorr r__annotations__rrrrrrwsT  SFNO1,I5H(-(q 1q r