]jCdZddlZddlmZddlmZmZddlmZddl m Z ddl m Z ddl mZdd lmZmZmZdd lmZdd lmZmZdd lmZmZdd lmZmZddlmZm Z m!Z!ddl"m#Z#ddl$m%Z%ejLe'Z(ddddddddddddddddddddgdddddd ddd!ddd"dd# gd$Z)e d%ed&gd'd(jUejVxsd)*d+e)id,d-d.gd/01e d2ed&gd3d4jUejVxsd)*ed5ejXd6d7gd,d8d.gd/91Gd:d;ejZZ.e d<ed&gd=d>jUejVxsd)*d+e)id,d?d.gd/01e d2ed&gd@dAjUejVxsd)*dd,d2d.gd/01e dBed&gdCdDjUejVxsd)*dd,dBd.gd/01e dEedFG1GdHdIej^Z0e d%ed&gdJdKedLejXdMdN7gd+ddOddPdidQiedRSedTdUddVdWXdYd,dZd.gd/[1Gd\d]e%Z1e d%ed&gd^d_edLejXdMdN7ged`SedadUddVdWXdYd,dbdcgd/d1Gdedfe%Z2e d%ed&gdgdhedLejXdMdN7gediedjSid,dkd.gd/[1Gdldme%Z3e d2ed&gdndodiedpddqdrdidsdrdidtdQXid,dud.gd/v1GdwdxejhZ5y)yzThis file and its contents are licensed under the Apache License 2.0. Please see the included NOTICE for copyright information and LICENSE for a copy of the license. N)flag_set)ViewClassPermissionall_permissions)settings)method_decorator)DjangoFilterBackend) OpenApiTypes)OpenApiParameterOpenApiResponse extend_schema) MLBackend)MLBackendSerializerMLInteractiveAnnotatingRequest)ProjectTask)genericsstatus) FormParser JSONParserMultiPartParser)Response)APIViewobjectstringzML backend URL)type descriptionintegerz Project IDbooleanzIs interactiveTitle Descriptionz Auth methodNONE BASIC_AUTH)rrenumzBasic auth userzBasic auth passwordzExtra parameterszResponse model timeout) urlprojectis_interactivetitler auth_methodbasic_auth_userbasic_auth_pass extra_paramstimeout)r propertiesrequiredpostzMachine LearningzAdd ML BackendaC Add an ML backend to a project using the Label Studio UI or by sending a POST request using the following cURL command: ```bash curl -X POST -H 'Content-type: application/json' {host}/api/ml -H 'Authorization: Token abc123'\ --data '{{"url": "http://localhost:9090", "project": {{project_id}}}}' zhttps://localhost:8080)hostzapplication/jsonmlcreatepublic)zx-fern-sdk-group-namezx-fern-sdk-method-namezx-fern-audiences)tagssummaryrrequest extensions)name decoratorgetzList ML backendsz List all configured ML backends for a specific project by ID. Use the following cURL command: ```bash curl {host}/api/ml?project={{project_id}} -H 'Authorization: Token abc123' r%query)r8rlocationrlist)r4r5r parametersr7cneZdZeeefZeejejZ e Z egZdgZdZdZy)MLBackendListAPI)GETPOSTr&c|jjjd}tjt |}|j |j||j}|S)Nr%pk)r6 query_paramsr:rget_object_or_404rcheck_object_permissionsupdate_ml_backends_state)self project_pkr% ml_backendss =/root/env/lib/python3.12/site-packages/label_studio/ml/api.py get_querysetzMLBackendListAPI.get_querysetsV\\..229= ,,WD %%dllG<668 c|j}|j|j}|jr2|js%|j |_|jdgyyy)N model_version) update_fields)save update_stater%show_collab_predictionsrQr')rJ serializer ml_backendr%s rMperform_createzMLBackendListAPI.perform_creates`__& !$$  * *73H3H$.$4$4G ! LL'8L 94I *rON)__name__ __module__ __qualname__rrrparser_classesrr projects_viewprojects_changepermission_requiredrserializer_classrfilter_backendsfilterset_fieldsrNrXrOrMr@r@GsQ`!*o>N-  ) )  , ,+*+O() :rOr@patchzUpdate ML Backenda5 Update ML backend parameters using the Label Studio UI or by sending a PATCH request using the following cURL command: ```bash curl -X PATCH -H 'Content-type: application/json' {host}/api/ml/{{ml_backend_ID}} -H 'Authorization: Token abc123'\ --data '{{"url": "http://localhost:9091"}}' updatezGet ML Backendz Get details about a specific ML backend connection by ID. For example, make a GET request using the following cURL command: ```bash curl {host}/api/ml/{{ml_backend_ID}} -H 'Authorization: Token abc123' deletezRemove ML Backendz Remove an existing ML backend connection by ID. For example, use the following cURL command: ```bash curl -X DELETE {host}/api/ml/{{ml_backend_ID}} -H 'Authorization: Token abc123' putT)excludec~eZdZeeefZeZe jZ e jjZfdZdZxZS)MLBackendDetailAPIcLtt| }|j|SN)superrj get_objectrT)rJrW __class__s rMrnzMLBackendDetailAPI.get_objects%-t?A !rOcD|j}|jyrl)rSrT)rJrVrWs rMperform_updatez!MLBackendDetailAPI.perform_updates__& !rO)rYrZr[rrrr\rr`rr^r_r objectsallquerysetrnrq __classcell__)ros@rMrjrjsBF!*o>N*)99  $$&H "rOrjTraina After you add an ML backend, call this API with the ML backend ID to start training with already-labeled tasks. Get the ML backend ID by [listing the ML backends for a project](https://labelstud.io/api/#operation/api_ml_list). idpathz3A unique integer value identifying this ML backend.use_ground_truthz7Whether to include ground truth annotations in training)rr-z"Training has successfully started.)rzTraining errorz Error messagezServer responded with an error.)rrexample)rresponse)train)r4r5rr>r6 responsesr7c*eZdZejZdZy)MLBackendTrainAPIctjt|jd}|j |j ||j ttjS)NrErDr) rrGr kwargsrHr6r~rr HTTP_200_OK)rJr6argsrrWs rMr/zMLBackendTrainAPI.postsN// dkk$>OP  %%dllJ?v1122rON)rYrZr[rr^r_r/rcrOrMrrsd*993rOrzTest predictionz After you add an ML backend, call this API with the ML backend ID to run a test prediction on specific task data z$Predicting has successfully started.zPredicting error test_predictinternal)r4r5rr>rr7c.eZdZeZej ZdZy)MLBackendPredictTestAPIctjt|jd}|j |j ||j jdd}|rztj|j}|sttjddiS|j|}|sttjdd iStd i|Sttjd d iS) NrErDrandomF)r%detailzSProject has no tasks to run prediction on, import at least 1 task to run prediction)rdatazQML backend did not return any predictions, check ML backend logs for more detailserrorz>Not implemented - you must provide random=true query parameterrc)rrGr rrHr6rFr:r get_randomr%rrHTTP_500_INTERNAL_SERVER_ERROR_predictHTTP_501_NOT_IMPLEMENTED)rJr6rrrWrtasks rMr/zMLBackendPredictTestAPI.postLs// dkk$>OP  %%dllJ?%%))(E: ??:+=+=>D!@@ "w ((.F!@@ "u %f% %66_` rON) rYrZr[rr`rr^r_r/rcrOrMrr&sF+)99rOrzRequest Interactive Annotationa& Send a request to the machine learning backend set up to be used for interactive preannotations to retrieve a predicted region based on annotator input. See [set up machine learning](https://labelstud.io/guide/ml.html#Get-interactive-preannotations) for more. r|z%Interactive annotation has succeeded.predict_interactivecXeZdZdZej ZejfdZ dZ dZ dZ y)MLBackendInteractiveAnnotatingz Send a request to the machine learning backend set up to be used for interactive preannotations to retrieve a predicted region based on annotator input. cN||td|gitjS)Nerrorsr)rrr)rJmessage log_functions rM_error_responsez.MLBackendInteractiveAnnotating._error_responses$WG9-f6H6HIIrOcTtjt|d|jS)Nr)rEr%)rrGrr%)rJrWvalidated_datas rM _get_taskz(MLBackendInteractiveAnnotating._get_tasks#))$>&3IS]SeSeffrOctd|jr'|j|j|j|S)N1ff_back_dev_2362_project_credentials_060722_short)project_credentials_loginproject_credentials_password)ruserretask_data_logintask_data_password)rJr6contextr%s rM_get_credentialsz/MLBackendInteractiveAnnotating._get_credentialss; G V NN*1*A*A-4-G-G  rOctjt|jd}|j |j |t |j}|jd|j||j}|j||jjdi|j}|j|||j}t!|t"j$S) z Send a request to the machine learning backend set up to be used for interactive preannotations to retrieve a predicted region based on annotator input. rErD)rT)raise_exceptionr)rr)rrGr rrHr6rris_validrrrr:r%interactive_annotatingrrrr) rJr6rrrWrVrrresults rMr/z#MLBackendInteractiveAnnotating.posts // dkk$>OP  %%dllJ?3F D1~~j**C*CD''1J1J1N1NyZ\1]_c_k_kl224w||2T %%  rON) rYrZr[__doc__r tasks_viewr_loggerinforrrr/rcrOrMrrls2< *444:KKJg rOrzGet model versionsz$Get available versions of the model.zList of available versions.arrayr)ritemsversionsrlist_model_versions)r4r5rrr7c*eZdZejZdZy)MLBackendVersionsAPIctjt|jd}|j |j ||j }|jdk(r+d|jjdgi}t|dS|jdk(r|jgdd}t|dSd t|ji}|jd kDr |jnd }t||S) NrErDr|r)rriz*Upgrade your ML backend version to latest.rrrr})rrGr rrHr6 get_versions status_coder{r:rrQstr error_message)rJr6rrrWversions_responserrs rMr:zMLBackendVersionsAPI.gets// dkk$>OP  %%dllJ?&335  ( (C / "3"<"<"@"@R"PQF4 4  * *c 1#-#;#;"rs'A 4=.RRN)+JJ,(   8 $ + ' + " ( (\* , 0 - 3 M*V[.b  !  ##?'?    2 &*&.!)  0  !" ##?'?   ),2B2BWbn o &*&,!)  .:x11:/1^:D  !# ##?'?    2 &*&.!)  .  !  ##?'?  %)&+!)  *  !# ##?'?  %)&.!)  *u d(CD ">> "E++/D "   !  !%%Q     & )'`)!  !-QR ,#2$@  &*&-!)  O,/` 3 3a/` 3  !!  !%%Q   !-ST .#2$@  &*&4!+  3!D!g!E!D!H  !0 !%%Q   / -TU &*&;!)  ):) W) ;:) X  !$: 9$%, &&%#H$ #  (&*&;!)  /@=833=A@=rO