]jCddlZddlmZmZddlmZddlmZddlm Z m Z ejejdZ Gd d ZGd d Zy) N)AsyncClientWrapperSyncClientWrapper)RequestOptions) Prediction)AsyncRawPredictionsClientRawPredictionsClient.ceZdZdefdZedefdZdddddeje deje d eje deje fd Zeeeedd d ejed ejej"ej$eej&fdejedeje d eje de f dZddde d eje de fdZddde d eje ddfdZeeeedd de d ejed ejej"ej$eej&fdejedeje d eje de fdZy)PredictionsClientclient_wrapperc&t||_yN)r )r _raw_clientselfr s M/root/env/lib/python3.12/site-packages/label_studio_sdk/predictions/client.py__init__zPredictionsClient.__init__s/~Nreturnc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- RawPredictionsClient rrs rwith_raw_responsez#PredictionsClient.with_raw_responserNprojecttaskrequest_optionsrrrcV|jj|||}|jS)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 ------- typing.List[Prediction] Predictions list Examples -------- from label_studio_sdk import LabelStudio client = LabelStudio( api_key="YOUR_API_KEY", ) client.predictions.list() rrlistdatarrrr _responses rr"zPredictionsClient.lists-F$$))'Ve)f ~~r model_versionresultscorerrr'r(r)cZ|jj|||||}|jS)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 ------- Prediction Created prediction Examples -------- from label_studio_sdk import LabelStudio client = LabelStudio( api_key="YOUR_API_KEY", ) client.predictions.create( model_version="yolo-v8", result=[ { "from_name": "bboxes", "image_rotation": 0, "original_height": 1080, "original_width": 1920, "to_name": "image", "type": "rectanglelabels", "value": { "height": 60, "rotation": 0, "values": {"rectanglelabels": ["Person"]}, "width": 50, "x": 20, "y": 30, }, } ], score=0.95, ) r&rcreater#rr'r(r)rrr%s rr,zPredictionsClient.createCs8@$$++'e$`o, ~~rridcT|jj||}|jS)a- Get details about a specific prediction by its ID. Parameters ---------- id : int Prediction ID request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- Prediction Prediction details Examples -------- from label_studio_sdk import LabelStudio client = LabelStudio( api_key="YOUR_API_KEY", ) client.predictions.get( id=1, ) r.rgetr#rr/rr%s rr2zPredictionsClient.gets)8$$((_(M ~~rcT|jj||}|jS)a Delete a prediction by prediction ID. Parameters ---------- id : int Prediction ID request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from label_studio_sdk import LabelStudio client = LabelStudio( api_key="YOUR_API_KEY", ) client.predictions.delete( id=1, ) r.rdeleter#r3s rr6zPredictionsClient.deletes)6$$++B+P ~~rc\|jj||||||}|jS)a Update prediction data by prediction ID. Parameters ---------- id : int Prediction ID 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 ------- Prediction Updated prediction Examples -------- from label_studio_sdk import LabelStudio client = LabelStudio( api_key="YOUR_API_KEY", ) client.predictions.update( id=1, model_version="yolo-v8", result=[ { "from_name": "bboxes", "image_rotation": 0, "original_height": 1080, "original_width": 1920, "to_name": "image", "type": "rectanglelabels", "value": { "height": 60, "rotation": 0, "values": {"rectanglelabels": ["Person"]}, "width": 50, "x": 20, "y": 30, }, } ], score=0.95, ) r&rupdater#rr/r'r(r)rrr%s rr9zPredictionsClient.updates:J$$++ mF%dds, ~~r)__name__ __module__ __qualname__rrpropertyr rtypingOptionalintrListrr"OMITstrSequenceDictAnyfloatr,r2r6r9rrr r s+O*;O #7  )-%);? $%$ooc" $  8 $ Z $R/3QU(,%);?Cs+C CO0L MN C u% C ooc" C 8C CJRVcv~/NZd>UY&//.2Q]aD/3QU(,%);?H Hs+ H  CO0L MN H u% Hooc"H 8H Hrr ceZdZdefdZedefdZdddddeje deje d eje deje fd Zeeeedd d ejed ejej"ej$eej&fdejedeje d eje de f dZddde d eje de fdZddde d eje ddfdZeeeedd de d ejed ejej"ej$eej&fdejedeje d eje de fdZy)AsyncPredictionsClientr c&t||_yr)r rrs rrzAsyncPredictionsClient.__init__s4NSrrc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- AsyncRawPredictionsClient rrs rrz(AsyncPredictionsClient.with_raw_responserrNrrrrcrK|jj|||d{}|jS7w)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 ------- typing.List[Prediction] Predictions list Examples -------- import asyncio from label_studio_sdk import AsyncLabelStudio client = AsyncLabelStudio( api_key="YOUR_API_KEY", ) async def main() -> None: await client.predictions.list() asyncio.run(main()) rNr!r$s rr"zAsyncPredictionsClient.lists:V**//d\k/ll ~~ms "757r&r'r(r)cvK|jj|||||d{}|jS7w)aF 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 ------- Prediction Created prediction Examples -------- import asyncio from label_studio_sdk import AsyncLabelStudio client = AsyncLabelStudio( api_key="YOUR_API_KEY", ) async def main() -> None: await client.predictions.create( model_version="yolo-v8", result=[ { "from_name": "bboxes", "image_rotation": 0, "original_height": 1080, "original_width": 1920, "to_name": "image", "type": "rectanglelabels", "value": { "height": 60, "rotation": 0, "values": {"rectanglelabels": ["Person"]}, "width": 50, "x": 20, "y": 30, }, } ], score=0.95, ) asyncio.run(main()) r&Nr+r-s rr,zAsyncPredictionsClient.createMsFP**11'e$`o2  ~~ s $979r.r/cpK|jj||d{}|jS7w)a Get details about a specific prediction by its ID. Parameters ---------- id : int Prediction ID request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- Prediction Prediction details Examples -------- import asyncio from label_studio_sdk import AsyncLabelStudio client = AsyncLabelStudio( api_key="YOUR_API_KEY", ) async def main() -> None: await client.predictions.get( id=1, ) asyncio.run(main()) r.Nr1r3s rr2zAsyncPredictionsClient.gets7H**..r?.SS ~~T !646cpK|jj||d{}|jS7w)at Delete a prediction by prediction ID. Parameters ---------- id : int Prediction ID request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- import asyncio from label_studio_sdk import AsyncLabelStudio client = AsyncLabelStudio( api_key="YOUR_API_KEY", ) async def main() -> None: await client.predictions.delete( id=1, ) asyncio.run(main()) r.Nr5r3s rr6zAsyncPredictionsClient.deletes7F**11"o1VV ~~WrQcxK|jj||||||d{}|jS7w)a Update prediction data by prediction ID. Parameters ---------- id : int Prediction ID 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 ------- Prediction Updated prediction Examples -------- import asyncio from label_studio_sdk import AsyncLabelStudio client = AsyncLabelStudio( api_key="YOUR_API_KEY", ) async def main() -> None: await client.predictions.update( id=1, model_version="yolo-v8", result=[ { "from_name": "bboxes", "image_rotation": 0, "original_height": 1080, "original_width": 1920, "to_name": "image", "type": "rectanglelabels", "value": { "height": 60, "rotation": 0, "values": {"rectanglelabels": ["Person"]}, "width": 50, "x": 20, "y": 30, }, } ], score=0.95, ) asyncio.run(main()) r&Nr8r:s rr9zAsyncPredictionsClient.updatesHZ**11 mF%dds2  ~~ s %:8:)r;r<r=rrr>r rr?r@rArrBrr"rCrDrErFrGrHr,r2r6r9rIrrrKrKs,T*<T #<  )-%);? ,%,ooc" ,  8 , Z ,b/3QU(,%);?Ks+K CO0L MN K u% K ooc" K 8K KZX\%C%V__^5T%`j%N[_$s$8W$cg$T/3QU(,%);?P Ps+ P  CO0L MN P u% Pooc"P 8P PrrK)r?core.client_wrapperrrcore.request_optionsrtypes.predictionr raw_clientr r castrGrCr rKrIrrrYsGG1)Gv{{6::s#Dggr