o c۷i @sxUddlmZddlmZmZmZddlmZddlm Z m Z ddl m Z ddl mZmZmZmZmZddlmZGdd d eeZGd d d e ZGd d d eZGdddeZGdddeZGdddeZGdddeZe eeBeBeBeddfZe eeBeBeBeddfZededZededZ ededZ!ededZ"ededZ#e e!Be"Be#BZ$e e%d<dS) )Enum)AnyLiteralTypeVar)Field) Annotated TypeAlias) MistralBase) ContentChunk TextChunk ThinkChunkUserContentChunk_convert_openai_content_chunks)ToolCallc@s eZdZdZdZdZdZdZdS)RoleszEnum for the roles of the messages. Attributes: system: The system role. user: The user role. assistant: The assistant role. tool: The tool role. Examples: >>> role = Roles.user systemuser assistanttoolN)__name__ __module__ __qualname____doc__rrrrrr_/home/ubuntu/vllm_env/lib/python3.10/site-packages/mistral_common/protocol/instruct/messages.pyrs  rc @seZdZUdZeejejejej fe d<de e e e e e e e e efBfBffddZede e e e e e e e e efBfBfddfddZd S) BaseMessagezXBase class for all messages. Attributes: role: The role of the message. rolereturncCstdt|j)zaConverts the message to the OpenAI format. Should be implemented by subclasses. z%to_openai method not implemented for )NotImplementedErrortyperselfrrr to_openai.szBaseMessage.to_openaiopenai_messagecCstd|jd)ziConverts the OpenAI message to the Mistral format. Should be implemented by subclasses. z'from_openai method not implemented for .)rrclsr#rrr from_openai5szBaseMessage.from_openaiN)rrrrrrrrrr__annotations__dictstrlistrr" classmethodr'rrrrr%s 2<rc @seZdZUdZejZeejed<e e e Bed<de e e e e e e e e e fBfBffddZede e e e e e e e e e fBfBfddfdd Zd S) UserMessagezUser message. Attributes: content: The content of the message. Examples: >>> message = UserMessage(content="Can you help me to write a poem?") rcontentrcCs2t|jtr |j|jdS|jdd|jDdS)*Converts the message to the OpenAI format.rr.cSg|]}|qSrr".0chunkrrr Oz)UserMessage.to_openai..) isinstancer.r*rr rrrr"Ks zUserMessage.to_openair#cCs:t|dtr ||S||ddd|dDdS)2Converts the OpenAI message to the Mistral format.r.rcSg|]}t|qSrrr3rrrr6Yr7z+UserMessage.from_openai..r0)r8r*model_validater%rrrr'Qs zUserMessage.from_openaiN)rrrrrrrrr(r*r+r r)rr"r,r'rrrrr->s  2<r-c @seZdZUdZejZeejed<e e e e BBed<de e e e e e e e e efBfBffddZede e e e e e e e e efBfBfddfdd Zd S) SystemMessagezSystem message. Attributes: content: The content of the message. Examples: >>> message = SystemMessage(content="You are a helpful assistant.") rr.rcCs|S)r/) model_dumpr rrrr"kszSystemMessage.to_openair#cCs ||S)r9)r<r%rrrr'os zSystemMessage.from_openaiN)rrrrrrrrr(r*r+r r r)rr"r,r'rrrrr=^s  2<r=c @seZdZUdZejZeejed<dZ e e e e BBdBed<dZe edBed<dZeed<dee e e ee e ee efBfBffd d Zed ee e e ee e ee efBfBfddfd d ZdS)AssistantMessagea<Assistant message. Attributes: role: The role of the message. content: The content of the message. tool_calls: The tool calls of the message. prefix: Whether the message is a prefix. Examples: >>> message = AssistantMessage(content="Hello, how can I help you?") rNr. tool_callsFprefixrcCsdd|ji}|jdur nt|jtr|j|d<n dd|jD|d<|jdur0dd|jD|d<|S)r/rNr.cSr1rr2r3rrrr6r7z.AssistantMessage.to_openai..cSr1rr2)r4 tool_callrrrr6r7r@)rr.r8r*r@)r!out_dictrrrr"s    zAssistantMessage.to_openair#cCs|dd}|dur d}n t|tr$g}|pgD] }|t|qn tdt||dd}d}|dus>t|trA|}nt|trNdd|D}n tdt|| |d||d S) r9r@Nztool_calls must be a list, got r.cSr:rr;r3rrrr6r7z0AssistantMessage.from_openai..zUnknown content type: r)rr.r@) getr8r+appendrr' ValueErrorrr*r<)r&r#openai_tool_calls tools_callsopenai_tool_callopenai_contentr.rrrr's,     zAssistantMessage.from_openai)rrrrrrrrr(r.r*r+r r r@rrAboolr)rr"r,r'rrrrr?us   2<r?c@s"eZdZUdZdZedBed<dS)FinetuningAssistantMessagezAssistant message for finetuning. Attributes: weight: The weight of the message to train on. Examples: >>> message = FinetuningAssistantMessage(content="Hello, how can I help you?", weight=0.5) Nweight)rrrrrMfloatr(rrrrrLs  rLc @seZdZUdZeeeBed<ej Z e ej ed<dZ edBed<dZ edBed<deeeeeeeeeefBfBffdd Zed eeeeeeeeeefBfBfddfd d ZdS) ToolMessagea5Tool message. Attributes: content: The content of the message. tool_call_id: The tool call id of the message. name: The name of the tool. (Deprecated in V3 tokenization) Examples: >>> message = ToolMessage(content="Hello, how can I help you?", tool_call_id="123") r.rN tool_call_idnamercCs |jdus Jd|jdhdS)r/N0tool_call_id must be provided for tool messages.rQ)exclude)rPr>r rrrr"szToolMessage.to_openaimessagescCs ||}|jdusJd|S)r9NrR)r<rP)r&rT tool_messagerrrr's zToolMessage.from_openai)rrrrr*r+r r(rrrrrPrQr)rr"r,r'rrrrrOs  2<rOr) discriminatorChatMessageType)boundUserMessageTypeAssistantMessageTypeToolMessageTypeSystemMessageTypeUATSN)&enumrtypingrrrpydanticrtyping_extensionsrrmistral_common.baser &mistral_common.protocol.instruct.chunkr r r r r+mistral_common.protocol.instruct.tool_callsrr*rrr-r=r?rLrO ChatMessageFinetuningMessagerWrYrZr[r\r]r(rrrrs4    @