o 0Xxi@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) batch_jobsmodels tuning_jobsfilescached_contentsc@seZdZdZdedeeefdedefddZede e fd d Z ede fd d Z edefd dZedee effddZdefddZdede fddZdeddfddZdS) _BasePagerz9Base pager class for iterating through paginated results.namerequestresponseconfigcCsx||_||_t||jp g|_d|_|si}nt|tr"t|}nt|}t|d|d<||_ | dt |j|_ dS)Nrnext_page_token page_token page_size) _name_requestgetattr_page_idx isinstancedictcopydeepcopy_configgetlen _page_size)selfrrrrrequest_configr'R/home/air/biblejyuku/back/venv/lib/python3.10/site-packages/google/genai/pagers.py__init__!s  z_BasePager.__init__returncC|jS)alReturns the current page, which is a list of items. The returned list of items is a subset of the entire list. 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(page9z_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(rJs z_BasePager.namecCr+)aOReturns the length of the page fetched each time by this pager. The number of items in the page is less than or equal to the page length. 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(rYr.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'} )r!r,r'r'r(rjsz_BasePager.configcCs t|jS)z6Returns the total number of items in the current page.)r#r-r,r'r'r(__len__} z_BasePager.__len__indexcCs |j|S)z$Returns the item at the given index.)r-)r%r1r'r'r( __getitem__r0z_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)propertylistr r-strrintrrrr/r2r4r'r'r'r(rs,  rc@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.rrr"rArr4r-r3r'r'r(r@s  zPager.next_pageN) r5r6r7r8r rErrHr;r@r'r'r'r(r>s  r>csteZdZdZdedeeeefdedeffdd Zde e fd d Z dee fd d Z de e fd dZZS) AsyncPagerz9AsyncPager class for iterating through paginated results.rrrrcst||||dS)N)superr))r%rrrr __class__r'r(r)szAsyncPager.__init__r*cCrF)z)Returns an async iterator over the items.rrGr,r'r'r( __aiter__rIzAsyncPager.__aiter__csT|jt|krz |IdHWn tytw|j|j}|jd7_|S)z%Returns the next item asynchronously.Nr?)rr#r@rAStopAsyncIterationr-rCr'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 rrJrKNrLr3r'r'r(r@s   zAsyncPager.next_page)r5r6r7r8r9rrrr)rr rQrSr;r@ __classcell__r'r'rOr(rMs  rM)r8rtypingrrrrrrrr r r9rr>rMr'r'r'r(s(s+