LidZddlZddlmZmZmZmZmZmZm Z m Z m Z ddl m Z ddl mZe dZe dZGd d eeZGd d eeZGd deeZy)zPagers for the GenAI List APIs.N) Any AsyncIterator AwaitableCallableGenericIteratorLiteralTypeVarUnion)_common)typesT) batch_jobsmodels tuning_jobsfilescached_contentsfile_search_stores documentsc *eZdZdZdededefdededdf d Zdededefdedefd Ze de e fd Z e defd Z e defd Ze deej$dffdZe dej*fdZdefdZdede fdZdeddfdZy) _BasePagerz9Base pager class for iterating through paginated results.namerequest.responseconfigreturnNc|||_||_t||jxsg|_d|_t|dd|_|si}n1t |trtj|}n t|}t|d|d<||_ |jdt|j|_ y)Nrsdk_http_responsenext_page_token page_token page_size)_name_requestgetattr_page_idx_sdk_http_response isinstancedictcopydeepcopy_configgetlen _page_size)selfrrrrrequest_configs Q/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/google/genai/pagers.py _init_pagez_BasePager._init_page)sDJDM4::.4"DJDI%h0CTJD n FD !}}V,nF|n#*85F#GN< !DL)--k3tzz?KDOc,|j||||yN)r4)r1rrrrs r3__init__z_BasePager.__init__Cs OOD'8V4r5c|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 )r&r1s r3pagez_BasePager.pageLs ::r5c|jS)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 )r#r:s r3rz_BasePager.name]s ::r5c|jS)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 )r0r:s r3r"z_BasePager.page_sizels ??r5c|jS)z.Returns the http response of the API response.)r(r:s r3rz_BasePager.sdk_http_response{s  " ""r5c|jS)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:s r3rz_BasePager.configs" <<r5c,t|jS)z6Returns the total number of items in the current page.)r/r;r:s r3__len__z_BasePager.__len__s tyy>r5indexc |j|S)z$Returns the item at the given index.)r;)r1rBs r3 __getitem__z_BasePager.__getitem__s 99U r5ch|j|j|j||jy)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)r4rr$rr1rs r3_init_next_pagez_BasePager._init_next_pages" OODIIt}}h Dr5)__name__ __module__ __qualname____doc__ PagedItemrrr4r8propertylistrr;rintr"r r HttpResponserr StringDictrrArDrGr5r3rr&sSAL LS!L L  L  L45 5S!5 5  5 DG    I         #u'9'94'?!@# #  g(( $ssq Ec Ed Er5rc@eZdZdZdefdZdeefdZdeefdZ y)Pagerz4Pager class for iterating through paginated results.rc|jt|k\r |j|j |j}|xjdz c_|S#t$rtwxYw)zReturns the next item.r )r'r/ next_page IndexError StopIterationr;r1items r3__next__zPager.__next__s^ yyCI  99TYY DIINI K s AA*cd|_|S)z#Returns an iterator over the items.rr'r:s r3__iter__zPager.__iter__DI Kr5c|jjds td|j|j}|j ||j S)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 r!No more pages to fetch.rrr.rWr$rGr;rFs r3rVzPager.next_pagesL ;;??< ( 0 11}}DKK}0H" 99r5N) rHrIrJrKrr[rr^rNrVrRr5r3rTrTs1<    ar5rTcteZdZdZdededeefdedeffd Zde e fd Z de fd Z de e fd ZxZS) AsyncPagerz9AsyncPager class for iterating through paginated results.rr.rrc*t|||||yr7)superr8)r1rrrr __class__s r3r8zAsyncPager.__init__s GT7Hf5r5rcd|_|S)z)Returns an async iterator over the items.rr]r:s r3 __aiter__zAsyncPager.__aiter__r_r5cK|jt|k\r |jd{|j |j}|xjdz c_|S74#t$rtwxYww)z%Returns the next item asynchronously.Nr )r'r/rVrWStopAsyncIterationr;rYs r3 __anext__zAsyncPager.__anext__sj yyCI!nn 99TYY DIINI K  !  !s+A9A&A$A&0A9$A&&A66A9cK|jjds td|j|jd{}|j ||j S7!w)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 r!rarbNrcrFs r3rVzAsyncPager.next_pagesX. ;;??< ( 0 11]]$++]66H" 997sAA,A* "A,)rHrIrJrKrLrrrr8rrrjrmrNrV __classcell__)rhs@r3rereslA6 6Ys^+,6 6  6q) q tAwr5re)rKr+typingrrrrrrr r r r rrrLrrTrerRr5r3rrsl & fff CL   EED(JqM(V9A9r5