o i)@sUddlZddlmZddlmZmZmZddlmZddl m Z m Z ddl m Z ddlmZmZmZmZmZddlmZejd ed d d ed efddZGdddeeZGddde ZGdddeZGdddeZGdddeZGdddeZGdddeZ e eeBeBe BeddfZ!e eeBeBe BeddfZ"ed e!d!Z#ed"ed!Z$ed#ed!Z%ed$e d!Z&ed%ed!Z'e$e%Be&Be'BZ(e e)d&<dS)'N)Enum)AnyLiteralTypeVar)Field) Annotated TypeAlias) MistralBase) ContentChunk TextChunk ThinkChunkUserContentChunk_convert_openai_content_chunks)ToolCalloncezN.*has moved to 'mistral_common.protocol.instruct.chunk'\. It will be removed.*)actioncategorymessagenamereturnc Cs|dtd}|dkrddlm}t|t|S|dkr,ddlm}t|t|S|dkr>dd lm}t|t|S|d krPdd lm}t|t|S|d krbdd lm }t|t|S|dkrtddlm }t|t|S|dkrddlm }t|t|S|dkrddlm } t|t| S|dkrddlm } t|t| S|dkrddlm} t|t| Stdtd|)NzQ has moved to 'mistral_common.protocol.instruct.chunk'. It will be removed from 'z ' in 1.10.0.AudioURLr)r AudioURLChunk)r AudioURLType)r AudioChunk)rBaseContentChunk)r ChunkTypes)rImageURL)r ImageURLChunk)r ImageChunk)rRawAudio)rzmodule z has no attribute )__name__&mistral_common.protocol.instruct.chunkrwarningswarn FutureWarningrrrrrrrrrAttributeError) rmsgrrrrrrrrrrr']/home/ubuntu/.local/lib/python3.10/site-packages/mistral_common/protocol/instruct/messages.py __getattr__sV                    r)c@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)r __module__ __qualname____doc__r+r,r-r.r'r'r'r(r*Ts  r*c @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. rolercCstdt|j)zaConverts the message to the OpenAI format. Should be implemented by subclasses. z%to_openai method not implemented for )NotImplementedErrortyper selfr'r'r( to_openaipszBaseMessage.to_openaiopenai_messagecCstd|jd)ziConverts the OpenAI message to the Mistral format. Should be implemented by subclasses. z'from_openai method not implemented for .)r4r clsr9r'r'r( from_openaiwszBaseMessage.from_openaiN)r r/r0r1rr*r+r,r-r.__annotations__dictstrlistrr8 classmethodr=r'r'r'r(r2gs 2<r2c @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?") r3contentrcCs2t|jtr |j|jdS|jdd|jDdS)*Converts the message to the OpenAI format.r3rDcSg|]}|qSr'r8.0chunkr'r'r( z)UserMessage.to_openai..) isinstancerDr@r3r6r'r'r(r8s zUserMessage.to_openair9cCs:t|dtr ||S||ddd|dDdS)2Converts the OpenAI message to the Mistral format.rDr3cSg|]}t|qSr'rrIr'r'r(rLrMz+UserMessage.from_openai..rF)rNr@model_validater;r'r'r(r=s zUserMessage.from_openaiN)r r/r0r1r*r,r3rr>r@rAr r?rr8rBr=r'r'r'r(rCs  2<rCc @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.") r3rDrcCs|S)rE) model_dumpr6r'r'r(r8szSystemMessage.to_openair9cCs ||S)rO)rRr;r'r'r(r=s zSystemMessage.from_openaiN)r r/r0r1r*r+r3rr>r@rAr r r?rr8rBr=r'r'r'r(rSs  2<rSc @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?") r3NrD tool_callsFprefixrcCsdd|ji}|jdur nt|jtr|j|d<n dd|jD|d<|jdur0dd|jD|d<|S)rEr3NrDcSrGr'rHrIr'r'r(rLrMz.AssistantMessage.to_openai..cSrGr'rH)rJ tool_callr'r'r(rLrMrV)r3rDrNr@rV)r7out_dictr'r'r(r8s    zAssistantMessage.to_openair9cCs|dd}|dur d}n t|tr$g}|pgD] }|t|qn tdt||dd}d}|dus>t|trA|}nt|trNdd|D}n tdt|| |d||d S) rOrVNztool_calls must be a list, got rDcSrPr'rQrIr'r'r(rLrMz0AssistantMessage.from_openai..zUnknown content type: r3)r3rDrV) getrNrAappendrr= ValueErrorr5r@rR)r<r9openai_tool_calls tools_callsopenai_tool_callopenai_contentrDr'r'r(r=s,     zAssistantMessage.from_openai)r r/r0r1r*r-r3rr>rDr@rAr r rVrrWboolr?rr8rBr=r'r'r'r(rUs   2<rUc@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)r r/r0r1rcfloatr>r'r'r'r(rbs  rbc @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") rDr3N tool_call_idrrcCs |jdus Jd|jdhdS)rEN0tool_call_id must be provided for tool messages.r)exclude)rfrTr6r'r'r(r8szToolMessage.to_openaimessagescCs ||}|jdusJd|S)rONrg)rRrf)r<ri tool_messager'r'r(r=s zToolMessage.from_openai)r r/r0r1r@rAr r>r*r.r3rrfrr?rr8rBr=r'r'r'r(res  2<rer3) discriminatorChatMessageType)boundUserMessageTypeAssistantMessageTypeToolMessageTypeSystemMessageTypeUATS)*r"enumrtypingrrrpydanticrtyping_extensionsrrmistral_common.baser r!r r r r r+mistral_common.protocol.instruct.tool_callsrfilterwarningsr$r@r)r*r2rCrSrUrbre ChatMessageFinetuningMessagerlrnrorprqrrr>r'r'r'r(sB     9 @