add total_tokens to LLMUsage

Author: darjeeling

pythonkr_backend/curation/migrations/0012_llmusage_total_tokens.py

@@ -0,0 +1,19 @@
+# Generated by Django 5.2.1 on 2025-06-14 04:17
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('curation', '0011_alter_llmusage_date'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='llmusage',
+            name='total_tokens',
+            field=models.PositiveIntegerField(default=0, help_text='총 토큰 수'),
+            preserve_default=False,
+        ),
+    ]

pythonkr_backend/curation/models.py

@@ -511,7 +511,7 @@ def _get_available_models_for_provider(cls, provider, model_configs):
                     model_name=model_key,
                     date__gte=start_of_day_utc
                 ).aggregate(
-                    total_tokens=models.Sum(models.F('input_tokens') + models.F('output_tokens')),
+                    total_tokens=models.Sum(models.F('total_tokens')),
                     total_requests=models.Count('id')
                 )
 
@@ -536,7 +536,7 @@ def _get_available_models_for_provider(cls, provider, model_configs):
                 model_name__in=combined_models,
                 date__gte=today_start
             ).aggregate(
-                total_tokens=models.Sum(models.F('input_tokens') + models.F('output_tokens'))
+                total_tokens=models.Sum(models.F('total_tokens'))
             )
             combined_tokens = combined_usage['total_tokens'] or 0
 
@@ -557,7 +557,7 @@ def _get_available_models_for_provider(cls, provider, model_configs):
                         model_name=model_key,
                         date__gte=today_start
                     ).aggregate(
-                        total_tokens=models.Sum(models.F('input_tokens') + models.F('output_tokens'))
+                        total_tokens=models.Sum(models.F('total_tokens'))
                     )
                     total_tokens = today_usage['total_tokens'] or 0
 
@@ -591,6 +591,10 @@ class LLMUsage(models.Model):
     output_tokens = models.PositiveIntegerField(
         help_text="출력 토큰 수"
     )
+    total_tokens = models.PositiveIntegerField(
+        help_text="총 토큰 수"
+        # gemini 가 다른 경우가 있어서 별도로 저장.
+    )
     created_at = models.DateTimeField(auto_now_add=True)
 
     def __str__(self):