uPjGUdZddlZddlZddlZddlmZmZddlmZm Z m Z ddl Z ddl m Z ddlmZmZddlmZddlmZdd lmZmZmZmZd d lmZd d lmZer ddlZdd lm Z!e!Z"ne Z"dZ#e e$d<de%de&fdZ'GddeZ(GddeZ)y)z^ GigaChat Chat Transformation Transforms OpenAI-format requests to GigaChat format and back. N) AsyncIteratorIterator) TYPE_CHECKINGAnyFinal)verbose_logger) BaseConfigBaseLLMException)get_secret_str)AllMessageValues)ChoicesMessage ModelResponseUsage)get_access_token)upload_file_sync)Loggingz+https://gigachat.devices.sberbank.ru/api/v1GIGACHAT_BASE_URLvaluereturncb tj|y#tj$rYywxYw)zAChecks whether the value passed is a valid serialized JSON stringTF)jsonloadsJSONDecodeError)rs n/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/gigachat/chat/transformation.py is_valid_jsonr%s2 5   s ..ceZdZdZy) GigaChatErrorzGigaChat API error.N)__name__ __module__ __qualname____doc__rrr/sr%rceZdZUdZdZedzed<dZedzed<dZe dzed<dZ edzed<dZ e dzed< d.dedzdedzde dzdedzde dzddf d Z d/d edzd edzd ed edede dzdefdZ d0ded edeed eded edzd edzdefdZd edeefdZded ed ede def dZdeedeefdZdeezdeezdzfdZdededzfdZd edeed edededef dZdeedeefdZ d0d ed ej8d!ed"ed#edeed eded$d%d edzd&e dzdefd'Zd(ed)e deej@zde!fd*Z" d1d+e#ee$ezezd,e d&e dzfd-Z%y)2GigaChatConfiga Configuration class for GigaChat API. GigaChat is Sber's (Russia's largest bank) LLM API. Supported parameters: temperature: Sampling temperature (0-2, default 0.87) top_p: Nucleus sampling parameter max_tokens: Maximum tokens to generate repetition_penalty: Repetition penalty factor profanity_check: Enable content filtering stream: Enable streaming N temperaturetop_p max_tokensrepetition_penaltyprofanity_checkrctj}|jD]%\}}|dk7s |t|j||'d|_d|_y)Nself)localscopyitemssetattr __class___current_credentials_current_api_base) r.r(r)r*r+r,locals_keyrs r__init__zGigaChatConfig.__init__HsX !--/ 4JCf}!2U3 415!-1r%api_baseapi_keymodeloptional_paramslitellm_paramsstreamc:|xstdxst}|dS)z*Get complete API URL for chat completions.GIGACHAT_API_BASEz/chat/completions)r r)r.r9r:r;r<r=r>bases rget_complete_urlzGigaChatConfig.get_complete_urlXs(Z.1D"EZIZ())r%headersmessagesc|xstdxs td}t|} ||_||_d| |d<d|d<d|d<|S) z2 Set up headers with OAuth token. GIGACHAT_CREDENTIALSGIGACHAT_API_KEY) credentialszBearer Authorizationzapplication/jsonz Content-TypeAccept)r rr4r5) r.rCr;rDr<r=r:r9rH access_tokens rvalidate_environmentz#GigaChatConfig.validate_environmentesf%t7M(NtR`asRt .;G %0!!)%,\N#; "4.r%c gdS)z+Return list of supported OpenAI parameters.) r>r(r)r*max_completion_tokensstoptools tool_choice functions function_callresponse_formatr$)r.r;s rget_supported_openai_paramsz*GigaChatConfig.get_supported_openai_paramss   r%non_default_params drop_paramscV|jD]\}}|dk(r||d<|dk(r|dk(rd|d<"||d<(|dk(r||d<3|dvr||d<=|dk(rC|dk(r|j||d <]|d k(r|j|}|v||d <||d k(r||d <|d k(r||d <|d k(s|jd dk(s|jdi}|jdd} |jdi} | d| | d} d |vrg|d <|d j | d| i|d <d|d<|S)z-Map OpenAI parameters to GigaChat parameters.r>r(rr))r*rNr*rOrPrRrQrSrTtype json_schemanamestructured_outputschemazOutput structured response: r[ description parametersT_structured_output)r1_convert_tools_to_functions_map_tool_choicegetappend) r.rVr<r;rWparamr mapped_choicerZ schema_namer] function_defs rmap_openai_paramsz GigaChatConfig.map_openai_paramss/446- ALE5 ,1)-'A:/0OG,5:OM2'!+0(AA05 -&'!/3/O/OPU/V ,-' $ 5 5e <  ,7DOO4+%/4 ,/)380++99V$ 5"'))M2">K"-//&:M"NK(__Xr:F!,)Ek]'S&,$L #/979 4#K077 E8> 7LOO4<@O$89[- A^r%rPc g}|D]n}|jddk(s|jdi}|j|jdd|jdd|jdidp|S)z9Convert OpenAI tools format to GigaChat functions format.rYfunctionr[r_r`r^)rdre)r.rPrRtoolfuncs rrbz*GigaChatConfig._convert_tools_to_functionss|  Dxx:-xx B/  $ 4'+xx r'B&*hh|R&@ r%rQc|dk(ry|dk(ry|dk(ryt|tr;|jddk(r'|jdijd}|rd|iSy)a Map OpenAI tool_choice to GigaChat function_call format. OpenAI format: - "auto": Call zero, one, or multiple functions (default) - "required": Call one or more functions - "none": Don't call any functions - {"type": "function", "function": {"name": "get_weather"}}: Force specific function GigaChat format: - "none": Disable function calls - "auto": Automatic mode (default) - {"name": "get_weather"}: Force specific function Args: tool_choice: OpenAI tool_choice value Returns: GigaChat function_call value or None noneautorequiredrYrlr[N) isinstancedictrd)r.rQ func_names rrczGigaChatConfig._map_tool_choicesr* &  F " J &  T *v&*4#.??:r#B#F#Fv#N "I..r% image_urlc t||j|jS#t$r }t j d|Yd}~yd}~wwxYw)z Upload image to GigaChat and return file_id. Args: image_url: URL or base64 data URL of the image Returns: file_id string or None if upload failed )rwrHr9zFailed to upload image: %sN)rr4r5 Exceptionrerror)r.rwes r _upload_imagezGigaChatConfig._upload_imagesO ## 55//     !=q A s!$ A AA c|j|}|jdd|d}dD]}||vs||||<d|vr|d|d<d|vr|d|d<|S)z,Transform OpenAI request to GigaChat format.z gigachat/rm)r;rD)r(r)r*r>r+r,rRrS)_transform_messagesreplace) r.r;rDr<r=rC giga_messages request_datar7s rtransform_requestz GigaChatConfig.transform_requests $77A ]];3%  9Co%$3C$8 S! 9 / )(7 (DL % o -,;O,LL )r%cNg}t|D]x\}}t|}|jdd|jdd}|dk(rd|d<na|dk(r |dkDrd|d<nQ|dk(rLd |d<|jd d }t |t r t |stj|d |d <|jd d |d <|jd }t |trg}g} |D]} t | ts| jddk(r"|j| jdd J| jddk(s_| jdi} t | t r| } n| jdd } | s|j| } | s| j| |rdj|nd |d <| r| |d<|jd}|rt |trt|dkDry|d}|jd i}|jdd}t |t r tj|}|jdd |d|d<|jdd|j|{|S#tj$ri}YWwxYw)z-Transform OpenAI messages to GigaChat format.r[Nroleuser developersystemrrnrlcontentrmF ensure_asciirYtextrwurl attachments tool_calls argumentsz{}r[rrS) enumeraterupoprdrtstrrrdumpslistrer|joinlenrr)r.rD transformedimsgmessagerrtextsrpartrwrfile_idr tool_callroargss rr~z"GigaChatConfig._transform_messages:s )@ (FAs3iG KK %;;vv.D{""*!a!e"(",!++i4!'3/}W7M)-G%)PGI&{{9%-%' "kk),G'4( #@D!$-88F+v5!LL&")=>!XXf-<(,b(AI))S9&/&/mmE2&>"*.*<*N!O$!(/U (441,2C2C1DE  s J==.K+ error_messagerct|||S)zReturn GigaChat error class.)rrrC)r)r.rrrCs rget_error_classzGigaChatConfig.get_error_classs#!  r%streaming_response sync_streamc$ddlm}||||S)z#Return streaming response iterator.)GigaChatModelResponseIterator)rrr) streamingr)r.rrrrs rget_model_response_iteratorz*GigaChatConfig.get_model_response_iterators =,1#  r%)NNNNN)N)NN)F)&r r!r"r#r(float__annotations__r)r*rr+r,boolr8rrurBrr rLrUrjrbrcr|rr~httpxResponserLiteLLMLoggingObjrHeadersr rrrrr$r%rr'r'3s !%K$E54<!Jd !'+ +#'OTD['%)"!%+/'+ 2T\2t|2$J 2 "DL 2  2 2.# ** *t * *  *  *t  *  *(##'(     t* 6  c 7 77 7  7  7r d T &C$J&3:;L&PssTz($$'($ $  $  $ $LFD1A,BFtDzFf#!%[[nn[& [ ' [  ['([[[-[t[$;[ [z       %    ""'  $SMM#,>>N    $;  r%r')*r#rrrcollections.abcrrtypingrrrrlitellm._loggingr)litellm.llms.base_llm.chat.transformationr r litellm.secret_managers.mainr litellm.types.llms.openair litellm.types.utilsr rrr authenticatorr file_handlerrtiktoken*litellm.litellm_core_utils.litellm_loggingr_LiteLLMLoggingObjrrrrrrrr'r$r%rrs  3,, +R76FF,+X*I5H$F ZF r%