]jdZddlmZddlmZddlmZddlmZm Z ddl m Z Gdde jZ Gd d e jZy ) zThis 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. )validate_upload_url)settings)extend_schema_field) MLBackend MLBackendAuth) serializersceZdZdZej ZejddddZej Z e e dZ dZ dZdZd Zd Zd Zfd ZGd dZxZS)MLBackendSerializerz) Serializer for MLBackend model. TF) write_onlyrequired allow_null allow_blankc,t|jSN)boolbasic_auth_passselfobjs E/root/env/lib/python3.12/site-packages/label_studio/ml/serializers.pyget_basic_auth_pass_is_setz.MLBackendSerializer.get_basic_auth_pass_is_setsC''((c"|jSr)get_state_displayrs rget_readable_statez&MLBackendSerializer.get_readable_states$$&&rc|sA|jjstjd|jjS|S)Nz=Authentication password is required for Basic Authentication.)instancerrValidationErrorrvalues rvalidate_basic_auth_passz,MLBackendSerializer.validate_basic_auth_passs;==00!112qrr}}444 rc<t|tj|S)N)block_local_urls)rrML_BLOCK_LOCAL_IPrs r validate_urlz MLBackendSerializer.validate_url$sEH4N4NO rcjdtjk(r.ddg}tfd|Drt j dyy)N auth_methodbasic_auth_userrc3&K|]}|v ywr).0fieldattrss r z?MLBackendSerializer._validate_authentication..-sC%5%CszJAuthentication username and password is required for Basic Authentication.)getr BASIC_AUTHanyrr)rr-required_fieldss ` r_validate_authenticationz,MLBackendSerializer._validate_authentication)sU 99] #}'?'? ?02CDOC?CC!11`D @rctjdi|}|jr<|jdk(rd}nd|dd|j}t j |y)NizfAble to connect to ML Server, but authentication parameters were either not provided or are incorrect.zCan't connect to ML backend urlz, health check failed. Make sure it is up and your firewall is properly configured. Learn more about how to set up an ML backend. Additional info: r*)r healthcheck_is_error status_code error_messagerr)rr-healthcheck_responsemessages r_validate_healthcheckz)MLBackendSerializer._validate_healthcheck2s~(55>>  ( (#//36< 35<.AKL`KmKmJnp--g6 6 )rctjdi|}|jr+d|dd|jd}t j |y)NzSuccessfully connected to r5z6 but it doesn't look like a valid ML backend. Reason: z. Check the ML backend server console logs to check the status.There might be something wrong with your model or it might be incompatible with the current labeling configuration.r*)rsetup_r7r9rr)rr-setup_responser;s r_validate_setupz#MLBackendSerializer._validate_setupEsc"))2E2  " ",U5\N;)7789FF --g6 6 #rct||}|j||j||j ||Sr)supervalidater3r<r@)rr- __class__s rrCzMLBackendSerializer.validateRsA ' %%e, ""5) U# rceZdZeZgdZy)MLBackendSerializer.Meta)idstatereadable_stateis_interactiver5r9titler'r(rbasic_auth_pass_is_set description extra_params model_versiontimeout created_at updated_at auto_updateprojectN)__name__ __module__ __qualname__rmodelfieldsr*rrMetarF[s rrZ)rUrVrW__doc__rSerializerMethodFieldrI CharFieldrrLrrrrr!r%r3r<r@rCrZ __classcell__)rDs@rr r s7[668N+k++teX\jnoO>[>>@))' 7& 7  rr cbeZdZdZej ddZejdddZy)MLInteractiveAnnotatingRequestz; Serializer for ML interactive annotating request. zID of task to annotateT) help_textr zContext for ML modelN)rar default) rUrVrWr[r IntegerFieldtask JSONFieldcontextr*rrr`r`ts8 $; # #.FQU VD#k##.DQU_cdGrr`N)r[ core.utils.ior django.confrdrf_spectacular.utilsr ml.modelsrrrest_frameworkrModelSerializerr Serializerr`r*rrrnsB- 5.&g +55g Te[%;%;er