
    JiN                        d dl Z d dlZd dl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 dd	lmZ 	 d dlZej&                  j)                  d
      red dlmZ  G d d      Z G d d      Zy# e$ r d dlZY  w xY w)    N)JSONDecodeError   )ApiError)AsyncClientWrapperSyncClientWrapper)remove_none_from_dict)UnprocessableEntityError)AgentDeploymentList)HttpValidationErrorz1.c                       e Zd ZdefdZddddej                  e   dej                  e   defdZ	ddddej                  e   dej                  e   defd	Z
y)
AgentDeploymentsClientclient_wrapperc                    || _         y N_client_wrapperselfr   s     /var/www/html/BankruptcyAI-uat/bankruptcy-ai/venv/lib/python3.12/site-packages/llama_cloud/resources/agent_deployments/client.py__init__zAgentDeploymentsClient.__init__   
    -    N
project_idorganization_idr   r   returnc          	         | j                   j                  j                  dt        j                  j                  | j                   j                          dd      t        ||d      | j                   j                         d      }d|j                  cxk  rdk  r+n n(t        j                  t        |j                               S |j                  d	k(  r1t        t        j                  t        |j                                     	 |j                         }t#        |j                  |
      # t         $ r" t#        |j                  |j$                  
      w xY w)a  
        List all deployments for a project.

        Parameters:
            - project_id: typing.Optional[str].

            - organization_id: typing.Optional[str].
        ---
        from llama_cloud.client import LlamaCloud

        client = LlamaCloud(
            token="YOUR_TOKEN",
        )
        client.agent_deployments.list_deployments_api_v_1_agent_deployments_get()
        GET/api/v1/agent-deploymentsr   <   paramsheaderstimeout   ,    status_codebodyr   httpx_clientrequesturllibparseurljoinget_base_urlr   get_headersr*   pydanticparse_obj_asr
   jsonr	   r   r   r   textr   r   r   	_response_response_jsons        r   .list_deployments_api_v_1_agent_deployments_getzEAgentDeploymentsClient.list_deployments_api_v_1_agent_deployments_get   s)   $ ((55==LL  D$8$8$E$E$G#H!JLfg(
Wf)gh((446 > 
	 )''-#-(()<inn>NOO  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN  	Sy'<'<9>>RR	S   	D0 0+Ec          	         | j                   j                  j                  dt        j                  j                  | j                   j                          dd      t        ||d      | j                   j                         d      }d|j                  cxk  rdk  r+n n(t        j                  t        |j                               S |j                  d	k(  r1t        t        j                  t        |j                                     	 |j                         }t#        |j                  |
      # t         $ r" t#        |j                  |j$                  
      w xY w)a  
        Sync deployments for a project.

        Parameters:
            - project_id: typing.Optional[str].

            - organization_id: typing.Optional[str].
        ---
        from llama_cloud.client import LlamaCloud

        client = LlamaCloud(
            token="YOUR_TOKEN",
        )
        client.agent_deployments.sync_deployments_api_v_1_agent_deployments_sync_post()
        POSTr   api/v1/agent-deployments/:syncr   r!   r"   r&   r'   r(   r)   r,   r8   s        r   4sync_deployments_api_v_1_agent_deployments_sync_postzKAgentDeploymentsClient.sync_deployments_api_v_1_agent_deployments_sync_post>   s)   $ ((55==LL  D$8$8$E$E$G#H!JLlm(
Wf)gh((446 > 
	 )''-#-(()<inn>NOO  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN  	Sy'<'<9>>RR	Sr<   )__name__
__module____qualname__r   r   typingOptionalstrr
   r;   r@    r   r   r   r      s    .*; . 59bf!O#__S1!OKQ??[^K_!O	!OH 59bf!O#__S1!OKQ??[^K_!O	!Or   r   c                       e Zd ZdefdZddddej                  e   dej                  e   defdZ	ddddej                  e   dej                  e   defd	Z
y)
AsyncAgentDeploymentsClientr   c                    || _         y r   r   r   s     r   r   z$AsyncAgentDeploymentsClient.__init__c   r   r   Nr   r   r   r   c          	        K   | j                   j                  j                  dt        j                  j                  | j                   j                          dd      t        ||d      | j                   j                         d       d{   }d|j                  cxk  rd	k  r+n n(t        j                  t        |j                               S |j                  d
k(  r1t        t        j                  t        |j                                     	 |j                         }t#        |j                  |      7 # t         $ r" t#        |j                  |j$                        w xY ww)a  
        List all deployments for a project.

        Parameters:
            - project_id: typing.Optional[str].

            - organization_id: typing.Optional[str].
        ---
        from llama_cloud.client import AsyncLlamaCloud

        client = AsyncLlamaCloud(
            token="YOUR_TOKEN",
        )
        await client.agent_deployments.list_deployments_api_v_1_agent_deployments_get()
        r   r   r    r   r!   r"   Nr&   r'   r(   r)   r,   r8   s        r   r;   zJAsyncAgentDeploymentsClient.list_deployments_api_v_1_agent_deployments_getf   s7    $ ..;;CCLL  D$8$8$E$E$G#H!JLfg(
Wf)gh((446 D 
 
	 )''-#-(()<inn>NOO  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN
  	Sy'<'<9>>RR	S+   BE*D:BE*D< #E*<+E''E*c          	        K   | j                   j                  j                  dt        j                  j                  | j                   j                          dd      t        ||d      | j                   j                         d       d{   }d|j                  cxk  rd	k  r+n n(t        j                  t        |j                               S |j                  d
k(  r1t        t        j                  t        |j                                     	 |j                         }t#        |j                  |      7 # t         $ r" t#        |j                  |j$                        w xY ww)a  
        Sync deployments for a project.

        Parameters:
            - project_id: typing.Optional[str].

            - organization_id: typing.Optional[str].
        ---
        from llama_cloud.client import AsyncLlamaCloud

        client = AsyncLlamaCloud(
            token="YOUR_TOKEN",
        )
        await client.agent_deployments.sync_deployments_api_v_1_agent_deployments_sync_post()
        r>   r   r?   r   r!   r"   Nr&   r'   r(   r)   r,   r8   s        r   r@   zPAsyncAgentDeploymentsClient.sync_deployments_api_v_1_agent_deployments_sync_post   s7    $ ..;;CCLL  D$8$8$E$E$G#H!JLlm(
Wf)gh((446 D 
 
	 )''-#-(()<inn>NOO  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN
  	Sy'<'<9>>RR	SrL   )rA   rB   rC   r   r   rD   rE   rF   r
   r;   r@   rG   r   r   rI   rI   b   s    .*< . 59bf!O#__S1!OKQ??[^K_!O	!OH 59bf!O#__S1!OKQ??[^K_!O	!Or   rI   )rD   urllib.parser/   json.decoderr   core.api_errorr   core.client_wrapperr   r   core.remove_none_from_dictr   !errors.unprocessable_entity_errorr	   types.agent_deployment_listr
   types.http_validation_errorr   r4   __version__
startswithImportErrorpydantic.v1v1r   rI   rG   r   r   <module>r[      sy      ( & H ? I > >&&t,"
HO HOVHO HO_  s   'A2 2	A>=A>