o c۷i @sUddlmZddlmZmZmZmZmZddlm Z de e e e e e e e e e efBfBfde efddZ de e e efde e fd d Zd ee d ee dd fddZde defddZhdZee ed<d S))Any)AssistantMessage ChatMessage SystemMessage ToolMessage UserMessage)ToolmessagesreturncCsg}|D];}|d}|dkrt|}n%|dkrt|}n|dkr)t|}n|dkr3t|}ntd|||q|S)zConvert OpenAI messages to Mistral messages. Args: messages: The OpenAI messages to convert. Returns: The Mistral messages. roleuser assistanttoolsystemzUnknown message role: )getr from_openairrr ValueErrorappend)r converted_messagesopenai_message message_rolemessagera/home/ubuntu/vllm_env/lib/python3.10/site-packages/mistral_common/protocol/instruct/converters.pyconvert_openai_messages s       rtoolscCsdd|D}|S)zConvert OpenAI tools to Mistral tools. Args: tools: The OpenAI tools to convert. Returns: The Mistral tools. cSsg|]}t|qSr)rr).0 openai_toolrrr 5sz(convert_openai_tools..r)rconverted_toolsrrrconvert_openai_tools*s r valid_fields_namesnamesNcCs^t}t}|D]}||vrq|tvr||q||q|s#|r-td|d|dS)aCheck if the names are valid field names. Names are valid if they are inside the `valid_fields_names` set or chat completion OpenAI fields. If the names are not valid field names, raise a ValueError. The error message will contain the invalid field names sorted by if they are openAI valid field names or not. Args: valid_fields_names: The valid field names. names: The names to check. Raises: ValueError: If the names are not valid field names. z~Invalid parameters passed to `ChatCompletionRequest.from_openai`: OpenAI valid parameters but not in `ChatCompletionRequest`: z Non valid parameters: N)set_OPENAI_COMPLETION_FIELDSaddr)r!r"openai_valid_paramsnon_valid_paramsnamerrr_check_openai_fields_names9s"  r)r(cCs|tvS)N)r$)r(rrr_is_openai_field_name\sr*>#nseedstopr audiomodelstorertop_pstreamtimeoutlogprobsr metadata functions extra_body logit_bias max_tokens modalities prediction extra_query temperature tool_choice service_tier top_logprobs extra_headers function_callstream_optionsresponse_formatpresence_penaltyreasoning_effortfrequency_penaltyweb_search_optionsparallel_tool_callsmax_completion_tokensr$)typingr)mistral_common.protocol.instruct.messagesrrrrr+mistral_common.protocol.instruct.tool_callsrlistdictstrrr r#r)boolr*r$__annotations__rrrrs  *  #