
    i                        S SK r S SKrS SKJr  S SKrSSKJr  SSKJ	r	J
r
  SSKJr  SSKJr  SSKJr  SS	KJr  SS
KJr  SSKJr  SSKJr  SSKJr  SSKJr  SSKJr  SSKJ r   SSK!J"r"  SSK#J$r$   S SK%r%\%RL                  RO                  S5      (       a  \(eS SK)J*r%  \ RV                  " \ RX                  S5      r- " S S5      r. " S S5      r/g! \( a    S SK%r% N>f = f)    N)JSONDecodeError   )ApiError)AsyncClientWrapperSyncClientWrapper)jsonable_encoder)remove_none_from_dict)UnprocessableEntityError)	AgentData)Batch)BatchPaginatedList)BatchPublicOutput)FilterOperation)HttpValidationError)LlamaParseParameters)PaginatedResponseAgentData)PaginatedResponseAggregateGroup)#PaginatedResponseQuotaConfigurationz1..c                      \ rS rSrS\4S jrSSSSS.S\R                  \   S\R                  \   S\R                  \	   S	\R                  \	   S
\
4
S jjrSS\\\\S.S	\R                  \	   S\R                  \	   S\	S\R                  \   S\	S\	S\R                  \	   S\R                  \	   S\	S\	S\R                  \   S
\4S jjrSS.S\	S	\R                  \	   S
\4S jjrSSS.S\	S\R                  \	   S	\R                  \	   S
\4S jjrSSS.S\	S\R                  \	   S	\R                  \	   S\R(                  \	\R*                  4   S
\4
S jjrSSS.S\	S\R                  \	   S	\R                  \	   S
\R(                  \	\	4   4S jjrSS\S .S\R                  \	   S	\R                  \	   S!\	S"\R                  \	   S\R(                  \	\R*                  4   S
\4S# jjrSS\\\\\\\S$.	S\R                  \	   S	\R                  \	   S%\R                  \   S&\R                  \	   S'\R                  \R(                  \	\R                  \   4      S(\R                  \	   S!\	S"\R                  \	   S)\R                  \   S\R                  \   S
\4S* jjrSS\\\\\\\\\S+.S\R                  \	   S	\R                  \	   S%\R                  \   S&\R                  \	   S'\R                  \R(                  \	\R                  \   4      S(\R                  \	   S!\	S"\R                  \	   S,\R                  \R:                  \	      S-\R                  \   S.\R                  \   S\R                  \   S
\4S/ jjrSSS0.S1\ RB                  S2   S3\	S4\R                  \   S%\R                  \   S
\"4
S5 jjr#S6r$g)7
BetaClient%   client_wrapperc                    Xl         g N_client_wrapperselfr   s     [/home/james-whalen/.local/lib/python3.13/site-packages/llama_cloud/resources/beta/client.py__init__BetaClient.__init__&       -    Nlimitoffset
project_idorganization_idr%   r&   r'   r(   returnc                   U R                   R                  R                  S[        R                  R                  U R                   R                  5        S3S5      [        XX4S.5      U R                   R                  5       SS9nSUR                  s=::  a  S:  a,  O  O)[        R                  " [        UR                  5       5      $ UR                  S	:X  a2  [        [        R                  " [        UR                  5       5      5      e UR                  5       n[#        UR                  US
9e! [          a    [#        UR                  UR$                  S
9ef = f)a%  
Parameters:
    - limit: typing.Optional[int].

    - offset: typing.Optional[int].

    - project_id: typing.Optional[str].

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

client = LlamaCloud(
    token="YOUR_TOKEN",
)
client.beta.list_batches()
GET/api/v1/beta/batchesr$   <   paramsheaderstimeout   ,    status_codebodyr   httpx_clientrequesturllibparseurljoinget_base_urlr	   get_headersr7   pydanticparse_obj_asr   jsonr
   r   r   r   textr   r%   r&   r'   r(   	_response_response_jsons          r   list_batchesBetaClient.list_batches)   s-   2 ((55==LL  D$8$8$E$E$G#H!JLab(p ((446 > 
	 )''-#-(();Y^^=MNN  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN  	Sy'<'<9>>RR	S   D/ /)Er(   r'   	tool_dataoutput_type	output_idcompletion_windowtoolrL   
input_typeinput_idrM   rN   batch_create_project_idexternal_idrO   c          	      @   UUUU	U
S.nU[         La  XLS'   U[         La  X|S'   U[         La  XS'   U[         La  XS'   U R                  R                  R                  S[        R
                  R                  U R                  R                  5        S3S5      [        XS	.5      [        U5      U R                  R                  5       S
S9nSUR                  s=::  a  S:  a,  O  O)[        R                  " [        UR                  5       5      $ UR                  S:X  a2  [!        [        R                  " ["        UR                  5       5      5      e UR                  5       n['        UR                  US9e! [$         a    ['        UR                  UR(                  S9ef = f)a\  
Parameters:
    - organization_id: typing.Optional[str].

    - project_id: typing.Optional[str].

    - tool: str. The tool to be used for all requests in the batch.

    - tool_data: typing.Optional[LlamaParseParameters].

    - input_type: str. The type of input file. Currently only 'datasource' is supported.

    - input_id: str. The ID of the input file for the batch.

    - output_type: typing.Optional[str].

    - output_id: typing.Optional[str].

    - batch_create_project_id: str. The ID of the project to which the batch belongs

    - external_id: str. A developer-provided ID for the batch. This ID will be returned in the response.

    - completion_window: typing.Optional[int]. The time frame within which the batch should be processed. Currently only 24h is supported.
---
from llama_cloud import (
    FailPageMode,
    LlamaParseParameters,
    LlamaParseParametersPriority,
    ParsingMode,
)
from llama_cloud.client import LlamaCloud

client = LlamaCloud(
    token="YOUR_TOKEN",
)
client.beta.create_batch(
    tool="string",
    tool_data=LlamaParseParameters(
        priority=LlamaParseParametersPriority.LOW,
        parse_mode=ParsingMode.PARSE_PAGE_WITHOUT_LLM,
        replace_failed_page_mode=FailPageMode.RAW_TEXT,
    ),
    input_type="string",
    input_id="string",
    batch_create_project_id="string",
    external_id="string",
)
rP   rQ   rR   r'   rT   rL   rM   rN   rO   POSTr,   r-   r(   r'   r.   r0   rC   r1   r2   r3   r4   r5   r6   OMITr   r:   r;   r<   r=   r>   r?   r	   r   r@   r7   rA   rB   r   rC   r
   r   r   r   rD   r   r(   r'   rP   rL   rQ   rR   rM   rN   rS   rT   rO   _requestrF   rG   s                  r   create_batchBetaClient.create_batchU   s   @ $ 1&2
 D $-[!d"&1]#D $-[!D(,=()((55==LL  D$8$8$E$E$G#H!JLab(_)gh!(+((446 > 
	 )''-#-((	0@AA  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN  	Sy'<'<9>>RR	Ss   E4 4)Fr(   batch_idc                   U R                   R                  R                  S[        R                  R                  U R                   R                  5        S3SU 35      [        SU05      U R                   R                  5       SS9nSUR                  s=::  a  S:  a,  O  O)[        R                  " [        UR                  5       5      $ UR                  S	:X  a2  [        [        R                  " [        UR                  5       5      5      e UR                  5       n[#        UR                  US
9e! [          a    [#        UR                  UR$                  S
9ef = f)z
Parameters:
    - batch_id: str.

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

client = LlamaCloud(
    token="YOUR_TOKEN",
)
client.beta.get_batch(
    batch_id="string",
)
r+   r,   api/v1/beta/batches/r(   r.   r/   r3   r4   r5   r6   r   r:   r;   r<   r=   r>   r?   r	   r@   r7   rA   rB   r   rC   r
   r   r   r   rD   r   ra   r(   rF   rG   s        r   	get_batchBetaClient.get_batch   s2     ((55==LL  D$8$8$E$E$G#H!JNbckblLmn(*;_)MN((446 > 
	 )''-#-(():INN<LMM  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN  	Sy'<'<9>>RR	S   D1 1)Er'   r(   item_idc          	         U R                   R                  R                  S[        R                  R                  U R                   R                  5        S3SU 35      [        X#S.5      U R                   R                  5       SS9nSUR                  s=::  a  S:  a,  O  O)[        R                  " [        UR                  5       5      $ UR                  S	:X  a2  [        [        R                  " [        UR                  5       5      5      e UR                  5       n[#        UR                  US
9e! [          a    [#        UR                  UR$                  S
9ef = f)a!  
Get agent data by ID.

Parameters:
    - item_id: str.

    - project_id: typing.Optional[str].

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

client = LlamaCloud(
    token="YOUR_TOKEN",
)
client.beta.get_agent_data(
    item_id="string",
)
r+   r,   api/v1/beta/agent-data/ri   r.   r/   r3   r4   r5   r6   r   r:   r;   r<   r=   r>   r?   r	   r@   r7   rA   rB   r   rC   r
   r   r   r   rD   r   rj   r'   r(   rF   rG   s         r   get_agent_dataBetaClient.get_agent_data   s.   , ((55==LL  D$8$8$E$E$G#H!JNefmenLop(
)gh((446 > 
	 )''-#-((INN4DEE  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN  	Sy'<'<9>>RR	Srh   datac          	         U R                   R                  R                  S[        R                  R                  U R                   R                  5        S3SU 35      [        X#S.5      [        SU05      U R                   R                  5       SS9nSUR                  s=::  a  S	:  a,  O  O)[        R                  " [        UR                  5       5      $ UR                  S
:X  a2  [        [        R                  " [         UR                  5       5      5      e UR                  5       n[%        UR                  US9e! ["         a    [%        UR                  UR&                  S9ef = f)ax  
Update agent data by ID (overwrites).

Parameters:
    - item_id: str.

    - project_id: typing.Optional[str].

    - organization_id: typing.Optional[str].

    - data: typing.Dict[str, typing.Any].
---
from llama_cloud.client import LlamaCloud

client = LlamaCloud(
    token="YOUR_TOKEN",
)
client.beta.update_agent_data(
    item_id="string",
    data={"string": {}},
)
PUTr,   rl   ri   rq   r.   rY   r3   r4   r5   r6   r   r:   r;   r<   r=   r>   r?   r	   r   r@   r7   rA   rB   r   rC   r
   r   r   r   rD   r   rj   r'   r(   rq   rF   rG   s          r   update_agent_dataBetaClient.update_agent_data   s:   < ((55==LL  D$8$8$E$E$G#H!JNefmenLop(
)gh!64.1((446 > 
	 )''-#-((INN4DEE  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN  	Sy'<'<9>>RR	Ss   D= =)E&c          	         U R                   R                  R                  S[        R                  R                  U R                   R                  5        S3SU 35      [        X#S.5      U R                   R                  5       SS9nSUR                  s=::  a  S:  aC  O  O@[        R                  " [        R                  [        [        4   UR                  5       5      $ UR                  S	:X  a2  [!        [        R                  " ["        UR                  5       5      5      e UR                  5       n['        UR                  US
9e! [$         a    ['        UR                  UR(                  S
9ef = f)a'  
Delete agent data by ID.

Parameters:
    - item_id: str.

    - project_id: typing.Optional[str].

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

client = LlamaCloud(
    token="YOUR_TOKEN",
)
client.beta.delete_agent_data(
    item_id="string",
)
DELETEr,   rl   ri   r.   r/   r3   r4   r5   r6   r   r:   r;   r<   r=   r>   r?   r	   r@   r7   rA   rB   typingDictstrrC   r
   r   r   r   rD   rn   s         r   delete_agent_dataBetaClient.delete_agent_data-  s;   , ((55==LL  D$8$8$E$E$G#H!JNefmenLop(
)gh((446 > 
	 )''-#-((S#X)>	@PQQ  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN  	Sy'<'<9>>RR	Ss   $E )E1r'   r(   
collection
agent_slugr   c          	         X5S.nU[         La  XFS'   U R                  R                  R                  S[        R
                  R                  U R                  R                  5        S3S5      [        XS.5      [        U5      U R                  R                  5       SS9nS	UR                  s=::  a  S
:  a,  O  O)[        R                  " [        UR                  5       5      $ UR                  S:X  a2  [!        [        R                  " ["        UR                  5       5      5      e UR                  5       n['        UR                  US9e! [$         a    ['        UR                  UR(                  S9ef = f)a  
Create new agent data.

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

    - organization_id: typing.Optional[str].

    - agent_slug: str.

    - collection: typing.Optional[str].

    - data: typing.Dict[str, typing.Any].
---
from llama_cloud.client import LlamaCloud

client = LlamaCloud(
    token="YOUR_TOKEN",
)
client.beta.create_agent_data(
    agent_slug="string",
    data={"string": {}},
)
r   rq   r   rW   r,   api/v1/beta/agent-datari   r.   rY   r3   r4   r5   r6   r[   r   r:   r;   r<   r=   r>   r?   r	   r   r@   r7   rA   rB   r   rC   r
   r   r   r   rD   	r   r'   r(   r   r   rq   r]   rF   rG   s	            r   create_agent_dataBetaClient.create_agent_dataT  sH   B AK1YT!%/\"((55==LL  D$8$8$E$E$G#H!JLde(
)gh!(+((446 > 
	 )''-#-((INN4DEE  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN  	Sy'<'<9>>RR	Ss   %E	 	)E2	r'   r(   	page_size
page_tokenfilterorder_byr   include_totalr&   r   r   r   r   r   c       
   	         SU0nU[         La  X;S'   U[         La  XKS'   U[         La  X[S'   U[         La  XkS'   U[         La  XS'   U	[         La  XS'   U
[         La  XS'   U R                  R                  R                  S	[        R
                  R                  U R                  R                  5        S
3S5      [        XS.5      [        U5      U R                  R                  5       SS9nSUR                  s=::  a  S:  a,  O  O)[        R                  " [        UR                  5       5      $ UR                  S:X  a2  [!        [        R                  " ["        UR                  5       5      5      e UR                  5       n['        UR                  US9e! [$         a    ['        UR                  UR(                  S9ef = f)a]  
Search agent data with filtering, sorting, and pagination.

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

    - organization_id: typing.Optional[str].

    - page_size: typing.Optional[int].

    - page_token: typing.Optional[str].

    - filter: typing.Optional[typing.Dict[str, typing.Optional[FilterOperation]]].

    - order_by: typing.Optional[str].

    - agent_slug: str. The agent deployment's agent_slug to search within

    - collection: typing.Optional[str]. The logical agent data collection to search within

    - include_total: typing.Optional[bool]. Whether to include the total number of items in the response

    - offset: typing.Optional[int].
---
from llama_cloud.client import LlamaCloud

client = LlamaCloud(
    token="YOUR_TOKEN",
)
client.beta.search_agent_data_api_v_1_beta_agent_data_search_post(
    agent_slug="string",
)
r   r   r   r   r   r   r   r&   rW   r,   api/v1/beta/agent-data/:searchri   r.   rY   r3   r4   r5   r6   r[   r   r:   r;   r<   r=   r>   r?   r	   r   r@   r7   rA   rB   r   rC   r
   r   r   r   rD   r   r'   r(   r   r   r   r   r   r   r   r&   r]   rF   rG   s                 r   5search_agent_data_api_v_1_beta_agent_data_search_post@BetaClient.search_agent_data_api_v_1_beta_agent_data_search_post  s   ^ 3?
1KD $-[!T!%/\"!'X4#+Z T!%/\"$(5_%!'X((55==LL  D$8$8$E$E$G#H!JLlm(
)gh!(+((446 > 
	 )''-#-(()CY^^EUVV  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN  	Sy'<'<9>>RR	Ss   3F )G r'   r(   r   r   r   r   r   group_bycountfirstr&   r   r   r   c          	         SU0nU[         La  X=S'   U[         La  XMS'   U[         La  X]S'   U[         La  XmS'   U[         La  XS'   U	[         La  XS'   U
[         La  XS'   U[         La  XS	'   U[         La  XS
'   U R                  R                  R                  S[        R
                  R                  U R                  R                  5        S3S5      [        XS.5      [        U5      U R                  R                  5       SS9nSUR                  s=::  a  S:  a,  O  O)[        R                  " [        UR                  5       5      $ UR                  S:X  a2  [!        [        R                  " ["        UR                  5       5      5      e UR                  5       n['        UR                  US9e! [$         a    ['        UR                  UR(                  S9ef = f)a  
Aggregate agent data with grouping and optional counting/first item retrieval.

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

    - organization_id: typing.Optional[str].

    - page_size: typing.Optional[int].

    - page_token: typing.Optional[str].

    - filter: typing.Optional[typing.Dict[str, typing.Optional[FilterOperation]]].

    - order_by: typing.Optional[str].

    - agent_slug: str. The agent deployment's agent_slug to aggregate data for

    - collection: typing.Optional[str]. The logical agent data collection to aggregate data for

    - group_by: typing.Optional[typing.List[str]].

    - count: typing.Optional[bool].

    - first: typing.Optional[bool].

    - offset: typing.Optional[int].
---
from llama_cloud.client import LlamaCloud

client = LlamaCloud(
    token="YOUR_TOKEN",
)
client.beta.aggregate_agent_data_api_v_1_beta_agent_data_aggregate_post(
    agent_slug="string",
)
r   r   r   r   r   r   r   r   r   r&   rW   r,   !api/v1/beta/agent-data/:aggregateri   r.   rY   r3   r4   r5   r6   r[   r   r:   r;   r<   r=   r>   r?   r	   r   r@   r7   rA   rB   r   rC   r
   r   r   r   rD   r   r'   r(   r   r   r   r   r   r   r   r   r   r&   r]   rF   rG   s                   r   ;aggregate_agent_data_api_v_1_beta_agent_data_aggregate_postFBetaClient.aggregate_agent_data_api_v_1_beta_agent_data_aggregate_post  s   j 3?
1KD $-[!T!%/\"!'X4#+Z T!%/\"4#+Z  %W %W!'X((55==LL  D$8$8$E$E$G#H!JLop(
)gh!(+((446 > 
	 )''-#-(()H)..JZ[[  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN  	Sy'<'<9>>RR	Ss   F1 1)Gpager   source_typeorganization	source_idr   c                   U R                   R                  R                  S[        R                  R                  U R                   R                  5        S3S5      [        XX4S.5      U R                   R                  5       SS9nSUR                  s=::  a  S:  a,  O  O)[        R                  " [        UR                  5       5      $ UR                  S	:X  a2  [        [        R                  " [        UR                  5       5      5      e UR                  5       n[#        UR                  US
9e! [          a    [#        UR                  UR$                  S
9ef = f)a  
Retrieve a paginated list of quota configurations with optional filtering.

Parameters:
    - source_type: typing_extensions.Literal["organization"].

    - source_id: str.

    - page: typing.Optional[int].

    - page_size: typing.Optional[int].
---
from llama_cloud.client import LlamaCloud

client = LlamaCloud(
    token="YOUR_TOKEN",
)
client.beta.list_quota_configurations(
    source_type="organization",
    source_id="string",
)
r+   r,   api/v1/beta/quota-managementr   r   r   r   r.   r/   r3   r4   r5   r6   r   r:   r;   r<   r=   r>   r?   r	   r@   r7   rA   rB   r   rC   r
   r   r   r   rD   r   r   r   r   r   rF   rG   s          r   list_quota_configurations$BetaClient.list_quota_configurations4  s-   < ((55==LL  D$8$8$E$E$G#H!JLjk( +Tj ((446 > 
	 )''-#-(()LinnN^__  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN  	Sy'<'<9>>RR	SrJ   r   )%__name__
__module____qualname____firstlineno__r   r    r{   Optionalintr}   r   rH   r[   r   r   r^   r   rf   r   ro   r|   Anyrv   r~   r   r   boolr   r   Listr   r   typing_extensionsLiteralr   r   __static_attributes__ r#   r   r   r   %   s   .*; . '+'++/04*O s#*O $	*O
 OOC(*O  -*O 
*O^ 15+/;? -1*. 37^O  -^O OOC(	^O
 ^O ??#78^O ^O ^O __S)^O ??3'^O "%^O ^O "??3/^O 
^O@ SW O# O6??3;O O[l OD CGpt%O%O+1??3+?%OY_YhYhilYm%O	%OV ,004.O.O OOC(	.O
  -.O kk#vzz/*.O 
.Ob CGpt%O%O+1??3+?%OY_YhYhilYm%O	S#X	%OT ,004+/4O OOC(4O  -	4O
 4O OOC(4O kk#vzz/*4O 
4Or ,004*.+/VZ)-+//3'+NO OOC(NO  -	NO
 ??3'NO OOC(NO C1Q,Q RSNO //#&NO NO OOC(NO t,NO $NO 
$NOf ,004*.+/VZ)-+/6:'+'+'+XO OOC(XO  -	XO
 ??3'XO OOC(XO C1Q,Q RSXO //#&XO XO OOC(XO //&++c"23XO t$XO t$XO $XO 
)XO~ &**./O '..~>/O 	/O
 ooc"/O ??3'/O 
-/O /Or#   r   c                      \ rS rSrS\4S jrSSSSS.S\R                  \   S\R                  \   S\R                  \	   S	\R                  \	   S
\
4
S jjrSS\\\\S.S	\R                  \	   S\R                  \	   S\	S\R                  \   S\	S\	S\R                  \	   S\R                  \	   S\	S\	S\R                  \   S
\4S jjrSS.S\	S	\R                  \	   S
\4S jjrSSS.S\	S\R                  \	   S	\R                  \	   S
\4S jjrSSS.S\	S\R                  \	   S	\R                  \	   S\R(                  \	\R*                  4   S
\4
S jjrSSS.S\	S\R                  \	   S	\R                  \	   S
\R(                  \	\	4   4S jjrSS\S .S\R                  \	   S	\R                  \	   S!\	S"\R                  \	   S\R(                  \	\R*                  4   S
\4S# jjrSS\\\\\\\S$.	S\R                  \	   S	\R                  \	   S%\R                  \   S&\R                  \	   S'\R                  \R(                  \	\R                  \   4      S(\R                  \	   S!\	S"\R                  \	   S)\R                  \   S\R                  \   S
\4S* jjrSS\\\\\\\\\S+.S\R                  \	   S	\R                  \	   S%\R                  \   S&\R                  \	   S'\R                  \R(                  \	\R                  \   4      S(\R                  \	   S!\	S"\R                  \	   S,\R                  \R:                  \	      S-\R                  \   S.\R                  \   S\R                  \   S
\4S/ jjrSSS0.S1\ RB                  S2   S3\	S4\R                  \   S%\R                  \   S
\"4
S5 jjr#S6r$g)7AsyncBetaClientif  r   c                    Xl         g r   r   r   s     r   r    AsyncBetaClient.__init__g  r"   r#   Nr$   r%   r&   r'   r(   r)   c                  #    U R                   R                  R                  S[        R                  R                  U R                   R                  5        S3S5      [        XX4S.5      U R                   R                  5       SS9I Sh  vN nSUR                  s=::  a  S	:  a,  O  O)[        R                  " [        UR                  5       5      $ UR                  S
:X  a2  [        [        R                  " [        UR                  5       5      5      e UR                  5       n[#        UR                  US9e N! [          a    [#        UR                  UR$                  S9ef = f7f)a5  
Parameters:
    - limit: typing.Optional[int].

    - offset: typing.Optional[int].

    - project_id: typing.Optional[str].

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

client = AsyncLlamaCloud(
    token="YOUR_TOKEN",
)
await client.beta.list_batches()
r+   r,   r-   r$   r.   r/   Nr3   r4   r5   r6   r9   rE   s          r   rH   AsyncBetaClient.list_batchesj  s:    2 ..;;CCLL  D$8$8$E$E$G#H!JLab(p ((446 D 
 
	 )''-#-(();Y^^=MNN  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN#
  	Sy'<'<9>>RR	S+   BE'
D9B	E'D; %E';)E$$E'rK   rP   rL   rQ   rR   rM   rN   rS   rT   rO   c          	      \  #    UUUU	U
S.nU[         La  XLS'   U[         La  X|S'   U[         La  XS'   U[         La  XS'   U R                  R                  R                  S[        R
                  R                  U R                  R                  5        S3S5      [        XS	.5      [        U5      U R                  R                  5       S
S9I Sh  vN nSUR                  s=::  a  S:  a,  O  O)[        R                  " [        UR                  5       5      $ UR                  S:X  a2  [!        [        R                  " ["        UR                  5       5      5      e UR                  5       n['        UR                  US9e N! [$         a    ['        UR                  UR(                  S9ef = f7f)al  
Parameters:
    - organization_id: typing.Optional[str].

    - project_id: typing.Optional[str].

    - tool: str. The tool to be used for all requests in the batch.

    - tool_data: typing.Optional[LlamaParseParameters].

    - input_type: str. The type of input file. Currently only 'datasource' is supported.

    - input_id: str. The ID of the input file for the batch.

    - output_type: typing.Optional[str].

    - output_id: typing.Optional[str].

    - batch_create_project_id: str. The ID of the project to which the batch belongs

    - external_id: str. A developer-provided ID for the batch. This ID will be returned in the response.

    - completion_window: typing.Optional[int]. The time frame within which the batch should be processed. Currently only 24h is supported.
---
from llama_cloud import (
    FailPageMode,
    LlamaParseParameters,
    LlamaParseParametersPriority,
    ParsingMode,
)
from llama_cloud.client import AsyncLlamaCloud

client = AsyncLlamaCloud(
    token="YOUR_TOKEN",
)
await client.beta.create_batch(
    tool="string",
    tool_data=LlamaParseParameters(
        priority=LlamaParseParametersPriority.LOW,
        parse_mode=ParsingMode.PARSE_PAGE_WITHOUT_LLM,
        replace_failed_page_mode=FailPageMode.RAW_TEXT,
    ),
    input_type="string",
    input_id="string",
    batch_create_project_id="string",
    external_id="string",
)
rV   rL   rM   rN   rO   rW   r,   r-   rX   r.   rY   Nr3   r4   r5   r6   rZ   r\   s                  r   r^   AsyncBetaClient.create_batch  s    @ $ 1&2
 D $-[!d"&1]#D $-[!D(,=()..;;CCLL  D$8$8$E$E$G#H!JLab(_)gh!(+((446 D 
 
	 )''-#-((	0@AA  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN!
  	Sy'<'<9>>RR	Ss+   CF,E>B	F,F  *F, )F))F,r`   ra   c                  #    U R                   R                  R                  S[        R                  R                  U R                   R                  5        S3SU 35      [        SU05      U R                   R                  5       SS9I Sh  vN nSUR                  s=::  a  S	:  a,  O  O)[        R                  " [        UR                  5       5      $ UR                  S
:X  a2  [        [        R                  " [        UR                  5       5      5      e UR                  5       n[#        UR                  US9e N! [          a    [#        UR                  UR$                  S9ef = f7f)z
Parameters:
    - batch_id: str.

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

client = AsyncLlamaCloud(
    token="YOUR_TOKEN",
)
await client.beta.get_batch(
    batch_id="string",
)
r+   r,   rc   r(   r.   r/   Nr3   r4   r5   r6   rd   re   s        r   rf   AsyncBetaClient.get_batch  s?      ..;;CCLL  D$8$8$E$E$G#H!JNbckblLmn(*;_)MN((446 D 
 
	 )''-#-(():INN<LMM  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)ri   rj   c          	        #    U R                   R                  R                  S[        R                  R                  U R                   R                  5        S3SU 35      [        X#S.5      U R                   R                  5       SS9I Sh  vN nSUR                  s=::  a  S	:  a,  O  O)[        R                  " [        UR                  5       5      $ UR                  S
:X  a2  [        [        R                  " [        UR                  5       5      5      e UR                  5       n[#        UR                  US9e N! [          a    [#        UR                  UR$                  S9ef = f7f)a1  
Get agent data by ID.

Parameters:
    - item_id: str.

    - project_id: typing.Optional[str].

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

client = AsyncLlamaCloud(
    token="YOUR_TOKEN",
)
await client.beta.get_agent_data(
    item_id="string",
)
r+   r,   rl   ri   r.   r/   Nr3   r4   r5   r6   rm   rn   s         r   ro   AsyncBetaClient.get_agent_data  s;    , ..;;CCLL  D$8$8$E$E$G#H!JNefmenLop(
)gh((446 D 
 
	 )''-#-((INN4DEE  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN
  	Sy'<'<9>>RR	Sr   rq   c          	        #    U R                   R                  R                  S[        R                  R                  U R                   R                  5        S3SU 35      [        X#S.5      [        SU05      U R                   R                  5       SS9I Sh  vN nS	UR                  s=::  a  S
:  a,  O  O)[        R                  " [        UR                  5       5      $ UR                  S:X  a2  [        [        R                  " [         UR                  5       5      5      e UR                  5       n[%        UR                  US9e N! ["         a    [%        UR                  UR&                  S9ef = f7f)a  
Update agent data by ID (overwrites).

Parameters:
    - item_id: str.

    - project_id: typing.Optional[str].

    - organization_id: typing.Optional[str].

    - data: typing.Dict[str, typing.Any].
---
from llama_cloud.client import AsyncLlamaCloud

client = AsyncLlamaCloud(
    token="YOUR_TOKEN",
)
await client.beta.update_agent_data(
    item_id="string",
    data={"string": {}},
)
rs   r,   rl   ri   rq   r.   rY   Nr3   r4   r5   r6   rt   ru   s          r   rv   !AsyncBetaClient.update_agent_data>  sG    < ..;;CCLL  D$8$8$E$E$G#H!JNefmenLop(
)gh!64.1((446 D 
 
	 )''-#-((INN4DEE  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN!
  	Sy'<'<9>>RR	Ss+   BE5EB	E5#E	 3E5	)E22E5c          	        #    U R                   R                  R                  S[        R                  R                  U R                   R                  5        S3SU 35      [        X#S.5      U R                   R                  5       SS9I Sh  vN nSUR                  s=::  a  S	:  aC  O  O@[        R                  " [        R                  [        [        4   UR                  5       5      $ UR                  S
:X  a2  [!        [        R                  " ["        UR                  5       5      5      e UR                  5       n['        UR                  US9e N! [$         a    ['        UR                  UR(                  S9ef = f7f)a7  
Delete agent data by ID.

Parameters:
    - item_id: str.

    - project_id: typing.Optional[str].

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

client = AsyncLlamaCloud(
    token="YOUR_TOKEN",
)
await client.beta.delete_agent_data(
    item_id="string",
)
ry   r,   rl   ri   r.   r/   Nr3   r4   r5   r6   rz   rn   s         r   r~   !AsyncBetaClient.delete_agent_datan  sH    , ..;;CCLL  D$8$8$E$E$G#H!JNefmenLop(
)gh((446 D 
 
	 )''-#-((S#X)>	@PQQ  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN
  	Sy'<'<9>>RR	Ss+   B
F EB F .E >F )E==F r   r   r   c          	        #    X5S.nU[         La  XFS'   U R                  R                  R                  S[        R
                  R                  U R                  R                  5        S3S5      [        XS.5      [        U5      U R                  R                  5       SS9I S	h  vN nS
UR                  s=::  a  S:  a,  O  O)[        R                  " [        UR                  5       5      $ UR                  S:X  a2  [!        [        R                  " ["        UR                  5       5      5      e UR                  5       n['        UR                  US9e N! [$         a    ['        UR                  UR(                  S9ef = f7f)a  
Create new agent data.

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

    - organization_id: typing.Optional[str].

    - agent_slug: str.

    - collection: typing.Optional[str].

    - data: typing.Dict[str, typing.Any].
---
from llama_cloud.client import AsyncLlamaCloud

client = AsyncLlamaCloud(
    token="YOUR_TOKEN",
)
await client.beta.create_agent_data(
    agent_slug="string",
    data={"string": {}},
)
r   r   rW   r,   r   ri   r.   rY   Nr3   r4   r5   r6   r   r   s	            r   r   !AsyncBetaClient.create_agent_data  sU    B AK1YT!%/\"..;;CCLL  D$8$8$E$E$G#H!JLde(
)gh!(+((446 D 
 
	 )''-#-((INN4DEE  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN!
  	Sy'<'<9>>RR	Ss+   B"F$E%B	F/E ?F)E>>Fr   r   r   r   r   r   c       
   	        #    SU0nU[         La  X;S'   U[         La  XKS'   U[         La  X[S'   U[         La  XkS'   U[         La  XS'   U	[         La  XS'   U
[         La  XS'   U R                  R                  R                  S	[        R
                  R                  U R                  R                  5        S
3S5      [        XS.5      [        U5      U R                  R                  5       SS9I Sh  vN nSUR                  s=::  a  S:  a,  O  O)[        R                  " [        UR                  5       5      $ UR                  S:X  a2  [!        [        R                  " ["        UR                  5       5      5      e UR                  5       n['        UR                  US9e N! [$         a    ['        UR                  UR(                  S9ef = f7f)am  
Search agent data with filtering, sorting, and pagination.

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

    - organization_id: typing.Optional[str].

    - page_size: typing.Optional[int].

    - page_token: typing.Optional[str].

    - filter: typing.Optional[typing.Dict[str, typing.Optional[FilterOperation]]].

    - order_by: typing.Optional[str].

    - agent_slug: str. The agent deployment's agent_slug to search within

    - collection: typing.Optional[str]. The logical agent data collection to search within

    - include_total: typing.Optional[bool]. Whether to include the total number of items in the response

    - offset: typing.Optional[int].
---
from llama_cloud.client import AsyncLlamaCloud

client = AsyncLlamaCloud(
    token="YOUR_TOKEN",
)
await client.beta.search_agent_data_api_v_1_beta_agent_data_search_post(
    agent_slug="string",
)
r   r   r   r   r   r   r   r&   rW   r,   r   ri   r.   rY   Nr3   r4   r5   r6   r   r   s                 r   r   EAsyncBetaClient.search_agent_data_api_v_1_beta_agent_data_search_post  s    ^ 3?
1KD $-[!T!%/\"!'X4#+Z T!%/\"$(5_%!'X..;;CCLL  D$8$8$E$E$G#H!JLlm(
)gh!(+((446 D 
 
	 )''-#-(()CY^^EUVV  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN!
  	Sy'<'<9>>RR	Ss+   C0G2F!3B	G=F# G#)GGr   r   r   r   c          	        #    SU0nU[         La  X=S'   U[         La  XMS'   U[         La  X]S'   U[         La  XmS'   U[         La  XS'   U	[         La  XS'   U
[         La  XS'   U[         La  XS	'   U[         La  XS
'   U R                  R                  R                  S[        R
                  R                  U R                  R                  5        S3S5      [        XS.5      [        U5      U R                  R                  5       SS9I Sh  vN nSUR                  s=::  a  S:  a,  O  O)[        R                  " [        UR                  5       5      $ UR                  S:X  a2  [!        [        R                  " ["        UR                  5       5      5      e UR                  5       n['        UR                  US9e N! [$         a    ['        UR                  UR(                  S9ef = f7f)a  
Aggregate agent data with grouping and optional counting/first item retrieval.

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

    - organization_id: typing.Optional[str].

    - page_size: typing.Optional[int].

    - page_token: typing.Optional[str].

    - filter: typing.Optional[typing.Dict[str, typing.Optional[FilterOperation]]].

    - order_by: typing.Optional[str].

    - agent_slug: str. The agent deployment's agent_slug to aggregate data for

    - collection: typing.Optional[str]. The logical agent data collection to aggregate data for

    - group_by: typing.Optional[typing.List[str]].

    - count: typing.Optional[bool].

    - first: typing.Optional[bool].

    - offset: typing.Optional[int].
---
from llama_cloud.client import AsyncLlamaCloud

client = AsyncLlamaCloud(
    token="YOUR_TOKEN",
)
await client.beta.aggregate_agent_data_api_v_1_beta_agent_data_aggregate_post(
    agent_slug="string",
)
r   r   r   r   r   r   r   r   r   r&   rW   r,   r   ri   r.   rY   Nr3   r4   r5   r6   r   r   s                   r   r   KAsyncBetaClient.aggregate_agent_data_api_v_1_beta_agent_data_aggregate_post  s    j 3?
1KD $-[!T!%/\"!'X4#+Z T!%/\"4#+Z  %W %W!'X..;;CCLL  D$8$8$E$E$G#H!JLop(
)gh!(+((446 D 
 
	 )''-#-(()H)..JZ[[  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN!
  	Sy'<'<9>>RR	Ss+   D
G)F;B	G)F= 'G)=)G&&G)r   r   r   r   r   c                  #    U R                   R                  R                  S[        R                  R                  U R                   R                  5        S3S5      [        XX4S.5      U R                   R                  5       SS9I Sh  vN nSUR                  s=::  a  S	:  a,  O  O)[        R                  " [        UR                  5       5      $ UR                  S
:X  a2  [        [        R                  " [        UR                  5       5      5      e UR                  5       n[#        UR                  US9e N! [          a    [#        UR                  UR$                  S9ef = f7f)a  
Retrieve a paginated list of quota configurations with optional filtering.

Parameters:
    - source_type: typing_extensions.Literal["organization"].

    - source_id: str.

    - page: typing.Optional[int].

    - page_size: typing.Optional[int].
---
from llama_cloud.client import AsyncLlamaCloud

client = AsyncLlamaCloud(
    token="YOUR_TOKEN",
)
await client.beta.list_quota_configurations(
    source_type="organization",
    source_id="string",
)
r+   r,   r   r   r.   r/   Nr3   r4   r5   r6   r   r   s          r   r   )AsyncBetaClient.list_quota_configurationsu  s:    < ..;;CCLL  D$8$8$E$E$G#H!JLjk( +Tj ((446 D 
 
	 )''-#-(()LinnN^__  C'*8+@+@ATV_VdVdVf+ghh	S&^^-N 9#8#8~NN#
  	Sy'<'<9>>RR	Sr   r   )%r   r   r   r   r   r    r{   r   r   r}   r   rH   r[   r   r   r^   r   rf   r   ro   r|   r   rv   r~   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r#   r   r   r   f  s   .*< . '+'++/04*O s#*O $	*O
 OOC(*O  -*O 
*O^ 15+/;? -1*. 37^O  -^O OOC(	^O
 ^O ??#78^O ^O ^O __S)^O ??3'^O "%^O ^O "??3/^O 
^O@ Y] O OQTAU Oar OD CGpt%O%O+1??3+?%OY_YhYhilYm%O	%OV ,004.O.O OOC(	.O
  -.O kk#vzz/*.O 
.Ob CGpt%O%O+1??3+?%OY_YhYhilYm%O	S#X	%OT ,004+/4O OOC(4O  -	4O
 4O OOC(4O kk#vzz/*4O 
4Or ,004*.+/VZ)-+//3'+NO OOC(NO  -	NO
 ??3'NO OOC(NO C1Q,Q RSNO //#&NO NO OOC(NO t,NO $NO 
$NOf ,004*.+/VZ)-+/6:'+'+'+XO OOC(XO  -	XO
 ??3'XO OOC(XO C1Q,Q RSXO //#&XO XO OOC(XO //&++c"23XO t$XO t$XO $XO 
)XO~ &**./O '..~>/O 	/O
 ooc"/O ??3'/O 
-/O /Or#   r   )0r{   urllib.parser<   json.decoderr   r   core.api_errorr   core.client_wrapperr   r   core.jsonable_encoderr   core.remove_none_from_dictr	   !errors.unprocessable_entity_errorr
   types.agent_datar   types.batchr   types.batch_paginated_listr   types.batch_public_outputr   types.filter_operationr   types.http_validation_errorr   types.llama_parse_parametersr   #types.paginated_response_agent_datar   (types.paginated_response_aggregate_groupr   ,types.paginated_response_quota_configurationr   rA   __version__
startswithImportErrorpydantic.v1v1castr   r[   r   r   r   r#   r   <module>r      s      (  & H 5 ? I )   < : 5 > @ M W _&&t,,"
 {{6::s#~O ~OB~O ~OQ  s   0,C 
CC