o #1 i=*@sUddlZddlmZddlmZmZmZmZmZm Z m Z ddl m Z ddl mZmZddlmZddlmZmZmZmZmZddlmZejd ed d d ed efddZGdddeeZGdddeZGdddeZ GdddeZ!GdddeZ"Gddde"Z#GdddeZ$ee e!e e"e$fe ddfZ%ee e!e e#e$fe ddfZ&e d e%d!Z'e d"e d!Z(e d#e"d!Z)e d$e$d!Z*e d%e!d!Z+e e(e)e*e+fZ,ee-d&<dS)'N)Enum)AnyDictListLiteralOptionalTypeVarUnion)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)r"RawAudio)r#zmodule z has no attribute )__name__&mistral_common.protocol.instruct.chunkrwarningswarn FutureWarningrrrrrr r!r"r#AttributeError) rmsgrrrrrrr r!r"r#r+u/home/app/PaddleOCR-VL-test/.venv_paddleocr/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/r0r1r2r+r+r+r,r.Ts  r.c@seZdZUdZeejejejej fe d<de e e e ee e e e e e effffffddZede e e e ee e e e e e efffffddfddZd 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 .)r8r$clsr=r+r+r, from_openaiwszBaseMessage.from_openaiN)r$r3r4r5rr.r/r0r1r2__annotations__rstrr rrr< classmethodrAr+r+r+r,r6gs :.r6c@seZdZUdZejZeejed<e e e e fed<de e e e e e e e e e e effffffddZede e e e e e e e e e e efffffddfdd Zd S) UserMessagezUser message. Attributes: content: The content of the message. Examples: >>> message = UserMessage(content="Can you help me to write a poem?") r7contentrcCs2t|jtr |j|jdS|jdd|jDdS)*Converts the message to the OpenAI format.r7rFcSg|]}|qSr+r<.0chunkr+r+r, z)UserMessage.to_openai..) isinstancerFrCr7r:r+r+r,r<s zUserMessage.to_openair=cCs:t|dtr ||S||ddd|dDdS)2Converts the OpenAI message to the Mistral format.rFr7cSg|]}t|qSr+rrKr+r+r,rNrOz+UserMessage.from_openai..rH)rPrCmodel_validater?r+r+r,rAs zUserMessage.from_openaiN)r$r3r4r5r.r0r7rrBr rCrrrrr<rDrAr+r+r+r,rEs  :.rEc@seZdZUdZejZeejed<e e e e e e ffed<dee e e e ee e e ee effffffddZedee e e e ee e e ee efffffddfdd Zd S) SystemMessagezSystem message. Attributes: content: The content of the message. Examples: >>> message = SystemMessage(content="You are a helpful assistant.") r7rFrcCs|S)rG) model_dumpr:r+r+r,r<szSystemMessage.to_openair=cCs ||S)rQ)rTr?r+r+r,rAs zSystemMessage.from_openaiN)r$r3r4r5r.r/r7rrBr rCrrrrrr<rDrAr+r+r+r,rUs  :.rUc@seZdZUdZejZeejed<dZ e e e e e eeffed<dZe e eed<dZeed<dee e e e ee e e ee effffffd d Zed ee e e e ee e e ee efffffddfd 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?") r7NrF tool_callsFprefixrcCsdd|ji}|jdur nt|jtr|j|d<n dd|jD|d<|jdur0dd|jD|d<|S)rGr7NrFcSrIr+rJrKr+r+r,rNrOz.AssistantMessage.to_openai..cSrIr+rJ)rLZ tool_callr+r+r,rNrOrX)r7rFrPrCrX)r;Zout_dictr+r+r,r<s    zAssistantMessage.to_openair=cCs|dd}|durdd|Dnd}|dd}d}|dus$t|tr'|}nt|tr4dd|D}n tdt|||d||d S) rQrXNcSsg|]}t|qSr+)rrA)rLZopenai_tool_callr+r+r,rNsz0AssistantMessage.from_openai..rFcSrRr+rSrKr+r+r,rNrOzUnknown content type: r7)r7rFrX)getrPrClist ValueErrorr9rT)r@r=Zopenai_tool_callsZ tools_callsZopenai_contentrFr+r+r,rAs(   zAssistantMessage.from_openai)r$r3r4r5r.r1r7rrBrFrr rCrrrrXrrYboolrrr<rDrAr+r+r+r,rWs  $ :.rWc@s"eZdZUdZdZeeed<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$r3r4r5r_rfloatrBr+r+r+r,r^s  r^c@seZdZUdZeed<ejZe ejed<dZ e eed<dZ e eed<de eeeee eeee eeffffffdd Zed e eeeee eeee eefffffddfd 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") rFr7N tool_call_idrrcCs |jdus Jd|jdhdS)rGN0tool_call_id must be provided for tool messages.r)exclude)rbrVr:r+r+r,r<szToolMessage.to_openaimessagescCs ||}|jdusJd|S)rQNrc)rTrb)r@reZ tool_messager+r+r,rA$s zToolMessage.from_openai)r$r3r4r5rCrBr.r2r7rrbrrrr rrr<rDrAr+r+r+r,ra s  :Drar7)Z discriminatorChatMessageType)boundUserMessageTypeAssistantMessageTypeToolMessageTypeSystemMessageTypeUATS).r&enumrtypingrrrrrrr Zpydanticr Ztyping_extensionsr r Zmistral_common.baser r%rrrrrZ+mistral_common.protocol.instruct.tool_callsrfilterwarningsr(rCr-r.r6rErUrWr^raZ ChatMessageZFinetuningMessagerfrhrirjrkrlrBr+r+r+r,sB $    9"B