o ưi@spdZddlmZmZmZmZmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZGdd d eZd S) zY Translates from OpenAI's `/v1/chat/completions` to Moonshot AI's `/v1/chat/completions` )Any CoroutineListLiteralOptionalTupleUnionoverload)3handle_messages_with_content_list_to_str_conversionget_secret_str)AllMessageValues)OpenAIGPTConfigcseZdZedeedededdee e eeffddZ e d deedededdeefd dZ d deedede de eeee e eeffffd d Z d e ed e edee ee effd dZ d!d e ed e edededede e defddZdedeffdd Zdededede def ddZdedeededededef fdd ZdeededeefddZZS)"MoonshotChatConfigmessagesmodelis_asyncTreturncCdSNselfrrrrr`/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/moonshot/chat/transformation.py_transform_messagessz&MoonshotChatConfig._transform_messagesFcCrrrrrrrrscstd}|D]}|d}|rt|trtdd|Drd}nq|s&t|}|r1tj||ddStj||ddS)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. Fcontentcss|] }|ddkVqdS)typetextN)get).0crrr 0sz9MoonshotChatConfig._transform_messages..T)rrr)r isinstancelistanyr superr)rrrrZ has_non_textm_content __class__rrr s"  api_baseapi_keycCs$|ptdpd}|p td}||fS)NZMOONSHOT_API_BASEhttps://api.moonshot.ai/v1ZMOONSHOT_API_KEYr )rr+r,Zdynamic_api_keyrrr$_get_openai_compatible_provider_info@s z7MoonshotChatConfig._get_openai_compatible_provider_infoNoptional_paramslitellm_paramsstreamcCs |sd}|ds|d}|S)zf If api_base is not provided, use the default Moonshot AI /chat/completions endpoint. r-z/chat/completions)endswith)rr+r,rr/r0r1rrrget_complete_urlKs    z#MoonshotChatConfig.get_complete_urlcsNdg}d|vr|ddgtj|d}g}|D] }||vr$||q|S)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 Z functionszkimi-thinking-previewZtools tool_choice)r)extendr&get_supported_openai_paramsappend)rrZexcluded_paramsZbase_openai_paramsZ final_paramsparamr)rrr6_s  z.MoonshotChatConfig.get_supported_openai_paramsnon_default_params drop_paramscCs||}|D]\}}|dkr||d<q ||vr|||<q d|vr?|ddkr-d|d<|ddkr?|dddkr?d|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 Zmax_completion_tokensZ max_tokensZ temperatureg333333?n)r6itemsr)rr9r/rr:Zsupported_openai_paramsr8valuerrrmap_openai_paramsvs   z$MoonshotChatConfig.map_openai_paramsheaderscs4|dddkr|j||d}tj|||||dS)z Transform the overall request to be sent to the API. Returns: dict: The transformed request. Sent as the body of the API call. r4Nrequired)rr/)rrr/r0r@)r!_add_tool_choice_required_messager&transform_request)rrrr/r0r@r)rrrCsz$MoonshotChatConfig.transform_requestcCs|ddd|d|S)z Add a message to the messages list to indicate that the tool choice is required. https://platform.moonshot.ai/docs/guide/migrating-from-openai-to-kimi#about-tool_choice userz1Please select a tool to handle the current issue.)Zrolerr4)r7pop)rrr/rrrrBs  z4MoonshotChatConfig._add_tool_choice_required_message)Fr)__name__ __module__ __qualname__r rr strrrrrboolrrrr.dictr3r$r6r?rCrB __classcell__rrr)rrs       #&rN)__doc__typingrrrrrrrr Z8litellm.litellm_core_utils.prompt_templates.common_utilsr Zlitellm.secret_managers.mainr Zlitellm.types.llms.openair Zopenai.chat.gpt_transformationrrrrrrs(