+jdZddlZddlmZmZmZmZmZmZm Z m Z m Z ddl m Z ddl mZe dZe dZGd d eeZGd d eeZGd deeZdS)zPagers for the GenAI List APIs.N) Any AsyncIterator AwaitableCallableGenericIteratorLiteralTypeVarUnion)_common)typesT) batch_jobsmodels tuning_jobsfilescached_contentsfile_search_stores documentsc NeZdZdZdededefdededdf d Zdededefdedefd Ze de e fd Z e defd Z e defd Ze deejdffdZe dejfdZdefdZdede fdZdeddfdZdS) _BasePagerz9Base pager class for iterating through paginated results.namerequest.responseconfigreturnNc||_||_t||jpg|_d|_t|dd|_|si}n9t |trtj |}nt|}t|d|d<||_ | dt|j|_ dS)Nrsdk_http_responsenext_page_token page_token page_size)_name_requestgetattr_page_idx_sdk_http_response isinstancedictcopydeepcopy_configgetlen _page_size)selfrrrrrequest_configs W/home/wildlama/visual-decline/.venv/lib/python3.11/site-packages/google/genai/pagers.py _init_pagez_BasePager._init_page)sDJDM4:..4"DJDI%h0CTJJD $nn FD ! !$}V,,nnF||n#*85F#G#GN< !DL)--k3tz??KKDOOOc6|||||dSN)r4)r1rrrrs r3__init__z_BasePager.__init__Cs" OOD'8V44444r5c|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 ty>>r5indexc|j|S)z$Returns the item at the given index.)r;)r1rBs r3 __getitem__z_BasePager.__getitem__s 9U r5cT||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)r4rr$rr1rs r3_init_next_pagez_BasePager._init_next_pages( OODIt}h DDDDDr5)__name__ __module__ __qualname____doc__ PagedItemrrr4r8propertylistrr;rintr"r r HttpResponserr StringDictrrArDrGr5r3rr&sAAL LS!L L  L  LLLL45 5S!5 5  5555 DG 8   I    8       8  #u'94'?!@### 8#  g( 8$ssq Ec Ed E E E E E Er5rcNeZdZdZdefdZdeefdZdeefdZ dS)Pagerz4Pager class for iterating through paginated results.rc|jt|kr+ |n#t$rtwxYw|j|j}|xjdz c_|S)zReturns the next item.r )r'r/ next_page IndexError StopIterationr;r1items r3__next__zPager.__next__sr yCII   9TY DIINII Ks /Acd|_|S)z#Returns an iterator over the items.rr'r:s r3__iter__zPager.__iter__DI Kr5c|jdstd||j}|||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 r!No more pages to fetch.rrr.rWr$rGr;rFs r3rVzPager.next_pagesZ ;??< ( (2 0 1 11}}DK}00H""" 9r5N) rHrIrJrKrr[rr^rNrVrRr5r3rTrTss<<       ar5rTceZdZdZdededeefdedeffd Zde e fd Z de fd Z de e fd ZxZS) AsyncPagerz9AsyncPager class for iterating through paginated results.rr.rrcPt||||dSr7)superr8)r1rrrr __class__s r3r8zAsyncPager.__init__s) GGT7Hf55555r5rcd|_|S)z)Returns an async iterator over the items.rr]r:s r3 __aiter__zAsyncPager.__aiter__r_r5cK|jt|kr1 |d{Vn#t$rtwxYw|j|j}|xjdz c_|S)z%Returns the next item asynchronously.Nr )r'r/rVrWStopAsyncIterationr;rYs r3 __anext__zAsyncPager.__anext__s yCII!nn !!!  ! 9TY DIINII Ks 7A cK|jdstd||jd{V}|||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 r!rarbNrcrFs r3rVzAsyncPager.next_pagesp. ;??< ( (2 0 1 11]]$+]66666666H""" 9r5)rHrIrJrKrLrrrr8rrrjrmrNrV __classcell__)rhs@r3reresAA6 6Ys^+,6 6  666666q) q    tAwr5re)rKr+typingrrrrrrr r r r rrrLrrTrerRr5r3rrs@ &% ffffffffffffffffffffff GCLL   EEEEEEEED(((((JqM(((V99999A99999r5