
    	]j                         d dl mZ ddlmZmZmZ  ej                  e       G d dej                               Z ej                  e       G d dej                               Z	 ej                  e       G d d	ej                               Z
y
)    )admin   )ModelInterfaceModelRunThirdPartyModelVersionc                   L    e Zd ZdZdZdZdZdZdddifd	dd
ifdddifdddiffZy)ModelInterfaceAdmin)title
skill_nameorganization
created_by
created_at
updated_at)r   r   )r
   descriptionr   r   )associated_projectszModel Detailsfields)r
   r   r   Organizationr   r   Configuration)input_fieldsoutput_classesr   
TimestampsN)	__name__
__module____qualname__list_displaylist_filtersearch_fieldsreadonly_fieldsfilter_horizontal	fieldsets     F/root/env/lib/python3.12/site-packages/label_studio/ml_models/admin.pyr	   r	      sd    L 1K,M2O0	8%KLM	($BCDPQ	
 
">?@Ir$   r	   c                   H    e Zd ZdZdZdZdZdddifddd	ifd
ddifdddiffZy)ThirdPartyModelVersionAdmin)r
   parent_modelproviderprovider_model_idr   r   )r)   r   )r
   r*   promptr   zVersion Detailsr   )r
   r(   r+   zProvider Details)r)   r*   model_provider_connectionr   r   r   N)r   r   r   r   r   r   r    r"   r#   r$   r%   r'   r'      s^    L /K<M2O	X'JKLUV	
 
($BCD	">?@Ir$   r'   c                   b    e Zd ZdZdZdZdZdddifddd	ifd
ddifdddiffZdgZd Z	de	_
        y)ModelRunAdmin)idmodel_versionprojectstatusr   r   completed_at)r2   project_subset)job_id)r   triggered_atpredictions_updated_atr3   total_predictionstotal_correct_predictionstotal_taskszRun Detailsr   )r0   r1   r4   r2   r5   r   r   
Statistics)r8   r9   r:   r   )r   r6   r7   r3   delete_model_run_predictionsc                 z    |D ]  }|j                           | j                  |d|j                          d       y )NzDeleted predictions for z model runs.)delete_predictionsmessage_usercount)selfrequestqueryset	model_runs       r%   r<   z*ModelRunAdmin.delete_model_run_predictionst   s?    ! 	+I((*	+'%=hnn>N=O|#\]r$   z*Delete predictions for selected model runsN)r   r   r   r   r   r   r    r"   actionsr<   short_descriptionr#   r$   r%   r.   r.   7   s    L /KMO  	
 
($BCD 		
  
	
1#IJ ..G^
 6b 2r$   r.   N)django.contribr   modelsr   r   r   register
ModelAdminr	   r'   r.   r#   r$   r%   <module>rK      s      D D %**   0 &'%"2"2  (. AbE$$ Ab Abr$   