]jE~ddlZddlmZejjZddlmZddlm Z ddl m Z m Z dZ dZdZy) N)LABEL_CONFIG_AND_TASKS) LabelStudio)LabelInterface)PredictionValue TaskValuect||j}ttd}|jj d|j }|jj |jddi}t|jdjd ggd d }|jj dd |ji|j}|jj|j}|jdddd gk(sJ|j d k(sJ|j"d k(sJt|jdjdg|jdjdggdd }|jj dd |ji|j}|jj%|j} t'| dk(sJ|jj)|j|jj%|j} t'| dk(sJ| dj|jk(sJy)Nbase_urlapi_key label_config New Projecttitler my_textz Test taskprojectdatasentiment_classPositiveg?z1.0.0resultscore model_versiontaskidrvaluechoicesNeutralNegativeg?z1.0.1r)rr rrprojectscreateconfigtasksrr get_controllabel predictions model_dumpgetrrrlistlendelete) django_live_urlbusiness_clientlsliprpv predictionanother_predictionr+s Q/root/env/lib/python3.12/site-packages/label_studio/tests/sdk/test_predictions.pytest_predictions_CRUDr: s; o7N7N OB .~> ?B RYYGA 88??144y+.F? GD 0177 EF B '&& WW --/J ##z}}#5J   Q  ( 3 | CC C   s "" "  # #w .. .  NN, - 3 3YK @ NN, - 3 3ZL A   B/..ODGGOr}}O..%%477%3K { q  NNZ]]+..%%477%3K { q  q>   2 5 5 55 5c t||j}ttd}|jj d|j }|jj|jddddd dd ddgd g g}|jj|j D]Q}t|jj|jdk(sJ|j|jSt|dk(sJtddit|j!d j#dggddg}tddit|j!d j#d ggddg}|jj|j|j%|j%g|jj|j D]v}|jj|j} t| dk(sJ|j|vsH| dj&dk(sJ|j|jxt|dk(sJ|j&dk(sJ|jj)|jd|jj+|j} | j&dk(sJddlm} t1j2| 5} |jj)|jdddd j4j6dk(sJ| j4j8d d!dj;d"sJy#1swYUxYw)#Nr r r rz Hello worldr)rrzGoodbye Label Studior zWhat a beautiful dayr)rrequestpreannotated_from_fields)rr!r#rgffffff?z3.4.5r)rr+g333333?rr=r)rrr)ApiErrorz3.4.6ivalidation_errorsrzModel version doesn't exist)rr rrr%r&r' import_tasksrr(r.r/r+appendrrr)r*r,rupdater-label_studio_sdk.core.api_errorrCpytestraisesr status_codebody startswith) r1r2r3r4r5task_idsrtask1task2r+rrCes r9#test_create_predictions_with_importrR:s< o7N7N OB .~> ?B RYYGAKK 44%* E.: N.: N #4!4H add +!2>>&&DGG&45::: ! x=A    ' '89?? MN%   E / 0 '89?? MN%   EKK 44!!#U%5%5%78  add +%nn))tww)7 ;1$$$ 77( "q>//7: :: OODGG $ % x=A   ??b  KK!$$g6kkooo&G  G ++ +9 x ;A add':; 77  # %% % 77<<+ ,_ =a @ K KLi jj j;;s (NN ct||j}ttd}|jj d|j }|jj |jddi}gd}d gd gd gg}g}t||D]-\} } |jd d dd| idg| d|jd/|jj|j|} | jdk(sJ|jj|j} t| dk(sJt!| D cgc]} | j"c} }|t!|k(sJycc} w)a Import multiple predictions via projects.import_predictions Purpose: - Verify that the bulk predictions import endpoint creates predictions for a project Setup: - Create a project with a simple text classification config - Create one task in the project Actions: - Call ls.projects.import_predictions with three predictions for the same task Validations: - API returns created == 3 - Listing predictions for the task returns exactly three items with expected model versions r r zPredictions Import ProjectrrzClassify this sentencer)zhumor__gpt-5-minizhumor__gpt-4.1-minizhumor__gpt-4o-minirrr rmessager) from_nameto_nametyperr#)rrrrr@r?r!N)rr rrr%r&r'r(rziprFimport_predictionscreatedr+r.r/sortedr)r1r2r3r4rrmodel_versionsrpredictions_payloadmvchresponsepredsr5returned_versionss r9 test_projects_import_predictionsrcsn$ o7N7N OB .~> ?Bkk  'CRTR[R[ \G 88??7::Y@X4Y? ZDWN|i[:,7Gng. B""&7#, )"+R "$   "{{--EX-YH   q  NN  TWW  -E u:???A?@ ~ 6 66 6 @s;E&)rIlabel_studio.tests.sdk.commonrmark django_db pytestmarklabel_studio_sdk.clientr label_studio_sdk.label_interfacer(label_studio_sdk.label_interface.objectsrrr:rRrcr$r;r9rks6 @ [[ " " /;O,6^HkV37r;