Yj dZddlmZddlmZmZmZmZddlm Z m Z ddl m Z ddl mZede Zed e Zd d gZGd d e eefZGdd e eefZdS)z"Pagination wrapper for Honcho SDK.) annotations) AsyncIterator AwaitableCallableIterator)AnyGeneric) BaseModel)TypeVarT)boundU)defaultSyncPage AsyncPageceZdZdZ d d!d Zd"dZd#dZd$dZed%dZ ed&dZ ed&dZ ed&dZ ed&dZ d'dZd(dZdS))rz Paginated result wrapper that transforms objects from type T to type U. Provides iteration and transformation capabilities for paginated API responses. Ndatadict[str, Any] item_typetype[T]transform_funcCallable[[T], U] | None fetch_next(Callable[[int], 'SyncPage[T, U]'] | NonereturnNonec||_|_||_||_|dg}fd|D|_dS)a Initialize the page. Args: data: Raw paginated response data with items, page, size, total, pages item_type: Type to parse items as transform_func: Optional function to transform objects from type T to type U. If None, objects are passed through unchanged. fetch_next: Optional callback to fetch the next page. Takes page number. itemsc:g|]}|Smodel_validate.0itemrs X/home/ubuntu/.hermes/hermes-agent/venv/lib/python3.11/site-packages/honcho/pagination.py z%SyncPage.__init__..04$ $ $ /3I $ $T * *$ $ $ N_data _item_type_transform_func _fetch_nextget _raw_itemsselfrrrr raw_itemss ` r&__init__zSyncPage.__init__sf"&* #,8FEOHHWb)) $ $ $ $ 7@$ $ $ r)Iterator[U] | Iterator[T]c#K|}|E|jD]%}|j||V!|V&|}|CdSdS)aA Iterate over all transformed items across all pages. Warning: This iterator automatically fetches ALL subsequent pages as you iterate. For large datasets, this may result in many API calls. If you only need the current page, use the `items` property instead. Nr0r- get_next_pager2pager%s r&__iter__zSyncPage.__iter__4sz'+  '3..t444444JJJJ%%''D r)indexintU | TcX|j|}|j||S|Sz4Get a transformed item by index on the current page.r0r-r2r<r%s r& __getitem__zSyncPage.__getitem__F0u%   +''-- - r)c*t|jSz,Get the number of items on the current page.lenr0r2s r&__len__zSyncPage.__len__M4?###r)list[U] | list[T]c`jfdjDStjS).Get all transformed items on the current page.Nc:g|]}|Sr r-r$r%r2s r&r'z"SyncPage.items..U'KKK4D((..KKKr)r-r0listrIs`r&rzSyncPage.itemsQ:   +KKKK4?KKK KDO$$$r) int | Nonec6|jdSz/Get the total number of items across all pages.totalr+r/rIs r&rYzSyncPage.totalXz~~g&&&r)c6|jdSzGet the current page number.r:rZrIs r&r:z SyncPage.page]z~~f%%%r)c6|jdSzGet the page size.sizerZrIs r&raz SyncPage.sizebr^r)c6|jdSzGet the total number of pages.pagesrZrIs r&rdzSyncPage.pagesgr[r)boolc6|j}|j}||dS||kSzCheck if there's a next page.NFr:rdr2 current_page total_pagess r& has_next_pagezSyncPage.has_next_pagel+y j  ;#65k))r)'SyncPage[T, U] | None'c|sdS|jdS|j}|dS||dzSzx Fetch the next page of results. Returns None if there are no more pages or no fetch callback. Nrlr.r:r2rjs r&r8zSyncPage.get_next_pagetsV !!## 4   #4y  4 q 0111r)NN) rrrrrrrrrr)rr5r<r=rr>rr=rrLrrVrre)rrn)__name__ __module__ __qualname____doc__r4r;rCrJpropertyrrYr:rardrlr8r r)r&rrsC37?C      8(((($$$$$%%%X% '''X'&&&X&&&&X&'''X'****222222r)ceZdZdZ d d!d Zd"dZd#dZd$dZed%dZ ed&dZ ed&dZ ed&dZ ed&dZ d'dZd(dZdS))rz Async paginated result wrapper that transforms objects from type T to type U. Provides async iteration and transformation capabilities for paginated API responses. Nrrrrrrr4Callable[[int], Awaitable['AsyncPage[T, U]']] | Nonerrc||_|_||_||_|dg}fd|D|_dS)a Initialize the async page. Args: data: Raw paginated response data with items, page, size, total, pages item_type: Type to parse items as transform_func: Optional function to transform objects from type T to type U. If None, objects are passed through unchanged. fetch_next: Optional async callback to fetch the next page. Takes page number. rc:g|]}|Sr r!r#s r&r'z&AsyncPage.__init__..r(r)Nr*r1s ` r&r4zAsyncPage.__init__si"&* #,8F   HHWb)) $ $ $ $ 7@$ $ $ r)#AsyncIterator[U] | AsyncIterator[T]cK|}|M|jD]'}|j||WV"|WV(|d{V}|KdSdS)aG Async iterate over all transformed items across all pages. Warning: This iterator automatically fetches ALL subsequent pages as you iterate. For large datasets, this may result in many API calls. If you only need the current page, use the `items` property instead. Nr7r9s r& __aiter__zAsyncPage.__aiter__s(,  '3..t4444444JJJJJ++--------D r)r<r=r>cX|j|}|j||S|Sr@rArBs r&rCzAsyncPage.__getitem__rDr)c*t|jSrFrGrIs r&rJzAsyncPage.__len__rKr)rLc`jfdjDStjS)rNNc:g|]}|Sr rPrQs r&r'z#AsyncPage.items..rRr)rSrIs`r&rzAsyncPage.itemsrUr)rVc6|jdSrXrZrIs r&rYzAsyncPage.totalr[r)c6|jdSr]rZrIs r&r:zAsyncPage.pager^r)c6|jdSr`rZrIs r&razAsyncPage.sizer^r)c6|jdSrcrZrIs r&rdzAsyncPage.pagesr[r)rec6|j}|j}||dS||kSrgrhris r&rlzAsyncPage.has_next_pagermr)'AsyncPage[T, U] | None'cK|sdS|jdS|j}|dS||dzd{VSrprrrss r&r8zAsyncPage.get_next_pagesl !!## 4   #4y  4%%lQ&6777777777r)rt) rrrrrrrrrr)rrrurvrwrxry)rr)rzr{r|r}r4rrCrJr~rrYr:rardrlr8r r)r&rrsC37KO      <....$$$$$%%%X% '''X'&&&X&&&&X&'''X'****888888r)N)r} __future__rcollections.abcrrrrtypingrr pydanticr typing_extensionsr r r__all__rrr r)r&rs0((""""""HHHHHHHHHHHH%%%%%% GCy!!! GC { #r2r2r2r2r2wq!t}r2r2r2jt8t8t8t8t81 t8t8t8t8t8r)