jnwddlZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZmZddlmZej>ej@dZ!GddZ"GddZ#y)N)AsyncClientWrapperSyncClientWrapper)RequestOptions)JsonNode) LocalRunner)LocalRunnerConnectResponse)LocalRunnerHeartbeatResponse)LocalRunnerJob)LocalRunnerJobMetadata)LocalRunnerJobPage)LocalRunnerLogEntry)LocalRunnerPage)LocalRunnerPairResponse)AsyncRawRunnersClientRawRunnersClient)!LocalRunnerJobResultRequestStatus.ceZdZdefdZedefdZddddede je d e je de jefd Zdd ded e j ed e je ddfd Zdd ded e je ddfdZdd deded e je defdZeeedddedede jede jede jed e je ddfdZdd ded e je defdZdd ded e je defdZdd ded e je defdZdd ded e je defdZ ddddddede jede je d e je d e je de!f d!Z"dddd"dede je d e je d e je de#f d#Z$dd ded e je defd$Z%dd ded e jLee je jNfd e je ddfd%Z(eeedd&ded'e)d(e jed)e jed*e jed e je ddfd+Z*y), RunnersClientclient_wrapperc&t||_yN)r)r _raw_clientselfrs u/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/rest_api/runners/client.py__init__zRunnersClient.__init__s+>Jreturnc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- RawRunnersClient rrs rwith_raw_responsezRunnersClient.with_raw_responserNoffsetrequest_optionsjob_idr'r(cV|jj|||}|jS)a7 Get log entries for a local runner job Parameters ---------- job_id : str offset : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- typing.List[LocalRunnerLogEntry] Log entries Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.runners.get_job_logs(job_id='jobId', ) r&r get_job_logsdatarr)r'r( _responses rr,zRunnersClient.get_job_logs's,<$$11&Yh1i ~~rr(requestcV|jj|||}|jS)a Append log entries for a running local runner job Parameters ---------- job_id : str request : typing.Sequence[LocalRunnerLogEntry] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import OpikApi from Opik import LocalRunnerLogEntry client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.runners.append_job_logs(job_id='jobId', request=[LocalRunnerLogEntry(stream='stream', text='text', )], ) r1r(rappend_job_logsr-rr)r1r(r/s rr5zRunnersClient.append_job_logsHs,<$$44VW^m4n ~~rcT|jj||}|jS)a Cancel a pending or running local runner job Parameters ---------- job_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.runners.cancel_job(job_id='jobId', ) r0r cancel_jobr-rr)r(r/s rr9zRunnersClient.cancel_jobis)*$$///X ~~r pairing_code runner_namecV|jj|||}|jS)as Exchange a pairing code or API key for local runner credentials Parameters ---------- pairing_code : str runner_name : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- LocalRunnerConnectResponse Runner connected Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.runners.connect_runner(pairing_code='pairing_code', runner_name='runner_name', ) r;r<r(rconnect_runnerr-rr;r<r(r/s rr@zRunnersClient.connect_runners34$$33%;P_4 ~~rinputsmask_idmetadatar( agent_name project_idrCrDrEc\|jj||||||}|jS)a Create a local runner job and enqueue it for execution Parameters ---------- agent_name : str project_id : str inputs : typing.Optional[JsonNode] mask_id : typing.Optional[str] metadata : typing.Optional[LocalRunnerJobMetadata] 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.runners.create_job(agent_name='agent_name', project_id='project_id', ) rFrGrCrDrEr(r create_jobr-rrFrGrCrDrEr(r/s rrKzRunnersClient.create_jobs>L$$//!!+ 0 ~~rcT|jj||}|jS)aE Generate a pairing code for a local runner in the current workspace Parameters ---------- project_id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- LocalRunnerPairResponse Pairing code generated Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.runners.generate_pairing_code(project_id='project_id', ) rGr(rgenerate_pairing_coder-rrGr(r/s rrPz#RunnersClient.generate_pairing_codes*0$$::jbq:r ~~rcT|jj||}|jS)a Get a single local runner job's status and results Parameters ---------- job_id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- LocalRunnerJob Job details Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.runners.get_job(job_id='jobId', ) r0rget_jobr-r:s rrTzRunnersClient.get_jobs),$$,,V_,U ~~r runner_idcT|jj||}|jS)a Get a single local runner with its registered agents Parameters ---------- runner_id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- LocalRunner Runner details Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.runners.get_runner(runner_id='runnerId', ) r0r get_runnerr-rrUr(r/s rrXzRunnersClient.get_runners),$$// ?/[ ~~rcT|jj||}|jS)a Refresh local runner heartbeat Parameters ---------- runner_id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- LocalRunnerHeartbeatResponse Heartbeat response Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.runners.heartbeat(runner_id='runnerId', ) r0r heartbeatr-rYs rr\zRunnersClient.heartbeats)0$$..y/.Z ~~rrGpagesizer(r^r_cZ|jj|||||}|jS)aq List jobs for a local runner Parameters ---------- runner_id : str project_id : typing.Optional[str] page : typing.Optional[int] size : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- LocalRunnerJobPage Jobs list Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.runners.list_jobs(runner_id='runnerId', ) r]r list_jobsr-rrUrGr^r_r(r/s rrbzRunnersClient.list_jobs8s8H$$.. *4dTc/ ~~rr^r_r(cX|jj||||}|jS)an List local runners owned by the current user in the workspace Parameters ---------- project_id : str page : typing.Optional[int] size : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- LocalRunnerPage Runners list Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.runners.list_runners(project_id='project_id', ) r]r list_runnersr-rrGr^r_r(r/s rrgzRunnersClient.list_runnersas5B$$11!42 ~~rcT|jj||}|jS)a Long-poll for the next pending local runner job Parameters ---------- runner_id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- LocalRunnerJob Job available Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.runners.next_job(runner_id='runnerId', ) r0rnext_jobr-rYs rrkzRunnersClient.next_jobs),$$--i-Y ~~rcV|jj|||}|jS)aV Register or update the local runner's agent list Parameters ---------- runner_id : str request : typing.Dict[str, typing.Optional[typing.Any]] 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.runners.register_agents(runner_id='runnerId', request={'key': 'value' }, ) r3rregister_agentsr-rrUr1r(r/s rrnzRunnersClient.register_agentss,<$$44Yap4q ~~rresulterrortrace_idr(statusrqrrrsc\|jj||||||}|jS)a Report local runner job completion or failure Parameters ---------- job_id : str status : LocalRunnerJobResultRequestStatus result : typing.Optional[JsonNode] error : typing.Optional[str] trace_id : typing.Optional[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.runners.report_job_result(job_id='jobId', status="pending", ) rtrqrrrsr(rreport_job_resultr-rr)rtrqrrrsr(r/s rrxzRunnersClient.report_job_results:L$$66 6&bq7 ~~r)+__name__ __module__ __qualname__rrpropertyrr$strtypingOptionalintrListrr,Sequencer5r9r r@OMITrr rKrrPr rTrrXr r\r rbrrgrkDictAnyrnrrxrrrrsK*;K #3  (,;? $   8  ( ) L<@ !45   8   B]a&//.:Yei2jn"14GMWeGf #H-1(,<@;?.. . ) . % .//"89. 8. .bVZ 39??>3R 6Z^cv~7Vbp2`dCV__^=\hs4UY28//.2Q %>,0%)%);?''OOC( ' ooc" ' ooc" ' 8' 'Z&*%);? $$ooc" $ ooc" $  8 $ $L^b#6??>;Zft<<@ S&//&**"==>   8   L-1&*)-;?))2 ) ) ) s# )//#&) 8) )rrceZdZdefdZedefdZddddede je d e je de jefd Zdd ded e j ed e je ddfd Zdd ded e je ddfdZdd deded e je defdZeeedddedede jede jede jed e je ddfdZdd ded e je defdZdd ded e je defdZdd ded e je defdZdd ded e je defdZ ddddddede jede je d e je d e je de!f d!Z"dddd"dede je d e je d e je de#f d#Z$dd ded e je defd$Z%dd ded e jLee je jNfd e je ddfd%Z(eeedd&ded'e)d(e jed)e jed*e jed e je ddfd+Z*y),AsyncRunnersClientrc&t||_yr)rrrs rrzAsyncRunnersClient.__init__s0Orr c|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- AsyncRawRunnersClient r"r#s rr$z$AsyncRunnersClient.with_raw_responser%rNr&r)r'r(crK|jj|||d{}|jS7w)a Get log entries for a local runner job Parameters ---------- job_id : str offset : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- typing.List[LocalRunnerLogEntry] Log entries 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.runners.get_job_logs(job_id='jobId', ) asyncio.run(main()) r&Nr+r.s rr,zAsyncRunnersClient.get_job_logss:B**77v_n7oo ~~p "757r0r1crK|jj|||d{}|jS7w)a Append log entries for a running local runner job Parameters ---------- job_id : str request : typing.Sequence[LocalRunnerLogEntry] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import AsyncOpikApi from Opik import LocalRunnerLogEntry import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.runners.append_job_logs(job_id='jobId', request=[LocalRunnerLogEntry(stream='stream', text='text', )], ) asyncio.run(main()) r3Nr4r6s rr5z"AsyncRunnersClient.append_job_logs s:B**::67ds:tt ~~urcpK|jj||d{}|jS7w)aI Cancel a pending or running local runner job Parameters ---------- job_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.runners.cancel_job(job_id='jobId', ) asyncio.run(main()) r0Nr8r:s rr9zAsyncRunnersClient.cancel_jobDs60**55fo5^^ ~~_ !646r;r<crK|jj|||d{}|jS7w)a Exchange a pairing code or API key for local runner credentials Parameters ---------- pairing_code : str runner_name : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- LocalRunnerConnectResponse Runner connected 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.runners.connect_runner(pairing_code='pairing_code', runner_name='runner_name', ) asyncio.run(main()) r>Nr?rAs rr@z!AsyncRunnersClient.connect_runner_sA:**99%;P_:  ~~ rrBrFrGrCrDrEcxK|jj||||||d{}|jS7w)a# Create a local runner job and enqueue it for execution Parameters ---------- agent_name : str project_id : str inputs : typing.Optional[JsonNode] mask_id : typing.Optional[str] metadata : typing.Optional[LocalRunnerJobMetadata] 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.runners.create_job(agent_name='agent_name', project_id='project_id', ) asyncio.run(main()) rINrJrLs rrKzAsyncRunnersClient.create_jobsLR**55!!+ 6  ~~ %:8:cpK|jj||d{}|jS7w)a Generate a pairing code for a local runner in the current workspace Parameters ---------- project_id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- LocalRunnerPairResponse Pairing code generated 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.runners.generate_pairing_code(project_id='project_id', ) asyncio.run(main()) rNNrOrQs rrPz(AsyncRunnersClient.generate_pairing_codes76**@@Jhw@xx ~~yrcpK|jj||d{}|jS7w)an Get a single local runner job's status and results Parameters ---------- job_id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- LocalRunnerJob Job details 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.runners.get_job(job_id='jobId', ) asyncio.run(main()) r0NrSr:s rrTzAsyncRunnersClient.get_jobs62**226?2[[ ~~\rrUcpK|jj||d{}|jS7w)a| Get a single local runner with its registered agents Parameters ---------- runner_id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- LocalRunner Runner details 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.runners.get_runner(runner_id='runnerId', ) asyncio.run(main()) r0NrWrYs rrXzAsyncRunnersClient.get_runners76**55iQ`5aa ~~brcpK|jj||d{}|jS7w)az Refresh local runner heartbeat Parameters ---------- runner_id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- LocalRunnerHeartbeatResponse Heartbeat 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.runners.heartbeat(runner_id='runnerId', ) asyncio.run(main()) r0Nr[rYs rr\zAsyncRunnersClient.heartbeat s76**44YP_4`` ~~arr]r^r_cvK|jj|||||d{}|jS7w)a List jobs for a local runner Parameters ---------- runner_id : str project_id : typing.Optional[str] page : typing.Optional[int] size : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- LocalRunnerJobPage Jobs list 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.runners.list_jobs(runner_id='runnerId', ) asyncio.run(main()) r]Nrarcs rrbzAsyncRunnersClient.list_jobs*sFN**44 *4dTc5  ~~ s $979rdctK|jj||||d{}|jS7w)a List local runners owned by the current user in the workspace Parameters ---------- project_id : str page : typing.Optional[int] size : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- LocalRunnerPage Runners list 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.runners.list_runners(project_id='project_id', ) asyncio.run(main()) r]Nrfrhs rrgzAsyncRunnersClient.list_runnersVsCH**77!48  ~~ s #868cpK|jj||d{}|jS7w)aw Long-poll for the next pending local runner job Parameters ---------- runner_id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- LocalRunnerJob Job available 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.runners.next_job(runner_id='runnerId', ) asyncio.run(main()) r0NrjrYs rrkzAsyncRunnersClient.next_jobs66**33I3__ ~~`rcrK|jj|||d{}|jS7w)a Register or update the local runner's agent list Parameters ---------- runner_id : str request : typing.Dict[str, typing.Optional[typing.Any]] 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.runners.register_agents(runner_id='runnerId', request={'key': 'value' }, ) asyncio.run(main()) r3Nrmros rrnz"AsyncRunnersClient.register_agentss:B**::9ggv:ww ~~xrrprtrqrrrscxK|jj||||||d{}|jS7w)a Report local runner job completion or failure Parameters ---------- job_id : str status : LocalRunnerJobResultRequestStatus result : typing.Optional[JsonNode] error : typing.Optional[str] trace_id : typing.Optional[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.runners.report_job_result(job_id='jobId', status="pending", ) asyncio.run(main()) rvNrwrys rrxz$AsyncRunnersClient.report_job_resultsHR**<< 6&bq=  ~~ r)+rzr{r|rrr}rr$r~rrrrrrr,rr5r9r r@rrr rKrrPr rTrrXr r\r rbrrgrkrrrnrrxrrrrrs&P*<P #8  (,;? ""$ "  8 " ( ) "R<@ ""!45 "  8 "  "HcgsP^@_ko8jn " 14 GMWeGf  # N-1(,<@;?11 1 ) 1 % 1//"891 81 1hVZ 39??>3R <`dCV__^=\hv:UY28//.2Q >UY28//.2Q %D,0%)%);?**OOC( * ooc" * ooc" * 8* *`&*%);? ''ooc" ' ooc" '  8 ' 'TUY28//.2Q F<@ ""S&//&**"==> "  8 "  "R-1&*)-;?,,2 , ) , s# ,//#&, 8, ,rr)$rcore.client_wrapperrrcore.request_optionsrtypes.json_nodertypes.local_runnerr#types.local_runner_connect_responser %types.local_runner_heartbeat_responser types.local_runner_jobr types.local_runner_job_metadatar types.local_runner_job_pager types.local_runner_log_entryrtypes.local_runner_pager types.local_runner_pair_responser raw_clientrr,types.local_runner_job_result_request_statusrcastrrrrrrrrseG1&,LP3D<>5F?[v{{6::s#RRj@@r