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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"m#Z#m$Z$ ddl%m&Z&m'Z'm(Z(m)Z) ddl*m+Z+m,Z,m-Z-m.Z. e-de)eeeeeef Z/e+e)e+ee#df e+e
e#df e+ee#df e+ee#df e+ee#df e+ee#df f e"dd f Z0e'd Z1G dd de.Z2G dd deZ3dS ) zACode generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.    )annotations   )CodeInterpreterToolCodeInterpreterToolTypedDict)CompletionArgsCompletionArgsTypedDict)DocumentLibraryToolDocumentLibraryToolTypedDict)FunctionToolFunctionToolTypedDict)ImageGenerationToolImageGenerationToolTypedDict)WebSearchPremiumToolWebSearchPremiumToolTypedDict)WebSearchToolWebSearchToolTypedDict)datetime)	BaseModelNullableOptionalNullableUNSETUNSET_SENTINELget_discriminator)DiscriminatorTagmodel_serializer)ListLiteralOptionalUnion)	AnnotatedNotRequiredTypeAliasType	TypedDictAgentToolsTypedDictcode_interpreterdocument_libraryfunctionimage_generation
web_searchweb_search_premiumc                 C  s   t | ddS )Ntyper   )m r.   \/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/mistralai/models/agent.py<lambda>)   s    r0   agentc                   @  st   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S )AgentTypedDictstrmodelnameidintversionr   
created_at
updated_atzNotRequired[Nullable[str]]instructionsz&NotRequired[List[AgentToolsTypedDict]]toolsz$NotRequired[CompletionArgsTypedDict]completion_argsdescriptionz NotRequired[Nullable[List[str]]]handoffszNotRequired[AgentObject]objectN)__name__
__module____qualname____annotations__r.   r.   r.   r/   r2   0   s    
 r2   c                   @  s   e Zd ZU ded< ded< ded< ded< 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< eZ	ded< dZ
ded< edddd ZdS )Agentr3   r4   r5   r6   r7   r8   r   r9   r:   zOptionalNullable[str]r;   NzOptional[List[AgentTools]]r<   zOptional[CompletionArgs]r=   r>   zOptionalNullable[List[str]]r?   r1   zOptional[AgentObject]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,   model_fieldsitemsaliasgetpop__pydantic_fields_set__intersectionr   )selfhandleroptional_fieldsnullable_fieldsnull_default_fields
serializedr-   nfkvaloptional_nullableis_setr.   r.   r/   serialize_model^   s(   


zAgent.serialize_model)rA   rB   rC   rD   r   r;   r<   r=   r>   r?   r@   r   r[   r.   r.   r.   r/   rE   B   s$   
 rE   N)4__doc__
__future__r   codeinterpretertoolr   r   completionargsr   r   documentlibrarytoolr   r	   functiontoolr
   r   imagegenerationtoolr   r   websearchpremiumtoolr   r   websearchtoolr   r   r   mistralai.typesr   r   r   r   r   mistralai.utilsr   pydanticr   r   r   typingr   r   r   r    typing_extensionsr!   r"   r#   r$   r%   
AgentToolsAgentObjectr2   rE   r.   r.   r.   r/   <module>   sV    
