o
    i                     @  s   d 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 ddlmZmZmZmZmZ dd	lmZ dd
lmZ ddlmZmZ G dd deZG dd deZdS )zACode generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.    )annotations   )CompletionArgsStopCompletionArgsStopTypedDict)
PredictionPredictionTypedDict)ResponseFormatResponseFormatTypedDict)ToolChoiceEnum)	BaseModelNullableOptionalNullableUNSETUNSET_SENTINEL)model_serializer)Optional)NotRequired	TypedDictc                   @  sb   e Zd ZU dZ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 )CompletionArgsTypedDict.White-listed arguments from the completion APIz2NotRequired[Nullable[CompletionArgsStopTypedDict]]stopzNotRequired[Nullable[float]]presence_penaltyfrequency_penaltytemperaturetop_pzNotRequired[Nullable[int]]
max_tokensrandom_seedz*NotRequired[Nullable[PredictionTypedDict]]
predictionz.NotRequired[Nullable[ResponseFormatTypedDict]]response_formatzNotRequired[ToolChoiceEnum]tool_choiceN)__name__
__module____qualname____doc____annotations__ r%   r%   e/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/mistralai/models/completionargs.pyr      s   
 r   c                   @  s   e Zd ZU dZeZded< eZded< eZded< eZ	ded< eZ
ded< eZd	ed
< eZd	ed< eZded< eZded< dZded< edddd ZdS )CompletionArgsr   z$OptionalNullable[CompletionArgsStop]r   zOptionalNullable[float]r   r   r   r   zOptionalNullable[int]r   r   zOptionalNullable[Prediction]r   z OptionalNullable[ResponseFormat]r   NzOptional[ToolChoiceEnum]r   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   r   r   r   )	r   r   r   r   r   r   r   r   r   )	typemodel_fieldsitemsaliasgetpop__pydantic_fields_set__intersectionr   )selfhandleroptional_fieldsnullable_fieldsnull_default_fields
serializedmnfkvaloptional_nullableis_setr%   r%   r&   serialize_model4   s(   


zCompletionArgs.serialize_model)r    r!   r"   r#   r   r   r$   r   r   r   r   r   r   r   r   r   r   r?   r%   r%   r%   r&   r'      s   
 r'   N)r#   
__future__r   completionargsstopr   r   r   r   r   responseformatr   r	   toolchoiceenumr
   mistralai.typesr   r   r   r   r   pydanticr   typingr   typing_extensionsr   r   r   r'   r%   r%   r%   r&   <module>   s    