jddlZddlmZmZddlmZddlmZmZejejdZ Gdd Z Gd d Z y) N)AsyncClientWrapperSyncClientWrapper)RequestOptions)AsyncRawOllieStateClientRawOllieStateClient.c0eZdZdefdZedefdZdddeje deje fdZ ddd eje eje eje fdeje ddfd Zdddeje ddfd Zy) OllieStateClientclient_wrapperc&t||_yN)r )r _raw_clientselfr s y/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/rest_api/ollie_state/client.py__init__zOllieStateClient.__init__s.nMreturnc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- RawOllieStateClient rrs rwith_raw_responsez"OllieStateClient.with_raw_responserNrequest_optionsrc#K|jj|5}|jEd{dddy7 #1swYyxYww)aO Download stored ollie state file Parameters ---------- request_options : typing.Optional[RequestOptions] Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response. Returns ------- typing.Iterator[bytes] Ollie state file Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.ollie_state.download_ollie_state() rNrdownload_ollie_statedata)rrrs rrz%OllieStateClient.download_ollie_statesE,   2 2? 2 SWXvv  T S T Ss%A ?=? A ?AA requestcT|jj||}|jS)l Upload gzip-compressed SQLite DB file, replacing any existing state Parameters ---------- request : typing.Union[bytes, typing.Iterator[bytes], typing.AsyncIterator[bytes]] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None r"rrreplace_ollie_stater rr"r _responses rr'z$OllieStateClient.replace_ollie_state5s*($$88Zi8j ~~rcR|jj|}|jS)a Clear stored ollie state Parameters ---------- request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.ollie_state.delete_ollie_state() rrdelete_ollie_stater rrr)s rr,z#OllieStateClient.delete_ollie_stateLs'&$$777X ~~r)__name__ __module__ __qualname__rrpropertyr rtypingOptionalrIteratorbytesrUnion AsyncIteratorr'r,rrr r sN*;N #6  EI"(//."A  :<@ eV__U%;V=Q=QRW=XXY 8   .X\V__^5T`drr c0eZdZdefdZedefdZdddeje deje fdZ ddd eje eje eje fdeje ddfd Zdddeje ddfd Zy) AsyncOllieStateClientr c&t||_yr)rrrs rrzAsyncOllieStateClient.__init__ds3>Rrrc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- AsyncRawOllieStateClient rrs rrz'AsyncOllieStateClient.with_raw_responsegrrNrrcK|jj|4d{}|j23d{}|77 6dddd{7y#1d{7swYyxYww)a Download stored ollie state file Parameters ---------- request_options : typing.Optional[RequestOptions] Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response. Returns ------- typing.AsyncIterator[bytes] Ollie state file Examples -------- from Opik import AsyncOpikApi import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.ollie_state.download_ollie_state() asyncio.run(main()) rNr)rrr!r s rrz*AsyncOllieStateClient.download_ollie_statersY2##888YY]^ff d Z fZYYYYsd!A/AA/ AAAAAA/AA A/AA/A, A# !A,(A/r"cpK|jj||d{}|jS7w)r$r%Nr&r(s rr'z)AsyncOllieStateClient.replace_ollie_states7(**>>w`o>pp ~~qs !646cnK|jj|d{}|jS7w)a Clear stored ollie state Parameters ---------- request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import AsyncOpikApi import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.ollie_state.delete_ollie_state() asyncio.run(main()) rNr+r-s rr,z(AsyncOllieStateClient.delete_ollie_states4,**==o=^^ ~~_s 535)r.r/r0rrr1rrr2r3rr7r5rr6r4r'r,r8rrr:r:csS*<S #;  EI"(//."A   e $B<@ eV__U%;V=Q=QRW=XXY 8   .^b6??>;Zfjrr:)r2core.client_wrapperrrcore.request_optionsr raw_clientrr castAnyOMITr r:r8rrrFsDG1Ev{{6::s#SSlZZr