o Yi@sdZddlZddlmZmZmZmZmZmZm Z m Z m Z ddl m Z ddl mZe dZe dZGd d d eeZGd d d eeZGd ddeeZdS)zPagers for the GenAI List APIs.N) Any AsyncIterator AwaitableCallableGenericIteratorLiteralTypeVarUnion)_common)typesT)Z batch_jobsmodelsZ tuning_jobsfilesZcached_contentsZfile_search_storesZ documentsc @seZdZdZdededefdededdf d d Zdededefdedefd d Ze de e fd dZ e defddZ e defddZe deejdffddZe dejfddZdefddZdede fddZdeddfddZdS) _BasePagerz9Base pager class for iterating through paginated results.namerequest.responseconfigreturnNcCs||_||_t||jp g|_d|_t|dd|_|si}nt|tr)t |}nt|}t|d|d<||_ | dt |j|_ dS)Nrsdk_http_responseZnext_page_token page_token page_size)_name_requestgetattr_page_idx_sdk_http_response isinstancedictcopydeepcopy_configgetlen _page_size)selfrrrrZrequest_configr)M/home/kim/smarthome/.venv/lib/python3.10/site-packages/google/genai/pagers.py _init_page)s  z_BasePager._init_pagecCs|||||dSN)r+r(rrrrr)r)r*__init__Csz_BasePager.__init__cC|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*pageLsz_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 )rr0r)r)r*r] 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'r0r)r)r*rlr2z_BasePager.page_sizecCr/)z.Returns the http response of the API response.)rr0r)r)r*r{sz_BasePager.sdk_http_responsecCr/)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$r0r)r)r*rsz_BasePager.configcCs t|jS)z6Returns the total number of items in the current page.)r&r1r0r)r)r*__len__ z_BasePager.__len__indexcCs |j|S)z$Returns the item at the given index.)r1)r(r5r)r)r* __getitem__r4z_BasePager.__getitem__cCs||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.propertylistrr1rintrr r Z HttpResponserr Z StringDictrr3r6r8r)r)r)r*r&sF     rc@sBeZdZdZdefddZdeefddZdeefddZ d S) Pagerz4Pager class for iterating through paginated results.rcCsL|jt|krz|Wn tytw|j|j}|jd7_|S)zReturns the next item.r )rr& next_page IndexError StopIterationr1r(itemr)r)r*__next__s   zPager.__next__cC d|_|S)z#Returns an iterator over the items.rrr0r)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%rCrr8r1r7r)r)r*rBs  zPager.next_pageN) r9r:r;r<rrGrrJr?rBr)r)r)r*rAs  rAcspeZdZdZdededeefdedeffdd 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__rcCrH)z)Returns an async iterator over the items.rrIr0r)r)r* __aiter__rKzAsyncPager.__aiter__csT|jt|krz |IdHWn tytw|j|j}|jd7_|S)z%Returns the next item asynchronously.Nr )rr&rBrCStopAsyncIterationr1rEr)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 rrLrMNrNr7r)r)r*rBs   zAsyncPager.next_page)r9r:r;r<r=rrrr.rrrSrUr?rB __classcell__r)r)rQr*rOs  rO)r<r"typingrrrrrrrr r r r rr=rrArOr)r)r)r*s,   +