]jIddlZddlZddlmZddlmZddlmZm Z ddl m Z m Z ddl mZddlmZdd lmZdd lmZdd lmZdd lmZd dlmZej6ej8dZGddZGddZy)N)JSONDecodeError)ApiError)AsyncClientWrapperSyncClientWrapper)AsyncHttpResponse HttpResponse)jsonable_encoder)RequestOptions)construct_type)CancelModelRunResponse)ModelRun)ProjectSubsetEnum)ListRunsRequestProjectSubset.ceZdZdefdZdddddddededejedejed ejed eje d eje d e eje fd Zeeeeeeeedd deded edejedejedejedejej$d ejedejedejedejed eje d e e fdZdddededed eje d e ef dZy) RawRunsClientclient_wrapperc||_yN_client_wrapperselfrs R/root/env/lib/python3.12/site-packages/label_studio_sdk/prompts/runs/raw_client.py__init__zRawRunsClient.__init__ -Nordering parent_modelprojectproject_subsetrequest_options prompt_id version_idr r!r"r#r$returnc |jjjdt|dt|dd||||d|} d|jcxkrdkrinnft j t jttt jt|j } t|| S|j} t|jt|j| #t$r6t|jt|j|j wxYw) a Label Studio Enterprise badge

This endpoint is not available in Label Studio Community Edition. [Learn more about Label Studio Enterprise](https://humansignal.com/goenterprise)

Get information (status, metadata, etc) about an existing inference run Parameters ---------- prompt_id : int version_id : int ordering : typing.Optional[str] Which field to use when ordering the results. parent_model : typing.Optional[int] The ID of the parent model for this Inference Run project : typing.Optional[int] The ID of the project this Inference Run makes predictions on project_subset : typing.Optional[ListRunsRequestProjectSubset] Defines which tasks are operated on (e.g. HasGT will only operate on tasks with a ground truth annotation, but All will operate on all records) request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[typing.List[ModelRun]] api/prompts/ /versions//inference-runsGETr r!r"r#methodparamsr$,type_object_responsedata status_codeheadersbody)r httpx_clientrequestr r:typingcastListrr jsonr rrdictr;text rr%r&r r!r"r#r$ _response_data_response_jsons rlistzRawRunsClient.lists*\((55==+I67zBRS]B^A__n o$ ,""0  ,>   ti++1c1 KK)"$kk(3 ) 0$YUCC&^^-N9#8#8$yGXGXBY`noo ty'<' Label Studio Enterprise badge

This endpoint is not available in Label Studio Community Edition. [Learn more about Label Studio Enterprise](https://humansignal.com/goenterprise)

Run a prompt inference. Parameters ---------- prompt_id : int version_id : int project : int job_id : typing.Optional[str] Job ID for inference job for a ModelRun e.g. Adala job ID only_missing_predictions : typing.Optional[bool] When true, only tasks without successful predictions for this prompt version are submitted for inference. organization : typing.Optional[int] predictions_updated_at : typing.Optional[dt.datetime] project_subset : typing.Optional[ProjectSubsetEnum] total_correct_predictions : typing.Optional[int] total_predictions : typing.Optional[int] total_tasks : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[ModelRun] r)r*r+POST rKrLrMrNr"r#rOrPrQ content-typeapplication/jsonr/rBr;r$omitr1r2r3r6r9)rr=r>r OMITr:r?r@rr rBr rrrCr;rDrr%r&r"rKrLrMrNr#rOrPrQr$rFrGrHs rcreatezRawRunsClient.createbs7v((55==+I67zBRS]B^A__n o ,D ,*@""0-F%6*  2,%> ( ti++1c1 "& ) 0$YUCC&^^-N9#8#8$yGXGXBY`noo ty'<' Label Studio Enterprise badge

This endpoint is not available in Label Studio Community Edition. [Learn more about Label Studio Enterprise](https://humansignal.com/goenterprise)

Cancel the inference run for the given api Parameters ---------- prompt_id : int version_id : int inference_run_id : int request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[CancelModelRunResponse] r)r*/inference-runs//cancelrSr/r$r1r2r3r6r9)rr=r>r r:r?r@r r rBr rrrCr;rDrr%r&r]r$rFrGrHs rcancelzRawRunsClient.cancels+B((55==+I67zBRS]B^A__oqABRqSpTT[ \+> ti++1c1 *"4 ) 0$YUCC&^^-N9#8#8$yGXGXBY`noo ty'<'-1?C=A:>26,0;?\p\p\p  \p $ \p#)//$"7\pooc*\p!' <\p(9:\p$*??3#7\p"??3/\p__S)\p 8\p h \pH<@ 3p3p3p 3p  8 3p , -3prrceZdZdefdZdddddddededejedejed ejed eje d eje d e eje fd Zeeeeeeeedd deded edejedejedejedejej$d ejedejedejedejed eje d e e fdZdddededed eje d e ef dZy)AsyncRawRunsClientrc||_yrrrs rrzAsyncRawRunsClient.__init__rrNrr%r&r r!r"r#r$r'c K|jjjdt|dt|dd||||d|d{} d|jcxkrd krinnft j t jttt jt|j } t|| S|j} t|jt|j| 7#t$r6t|jt|j|j wxYww) a Label Studio Enterprise badge

This endpoint is not available in Label Studio Community Edition. [Learn more about Label Studio Enterprise](https://humansignal.com/goenterprise)

Get information (status, metadata, etc) about an existing inference run Parameters ---------- prompt_id : int version_id : int ordering : typing.Optional[str] Which field to use when ordering the results. parent_model : typing.Optional[int] The ID of the parent model for this Inference Run project : typing.Optional[int] The ID of the project this Inference Run makes predictions on project_subset : typing.Optional[ListRunsRequestProjectSubset] Defines which tasks are operated on (e.g. HasGT will only operate on tasks with a ground truth annotation, but All will operate on all records) request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[typing.List[ModelRun]] r)r*r+r,r-r.Nr1r2r3r6r9)rr=r>r r:r?r@rArr rBrrrrCr;rDrEs rrIzAsyncRawRunsClient.lists8\..;;CC+I67zBRS]B^A__n o$ ,""0  ,D    ti++1c1 KK)"$kk(3 ) 0))%HH&^^-N9#8#8$yGXGXBY`noo1  , ty'<' Label Studio Enterprise badge

This endpoint is not available in Label Studio Community Edition. [Learn more about Label Studio Enterprise](https://humansignal.com/goenterprise)

Run a prompt inference. Parameters ---------- prompt_id : int version_id : int project : int job_id : typing.Optional[str] Job ID for inference job for a ModelRun e.g. Adala job ID only_missing_predictions : typing.Optional[bool] When true, only tasks without successful predictions for this prompt version are submitted for inference. organization : typing.Optional[int] predictions_updated_at : typing.Optional[dt.datetime] project_subset : typing.Optional[ProjectSubsetEnum] total_correct_predictions : typing.Optional[int] total_predictions : typing.Optional[int] total_tasks : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[ModelRun] r)r*r+rSrTrUrVrWNr1r2r3r6r9)rr=r>r rYr:r?r@rr rBrrrrCr;rDrZs rr[zAsyncRawRunsClient.createBsFv..;;CC+I67zBRS]B^A__n o ,D ,*@""0-F%6*  2,%D  ( ti++1c1 "& ) 0))%HH&^^-N9#8#8$yGXGXBY`nooC > ty'<' Label Studio Enterprise badge

This endpoint is not available in Label Studio Community Edition. [Learn more about Label Studio Enterprise](https://humansignal.com/goenterprise)

Cancel the inference run for the given api Parameters ---------- prompt_id : int version_id : int inference_run_id : int request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[CancelModelRunResponse] r)r*r_r`rSraNr1r2r3r6r9)rr=r>r r:r?r@r r rBrrrrCr;rDrbs rrczAsyncRawRunsClient.cancels9B..;;CC+I67zBRS]B^A__oqABRqSpTT[ \+D  ti++1c1 *"4 ) 0))%HH&^^-N9#8#8$yGXGXBY`noo%  ty'<'-1?C=A:>26,0;?\p\p\p  \p $ \p#)//$"7\pooc*\p!' <\p(9:\p$*??3#7\p"??3/\p__S)\p 8\p 8 $\pH<@ 3p3p3p 3p  8 3p 1 23prro) rlrkr? json.decoderrcore.api_errorrcore.client_wrapperrrcore.http_responserr core.jsonable_encoderr core.request_optionsr core.unchecked_base_modelr types.cancel_model_run_responser types.model_runrtypes.project_subset_enumr&types.list_runs_request_project_subsetrr@AnyrYrrormrrrsc (&HA527E':Pv{{6::s#]p]p@]p]pr