:i8M FSSKrSSKJr SSKJr SSKJrJr SSKJ r J r J r J r J r JrJrJr SSKJrJrJrJrJrJrJrJr SSKJr SSKJrJrJrJ r SS K!J"r" "S S \5r#"S S \\\\\45r$"SS\$5r%"SS\%5r&"SS\&5r'S\"S\#S\$4Sjr(g)N)Enum)Generic)Draft7Validator SchemaError) InvalidAssistantMessageExceptionInvalidFunctionCallException InvalidMessageStructureExceptionInvalidRequestExceptionInvalidSystemPromptExceptionInvalidToolExceptionInvalidToolMessageExceptionInvalidToolSchemaException)UATSAssistantMessageAssistantMessageTypeFinetuningAssistantMessageRolesSystemMessageTypeToolMessageTypeUserMessageType)ChatCompletionRequest)Function FunctionCallToolToolCall)TokenizerVersionc$\rSrSrSrSrSrSrSrg)ValidationMode%zEnum for the validation mode. Attributes: serving: The serving mode. finetuning: The finetuning mode. test: The test mode. Examples: >>> mode = ValidationMode.serving serving finetuningtestN) __name__ __module__ __qualname____firstlineno____doc__r r!r"__static_attributes__r#d/home/james-whalen/.local/lib/python3.13/site-packages/mistral_common/protocol/instruct/validator.pyrr%s GJ Dr*rc\rSrSr%SrSr\\S'\R4S\4Sjjr S\ \ S\S S 4S jr S \S \\ 4S jrS\S S 4SjrS\ \S S 4SjrS\S S 4SjrS\S S 4SjrS\S S 4SjrS\S S 4SjrS\S\S S 4SjrS"S\S\S S 4SjjrS\ \ S S 4Sjr S\ \ S S 4Sjr!S\ S\S S 4Sjr"S\ \ S\S S 4Sjr#S\ \ S S 4S jr$S!r%g )#MistralRequestValidator6aoValidator for Mistral requests. This class validates the structure and content of Mistral requests. Examples: >>> from mistral_common.protocol.instruct.messages import UserMessage, AssistantMessage >>> validator = MistralRequestValidator() >>> messages = [UserMessage(content="Hello how are you ?")] >>> validator.validate_messages(messages, False) F_allow_tool_call_and_contentmodecXlg)zrInitializes the `MistralRequestValidator`. Args: mode: The validation mode. Defaults to ValidationMode.test. N_mode)selfr0s r+__init__ MistralRequestValidator.__init__Ds  r*messagescontinue_final_messagereturnNcDURXS9 URU5 g)aValidates the list of messages. Args: messages: The list of messages to validate. continue_final_message: Whether to continue the final message. Examples: >>> from mistral_common.protocol.instruct.messages import UserMessage, AssistantMessage >>> validator = MistralRequestValidator() >>> messages = [AssistantMessage(content="Hi"), UserMessage(content="Hello")] >>> validator.validate_messages(messages, False) r8N) _validate_message_list_structure_validate_message_list_contentr4r7r8s r+validate_messages)MistralRequestValidator.validate_messagesLs# --h-f ++H5r*requestcUR[R:XaURc [ S5eUR UR URS9 URUR=(d /5 U$)a|Validates the request Args: request: The request to validate. Returns: The validated request. Examples: >>> from mistral_common.protocol.instruct.messages import UserMessage >>> validator = MistralRequestValidator() >>> request = ChatCompletionRequest(messages=[UserMessage(content="Hello")]) >>> validated_request = validator.validate_request(request) z1Model name parameter is required for serving moder;) r3rr modelr r?r7r8_validate_toolstools)r4rAs r+validate_request(MistralRequestValidator.validate_request\sk ::// /}}$-.abb w//HfHfg W]]0b1r*functionc[R"UR5 [ R"SUR5(d[SURS35eg![an[ SUR 35eSnAff=f)z- Checks: - That the function schema is valid zInvalid tool schema: N^[a-zA-Z0-9_-]{1,64}$Function name was [ but must be a-z, A-Z, 0-9, or contain underscores and dashes, with a maximum length of 64.) r check_schema parametersrrmessagerematchnamer )r4rHes r+_validate_function*MistralRequestValidator._validate_functionxs  R  ( ()<)< =xx0(--@@&$X]]O4RR A R,/DQYYK-PQ Q Rs A"" B ,BB rEcLUHnURUR5 M g)z+ Checks: - That the tool schemas are valid N)rTrH)r4rEtools r+rD'MistralRequestValidator._validate_toolss D  # #DMM 2r*rOcg)Nr#r4rOs r+_validate_user_message.MistralRequestValidator._validate_user_messages r*cURb@[R"SUR5(d[SURS35egg)z" Checks: - The tool name is valid NrJrKrL)rRrPrQr rZs r+_validate_tool_message.MistralRequestValidator._validate_tool_messagesS << #884gllCC1(7VVD $r*c4URc [S5eg)z. Checks: - That the system prompt has content NzSystem prompt must have content)contentr rZs r+_validate_system_message0MistralRequestValidator._validate_system_messages ?? "./PQ Q #r* function_callc[R"SUR5(d[SURS35eg)z3 Checks: - That the function call has a valid name rJrKrLN)rPrQrRr)r4rds r+_validate_function_call/MistralRequestValidator._validate_function_callsI xx0-2D2DEE.$]%7%7$89RR Fr* tool_callis_last_messagec:URUR5 g)z3 Checks: - That the tool call has a valid function N)rfrH)r4rhris r+_validate_tool_call+MistralRequestValidator._validate_tool_calls $$Y%7%78r*cUR(d7[UR5[UR5:Xa [ S5eURb"URHnUR X2S9 M UR [R:Xa=[U[5(a(URbURS;a [ S5eUR(aU(d [ S5egg)zr Checks: - That the assistant message has either text or tool_calls, but not both - That the tool calls are valid zGAssistant message must have either content or tool_calls, but not both.Nri)rz.Assistant message weight must be either 0 or 1z7Assistant message with prefix True must be last message) r/boolra tool_callsrrkr3rr! isinstancerweightprefix)r4rOrirhs r+_validate_assistant_message3MistralRequestValidator._validate_assistant_messages11W__8MQUV]VhVhQi8i2Y     )$// (((T0 ::22 2z'Ke7f7f~~)gnnF.J67ghh >>"67pqq# r*cSnSnUHnUcURnMUR[R:XaUS-nOQUR[R:Xa3US:wa [ S5eUR b[ UR 5nURnM US:wa)UR[R:Xa [ S5eUS:a*UR[R:Xa [ S5egg)z| Checks: - That the number of tool calls and tool messages are the same - That the tool calls are followed by tool messages Nrro3Not the same number of function calls and responses#More tool responses than tool calls) rolerrW assistantr rqlenr3rr r!)r4r7 prev_roleexpected_tool_messagesrOs r+._validate_tool_calls_followed_by_tool_messagesFMistralRequestValidator._validate_tool_calls_followed_by_tool_messagess  !"G #LL ||uzz)&!+&0*Q.:;pqq%%1-01C1C-D* I# & "Q &4::9O9O+O23hi i #a 'DJJ.:S:S,S23XY Y-T 'r*crSnUGH.nURnUGbU[R:Xa0[R[R[R1nOU[R:Xa0[R[R[R1nOxU[R:Xa0[R[R[R 1nO4U[R :Xa [R[R 1nUW;a[ SUSUS35eUnGM1 g)z` Validates the order of the messages, for example user -> assistant -> user -> assistant -> ... NzUnexpected role 'z' after role '')rzrsystemuserr{rWr )r4r7 previous_rolerO current_roleexpected_roless r+_validate_message_order/MistralRequestValidator._validate_message_orders G"< VWX)M% r*c6URnUR[R:Xa5U[R :wa[ SU35eU(a [ S5eg[U[5=(a UR(+=(a U(+nUR[R[R1;nU(aU(a[ SU35eU(a4U[R :wdUR(a[ SU35egg)Nz4Expected last role Assistant for finetuning but got z0Cannot continue final message in finetuning modezuExpected last role User or Tool (or Assistant with prefix or continue_final_message set to True) for serving but got zkExpected last role Assistant with prefix False for serving with continue_final_message set to True but got ) rzr3rr!rr{r rrrrtrrW)r4rOr8last_message_role bad_assistantbad_roles r+_validate_last_message.MistralRequestValidator._validate_last_messages#LL ::22 2 EOO36JK\J]^&67ijj&'w0@Aw'..FXwaw]wM||EJJ +CCH6++<*=?(->%//-QU\UcUc6$$5#68Vd'r*c[U5S:Xa [S5e[U5S:Xa<USR[R[R 1;a [S5eUR [R:Xd[U5S:aURUSUS9 URU5 URU5 g)z Validates the structure of the list of messages For example the messages must be in the correct order of user/assistant/tool rz+Conversation must have at least one messageroz=Conversation must start with a user message or system messager;N) r|r rzrrrr3rr!rrrr>s r+r<8MistralRequestValidator._validate_message_list_structure)s x=A 23`a a x=A { ELL'AA67vww ::22 2c(ma6G  ' ' Mc ' d $$X. ;;HEr*c[U5Hup#UR[R:XaUR U5 M6UR[R :Xa UR X2[U5S- :HS9 MtUR[R:XaURU5 MUR[R:XaURU5 M[S[U535e g)z' Validates the content of the messages rornzUnsupported message type N) enumeraterzrrr[r{rur|rWr^rrbr type)r4r7idxrOs r+r=6MistralRequestValidator._validate_message_list_content?s &h/LC||uzz)++G4000QTU]Q^abQbJb0c+++G4---g6-0I$w-.YZZ0r*r2)F)&r$r%r&r'r(r/rp__annotations__rr"r5listrr?rrFrrTrrDrr[rr^rrbrrfrrkrrurrrr<r=r)r#r*r+r-r-6s */ $..<.A.A^6$t*6d6W[6 (=BWX\B]88 3T$Z3D3 o $  o $ R0ARdR \ d 9X99QU9r3GrZ^rkor:ZtDzZVZZ@)T )t)2dDUY0FdF]aFfjF,[tDz[d[r*r-cd^\rSrSrSrS\SS4SjrS\S\SS4S jr S\ S \SS4U4S jjr S r U=r $) MistralRequestValidatorV3iQzValidator for v3 Mistral requests. This validator adds additional validation for tool call IDs. Examples: >>> validator = MistralRequestValidatorV3() rOr9NcJURb?[R"SUR5(d[SURS35eURc [ S5e[R"SUR5(d[SURS35eg) z: Checks: - The tool name is valid - Tool call id is valid NrJrKrLzTool call id has to be defined.^[a-zA-Z0-9]{9}$Tool call id was / but must be a-z, A-Z, 0-9, with a length of 9.)rRrPrQr tool_call_idr rZs r+r^0MistralRequestValidatorV3._validate_tool_messageZs << #884gllCC1(7VV    ')*KL Lxx+W-A-ABB-#G$8$8#99hi Cr*rhricURS:wa?[R"SUR5(d[SURS35eUR[ R :Xa$U(dURS:Xa Sn[U5eUR[ R:XaURS:Xa [S5eURUR5 g)z, Validate that the tool call has a valid ID nullrrrzXTool call id of assistant message that is not last has to be defined in finetuning mode.z/Tool call id has to be defined in serving mode.N) idrPrQrr3rr!r rfrH)r4rhri err_messages r+rk-MistralRequestValidatorV3._validate_tool_callos <<6 !88/>>2' ~5de ::22 2?y||_eOetK.{; ; ::// /ILLF4J./`a a $$Y%7%78r*r8c>[TU]X5 UR[R:Xa2UR b$UR HnUR USS9 M ggg)NTrn)superrr3rr!rqrk)r4rOr8rh __class__s r+r0MistralRequestValidatorV3._validate_last_messages] &wG ::22 2!!-!(!3!3I,,Y,M"4. 3r*r#)r$r%r&r'r(rr^rrprkrrr) __classcell__)rs@r+rrQs[o$*9X99QU9$NdNDNUYNNr*rc(\rSrSr%SrSr\\S'Srg)MistralRequestValidatorV5izValidator for v5 Mistral requests. This validator allows for both tool calls and content in the assistant message. Examples: >>> validator = MistralRequestValidatorV5() Tr/r#N) r$r%r&r'r(r/rprr)r#r*r+rrs*. $-r*rc,\rSrSrS\\SS4SjrSrg)MistralRequestValidatorV13ir7r9NcSn[5n[5nUGHJnUcURnMUR[R:XaFURnXd;a[ SUS35eXc;a[ SUS35eUR U5 OUR[R:Xa[U5[U5:wa [ S5eUR5 UR5 URbWURHGnURU;a[ SURS35eUR UR5 MI URnGMM [U5[U5:wa)UR[R:Xa [ S5e[U5[U5:a*UR[R:Xa [ S5egg)z Checks: - That the number and ids of tool calls and tool messages are the same - That the tool calls are followed by tool messages - That tool calls have distinct ids for a given assistant message NzDuplicate tool call id z in tool resultszUnexpected tool call id rxz in assistant messagery)setrzrrWrr addr{r|clearrqrr3rr r!)r4r7r}expected_tool_idsobserved_tool_idsrOrrhs r+rIMistralRequestValidatorV13._validate_tool_calls_followed_by_tool_messagess &)e&)eG #LL ||uzz)&33 4:=TUaTbbr;stt8:=UVbUccs;tuu!%%l30()S1B-CC:;pqq!'')!'')%%1%,%7%7 $<<+<<"B"9),,G\ ]#*--ill; &8 I= @  !S):%; ; nNdNd@d23hi i " #c*;&< <~OhOhAh23XY YBi rs 3      K CT"X[go7K_^o&opX[v8N 78Nv . 9 ..Z!:.Zb + > F] r*