o
    ib                  	   @  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*AgentsCompletionStreamRequestStopTypedDict!AgentsCompletionStreamRequestStop.AgentsCompletionStreamRequestMessagesTypedDict	assistantsystemtooluserc                 C  s   t | ddS )Nrole)r   )m r1   t/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/mistralai/models/agentscompletionstreamrequest.py<lambda>4   s    r3   0AgentsCompletionStreamRequestToolChoiceTypedDict'AgentsCompletionStreamRequestToolChoicec                   @  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S )&AgentsCompletionStreamRequestTypedDictz4List[AgentsCompletionStreamRequestMessagesTypedDict]messagesstragent_idzNotRequired[Nullable[int]]
max_tokenszNotRequired[bool]streamz7NotRequired[AgentsCompletionStreamRequestStopTypedDict]stoprandom_seedz$NotRequired[ResponseFormatTypedDict]response_formatz*NotRequired[Nullable[List[ToolTypedDict]]]toolsz=NotRequired[AgentsCompletionStreamRequestToolChoiceTypedDict]tool_choicezNotRequired[float]presence_penaltyfrequency_penaltynz NotRequired[PredictionTypedDict]
predictionparallel_tool_callsz(NotRequired[Nullable[MistralPromptMode]]prompt_modeN)__name__
__module____qualname____annotations__r1   r1   r1   r2   r6   C   s2   
 r6   c                   @  s   e Zd ZU ded< 	 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< 	 edddd  Zd
S )!AgentsCompletionStreamRequestz+List[AgentsCompletionStreamRequestMessages]r7   r8   r9   zOptionalNullable[int]r:   TzOptional[bool]r;   Nz+Optional[AgentsCompletionStreamRequestStop]r<   r=   zOptional[ResponseFormat]r>   zOptionalNullable[List[Tool]]r?   z1Optional[AgentsCompletionStreamRequestToolChoice]r@   zOptional[float]rA   rB   rC   zOptional[Prediction]rD   rE   zYAnnotated[OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))]rF   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   )r:   r=   r?   rC   rF   )	typemodel_fieldsitemsaliasgetpop__pydantic_fields_set__intersectionr   )selfhandleroptional_fieldsnullable_fieldsnull_default_fields
serializedr0   rC   fkvaloptional_nullableis_setr1   r1   r2   serialize_model   s(   


z-AgentsCompletionStreamRequest.serialize_model)rG   rH   rI   rJ   r   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   r   ra   r1   r1   r1   r2   rK   ^   s8   
 
rK   N)@__doc__
__future__r   assistantmessager   r   mistralpromptmoder   rD   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'   r8   r(   r)   r*   %AgentsCompletionStreamRequestMessagesr4   r5   r6   rK   r1   r1   r1   r2   <module>   sp    

