uPj^+dZddlmZmZddlmZmZmZmZm Z ddl Z ddl m Z ddl mZddlmZddlmZd d lmZd ed edzfd ZGddeZy)zY Translates from OpenAI's `/v1/chat/completions` to Moonshot AI's `/v1/chat/completions` ) CoroutineMapping)AnyFinalLiteralcastoverloadN)3handle_messages_with_content_list_to_str_conversionget_secret_str)AllMessageValues)supports_reasoning)OpenAIGPTConfigvaluereturncpt|tr|jdn|}t|tr|SdS)a(The /v1/messages and /v1/responses bridges wrap the level as {"effort", "summary"} for providers with a reasoning-summary surface. Moonshot's API takes only the bare string and 400s on an object, so the level is unwrapped and the summary, which has no Moonshot equivalent, is dropped.effortN) isinstancergetstr)rrs n/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/moonshot/chat/transformation.py_reasoning_effort_stringrs1 ,6eW+EEIIh'5F,66$6ceZdZedeedededdee e eeffdZ e ddeedededdeefdZ ddeedede deeee e eefzffd Z d ed zd ed zde ed zed zffd Z dd ed zd ed zdededede d zdefdZdedeffd Zdededede def dZdeedeefdZdedeededededef fd ZdeededeefdZxZS)MoonshotChatConfigmessagesmodelis_asyncTrcyNselfrrrs r_transform_messagesz&MoonshotChatConfig._transform_messagess7:rFcyr!r"r#s rr%z&MoonshotChatConfig._transform_messages"s "%rcd}|D]=}|jd}|st|ts(td|Ds;d}n|s t |}|rt |||dSt |||dS)a Moonshot text-only models don't support content in list format. Multimodal models (kimi-k2.5, kimi-latest, etc.) accept the standard OpenAI content array with non-text blocks (image_url, input_audio, video_url, file, etc.). If any message contains a non-text content part, skip flattening so the multimodal payload is preserved. Fcontentc3DK|]}|jddk7yw)typetextN)r).0cs r z9MoonshotChatConfig._transform_messages..:sA1quuV}.As T)rrr)rrlistanyr superr%)r$rrr has_non_textm_content __class__s rr%z&MoonshotChatConfig._transform_messages*s  AuuY'HJx6AAA#'L  J8TH 7.X\.] ]7.X].^ ^rapi_baseNapi_keycN|xstdxsd}|xs td}||fS)NMOONSHOT_API_BASEhttps://api.moonshot.ai/v1MOONSHOT_API_KEYr )r$r6r7dynamic_api_keys r$_get_openai_compatible_provider_infoz7MoonshotChatConfig._get_openai_compatible_provider_infoFs5b~.ABbFb!(!NN;M,N((roptional_paramslitellm_paramsstreamc:|sd}|jds|d}|S)zf If api_base is not provided, use the default Moonshot AI /chat/completions endpoint. r:z/chat/completions)endswith)r$r6r7rr>r?r@s rget_complete_urlz#MoonshotChatConfig.get_complete_urlMs-3H  !45"#45Hrctd|vrdnd}t| |}|Dcgc] }||vs| }}t|drg|dS|Scc}w)a Get the supported OpenAI params for Moonshot AI models Moonshot AI limitations: - functions parameter is not supported (use tools instead) - tool_choice doesn't support "required" value - kimi-thinking-preview doesn't support tool calls at all A reasoning model additionally takes `reasoning_effort`, which the OpenAI base list this subtracts from does not carry, so it has to be added back rather than merely kept. zkimi-thinking-preview) functionstools tool_choice)rE)rmoonshotrcustom_llm_providerreasoning_effort) frozensetr1get_supported_openai_paramsr)r$rexcluded_paramsbase_openai_paramsparam supportedr5s rrMz.MoonshotChatConfig.get_supported_openai_paramsasu"+5LPU5U 1[i" %*G$Ge$G$T/AbeURaEaEb b Ez J3Y3 23 3cs A A non_default_params drop_paramscf|j|}|jD]3\}}|dk(r||d<||vr|dk(rt|x})||d</|||<5t|dr|j dd|Sd|vr4t |dd|d<|dd kr|j d ddkDrd |d<|S) z Map OpenAI parameters to Moonshot AI parameters Handles Moonshot AI specific limitations: - tool_choice doesn't support "required" value - Temperature <0.3 limitation for n>1 max_completion_tokens max_tokensrKNrHrI temperatureg333333?n)rMitemsrrpopminr) r$rRr>rrSsupported_openai_paramsrPrrs rmap_openai_paramsz$MoonshotChatConfig.map_openai_paramsvs*.)I)I%)P.446 /LE5//05 -55,,6u==FJ:@O$67).& /& Ez J    t 4  o --01OQR-SOM *}-38K8KCQR8SVW8W14 .rcg}|D]}|jddk(r|jdr|jdsttt|}|jdxsi}|jd}|r(||d<t|}|jdd||d<n$tj j dd|d<|jtt||j||S) u Moonshot reasoning models require `reasoning_content` on every assistant message that contains tool_calls (multi-turn tool-calling flows). For each such message that is missing the field: 1. Check if reasoning_content exists at the top level (for Pydantic models that have the attribute but don't support 'in' operator) 2. Promote provider_specific_fields["reasoning_content"] if present and non-empty (this is where LiteLLM stores it from a previous response) 3. Otherwise inject a single space — the minimum value the API accepts Messages that already carry the field, or are not assistant/tool-call messages, are appended as-is (no copy made). role assistant tool_callsreasoning_contentprovider_specific_fieldsNa Moonshot reasoning model: assistant tool-call message is missing `reasoning_content`. Injecting a placeholder to satisfy API validation. For best results, preserve `reasoning_content` from the original assistant response when building multi-turn conversation history. ) rdictrr[litellmverbose_loggerwarningappendr )r$rresultmsgpatchedprovider_fieldsstoredcleaned_provider_fieldss rfill_reasoning_contentz)MoonshotChatConfig.fill_reasoning_contents13 #C;.GGL)'tD#/")++.H"I"OR(,,-@A39G/0/3?.C++//0CTJ:QG67**22\ 47G/0 d#3W=> c"9 #: rheadersc|jdddk(r|j||}t|dr|j|}t||||||S)z Transform the overall request to be sent to the API. Returns: dict: The transformed request. Sent as the body of the API call. rGNrequired)rr>rHrI)rrr>r?rr)r!_add_tool_choice_required_messagerrqr1transform_request)r$rrr>r?rrr5s rrvz$MoonshotChatConfig.transform_requestsy   }d 3z A==! />H Ez J228s rruz4MoonshotChatConfig._add_tool_choice_required_messages6 M*  N   r)Fr!)__name__ __module__ __qualname__r r/r rrrrr%booltupler=rfrCrMr^rqrvru __classcell__)r5s@rrrs] :-.:7::FMdm: 3T"233 4:: $) %'(%%%. %   %%NS_-._7:_FJ_  )Cd;K6L,L"M M_8)d )-04Z) sTz3:% &)#*t     t  (*( (( (  (  (T,t4D/E,$O_J`,\  '(         B -. AE   rr)__doc__collections.abcrrtypingrrrrr rg8litellm.litellm_core_utils.prompt_templates.common_utilsr litellm.secret_managers.mainr litellm.types.llms.openair litellm.utilsropenai.chat.gpt_transformationrobjectrrrr"rrrsP/6686,=7F7sTz7b b r