i*SSKrSSKJrJrJrJrJrJr SSKJ r SSK J r J r J r JrJr SSKrSSKJrJrJrJrJrJr S\ S\ 4S jrS\4S jr"S S \\5rg) N)AnyDictGenericListOptionalType)AsyncLlamaCloud) WrappedFnretrystop_after_attemptwait_exponentialretry_if_exception_type) AgentDataTComparisonOperatorTypedAgentDataTypedAgentDataItemsTypedAggregateGroupTypedAggregateGroupItemsfuncreturnc [[S5[SSS9[[R [R [R45S9"U5$)a Decorator that adds automatic retry logic to agent data API calls. Applies exponential backoff retry strategy for common network-related exceptions: - Up to 3 retry attempts - Exponential wait time between 0.5s and 10s - Retries on timeout, connection, and HTTP status errors This ensures resilient API communication in distributed environments where temporary network issues or service unavailability may occur. g? )minmax)stopwaitr )r r r rhttpxTimeoutException ConnectErrorHTTPStatusError)rs e/home/james-whalen/.local/lib/python3.13/site-packages/llama_cloud_services/beta/agent_data/client.pyagent_data_retryr$sQ   " #2 .%  # #U%7%79N9N O      c@[R"S5=(d S$)aE Retrieve the default agent ID from environment variables. Returns: The value of LLAMA_DEPLOY_DEPLOYMENT_NAME environment variable, or None if not set Note: This provides a convenient way to configure agent ID globally via environment variables instead of passing it explicitly to each client instance. LLAMA_DEPLOY_DEPLOYMENT_NAME_public)osgetenvr%r#get_default_agent_idr,-s 993 4 A Ar%c\rSrSrSrSS\\S\S\\S\\ S\\S \\4 S jjr \ S \S \ \4S j5r \ S\S \ \4Sj5r\ S \S\S \ \4Sj5r\ S \S S4Sj5r\ SS\\\\\\44S\\S\\S\\S\S \\4 Sjj5r\ SS\\\\\\44S\\\S\\S\\S\\S\\S\\S \\4Sjj5rSrg) AsyncAgentDataClient=as Async client for managing agent-generated structured data with type safety. This client provides a high-level interface for CRUD operations, searching, and aggregation of structured data created by agents. It enforces type safety by validating all data against a specified Pydantic model type. The client is generic over AgentDataT, which must be a Pydantic BaseModel that defines the structure of your agent's data output. Example: ```python from pydantic import BaseModel from llama_cloud.client import AsyncLlamaCloud from llama_cloud_services.beta.agent_data import AsyncAgentDataClient class ExtractedPerson(BaseModel): name: str age: int email: str # Initialize client llama_client = AsyncLlamaCloud(token="your-api-key") agent_client = AsyncAgentDataClient( client=llama_client, type=ExtractedPerson, collection="extracted_people", agent_url_id="person-extraction-agent" ) # Create data person = ExtractedPerson(name="John Doe", age=30, email="john@example.com") result = await agent_client.create_agent_data(person) # Search data results = await agent_client.search( filter={"age": {"gt": 25}}, order_by="data.name", page_size=20 ) ``` Type Parameters: AgentDataT: Pydantic BaseModel type that defines the structure of agent data Ntype collection agent_url_idclienttokenbase_urlcU=(d [5UlX lU(dD[U=(d [R "S5U=(d [R "S5S9nX@lXlg)aR Initialize the AsyncAgentDataClient. Args: type: Pydantic BaseModel class that defines the data structure. All agent data will be validated against this type. collection: Named collection within the agent for organizing data. Defaults to "default". Collections allow logical separation of different data types or workflows within the same agent. agent_url_id: Unique identifier for the agent. This normally appears in the url of an agent within the llama cloud platform. If not provided, will attempt to use the LLAMA_DEPLOY_DEPLOYMENT_NAME environment variable. Data can only be added to an already existing agent in the platform. client: AsyncLlamaCloud client instance for API communication. If not provided, will construct one from the provided api token and base url token: Llama Cloud API token. Reads from LLAMA_CLOUD_API_KEY if not provided base_url: Llama Cloud API token. Reads from LLAMA_CLOUD_BASE_URL if not provided, and defaults to https://api.cloud.llamaindex.ai Raises: ValueError: If agent_url_id is not provided and the LLAMA_DEPLOY_DEPLOYMENT_NAME environment variable is not set Note: The client automatically applies retry logic to all API calls with exponential backoff for timeout, connection, and HTTP status errors. LLAMA_CLOUD_API_KEYLLAMA_CLOUD_BASE_URL)r4r5N)r,r2r1r r)r*r3r0)selfr0r1r2r3r4r5s r#__init__AsyncAgentDataClient.__init__lsXL)B,@,B$$?ryy)>?!FRYY/E%FF  r%item_idrc# URRRUS9IShvN n[R"X R S9$N"7f)Nr< validator)r3betaget_agent_datarfrom_rawr0)r9r<raw_datas r#get_itemAsyncAgentDataClient.get_itemsH))889  &&x99EE s'AA #Adatac# URRRURURUR 5S9IShvN n[ R"X RS9$N"7f)N) agent_slugr1rGr?) r3rAcreate_agent_datar2r1 model_dumprrCr0)r9rGrDs r# create_item AsyncAgentDataClient.create_items_));;(("<   &&x99EE  sA A2 A0#A2c# URRRUUR5S9IShvN n[R "X0R S9$N"7f)N)r<rGr?)r3rAupdate_agent_datarKrrCr0)r9r<rGrDs r# update_item AsyncAgentDataClient.update_itemsT));;"<  &&x99EE  s6AA#Acf# URRRUS9IShvN gN7f)Nr>)r3rAdelete_agent_data)r9r<s r# delete_item AsyncAgentDataClient.delete_items&kk000AAAs '1/1filterorder_byoffset page_size include_totalc R# URRRURURUUUUUS9IShvN n[ UR Vs/sH!n[R"XpRS9PM# snURSLURS9$N\s snf7f)a Search agent data with filtering, sorting, and pagination. Args: filter: Filter conditions to apply to the search. Dict mapping field names to FilterOperation objects. Filters only by data fields Examples: - {"age": {"gt": 18}} - age greater than 18 - {"status": {"eq": "active"}} - status equals "active" - {"tags": {"includes": ["python", "ml"]}} - tags include "python" or "ml" - {"created_at": {"gte": "2024-01-01"}} - created after date - {"score": {"lt": 100, "gte": 50}} - score between 50 and 100 order_by: Comma delimited list of fields to sort results by. Can order by standard agent fields like created_at, or by data fields. Data fields must be prefixed with "data.". If ordering desceding, use a " desc" suffix. Examples: - "data.name desc, created_at" - sort by name in descending order, and then by creation date page_size: Maximum number of items to return per page. Defaults to 10. offset: Number of items to skip from the beginning. Defaults to 0. include_total: Whether to include the total count in the response. Defaults to False to improve performance. It's recommended to only request on the first page. )rIr1rVrWrXrYrZNr?itemshas_moretotal) r3rA5search_agent_data_api_v_1_beta_agent_data_search_postr2r1rr]rrCr0next_page_token total_size)r9rVrWrXrYrZrawitems r#searchAsyncAgentDataClient.searchs4KK$$ZZ(('[  #ORyyOXt'' By((4..    s$AB'B B'(B"B'"B'group_bycountfirstc V# URRRURURUUUUUUUS9 IShvN n[ UR V s/sH!n [R"XRS9PM# sn URSLURS9$N\s sn f7f)a: Aggregate agent data into groups according to the group_by fields. Args: filter: Filter conditions to apply to the search. Dict mapping field names to FilterOperation objects. Filters only by data fields See search for more details on filtering. group_by: List of fields to group by. Groups strictly by equality. Can only group by data fields. Examples: - ["name"] - group by name - ["name", "age"] - group by name and age count: Whether to include the count of items in each group. first: Whether to include the first item in each group. order_by: Comma delimited list of fields to sort results by. See search for more details on ordering. offset: Number of groups to skip from the beginning. Defaults to 0. page_size: Maximum number of groups to return per page. ) rIr1rYrVrWrgrhrirXNr?r\) r3rA;aggregate_agent_data_api_v_1_beta_agent_data_aggregate_postr2r1rr]rrCr0rarb) r9rVrgrhrirWrXrYrcrds r# aggregateAsyncAgentDataClient.aggregates4KK$$``((a   ( II%D$,,TYYG%((4..     s$AB)B"B)(B$B)$B))r2r3r1r0)defaultNNNN)NNNNF)NNNNNNN)__name__ __module__ __qualname____firstlineno____doc__rrstrrr r:r$rrErLrPrTrrrintboolrrerrrl__static_attributes__r+r%r#r.r.=se,b$&*,0#"&/://sm / ) / } /3-/bFcFnZ.HFF FjF^J5OFFFF",F  #FFBBBBFJ"& $#'# ( c4(:C(?#@@AB( 3-(  ( C= (  (  Z (( ( TFJ(, $ $"& $#'+ c4(:C(?#@@AB+ 49%+ ~ + ~ + 3- +  + C=+  "* -+ + r%r.)r)typingrrrrrrllama_cloud.clientr tenacityr r r r rrschemarrrrrrr$rtr,r.r+r%r#r|s^ ;;.  9  * Bc B U 7:.U r%