jTddlZddlZddlmZmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZd d lmZmZd d lmZej2ej4dZGddZGddZy)N)AsyncClientWrapperSyncClientWrapper)RequestOptions)JsonListStringWrite)OptimizationPagePublic)OptimizationPublic)OptimizationStudioConfigWrite)OptimizationStudioLog)OptimizationWriteStatus)AsyncRawOptimizationsClientRawOptimizationsClient)OptimizationUpdateStatus.c|eZdZdefdZedefdZdddede je ddfd Z ddddddddd d e je d e je d e jede jede jede jede jede je defdZeeeeedddededede jede jede jede jede jej,de je ddfdZeeeeedddededede jede jede jede jede jej,de je ddfdZddde j2ede je ddfdZdddede je defdZeedddede jede jede je ddf dZdddede je defd Z y)!OptimizationsClientclient_wrapperc&t||_yN)r)r _raw_clientselfrs {/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/rest_api/optimizations/client.py__init__zOptimizationsClient.__init__s1Preturnc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- RawOptimizationsClient rrs rwith_raw_responsez%OptimizationsClient.with_raw_responserNrequest_optionsidr#cT|jj||}|jS)a Cancel Studio optimizations by id Parameters ---------- id : str 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.optimizations.cancel_studio_optimizations(id='id', ) r"rcancel_studio_optimizationsdatarr$r# _responses rr'z/OptimizationsClient.cancel_studio_optimizations$s**$$@@Ud@e ~~rpagesize dataset_idname dataset_namedataset_deletedfiltersr#r,r-r.r/r0r1r2c `|jj||||||||} | jS)a Find optimizations Parameters ---------- page : typing.Optional[int] size : typing.Optional[int] dataset_id : typing.Optional[str] name : typing.Optional[str] dataset_name : typing.Optional[str] dataset_deleted : typing.Optional[bool] filters : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- OptimizationPagePublic Optimizations resource Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.optimizations.find_optimizations() r+rfind_optimizationsr( rr,r-r.r/r0r1r2r#r*s rr5z&OptimizationsClient.find_optimizations<sDZ$$77!%++8  ~~rr$r/metadata studio_configlast_updated_atr#objective_namestatusr8r9r:c b|jj|||||||||  } | jS)ae Create optimization Parameters ---------- dataset_name : str objective_name : str status : OptimizationWriteStatus id : typing.Optional[str] name : typing.Optional[str] metadata : typing.Optional[JsonListStringWrite] studio_config : typing.Optional[OptimizationStudioConfigWrite] last_updated_at : typing.Optional[dt.datetime] 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.optimizations.create_optimization(dataset_name='dataset_name', objective_name='objective_name', status="running", ) r0r;r<r$r/r8r9r:r#rcreate_optimizationr( rr0r;r<r$r/r8r9r:r#r*s rr@z'OptimizationsClient.create_optimizationuG^$$88%)'++9  ~~rc b|jj|||||||||  } | jS)ae Upsert optimization Parameters ---------- dataset_name : str objective_name : str status : OptimizationWriteStatus id : typing.Optional[str] name : typing.Optional[str] metadata : typing.Optional[JsonListStringWrite] studio_config : typing.Optional[OptimizationStudioConfigWrite] last_updated_at : typing.Optional[dt.datetime] 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.optimizations.upsert_optimization(dataset_name='dataset_name', objective_name='objective_name', status="running", ) r>rupsert_optimizationr(rAs rrEz'OptimizationsClient.upsert_optimizationrBridscT|jj||}|jS)a Delete optimizations by id Parameters ---------- ids : typing.Sequence[str] 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.optimizations.delete_optimizations_by_id(ids=['ids'], ) rFr#rdelete_optimizations_by_idr(rrFr#r*s rrJz.OptimizationsClient.delete_optimizations_by_ids*.$$??CYh?i ~~rcT|jj||}|jS)a Get optimization by id Parameters ---------- id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- OptimizationPublic Optimization resource Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.optimizations.get_optimization_by_id(id='id', ) r"rget_optimization_by_idr(r)s rrNz*OptimizationsClient.get_optimization_by_ids*0$$;;BP_;` ~~rr/r<r#cX|jj||||}|jS)a9 Update optimization by id Parameters ---------- id : str name : typing.Optional[str] status : typing.Optional[OptimizationUpdateStatus] 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.optimizations.update_optimizations_by_id(id='id', ) rOrupdate_optimizations_by_idr(rr$r/r<r#r*s rrRz.OptimizationsClient.update_optimizations_by_id"s6@$$?? T&/@ ~~rcT|jj||}|jS)a" Get presigned S3 URL for downloading optimization logs Parameters ---------- id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- OptimizationStudioLog Logs response Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.optimizations.get_studio_optimization_logs(id='id', ) r"rget_studio_optimization_logsr(r)s rrVz0OptimizationsClient.get_studio_optimization_logsGs*0$$AA"VeAf ~~r)!__name__ __module__ __qualname__rrpropertyrr strtypingOptionalrr'intboolrr5OMITr rr dtdatetimer@rESequencerJr rNrrRr rVrrrrs|Q*;Q #9  jncvWeGfrv6&*%)+/%)-115(,;?7ooc"7ooc" 7 OOC( 7 ooc" 7ooc*7 .7%7 87 7~$(%)9=HL8<;?:: : ( : OOC :ooc"://"56:'DE: 5: 8: :D$(%)9=HL8<;?:: : ( : OOC :ooc"://"56:'DE: 5: 8: :z`dooc*=C__^=\ 6NR+1??>+J >&*<@;? # #ooc" #  89 #  8 # #LNR+1??>+J rrc|eZdZdefdZedefdZdddede je ddfd Z ddddddddd d e je d e je d e jede jede jede jede jede je defdZeeeeedddededede jede jede jede jede jej,de je ddfdZeeeeedddededede jede jede jede jede jej,de je ddfdZddde j2ede je ddfdZdddede je defdZeedddede jede jede je ddf dZdddede je defd Z y)!AsyncOptimizationsClientrc&t||_yr)rrrs rrz!AsyncOptimizationsClient.__init__ds6nUrrc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- AsyncRawOptimizationsClient rrs rr z*AsyncOptimizationsClient.with_raw_responsegr!rNr"r$r#cpK|jj||d{}|jS7w)aJ Cancel Studio optimizations by id Parameters ---------- id : str 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.optimizations.cancel_studio_optimizations(id='id', ) asyncio.run(main()) r"Nr&r)s rr'z4AsyncOptimizationsClient.cancel_studio_optimizationsrs74**FFr[jFkk ~~l !646r+r,r-r.r/r0r1r2c |K|jj||||||||d{} | jS7w)al Find optimizations Parameters ---------- page : typing.Optional[int] size : typing.Optional[int] dataset_id : typing.Optional[str] name : typing.Optional[str] dataset_name : typing.Optional[str] dataset_deleted : typing.Optional[bool] filters : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- OptimizationPagePublic Optimizations resource 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.optimizations.find_optimizations() asyncio.run(main()) r+Nr4r6s rr5z+AsyncOptimizationsClient.find_optimizationssR`**==!%++>   ~~  s '<:<r7r;r<r8r9r:c ~K|jj|||||||||  d{} | jS7w)a Create optimization Parameters ---------- dataset_name : str objective_name : str status : OptimizationWriteStatus id : typing.Optional[str] name : typing.Optional[str] metadata : typing.Optional[JsonListStringWrite] studio_config : typing.Optional[OptimizationStudioConfigWrite] last_updated_at : typing.Optional[dt.datetime] 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.optimizations.create_optimization(dataset_name='dataset_name', objective_name='objective_name', status="running", ) asyncio.run(main()) r>Nr?rAs rr@z,AsyncOptimizationsClient.create_optimizationUd**>>%)'++?   ~~  (=;=c ~K|jj|||||||||  d{} | jS7w)a Upsert optimization Parameters ---------- dataset_name : str objective_name : str status : OptimizationWriteStatus id : typing.Optional[str] name : typing.Optional[str] metadata : typing.Optional[JsonListStringWrite] studio_config : typing.Optional[OptimizationStudioConfigWrite] last_updated_at : typing.Optional[dt.datetime] 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.optimizations.upsert_optimization(dataset_name='dataset_name', objective_name='objective_name', status="running", ) asyncio.run(main()) r>NrDrAs rrEz,AsyncOptimizationsClient.upsert_optimization rmrnrFcpK|jj||d{}|jS7w)aX Delete optimizations by id Parameters ---------- ids : typing.Sequence[str] 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.optimizations.delete_optimizations_by_id(ids=['ids'], ) asyncio.run(main()) rHNrIrKs rrJz3AsyncOptimizationsClient.delete_optimizations_by_idIs74**EE#_nEoo ~~prjcpK|jj||d{}|jS7w)aj Get optimization by id Parameters ---------- id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- OptimizationPublic Optimization resource 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.optimizations.get_optimization_by_id(id='id', ) asyncio.run(main()) r"NrMr)s rrNz/AsyncOptimizationsClient.get_optimization_by_idfs76**AA"VeAff ~~grjrOctK|jj||||d{}|jS7w)a Update optimization by id Parameters ---------- id : str name : typing.Optional[str] status : typing.Optional[OptimizationUpdateStatus] 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.optimizations.update_optimizations_by_id(id='id', ) asyncio.run(main()) rONrQrSs rrRz3AsyncOptimizationsClient.update_optimizations_by_idsDF**EE T&/F  ~~ s #868cpK|jj||d{}|jS7w)a Get presigned S3 URL for downloading optimization logs Parameters ---------- id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- OptimizationStudioLog Logs response 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.optimizations.get_studio_optimization_logs(id='id', ) asyncio.run(main()) r"NrUr)s rrVz5AsyncOptimizationsClient.get_studio_optimization_logss76**GG\kGll ~~mrj)!rWrXrYrrrZrr r[r\r]rr'r^r_rr5r`r rr rarbr@rErcrJr rNrrRr rVrdrrrfrfcsV*<V #>  NR+1??>+J @&*%)+/%)-115(,;?:ooc":ooc" : OOC( : ooc" :ooc*: .:%: 8: :D$(%)9=HL8<;?== = ( = OOC =ooc"=//"56='DE= 5= 8= =J$(%)9=HL8<;?== = ( = OOC =ooc"=//"56='DE= 5= 8= =@`dooc*=C__^=\ <NR+1??>+J D&*<@;? & &ooc" &  89 &  8 & &RNR+1??>+J rrf)rbrar\core.client_wrapperrrcore.request_optionsrtypes.json_list_string_writertypes.optimization_page_publicrtypes.optimization_publicr &types.optimization_studio_config_writer types.optimization_studio_logr types.optimization_write_statusr raw_clientrr types.optimization_update_statusrcastAnyr`rrfrdrrrs\ G1>C:RAEKFv{{6::s#KK\ eer