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(ChatCompletionStreamRequestStopTypedDictChatCompletionStreamRequestStop,ChatCompletionStreamRequestMessagesTypedDict	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/models/chatcompletionstreamrequest.py<lambda>4   s    r3   .ChatCompletionStreamRequestToolChoiceTypedDict%ChatCompletionStreamRequestToolChoicec                   @  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 ) $ChatCompletionStreamRequestTypedDictstrmodelz2List[ChatCompletionStreamRequestMessagesTypedDict]messageszNotRequired[Nullable[float]]temperaturezNotRequired[float]top_pzNotRequired[Nullable[int]]
max_tokenszNotRequired[bool]streamz5NotRequired[ChatCompletionStreamRequestStopTypedDict]stoprandom_seedz$NotRequired[ResponseFormatTypedDict]response_formatz*NotRequired[Nullable[List[ToolTypedDict]]]toolsz;NotRequired[ChatCompletionStreamRequestToolChoiceTypedDict]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   C   s>   
 r6   c                   @  s  e Zd ZU ded< 	 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 )%ChatCompletionStreamRequestr7   r8   z)List[ChatCompletionStreamRequestMessages]r9   zOptionalNullable[float]r:   NzOptional[float]r;   zOptionalNullable[int]r<   TzOptional[bool]r=   z)Optional[ChatCompletionStreamRequestStop]r>   r?   zOptional[ResponseFormat]r@   zOptionalNullable[List[Tool]]rA   z/Optional[ChatCompletionStreamRequestToolChoice]rB   rC   rD   rE   zOptional[Prediction]rF   rG   zYAnnotated[OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))]rH   rI   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)r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   )r:   r<   r?   rA   rE   rH   )	typemodel_fieldsitemsaliasgetpop__pydantic_fields_set__intersectionr   )selfhandleroptional_fieldsnullable_fieldsnull_default_fields
serializedr0   rE   fkvaloptional_nullableis_setr1   r1   r2   serialize_model   s(   


z+ChatCompletionStreamRequest.serialize_model)rJ   rK   rL   rM   r   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   r   rd   r1   r1   r1   r2   rN   d   sD   
 
rN   N)@__doc__
__future__r   assistantmessager   r   mistralpromptmoder   rF   r   r   responseformatr	   r
   systemmessager   r   r-   r   r   
toolchoicer   r   toolchoiceenumr   toolmessager   r   usermessager   r   mistralai.typesr   r   r   r   r   mistralai.utilsr   r   pydanticr   r   r   pydantic.functional_validatorsr    typingr!   r"   r#   typing_extensionsr$   r%   r&   r'   r7   r(   r)   r*   #ChatCompletionStreamRequestMessagesr4   r5   r6   rN   r1   r1   r1   r2   <module>   sp    

!