
    i                         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 ddlmZ dd	lmZmZmZmZmZ  G d
 de      Zy)zACode generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.   )BaseSDK    )modelsutils)HookContext)OptionalNullableUNSET)get_security_from_env)unmarshal_json_response)AnyListMappingOptionalUnionc                      e Zd ZdZeddeeedddd	dededee   deee	e
j                     e	e
j                     f      deee
j                  e
j                  f      d	ee   d
ee	e      deej                      dee   dee   deeeef      de
j&                  fdZeddeeedddd	dededee   deee	e
j                     e	e
j                     f      deee
j                  e
j                  f      d	ee   d
ee	e      deej                      dee   dee   deeeef      de
j&                  fdZddedddddee   dee   deej                      dee   dee   deeeef      de	e
j&                     fdZddedddddee   dee   deej                      dee   dee   deeeef      de	e
j&                     fdZedddddedeej                      dee   dee   deeeef      de
j&                  fdZedddddedeej                      dee   dee   deeeef      de
j&                  fdZeddeeeeedddddedee   deee	e
j4                     e	e
j6                     f      deee
j                  e
j                  f      dee   dee   d	ee   d
ee	e      deej                      dee   dee   deeeef      de
j&                  fdZeddeeeeedddddedee   deee	e
j4                     e	e
j6                     f      deee
j                  e
j                  f      dee   dee   d	ee   d
ee	e      deej                      dee   dee   deeeef      de
j&                  fdZeddddded edeej                      dee   dee   deeeef      de
j&                  fd!Zeddddded edeej                      dee   dee   deeeef      de
j&                  fd"Zy)#MistralAgentsz(beta) Agents APIN)	instructionstoolscompletion_argsdescriptionhandoffsretries
server_url
timeout_mshttp_headersmodelnamer   r   r   r   r   r   r   r   r   returnc                d   d}d}|
| j                   j                  }
|	|	}n| j                  ||      }t        j                  |t        j                  |t        t        t        j                              t        j                  |t        t        j                           ||||      | j                  dd||ddddd|| j                   j                  fd	|

      }|t        k(  r2| j                   j                  t        ur| j                   j                  }d}t        |t
        j                         r|g df}| j#                  t%        | j                   |xs ddg t'        | j                   j                  t        j(                              |g d|      }d}t        j*                  |dd      rt-        t        j.                  |      S t        j*                  |dd      r0t-        t        j0                  |      }t        j2                  ||      t        j*                  |dd      r,t        j4                  |      }t        j6                  d||      t        j*                  |dd      r,t        j4                  |      }t        j6                  d||      t        j6                  d|      )  Create a agent that can be used within a conversation.

        Create a new agent giving it instructions, tools, description. The agent is then available to be used as a regular assistant in a conversation or as part of an agent pool from which it can be used.

        :param model:
        :param name:
        :param instructions: Instruction prompt the model will follow during the conversation.
        :param tools: List of tools which are available to the model during the conversation.
        :param completion_args: White-listed arguments from the completion API
        :param description:
        :param handoffs:
        :param retries: Override the default retry configuration for this method
        :param server_url: Override the default server URL for this method
        :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
        :param http_headers: Additional headers to set or replace on requests.
        Nr   r   r   r   r   r   r   POST
/v1/agentsTF
user-agentapplication/jsonc                  R    t        j                   dddt        j                        S NFjsonr   serialize_request_bodyr   AgentCreationRequestrequests   V/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/mistralai/mistral_agents.py<lambda>z&MistralAgents.create.<locals>.<lambda>Z   #    (D(Dvv/J/J)     methodpathbase_urlurl_variablesr-   request_body_requiredrequest_has_path_paramsrequest_has_query_paramsuser_agent_headeraccept_header_valuer   securityget_serialized_bodyr   429500502503504 agents_api_v1_agents_createconfigr5   operation_idoauth2_scopessecurity_source4224XX5XXhook_ctxr-   error_status_codesretry_config200rL   rM   *API error occurredrN   Unexpected response received)sdk_configurationr   _get_urlr   r+   r   get_pydantic_modelr   r   AgentCreationRequestToolsCompletionArgs_build_requestr<   r	   rR   
isinstanceRetryConfig
do_requestr   r
   Securitymatch_responser   AgentHTTPValidationErrorDataHTTPValidationErrorstream_to_textSDKErrorselfr   r   r   r   r   r   r   r   r   r   r   r5   r6   reqrR   http_resresponse_datahttp_res_textr-   s                      @r.   createzMistralAgents.create   s   L //::J!!H}}X}=H--%**xV%E%E FG "44&*?*?!@ #
 !!'"&$)%)* 2%++44! "! " 
& e%%22%?00==gu001#%HIL?? --!R:  5**33V__! 4% # 
 "%1CD*6<<BB%1CD3..M ,,]HEE%5!00:M//"6-PP%5!00:M//"6-PPoo<hGGr1   c                  K   d}d}|
| j                   j                  }
|	|	}n| j                  ||      }t        j                  |t        j                  |t        t        t        j                              t        j                  |t        t        j                           ||||      | j                  dd||ddddd|| j                   j                  fd	|

      }|t        k(  r2| j                   j                  t        ur| j                   j                  }d}t        |t
        j                         r|g df}| j#                  t%        | j                   |xs ddg t'        | j                   j                  t        j(                              |g d|       d{   }d}t        j*                  |dd      rt-        t        j.                  |      S t        j*                  |dd      r0t-        t        j0                  |      }t        j2                  ||      t        j*                  |dd      r4t        j4                  |       d{   }t        j6                  d||      t        j*                  |dd      r4t        j4                  |       d{   }t        j6                  d||      t        j6                  d|      7 +7 7 6w)r    Nr!   r"   r#   TFr$   r%   c                  R    t        j                   dddt        j                        S r'   r)   r,   s   r.   r/   z,MistralAgents.create_async.<locals>.<lambda>   r0   r1   r2   r>   rD   rE   rF   rK   rO   rS   rL   rM   rT   rU   rN   rV   )rW   r   rX   r   r+   r   rY   r   r   rZ   r[   _build_request_asyncr<   r	   rR   r]   r^   do_request_asyncr   r
   r`   ra   r   rb   rc   rd   stream_to_text_asyncrf   rg   s                      @r.   create_asynczMistralAgents.create_async   s    L //::J!!H}}X}=H--%**xV%E%E FG "44&*?*?!@ #
 '''"&$)%)* 2%++44! "! ( 
& e%%22%?00==gu001#%HIL.. --!R:  5**33V__! 4% / 
 
 "%1CD*6<<BB%1CD3..M ,,]HEE%5"'"<"<X"FFM//"6-PP%5"'"<"<X"FFM//"6-PPoo<hGG=
0 G Gs8   FK!K"B-KKA
KK2KKKr      )page	page_sizer   r   r   r   ru   rv   c                   d}d}|| j                   j                  }||}n| j                  ||      }t        j                  ||      }	| j                  dd|||	ddddd|| j                   j                  |	      }
|t        k(  r2| j                   j                  t        ur| j                   j                  }d}t        |t        j                        r|g d
f}| j                  t        | j                   |xs ddg t        | j                   j                  t        j                              |
g d|      }d}t        j                   |dd      r!t#        t$        t        j&                     |      S t        j                   |dd      r0t#        t        j(                  |      }t        j*                  ||      t        j                   |dd      r,t        j,                  |      }t        j.                  d||      t        j                   |dd      r,t        j,                  |      }t        j.                  d||      t        j.                  d|      a  List agent entities.

        Retrieve a list of agent entities sorted by creation time.

        :param page:
        :param page_size:
        :param retries: Override the default retry configuration for this method
        :param server_url: Override the default server URL for this method
        :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
        :param http_headers: Additional headers to set or replace on requests.
        N)ru   rv   GETr#   FTr$   r%   r3   r4   r5   r6   r-   r7   r8   r9   r:   r;   r   r<   r   r>   rD   agents_api_v1_agents_listrF   rK   rO   rS   rL   rM   rT   rU   rN   rV   )rW   r   rX   r   AgentsAPIV1AgentsListRequestr\   r<   r	   rR   r]   r   r^   r_   r   r
   r`   ra   r   r   rb   rc   rd   re   rf   rh   ru   rv   r   r   r   r   r5   r6   r-   ri   rR   rj   rk   rl   s                  r.   listzMistralAgents.list  s3   * //::J!!H}}X}=H55

 !!'"'$)%)* 2%++44! " 
  e%%22%?00==gu001#%HIL?? --!R8  5**33V__! 4% # 
 "%1CD*4+=xHH%1CD3..M ,,]HEE%5!00:M//"6-PP%5!00:M//"6-PPoo<hGGr1   c                  K   d}d}|| j                   j                  }||}n| j                  ||      }t        j                  ||      }	| j                  dd|||	ddddd|| j                   j                  |	      }
|t        k(  r2| j                   j                  t        ur| j                   j                  }d}t        |t        j                        r|g d
f}| j                  t        | j                   |xs ddg t        | j                   j                  t        j                              |
g d|       d{   }d}t        j                   |dd      r!t#        t$        t        j&                     |      S t        j                   |dd      r0t#        t        j(                  |      }t        j*                  ||      t        j                   |dd      r4t        j,                  |       d{   }t        j.                  d||      t        j                   |dd      r4t        j,                  |       d{   }t        j.                  d||      t        j.                  d|      7 27 7 6wrx   )rW   r   rX   r   r|   rp   r<   r	   rR   r]   r   r^   rq   r   r
   r`   ra   r   r   rb   rc   rd   rr   rf   r}   s                  r.   
list_asynczMistralAgents.list_async]  sW    * //::J!!H}}X}=H55

 '''"'$)%)* 2%++44! ( 
  e%%22%?00==gu001#%HIL.. --!R8  5**33V__! 4% / 
 
 "%1CD*4+=xHH%1CD3..M ,,]HEE%5"'"<"<X"FFM//"6-PP%5"'"<"<X"FFM//"6-PPoo<hGG=
0 G Gs8   D<I7>I0?B4I73I34A
I7>I5?2I73I75I7)r   r   r   r   agent_idc                   d}d}|| j                   j                  }||}n| j                  ||      }t        j                  |      }| j                  dd|||ddddd|| j                   j                  |	      }	|t        k(  r2| j                   j                  t        ur| j                   j                  }d}
t        |t        j                        r|g d
f}
| j                  t        | j                   |xs ddg t        | j                   j                  t        j                              |	g d|
      }d}t        j                   |dd      rt#        t        j$                  |      S t        j                   |dd      r0t#        t        j&                  |      }t        j(                  ||      t        j                   |dd      r,t        j*                  |      }t        j,                  d||      t        j                   |dd      r,t        j*                  |      }t        j,                  d||      t        j,                  d|      a  Retrieve an agent entity.

        Given an agent retrieve an agent entity with its attributes.

        :param agent_id:
        :param retries: Override the default retry configuration for this method
        :param server_url: Override the default server URL for this method
        :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
        :param http_headers: Additional headers to set or replace on requests.
        N)r   ry   /v1/agents/{agent_id}FTr$   r%   rz   r>   rD   agents_api_v1_agents_getrF   rK   rO   rS   rL   rM   rT   rU   rN   rV   )rW   r   rX   r   AgentsAPIV1AgentsGetRequestr\   r<   r	   rR   r]   r   r^   r_   r   r
   r`   ra   r   rb   rc   rd   re   rf   rh   r   r   r   r   r   r5   r6   r-   ri   rR   rj   rk   rl   s                 r.   getzMistralAgents.get  s+   & //::J!!H}}X}=H44
 !!('"'$(%)* 2%++44! " 
  e%%22%?00==gu001#%HIL?? --!R7  5**33V__! 4% # 
 "%1CD*6<<BB%1CD3..M ,,]HEE%5!00:M//"6-PP%5!00:M//"6-PPoo<hGGr1   c                  K   d}d}|| j                   j                  }||}n| j                  ||      }t        j                  |      }| j                  dd|||ddddd|| j                   j                  |	      }	|t        k(  r2| j                   j                  t        ur| j                   j                  }d}
t        |t        j                        r|g d
f}
| j                  t        | j                   |xs ddg t        | j                   j                  t        j                              |	g d|
       d{   }d}t        j                   |dd      rt#        t        j$                  |      S t        j                   |dd      r0t#        t        j&                  |      }t        j(                  ||      t        j                   |dd      r4t        j*                  |       d{   }t        j,                  d||      t        j                   |dd      r4t        j*                  |       d{   }t        j,                  d||      t        j,                  d|      7 +7 7 6wr   )rW   r   rX   r   r   rp   r<   r	   rR   r]   r   r^   rq   r   r
   r`   ra   r   rb   rc   rd   rr   rf   r   s                 r.   	get_asynczMistralAgents.get_async  sO    & //::J!!H}}X}=H44
 ''('"'$(%)* 2%++44! ( 
  e%%22%?00==gu001#%HIL.. --!R7  5**33V__! 4% / 
 
 "%1CD*6<<BB%1CD3..M ,,]HEE%5"'"<"<X"FFM//"6-PP%5"'"<"<X"FFM//"6-PPoo<hGG=
0 G Gs8   D;I/=I(>B-I/+I+,A
I/6I-72I/+I/-I/)r   r   r   r   r   r   r   r   r   r   r   c                   d}d}|| j                   j                  }|
|
}n| j                  ||      }t        j                  |t        j
                  |t        j                  |t        t        t        j                              t        j                  |t        t        j                           ||||            | j                  dd||ddddd|| j                   j                  fd	|
      }|	t        k(  r2| j                   j                  t        ur| j                   j                  }	d}t!        |	t        j"                        r|	g df}| j%                  t'        | j                   |xs ddg t)        | j                   j                  t        j*                              |g d|      }d}t        j,                  |dd      rt/        t        j0                  |      S t        j,                  |dd      r0t/        t        j2                  |      }t        j4                  ||      t        j,                  |dd      r,t        j6                  |      }t        j8                  d||      t        j,                  |dd      r,t        j6                  |      }t        j8                  d||      t        j8                  d|      )=  Update an agent entity.

        Update an agent attributes and create a new version.

        :param agent_id:
        :param instructions: Instruction prompt the model will follow during the conversation.
        :param tools: List of tools which are available to the model during the conversation.
        :param completion_args: White-listed arguments from the completion API
        :param model:
        :param name:
        :param description:
        :param handoffs:
        :param retries: Override the default retry configuration for this method
        :param server_url: Override the default server URL for this method
        :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
        :param http_headers: Additional headers to set or replace on requests.
        Nr!   r   agent_update_requestPATCHr   Tr$   r%   c                  f    t        j                   j                  dddt        j                        S r'   r   r*   r   r   AgentUpdateRequestr,   s   r.   r/   z&MistralAgents.update.<locals>.<lambda>  -    (D(D,,))) r1   r2   r>   rD   agents_api_v1_agents_updaterF   rK   rO   rS   rL   rM   rT   rU   rN   rV   )rW   r   rX   r   AgentsAPIV1AgentsUpdateRequestr   r   rY   r   r   AgentUpdateRequestToolsr[   r\   r<   r	   rR   r]   r^   r_   r   r
   r`   ra   r   rb   rc   rd   re   rf   rh   r   r   r   r   r   r   r   r   r   r   r   r   r5   r6   ri   rR   rj   rk   rl   r-   s                       @r.   updatezMistralAgents.updatek  s   P //::J!!H}}X}=H77!'!:!:)..8D)G)G$HI !& 8 8#Xf.C.C%D! '!"
" !!('"&$(%)* 2%++44! ") " 
. e%%22%?00==gu001#%HIL?? --!R:  5**33V__! 4% # 
 "%1CD*6<<BB%1CD3..M ,,]HEE%5!00:M//"6-PP%5!00:M//"6-PPoo<hGGr1   c                  K   d}d}|| j                   j                  }|
|
}n| j                  ||      }t        j                  |t        j
                  |t        j                  |t        t        t        j                              t        j                  |t        t        j                           ||||            | j                  dd||ddddd|| j                   j                  fd	|
      }|	t        k(  r2| j                   j                  t        ur| j                   j                  }	d}t!        |	t        j"                        r|	g df}| j%                  t'        | j                   |xs ddg t)        | j                   j                  t        j*                              |g d|       d{   }d}t        j,                  |dd      rt/        t        j0                  |      S t        j,                  |dd      r0t/        t        j2                  |      }t        j4                  ||      t        j,                  |dd      r4t        j6                  |       d{   }t        j8                  d||      t        j,                  |dd      r4t        j6                  |       d{   }t        j8                  d||      t        j8                  d|      7 +7 7 6w)r   Nr!   r   r   r   Tr$   r%   c                  f    t        j                   j                  dddt        j                        S r'   r   r,   s   r.   r/   z,MistralAgents.update_async.<locals>.<lambda>=  r   r1   r2   r>   rD   r   rF   rK   rO   rS   rL   rM   rT   rU   rN   rV   )rW   r   rX   r   r   r   r   rY   r   r   r   r[   rp   r<   r	   rR   r]   r^   rq   r   r
   r`   ra   r   rb   rc   rd   rr   rf   r   s                       @r.   update_asynczMistralAgents.update_async  s    P //::J!!H}}X}=H77!'!:!:)..8D)G)G$HI !& 8 8#Xf.C.C%D! '!"
" ''('"&$(%)* 2%++44! ") ( 
. e%%22%?00==gu001#%HIL.. --!R:  5**33V__! 4% / 
 
 "%1CD*6<<BB%1CD3..M ,,]HEE%5"'"<"<X"FFM//"6-PP%5"'"<"<X"FFM//"6-PPoo<hGG=
0 G Gs8   F3K(6K!7B-K($K$%A
K(/K&02K($K(&K(versionc                   d}d}|| j                   j                  }||}n| j                  ||      }t        j                  ||      }	| j                  dd|||	ddddd|| j                   j                  |	      }
|t        k(  r2| j                   j                  t        ur| j                   j                  }d}t        |t        j                        r|g d
f}| j                  t        | j                   |xs ddg t        | j                   j                  t        j                              |
g d|      }d}t        j                   |dd      rt#        t        j$                  |      S t        j                   |dd      r0t#        t        j&                  |      }t        j(                  ||      t        j                   |dd      r,t        j*                  |      }t        j,                  d||      t        j                   |dd      r,t        j*                  |      }t        j,                  d||      t        j,                  d|      a  Update an agent version.

        Switch the version of an agent.

        :param agent_id:
        :param version:
        :param retries: Override the default retry configuration for this method
        :param server_url: Override the default server URL for this method
        :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
        :param http_headers: Additional headers to set or replace on requests.
        N)r   r   r   z/v1/agents/{agent_id}/versionFTr$   r%   rz   r>   rD   #agents_api_v1_agents_update_versionrF   rK   rO   rS   rL   rM   rT   rU   rN   rV   )rW   r   rX   r   %AgentsAPIV1AgentsUpdateVersionRequestr\   r<   r	   rR   r]   r   r^   r_   r   r
   r`   ra   r   rb   rc   rd   re   rf   rh   r   r   r   r   r   r   r5   r6   r-   ri   rR   rj   rk   rl   s                  r.   update_versionzMistralAgents.update_versiono  s.   * //::J!!H}}X}=H>>

 !!0'"'$(%)* 2%++44! " 
  e%%22%?00==gu001#%HIL?? --!RB  5**33V__! 4% # 
 "%1CD*6<<BB%1CD3..M ,,]HEE%5!00:M//"6-PP%5!00:M//"6-PPoo<hGGr1   c                  K   d}d}|| j                   j                  }||}n| j                  ||      }t        j                  ||      }	| j                  dd|||	ddddd|| j                   j                  |	      }
|t        k(  r2| j                   j                  t        ur| j                   j                  }d}t        |t        j                        r|g d
f}| j                  t        | j                   |xs ddg t        | j                   j                  t        j                              |
g d|       d{   }d}t        j                   |dd      rt#        t        j$                  |      S t        j                   |dd      r0t#        t        j&                  |      }t        j(                  ||      t        j                   |dd      r4t        j*                  |       d{   }t        j,                  d||      t        j                   |dd      r4t        j*                  |       d{   }t        j,                  d||      t        j,                  d|      7 +7 7 6wr   )rW   r   rX   r   r   rp   r<   r	   rR   r]   r   r^   rq   r   r
   r`   ra   r   rb   rc   rd   rr   rf   r   s                  r.   update_version_asyncz"MistralAgents.update_version_async  sR    * //::J!!H}}X}=H>>

 ''0'"'$(%)* 2%++44! ( 
  e%%22%?00==gu001#%HIL.. --!RB  5**33V__! 4% / 
 
 "%1CD*6<<BB%1CD3..M ,,]HEE%5"'"<"<X"FFM//"6-PP%5"'"<"<X"FFM//"6-PPoo<hGG=
0 G Gs8   D<I0>I)?B-I0,I,-A
I07I.82I0,I0.I0) __name__
__module____qualname____doc__r	   strr   r   r   r   r   rZ   "AgentCreationRequestToolsTypedDictr[   CompletionArgsTypedDictr   r^   intr   rb   rm   rs   r~   r   r   r   r    AgentUpdateRequestToolsTypedDictr   r   r   r    r1   r.   r   r      s    /4  -2057<$($(48)wH wH 	wH
 's+wH V556V>>?A
wH "&'')G)GGH
wH &c*wH  #49-!wH" "%"3"34#wH$ SM%wH& SM'wH( wsCx01)wH* 
+wH| /4  -2057<$($(48)wH wH 	wH
 's+wH V556V>>?A
wH "&'')G)GGH
wH &c*wH  #49-!wH" "%"3"34#wH$ SM%wH& SM'wH( wsCx01)wH* 
+wHx  #%7<$($(48ZH smZH C=	ZH
 "%"3"34ZH SMZH SMZH wsCx01ZH 
fll	ZH~  #%7<$($(48ZH smZH C=	ZH
 "%"3"34ZH SMZH SMZH wsCx01ZH 
fll	ZH@ 8=$($(48WH WH "%"3"34	WH
 SMWH SMWH wsCx01WH 
WHz 8=$($(48WH WH "%"3"34	WH
 SMWH SMWH wsCx01WH 
WHz /4  ',&+-2057<$($(48+@H @H 's+	@H
 V334V<<=?
@H "&'')G)GGH
@H  $@H s#@H  &c*!@H" #49-#@H$ "%"3"34%@H& SM'@H( SM)@H* wsCx01+@H, 
-@HL /4  ',&+-2057<$($(48+@H @H 's+	@H
 V334V<<=?
@H "&'')G)GGH
@H  $@H s#@H  &c*!@H" #49-#@H$ "%"3"34%@H& SM'@H( SM)@H* wsCx01+@H, 
-@HN 8=$($(48ZH ZH 	ZH
 "%"3"34ZH SMZH SMZH wsCx01ZH 
ZHB 8=$($(48ZH ZH 	ZH
 "%"3"34ZH SMZH SMZH wsCx01ZH 
ZHr1   r   N)r   basesdkr   	mistralair   r   mistralai._hooksr   mistralai.typesr   r	   mistralai.utilsr
   'mistralai.utils.unmarshal_json_responser   typingr   r   r   r   r   r   r   r1   r.   <module>r      s0    G  # ( 3 1 K 6 6YHG YHr1   