37 lines
2.0 KiB
Python
37 lines
2.0 KiB
Python
from django.db import models
|
|
|
|
class AiModel(models.Model):
|
|
# 1. ข้อมูลพื้นฐานและการลงทะเบียน
|
|
name = models.CharField(max_length=255, unique=True, verbose_name="ชื่อโมเดล")
|
|
model_version = models.CharField(max_length=50, default="v1.0.0", verbose_name="เวอร์ชัน")
|
|
developer = models.CharField(max_length=255, blank=True, null=True, verbose_name="ผู้พัฒนา/ทีม")
|
|
|
|
# 2. ข้อมูลการเรียกใช้ (API Management)
|
|
base_url = models.URLField(max_length=500, verbose_name="Base URL ของ AI Service (Internal)")
|
|
inference_path = models.CharField(max_length=255, verbose_name="Endpoint Path สำหรับ Inference")
|
|
# ตัวอย่าง: base_url="http://ai_model_server:8001/", inference_path="inference/spleen/"
|
|
|
|
# 3. สถานะและการควบคุม
|
|
status_choices = [
|
|
('ACTIVE', 'ใช้งาน'),
|
|
('INACTIVE', 'ไม่ใช้งาน'),
|
|
('TESTING', 'กำลังทดสอบ'),
|
|
]
|
|
status = models.CharField(max_length=20, choices=status_choices, default='INACTIVE', verbose_name="สถานะบริการ")
|
|
|
|
auth_required = models.BooleanField(default=False, verbose_name="ต้องการ Internal Auth Key")
|
|
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
|
|
def full_inference_url(self):
|
|
# สร้าง Full URL สำหรับ DRF ใช้เรียก (Proxy)
|
|
return f"{self.base_url.rstrip('/')}/{self.inference_path.lstrip('/')}"
|
|
|
|
def __str__(self):
|
|
return f"{self.name} ({self.model_version}) - {self.status}"
|
|
|
|
class Meta:
|
|
verbose_name = "AI Model Registry"
|
|
verbose_name_plural = "AI Model Registry"
|
|
# สามารถเพิ่ม unique_together (name, model_version) ได้ในภายหลัง |