
    _j                    |    S SK Jr  S SKJrJrJr  SSKJrJrJ	r	  SSK
Jr  SSKJr   " S S\5      r " S	 S
\5      rg)    )annotations)DictAnyUnion   )CreateEnrichmentParametersUpdateEnrichmentParametersWebsetEnrichment)WebsetsBaseClient)WebsetsAsyncBaseClientc                  ^   ^  \ rS rSrSrU 4S jrS
S jrSS jrSS jrSS jr	SS jr
S	rU =r$ )WebsetEnrichmentsClient   z'Client for managing Webset Enrichments.c                $   > [         TU ]  U5        g Nsuper__init__selfclient	__class__s     V/home/rurouni/.local/lib/python3.13/site-packages/exa_py/websets/enrichments/client.pyr    WebsetEnrichmentsClient.__init__            c                V    U R                  SU S3US9n[        R                  " U5      $ )Create an Enrichment for a Webset.

Args:
    webset_id (str): The id of the Webset.
    params (CreateEnrichmentParameters): The parameters for creating an enrichment.

Returns:
    WebsetEnrichment: The created enrichment.
/v0/websets//enrichmentsdatarequestr
   model_validater   	webset_idparamsresponses       r   createWebsetEnrichmentsClient.create   s0     <<,yk FV<T..x88r   c                Z    U R                  SU SU 3SS9n[        R                  " U5      $ )Get an Enrichment by ID.

Args:
    webset_id (str): The id of the Webset.
    id (str): The id of the Enrichment.

Returns:
    WebsetEnrichment: The retrieved enrichment.
r   /enrichments/GETmethodr#   r   r'   idr)   s       r   getWebsetEnrichmentsClient.get    s5     <<,ykrd KTY<Z..x88r   c                \    U R                  SU SU 3USS9n[        R                  " U5      $ )Update an Enrichment.

Args:
    webset_id (str): The id of the Webset.
    id (str): The id of the Enrichment.
    params (UpdateEnrichmentParameters): The parameters for updating an enrichment.

Returns:
    WebsetEnrichment: The updated enrichment.
r   r.   PATCHr"   r1   r#   r   r'   r3   r(   r)   s        r   updateWebsetEnrichmentsClient.update-   s8     <<,ykrd KRXah<i..x88r   c                Z    U R                  SU SU 3SS9n[        R                  " U5      $ )Delete an Enrichment.

Args:
    webset_id (str): The id of the Webset.
    id (str): The id of the Enrichment.

Returns:
    WebsetEnrichment: The deleted enrichment.
r   r.   DELETEr0   r#   r2   s       r   deleteWebsetEnrichmentsClient.delete;   s5     <<,ykrd KT\<]..x88r   c                \    U R                  SU SU S3SS9n[        R                  " U5      $ )Cancel a running Enrichment.

Args:
    webset_id (str): The id of the Webset.
    id (str): The id of the Enrichment.

Returns:
    WebsetEnrichment: The canceled enrichment.
r   r.   /cancelPOSTr0   r#   r2   s       r   cancelWebsetEnrichmentsClient.cancelH   s7     <<,ykrd' R[a<b..x88r    r'   strr(   z1Union[Dict[str, Any], CreateEnrichmentParameters]returnr
   r'   rJ   r3   rJ   rK   r
   r'   rJ   r3   rJ   r(   z1Union[Dict[str, Any], UpdateEnrichmentParameters]rK   r
   __name__
__module____qualname____firstlineno____doc__r   r*   r4   r;   r@   rF   __static_attributes____classcell__r   s   @r   r   r      s)    1!99999 9r   r   c                  ^   ^  \ rS rSrSrU 4S jrS
S jrSS jrSS jrSS jr	SS jr
S	rU =r$ )AsyncWebsetEnrichmentsClientV   z-Async client for managing Webset Enrichments.c                $   > [         TU ]  U5        g r   r   r   s     r   r   %AsyncWebsetEnrichmentsClient.__init__Y   r   r   c                r   #    U R                  SU S3US9I Sh  vN n[        R                  " U5      $  N7f)r   r   r    r!   Nr#   r&   s       r   r*   #AsyncWebsetEnrichmentsClient.create\   s=      YK|&LSYZZ..x88 [s   757c                v   #    U R                  SU SU 3SS9I Sh  vN n[        R                  " U5      $  N7f)r-   r   r.   r/   r0   Nr#   r2   s       r   r4    AsyncWebsetEnrichmentsClient.geti   sA      YK}RD&QZ_``..x88 a   979c                x   #    U R                  SU SU 3USS9I Sh  vN n[        R                  " U5      $  N7f)r7   r   r.   r8   r9   Nr#   r:   s        r   r;   #AsyncWebsetEnrichmentsClient.updatev   sD      YK}RD&QX^gnoo..x88 p   :8:c                v   #    U R                  SU SU 3SS9I Sh  vN n[        R                  " U5      $  N7f)r>   r   r.   r?   r0   Nr#   r2   s       r   r@   #AsyncWebsetEnrichmentsClient.delete   sA      YK}RD&QZbcc..x88 dr`   c                x   #    U R                  SU SU S3SS9I Sh  vN n[        R                  " U5      $  N7f)rC   r   r.   rD   rE   r0   Nr#   r2   s       r   rF   #AsyncWebsetEnrichmentsClient.cancel   sD      YK}RDPW&Xaghh..x88 irc   rH   rI   rL   rM   rN   rV   s   @r   rX   rX   V   s)    7!99999 9r   rX   N)
__future__r   typingr   r   r   typesr   r	   r
   	core.baser   core.async_baser   r   rX   rH   r   r   <module>rm      s?    " # # 
 * 4F9/ F9RF9#9 F9r   