]jMddlZddlmZddlmZddlmZmZddlm Z m Z ddl m Z ddl mZdd lmZdd lmZej&ej(d ZGd d ZGddZy)N)JSONDecodeError)ApiError)AsyncClientWrapperSyncClientWrapper)AsyncHttpResponse HttpResponse)jsonable_encoder)RequestOptions)construct_type) Prediction.c$eZdZdefdZdddddej edej edej ede eje fd Z e e e e dd d ej ed ej ejej eej"fd ej edej edej ede e f dZdddedej ede e fdZdddedej ede dfdZe e e e dd ded ej ed ej ejej eej"fd ej edej edej ede e fdZy)RawPredictionsClientclient_wrapperc||_yN_client_wrapperselfrs Q/root/env/lib/python3.12/site-packages/label_studio_sdk/predictions/raw_client.py__init__zRawPredictionsClient.__init__ -Nprojecttaskrequest_optionsrrrreturncR|jjjdd||d|} d|jcxkrdkrinnft j tj tttj t|j}t||S|j}t|jt|j| #t$r6t|jt|j|j wxYw) a List all predictions and their IDs. Parameters ---------- project : typing.Optional[int] Filter predictions by project ID task : typing.Optional[int] Filter predictions by task ID request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[typing.List[Prediction]] Predictions list api/predictions/GETrrmethodparamsr,type_object_responsedata status_codeheadersbody)r httpx_clientrequestr0typingcastListr r jsonr rrdictr1textrrrr _response_data_response_jsons rlistzRawPredictionsClient.lists4((55== ",>  ti++1c1 KK +"$kk*5 ) 0$YUCC&^^-N9#8#8$yGXGXBY`noo ty'<'s rcreatezRawPredictionsClient.createHs D((55== !.   2,>  ti++1c1 "( ) 0$YUCC&^^-N9#8#8$yGXGXBY`noo ty'<'s rgetzRawPredictionsClient.gets"((55==/34A 6+> ti++1c1 "( ) 0$YUCC&^^-N9#8#8$yGXGXBY`noo ty'<'s rdeletezRawPredictionsClient.deletes ((55==/34A 6+>  ti++1c1#YTBB&^^-N9#8#8$yGXGXBY`noo ty'<'s rupdatezRawPredictionsClient.updatesL((55==/34A 6!.   2,>  ti++1c1 "( ) 0$YUCC&^^-N9#8#8$yGXGXBY`noo ty'<'.*;. )-%);? 0p%0pooc" 0p  8 0p fkk*- . 0pj/3QU(,%);?>ps+>p CO0L MN >p u% >p ooc" >p 8>p j !>p@RV#pc#pv~/N#pZfgqZr#pJUYpp&//.2Qp]ijn]opB/3QU(,%);?Bp Bps+ Bp  CO0L MN Bp u% Bpooc"Bp 8Bp j !Bprrc$eZdZdefdZdddddej edej edej ede eje fd Z e e e e dd d ej ed ej ejej eej"fd ej edej edej ede e f dZdddedej ede e fdZdddedej ede dfdZe e e e dd ded ej ed ej ejej eej"fd ej edej edej ede e fdZy)AsyncRawPredictionsClientrc||_yrrrs rrz"AsyncRawPredictionsClient.__init__rrNrrrrrcnK|jjjdd||d|d{} d|jcxkrdkrinnft j tj tttj t|j}t|| S|j}t|jt|j| 7#t$r6t|jt|j|j wxYww) a List all predictions and their IDs. Parameters ---------- project : typing.Optional[int] Filter predictions by project ID task : typing.Optional[int] Filter predictions by task ID request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[typing.List[Prediction]] Predictions list r!r"r#r$Nr'r(r)r,r/)rr3r4r0r5r6r7r r r8rrrr9r1r:r;s rr?zAsyncRawPredictionsClient.lists4..;;CC ",D   ti++1c1 KK +"$kk*5 ) 0))%HH&^^-N9#8#8$yGXGXBY`noo- ( ty'<'K|jjjdd||||dddi|td{} d|jcxkrd krGnnDt j ttt|j }t|| S|j}t|jt|j| 7#t$r6t|jt|j|j wxYww) a Create a prediction for a specific task. Parameters ---------- model_version : typing.Optional[str] Model version - tag for predictions that can be used to filter tasks in Data Manager, as well as select specific model version for showing preannotations in the labeling interface result : typing.Optional[typing.Sequence[typing.Dict[str, typing.Any]]] Prediction result in JSON format. Read more about the format in [the Label Studio documentation.](https://labelstud.io/guide/predictions) score : typing.Optional[float] Prediction score. Can be used in Data Manager to sort task by model confidence. Task with the lowest score will be shown first. task : typing.Optional[int] Task ID for which the prediction is created request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[Prediction] Created prediction r!rErFrGrHrINr'r(r)r,r/)rr3r4rKr0r5r6r r r8rrrr9r1r:rLs rrMz AsyncRawPredictionsClient.createEsD..;;CC !.   2,D   ti++1c1 "( ) 0))%HH&^^-N9#8#8$yGXGXBY`noo9 4 ty'<'ps+>p CO0L MN >p u% >p ooc" >p 8>p : &>pBNR%p%p+1??>+J%p : &%pPNRpp+1??>+Jp 4 pF/3QU(,%);?Bp Bps+ Bp  CO0L MN Bp u% Bpooc"Bp 8Bp : &Bprri)r5 json.decoderrcore.api_errorrcore.client_wrapperrrcore.http_responserr core.jsonable_encoderr core.request_optionsr core.unchecked_base_modelr types.predictionr r6rerKrrirgrrrxsW(%G@416)v{{6::s#zpzpz~p~pr