o Zh@sdZddlZddlmZmZmZmZmZmZm Z m Z e dZ e dZ Gdddee Z Gdd d e e ZGd d d e e ZdS) zPagers for the GenAI List APIs.N)Any AsyncIterator AwaitableCallableGenericIteratorLiteralTypeVarT)Z batch_jobsmodelsZ tuning_jobsfilesZcached_contentsc@seZdZdZdededefdedefddZdededefdedefd d Ze d e e fd d Z e d efddZ e d efddZe d eeeffddZd efddZded e fddZded dfddZdS) _BasePagerz9Base pager class for iterating through paginated results.namerequest.responseconfigcCsx||_||_t||jp g|_d|_|si}nt|tr"t|}nt|}t|d|d<||_ | dt |j|_ dS)NrZnext_page_token page_token page_size) _name_requestgetattr_page_idx isinstancedictcopydeepcopy_configgetlen _page_size)selfrrrrZrequest_configr"J/var/www/html/lang_env/lib/python3.10/site-packages/google/genai/pagers.py _init_page!s  z_BasePager._init_pagecCs|||||dSN)r$r!rrrrr"r"r#__init__9sz_BasePager.__init__returncC|jS)a`Returns a subset of the entire list of items. For the number of items returned, see `pageSize()`. Usage: .. code-block:: python batch_jobs_pager = client.batches.list(config={'page_size': 5}) print(f"first page: {batch_jobs_pager.page}") # first page: [BatchJob(name='projects/./locations/./batchPredictionJobs/1 )rr!r"r"r#pageBsz_BasePager.pagecCr))zReturns the type of paged item (for example, ``batch_jobs``). Usage: .. code-block:: python batch_jobs_pager = client.batches.list(config={'page_size': 5}) print(f"name: {batch_jobs_pager.name}") # name: batch_jobs )rr*r"r"r#rS z_BasePager.namecCr))aReturns the maximum number of items fetched by the pager at one time. Usage: .. code-block:: python batch_jobs_pager = client.batches.list(config={'page_size': 5}) print(f"page_size: {batch_jobs_pager.page_size}") # page_size: 5 )r r*r"r"r#rbr,z_BasePager.page_sizecCr))a Returns the configuration when making the API request for the next page. A configuration is a set of optional parameters and arguments that can be used to customize the API request. For example, the ``page_token`` parameter contains the token to request the next page. Usage: .. code-block:: python batch_jobs_pager = client.batches.list(config={'page_size': 5}) print(f"config: {batch_jobs_pager.config}") # config: {'page_size': 5, 'page_token': 'AMEw9yO5jnsGnZJLHSKDFHJJu'} )rr*r"r"r#rqsz_BasePager.configcCs t|jS)z6Returns the total number of items in the current page.)rr+r*r"r"r#__len__ z_BasePager.__len__indexcCs |j|S)z$Returns the item at the given index.)r+)r!r/r"r"r# __getitem__r.z_BasePager.__getitem__NcCs||j|j||jdS)zInitializes the next page from the response. This is an internal method that should be called by subclasses after fetching the next page. Args: response: The response object from the API request. N)r$rrrr!rr"r"r#_init_next_pages z_BasePager._init_next_page)__name__ __module__ __qualname____doc__ PagedItemrrr$r'propertylistr r+rintrrstrrr-r0r2r"r"r"r#r s>     r c@sBeZdZdZdefddZdeefddZdeefddZ d S) Pagerz4Pager class for iterating through paginated results.r(cCsL|jt|krz|Wn tytw|j|j}|jd7_|S)zReturns the next item.)rr next_page IndexError StopIterationr+r!itemr"r"r#__next__s   zPager.__next__cC d|_|S)z#Returns an iterator over the items.rrr*r"r"r#__iter__zPager.__iter__cCs2|jds td|j|jd}|||jS)aFetches the next page of items. This makes a new API request. Usage: .. code-block:: python batch_jobs_pager = client.batches.list(config={'page_size': 5}) print(f"current page: {batch_jobs_pager.page}") batch_jobs_pager.next_page() print(f"next page: {batch_jobs_pager.page}") # current page: [BatchJob(name='projects/.../batchPredictionJobs/1 # next page: [BatchJob(name='projects/.../batchPredictionJobs/6 rNo more pages to fetch.rrrr?rr2r+r1r"r"r#r>s  zPager.next_pageN) r3r4r5r6r rCrrFr9r>r"r"r"r#r<s  r<cspeZdZdZdededeefdedeffdd Zd e e fd d Z d e fd d Z d e e fddZZS) AsyncPagerz9AsyncPager class for iterating through paginated results.rr.rrcst||||dSr%)superr'r& __class__r"r#r'szAsyncPager.__init__r(cCrD)z)Returns an async iterator over the items.rrEr*r"r"r# __aiter__rGzAsyncPager.__aiter__csT|jt|krz |IdHWn tytw|j|j}|jd7_|S)z%Returns the next item asynchronously.Nr=)rrr>r?StopAsyncIterationr+rAr"r"r# __anext__s  zAsyncPager.__anext__cs:|jds td|j|jdIdH}|||jS)aSFetches the next page of items asynchronously. This makes a new API request. Returns: The next page of items. Raises: IndexError: No more pages to fetch. Usage: .. code-block:: python batch_jobs_pager = await client.aio.batches.list(config={'page_size': 5}) print(f"current page: {batch_jobs_pager.page}") await batch_jobs_pager.next_page() print(f"next page: {batch_jobs_pager.page}") # current page: [BatchJob(name='projects/.../batchPredictionJobs/1 # next page: [BatchJob(name='projects/.../batchPredictionJobs/6 rrHrINrJr1r"r"r#r>s   zAsyncPager.next_page)r3r4r5r6r7rrrr'rr rOrQr9r> __classcell__r"r"rMr#rKs  rK)r6rtypingrrrrrrrr r r7r r<rKr"r"r"r#s(z+