o
    i)                  	   @  s  d Z ddlmZ ddlmZmZ ddlmZ ddlm	Z	m
Z
 ddlmZmZ ddlmZmZ dd	lmZmZ dd
lmZmZ ddlmZ ddlmZmZ ddlmZmZ ddlm Z m!Z!m"Z"m#Z#m$Z$ ddl%m&Z&m'Z' ddl(m)Z)m*Z*m+Z+ ddl,m-Z- ddl.m/Z/m0Z0m1Z1 ddl2m3Z3m4Z4m5Z5m6Z6 e5de1e7e/e7 f Z8	 e5de1e7e/e7 f Z9	 e5de1eeeef Z:e3e1e3ee*df e3ee*df e3ee*df e3ee*df f e)dd f Z;e5de1eef Z<e5de1eef Z=G dd  d e6Z>G d!d" d"e Z?d#S )$zACode generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.    )annotations   )AssistantMessageAssistantMessageTypedDict)MistralPromptMode)
PredictionPredictionTypedDict)ResponseFormatResponseFormatTypedDict)SystemMessageSystemMessageTypedDict)ToolToolTypedDict)
ToolChoiceToolChoiceTypedDict)ToolChoiceEnum)ToolMessageToolMessageTypedDict)UserMessageUserMessageTypedDict)	BaseModelNullableOptionalNullableUNSETUNSET_SENTINEL)get_discriminatorvalidate_open_enum)DiscriminatorTagmodel_serializer)PlainValidator)ListOptionalUnion)	AnnotatedNotRequiredTypeAliasType	TypedDict"ChatCompletionRequestStopTypedDictChatCompletionRequestStop&ChatCompletionRequestMessagesTypedDict	assistantsystemtooluserc                 C  s   t | ddS )Nrole)r   )m r1   r/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/mistralai_azure/models/chatcompletionrequest.py<lambda>:   s    r3   (ChatCompletionRequestToolChoiceTypedDictChatCompletionRequestToolChoicec                   @  s   e Zd ZU ded< 	 ded< 	 ded< 	 ded< 	 d	ed
< 	 ded< 	 ded< 	 d	ed< 	 ded< ded< ded< ded< 	 ded< 	 d	ed< 	 ded< ded< ded< 	 ded< dS ) ChatCompletionRequestTypedDictz,List[ChatCompletionRequestMessagesTypedDict]messageszNotRequired[str]modelzNotRequired[Nullable[float]]temperaturezNotRequired[float]top_pzNotRequired[Nullable[int]]
max_tokenszNotRequired[bool]streamz/NotRequired[ChatCompletionRequestStopTypedDict]stoprandom_seedz$NotRequired[ResponseFormatTypedDict]response_formatz*NotRequired[Nullable[List[ToolTypedDict]]]toolsz5NotRequired[ChatCompletionRequestToolChoiceTypedDict]tool_choicepresence_penaltyfrequency_penaltynz NotRequired[PredictionTypedDict]
predictionparallel_tool_callsz(NotRequired[Nullable[MistralPromptMode]]prompt_modesafe_promptN)__name__
__module____qualname____annotations__r1   r1   r1   r2   r6   I   s@   
 r6   c                   @  s  e Zd ZU ded< 	 dZded< 	 eZded< 	 dZd	ed
< 	 eZded< 	 dZ	ded< 	 dZ
ded< 	 eZded< 	 dZded< eZded< dZded< dZd	ed< 	 dZd	ed< 	 eZded< 	 dZded< dZded< eZded < 	 dZded!< 	 ed"d#d$d% ZdS )&ChatCompletionRequestz#List[ChatCompletionRequestMessages]r7   azureaizOptional[str]r8   zOptionalNullable[float]r9   NzOptional[float]r:   zOptionalNullable[int]r;   FzOptional[bool]r<   z#Optional[ChatCompletionRequestStop]r=   r>   zOptional[ResponseFormat]r?   zOptionalNullable[List[Tool]]r@   z)Optional[ChatCompletionRequestToolChoice]rA   rB   rC   rD   zOptional[Prediction]rE   rF   zYAnnotated[OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))]rG   rH   wrap)modec                 C  s   g d}g d}g }|| }i }t | j D ]D\}}|jp|}	||	}
||	d  |	|v o2|	|v }| j|hp=|	|v }|
d urK|
tkrK|
||	< q|
tkr[|	|vsW|r[|r[|
||	< q|S )N)r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   )r9   r;   r>   r@   rD   rG   )	typemodel_fieldsitemsaliasgetpop__pydantic_fields_set__intersectionr   )selfhandleroptional_fieldsnullable_fieldsnull_default_fields
serializedr0   rD   fkvaloptional_nullableis_setr1   r1   r2   serialize_model   s(   


z%ChatCompletionRequest.serialize_model)rI   rJ   rK   rL   r8   r   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   r   rd   r1   r1   r1   r2   rM   k   sF   
 
rM   N)@__doc__
__future__r   assistantmessager   r   mistralpromptmoder   rE   r   r   responseformatr	   r
   systemmessager   r   r-   r   r   
toolchoicer   r   toolchoiceenumr   toolmessager   r   usermessager   r   mistralai_azure.typesr   r   r   r   r   mistralai_azure.utilsr   r   pydanticr   r   r   pydantic.functional_validatorsr    typingr!   r"   r#   typing_extensionsr$   r%   r&   r'   strr(   r)   r*   ChatCompletionRequestMessagesr4   r5   r6   rM   r1   r1   r1   r2   <module>   sp    

"