*`i)UddlZddlmZddlmZmZmZddlmZddl m Z m Z ddl m Z ddlmZmZmZmZmZddlmZejd ed d ed efdZGddeeZGdde ZGddeZGddeZGddeZGddeZGddeZ e eezeze zedfZ!e eezeze zedfZ"ede! Z#ed!e Z$ed"e Z%ed#e Z&ed$e Z'e$e%ze&ze'zZ(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.*)actioncategorymessagenamereturncp|dtd}|dkr"ddlm}tj|t |S|dkr"ddlm}tj|t |S|dkr"dd lm}tj|t |S|d kr"dd lm}tj|t |S|d kr"dd lm }tj|t |S|dkr"ddlm }tj|t |S|dkr"ddlm }tj|t |S|dkr"ddlm } tj|t | S|dkr"ddlm } tj|t | S|dkr"ddlm} tj|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 FutureWarningrrrrrrrr r!AttributeError) rmsgrrrrrrrrr r!s }/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/mistral_common/protocol/instruct/messages.py __getattr__r*sM xxbjxxx zCCCCCC c=)))  HHHHHH c=)))   GGGGGG c=)))   EEEEEE c=))) # # #KKKKKK c=)))   EEEEEE c=)))   CCCCCC c=)))  HHHHHH c=)))   EEEEEE c=)))   CCCCCC c=))) E8EEtEE F FFc"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/r0r1r+r)r-r-Ts.  F DI DDDr+r-c eZdZUdZeejejejej fe d<de e e e e e e e e efzfzffdZede e e e e e e e e efzfzfddfdZdS) BaseMessagezXBase class for all messages. Attributes: role: The role of the message. rolercJtdt|j)zaConverts the message to the OpenAI format. Should be implemented by subclasses. z%to_openai method not implemented for )NotImplementedErrortyper"selfs r) to_openaizBaseMessage.to_openaips$ ""_$t**J]"_"_```r+openai_messagec2td|jd)ziConverts the OpenAI message to the Mistral format. Should be implemented by subclasses. z'from_openai method not implemented for .)r:r"clsr?s r) from_openaizBaseMessage.from_openaiws ""[CL"["["[\\\r+N)r"r2r3r4rr-r.r/r0r1__annotations__dictstrlistrr> classmethodrDr5r+r)r7r7gs %, EOUZG HHHHa4S4S#S#X:N5N0O+P%P PQaaaa]c3d3dSVX[S[nH\C\>]9^3^.^)_]dq]]][]]]r+r7c eZdZUdZejZeejed<e e e zed<de e e e e e e e e e fzfzffdZede e e e e e e e e e fzfzfddfdZdS) UserMessagezUser message. Attributes: content: The content of the message. Examples: >>> message = UserMessage(content="Can you help me to write a poem?") r8contentrct|jtr|j|jdS|jd|jDdS)*Converts the message to the OpenAI format.r8rLc6g|]}|Sr5r>.0chunks r) z)UserMessage.to_openai..s".[.[.[Uu/@/@.[.[.[r+) isinstancerLrGr8r<s r)r>zUserMessage.to_openaisJ dlC ( ( @ I$,?? ? .[.[dl.[.[.[\\\r+r?ct|dtr||S||dd|dDdS)2Converts the OpenAI message to the Mistral format.rLr8c,g|]}t|Sr5rrRs r)rUz+UserMessage.from_openai..s!iiie:5AAiiir+rO)rVrGmodel_validaterBs r)rDzUserMessage.from_openaisp nY/ 5 5 6%%n55 5!!&v.ii~^gOhiii     r+N)r"r2r3r4r-r/r8rrErGrHrrFrr>rIrDr5r+r)rKrKs!& D'%* *** 4() ))))]4S4S#S#X:N5N0O+P%P PQ]]]]   c3d3dSVX[S[nH\C\>]9^3^.^)_  dq    [    r+rKc eZdZUdZejZeejed<e e e e zzed<de e e e e e e e e efzfzffdZede e e e e e e e e efzfzfddfdZdS) SystemMessagezSystem message. Attributes: content: The content of the message. Examples: >>> message = SystemMessage(content="You are a helpful assistant.") r8rLrc*|S)rN) model_dumpr<s r)r>zSystemMessage.to_openais   r+r?c,||S)rX)r[rBs r)rDzSystemMessage.from_openais!!.111r+N)r"r2r3r4r-r.r8rrErGrHr r rFrr>rIrDr5r+r)r]r]s#(,D'%, ... 4 J./ ////!4S4S#S#X:N5N0O+P%P PQ!!!!2c3d3dSVX[S[nH\C\>]9^3^.^)_2ds222[222r+r]c XeZdZUdZejZeejed<dZ e e e e zzdzed<dZe edzed<dZeed<dee e e ee e ee efzfzffd Zed ee e e ee e ee efzfzfddfd 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?") r8NrL tool_callsFprefixrcd|ji}|jn9t|jtr |j|d<nd|jD|d<|jd|jD|d<|S)rNr8NrLc6g|]}|Sr5rQrRs r)rUz.AssistantMessage.to_openai..s""O"O"O5??#4#4"O"O"Or+c6g|]}|Sr5rQ)rS tool_calls r)rUz.AssistantMessage.to_openai..s$%]%]%] i&9&9&;&;%]%]%]r+rc)r8rLrVrGrc)r=out_dicts r)r>zAssistantMessage.to_openais DIL  <    c * * P"&,HY  "O"O$,"O"O"OHY  ? &%]%]T_%]%]%]H\ "r+r?c&|dd}|d}net|tr1g}|pgD])}|t j|*nt dt||dd}d}|t|tr|}nAt|tr d|D}nt dt|| |d||dS) rXrcNztool_calls must be a list, got rLc,g|]}t|Sr5rZrRs r)rUz0AssistantMessage.from_openai..s!YYY5e<<YYYr+zUnknown content type: r8)r8rLrc) getrVrHappendrrD ValueErrorr;rGr[)rCr?openai_tool_calls tools_callsopenai_tool_callopenai_contentrLs r)rDzAssistantMessage.from_openaisO+..|TBB  $15KK )4 0 0 ZK$5$; K K ""8#78H#I#IJJJJ KXtDU?V?VXXYY Y'++It<<37  !Z%D%D !$GG  - - NYY.YYYGGLd>6J6JLLMM M!!&v.")     r+)r"r2r3r4r-r0r8rrErLrGrHr r rcrrdboolrFrr>rIrDr5r+r)rbrbs+  &+_D'%/ "4449=GS4 J./ /$ 6===(,JX%,,,FD4S4S#S#X:N5N0O+P%P PQ  c3d3dSVX[S[nH\C\>]9^3^.^)_ dv   [   r+rbc(eZdZUdZdZedzed<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"r2r3r4rvfloatrEr5r+r)rurus2 FEDLr+ruc BeZdZUdZeeezed<ej Z e ej ed<dZ edzed<dZ edzed<deeeeeeeeeefzfzffdZed eeeeeeeeeefzfzfddfd 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") rLr8N tool_call_idrrcR|j Jd|dhS)rNN0tool_call_id must be provided for tool messages.r)exclude)rzr_r<s r)r>zToolMessage.to_openais0 ,,.`,,,x000r+messagescR||}|j Jd|S)rXNr|)r[rz)rCr~ tool_messages r)rDzToolMessage.from_openais5))(33 (446h444r+)r"r2r3r4rGrHr rEr-r1r8rrzrrFrr>rIrDr5r+r)ryrys  4 ? """" % D'%* ***#L#*###D#*14S4S#S#X:N5N0O+P%P PQ1111 4S4S#SRUXBV=V8W3X-X(X#Y^k[r+ryr8) discriminatorChatMessageType)boundUserMessageTypeAssistantMessageTypeToolMessageTypeSystemMessageTypeUATS)*r$enumrtypingrrrpydanticrtyping_extensionsrr mistral_common.baser r#r r r rr+mistral_common.protocol.instruct.tool_callsrfilterwarningsr&rGr*r-r7rKr]rbrury ChatMessageFinetuningMessagerrrrrrrEr5r+r)rsT((((((((((22222222++++++A@@@@@   ]6Gc6Gc6G6G6G6GrC&]]]]]+]]]2     +   @22222K222.= = = = = {= = = @      !1    +@ 36FFTV[V[jpVqVqVqqr K"<<{J E! '+;???'+;???w5=MNNN'+;???G/}EEE!$88?JM^^i^^^^^r+