uPj^UddlmZmZddlmZddlmZddlmZdZ ee d<Gdd eZ y ) )FinalLiteral)get_secret_str)AllMessageValues)OpenAILikeChatConfigzhttps://inference.do-ai.runGRADIENT_AI_SERVERLESS_ENDPOINTc%eZdZUdZedzed<dZeedzed<dZ e dzed<dZ e dzed<dZ e dzed<dZe dzed<dZe dzed<dZe dzed <dZed dzed < d&d edzd edzdedzdedzd e dzde ee zdzde dzdedzdedzdedzdeedzde dzde dzde dzde dzde dzd e dzddf$dZefdZde defdZ d'dede deededede dzde dzfdZ d(de dzde dzde dedede dzde fd Zde dzde dzdee dze dzffd!Z d)d"edede d#e d$e def d%ZxZS)*GradientAIConfigNk kb_filters#filter_kb_content_by_query_metadatainstruction_overrideinclude_functions_infoinclude_retrieval_infoinclude_guardrails_infoprovide_citations)rewrite step_back sub_queriesnoneretrieval_methodfrequency_penalty max_tokensmax_completion_tokenspresence_penaltystopstream temperaturetop_preturnctj}|jD]%\}}|dk7s |t|j||'y)Nself)localscopyitemssetattr __class__)r#rrrrrrrrr r r rrrrrrlocals_keyvalues q/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/gradient_ai/chat/transformation.py__init__zGradientAIConfig.__init__sF( !--/ 4JCf}!2U3 4c t|SN)super get_config)clsr(s r,r2zGradientAIConfig.get_config2sw!##r.modelcgd}|S)N)rrrrrrstream_optionsrr r r rrrrrrr)r#r4supported_paramss r,get_supported_openai_paramsz,GradientAIConfig.get_supported_openai_params6s# * r.headersmessagesoptional_paramslitellm_paramsapi_keyapi_basec`|xs td}| td|i}d||d<d|d<|S)NGRADIENT_AI_API_KEYzGradientAI API key not foundzBearer Authorizationzapplication/jsonz Content-Type)r ValueError)r#r:r4r;r<r=r>r?s r,validate_environmentz%GradientAIConfig.validate_environmentNsOB^,AB ?;< < ?G%,WI#6 "4r.crtd}td}|r|tk7r|d}|S|r|tk7r|d}|S)NGRADIENT_AI_AGENT_ENDPOINTz/v1/chat/completionsz/api/v1/chat/completionsrr ) r#r?r>r4r<r=rgradient_ai_endpoint complete_urls r,get_complete_urlz!GradientAIConfig.get_complete_urlas`'55Q&R9::NO $CC&Z'?@L"&:>]&]233KLLr.c`td}|s |st}n|xs|}|xs td}||fS)NrFrArG)r#r?r>rHdynamic_api_keys r,$_get_openai_compatible_provider_infoz5GradientAIConfig._get_openai_compatible_provider_infotsA'55Q&R 46H7#7H!(!QN;P,Q((r.non_default_params drop_params-replace_max_completion_tokens_with_max_tokensc|j|}|jD]%\}}||vr|||<|rddlm} | dd|d|S)N)r4r)UnsupportedParamsErroriz'GradientAI does not support parameter 'z6'. To drop unsupported params, set `drop_params=True`.) status_codemessage)r9r& litellm.utilsrR) r#rNr<r4rOrPsupported_openai_paramsparamr+rRs r,map_openai_paramsz"GradientAIConfig.map_openai_paramssz*.)I)IPU)I)V.446 LE5//).& @, #EeWMCD r.)NNNNNNNNNNNNNNNNN)NNr0)FF)__name__ __module__ __qualname__r int__annotations__r listdictrboolrstrrrrrrrfloatr- classmethodr2r9rrDrJtuplerMrX __classcell__)r(s@r,r r s0AsTz$(JT T!(7;';'+#*+*.D4K.*.D4K.+/TD[/%)td{)VZgKLtSZ+/!%,0)-'+'+"$("(,;?+/.2.2/3)-%4 4<4$J4 #Tz 4  $, 4 * 4DIo$4t 4T\4t|4 :4J%4.2D[4"Dj4!%t 4 !%t !4""&#4$ $;%4& '42$$   >##'(     t*4#*t     t  & )d  )-04Z ) sTz3:% & )$">C      8<  r.r N) typingrrlitellm.secret_managers.mainrlitellm.types.llms.openairopenai_like.chat.transformationrr r]r r7r.r,rjs6!!7D*GFG+Gr.