]j#hdZddlZddlmZddlmZddlmZddlm Z ddl m Z m Z ddlmZdd lmZdd lmZmZmZmZej.eZGd d ej4Zd ZGddej:ZGddej:ZGddeZ Gddej:Z!y)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.N)settings)models)Q) gettext_lazy)ModelProviderConnectionModelProviders)Project)ValidationError) AnnotationFailedPrediction PredictionPredictionMetac4eZdZdedfZdedfZy) SkillNamesTextClassificationNamedEntityRecognitionN)__name__ __module__ __qualname___TEXT_CLASSIFICATIONNAMED_ENTITY_RECOGNITIONG/root/env/lib/python3.12/site-packages/label_studio/ml_models/models.pyrrs$.2F0GG7;S9TTrrc|s tdt|ts tdtd|Ds tdy)Nzlist should not be emptyzValue must be a listc3<K|]}t|tywN) isinstancestr).0items r z'validate_string_list..s7z$$7sz%All items in the list must be strings)r rlistall)values rvalidate_string_listr'sF 899 eT "455 77 7EFF 8rc:eZdZejedddddZejedddd Zeje jd ejd Z ejed d ZejeddZejdej"ddZejdej(dZej,eegZej,eegZej6ddZdZy)ModelInterfacetitleF Model name max_lengthnullblank help_text descriptionTzModel descriptionr/r0r1created_models related_name on_deleter/ created at auto_now_add updated atauto_noworganizations.Organizationmodel_interfacesr7r6r/)r.choicesr/)default validatorszprojects.Project)r0c4|j|jk(Sractive_organization organizationselfusers rhas_permissionzModelInterface.has_permission9''4+<+<<R=STL%V%%d@T?UVN0&001C4P=rr)c2eZdZGddZej edddddZeje dej Z ejed ddd Z ejedejd ZedZfdZxZS) ModelVersionceZdZdZy)ModelVersion.MetaTN)rrrabstractrrrMetara>srrcr*r+Fr,r-model_versions)r6r7promptzPrompt to executer3Tr5cL|jjd|jS)N__) parent_modelr*rJs r full_titlezModelVersion.full_titleKs$##))*"TZZL99rctjj|j}|D]}|j t ||i|y)zB Deletes Predictions associated with ModelVersion ) model_versionN)ModelRunobjectsfilteriddelete_predictionssuperdelete)rJargskwargs model_runs model_run __class__s rrszModelVersion.deleteOsN%%,,477,C # +I  ( ( * + ''r)rrrrcrrNrr*rPr)rWrhrOrerrRmodel_provider_connectionpropertyrjrs __classcell__rxs@rr_r_=s F  QwZCe5\h iE$6$$^BR^d^l^lmL V  akUNa bF 1 1 1.>&//`d!::((rr_ceZdZejdej ejdZejddddZ eje jdejd Zejed d Zejed d Zejdej&ddZedZdZy)ThirdPartyModelVersionrAz%The model provider to use e.g. OpenAI)r.rBrCr1Fz;The model ID to use within the given provider, e.g. gpt-3.5)r.r0r/r1"created_third_party_model_versionsTr5r8r9r;r<r>third_party_model_versionsr@c|jrH|jjjr$|jjjSyr)rhr]existsfirstris rprojectzThirdPartyModelVersion.projectwsB   !2!2!F!F!M!M!O$$88>>@ @rc4|j|jk(SrrFrIs rrLz%ThirdPartyModelVersion.has_permission~rMrN)rrrrrNrrBOPENAIproviderprovider_model_idrPrrQrRrSrTrrUrVrWrHrzrrLrrrr~r~Ysv&&%%9 H)(( O #""  9//  J&%%a oDIJ%%%a oEJ$6$$$MiptL =rr~ceZdZGddejZGddejZGddejZejdejdd Z eje ejd Z eje ejd Zejej dej"d Zej&d ej(ej*Zej&d ej(ej.Zej&d d d ddZej4eddZej4eddZej4eddZej>edd Z ej>edd dZ!ej>edd dZ"ej>edd dZ#dZ$dZ%fdZ&xZ'S) rmcHeZdZdedfZdedfZdedfZy)ModelRun.ProjectSubsetAllHasGTSampleN)rrrrALLHASGTSAMPLErrr ProjectSubsetrs+QuXo7#1X;&rrc4eZdZdedfZdedfZy)ModelRun.FileTypeInputOutputN)rrrrINPUTOUTPUTrrrFileTypers7#1X;&rrc@eZdZdedfZdedfZdZdZdZy)ModelRun.ModelRunStatusPending InProgress) Completedr)Failedr)CanceledrN) rrrrPENDING INPROGRESS COMPLETEDFAILEDCANCELEDrrrModelRunStatusrs-Qy\)!1\?2 . %+rrr>rvTr@)r7r6r5rA)r.rBrCNz9Job ID for inference job for a ModelRun e.g. Adala job ID)r.r/r0rCr1ztotal predictionsr)rCztotal correct predictionsz total tasksr8r9z triggered at)r/rCzpredictions updated atz completed atc4|j|jk(SrrFrIs rrLzModelRun.has_permissionrMrctjj|j}|Dcgc]}|j}}tjj|j d ddlm}m}|jj|}|j|jjt|t| z}|jtjj|j} | Dcgc]}|j} }tjj| jtjj| j|j!|j"| j!| j"ycc}w#t$r#}tjd |Yd}~d}~wwxYwcc}w) a Deletes any predictions that have originated from a ModelRun Executing a raw SQL query here for speed. This ignores any foreign key relationships so if another model has a Prediction fk and set to on_delete=CASCADE for example, it will not take affect. The only relationship like this that currently exists is in Annotation.parent_prediction, which we are handling here. )rw)parent_prediction__inN)parent_predictionr)PredictionPairStatsPredictionStats)prediction_to__in)prediction_to_id__in)prediction_from_id__inzHPredictionStats or PredictionPairStats model does not exist , exception:)prediction__in)failed_prediction__in)r rnrorpr update stats.modelsrrrsr Exceptionloggerinfor r _raw_deletedb) rJ predictionspprediction_idsrrprediction_stats_to_be_deleted#prediction_pair_stats_to_be_deletedefailed_predictionsfailed_predictions_idss rrqzModelRun.delete_predictionss!((//$''/B (341!$$44!!!GNNaeNf h I-<-D-D-K-K^l-K-m * * 1 1 32E2M2M2T2T~6R`9aa3 / 0 6 6 8 .55< Deletes Predictions associated with ModelRun N)rqrrrs)rJrtrurxs rrszModelRun.deletes" ! ''r)(rrrr TextChoicesrrrrPrWrHr rr~rlrrQrRrSrNrBrproject_subsetrstatusjob_id IntegerFieldrtotal_predictionstotal_correct_predictions total_tasksrTrU triggered_atpredictions_updated_at completed_atrLrqrsr{r|s@rrmrms'**' '6%%',++,%6$$$\`dL f6>>P\]G%F%%&<eqrM"""  !//  J&V%%m>S>S]j]p]pqN V  n6L6LVdVlVl mF V   M F,++A.A,BAN 3 3 3A6Q4R\] ^%&%%a &6BK%%%a oDIJ'6''.(9dSL1V11!4L2MTXbfg'6''.(9dSL=#>J((rrm)"__doc__logging django.confr django.dbrdjango.db.modelsrdjango.utils.translationrrml_model_providers.modelsrrprojects.modelsr rest_framework.exceptionsr tasks.modelsr r r r getLoggerrrrrr'Modelr)r_r~rmrrrrsl 6M#5QQ   8 $U##U G=V\\=:(6<<(8&=\&=Rg(v||g(r