diff --git a/ai/migrations/0007_aishow_prompt_version.py b/ai/migrations/0007_aishow_prompt_version.py new file mode 100644 index 0000000..0f2ec37 --- /dev/null +++ b/ai/migrations/0007_aishow_prompt_version.py @@ -0,0 +1,18 @@ +# Generated by Django 4.2.7 on 2026-06-10 08:37 + +from django.db import migrations, models + + +class Migration(migrations.Migration): + + dependencies = [ + ('ai', '0006_aishow_updated_at_alter_aishow_created_at'), + ] + + operations = [ + migrations.AddField( + model_name='aishow', + name='prompt_version', + field=models.CharField(default='', max_length=50), + ), + ] diff --git a/ai/models.py b/ai/models.py index e86a142..1ed4bf8 100644 --- a/ai/models.py +++ b/ai/models.py @@ -50,6 +50,7 @@ class AiShow(models.Model): updated_at = models.DateTimeField(auto_now=True) take_times = models.IntegerField(default=0) take_tokens = models.CharField(default='', max_length=2000) + prompt_version = models.CharField(max_length=50, default='') def __str__(self): return self.cinema diff --git a/ai/utils/show_process.py b/ai/utils/show_process.py index d766dd5..e35428f 100644 --- a/ai/utils/show_process.py +++ b/ai/utils/show_process.py @@ -14,6 +14,10 @@ def show_main_process(): # 获取影院列表 redis_conn.set(redis_key, 1, ex=60*60*20) test_cinema_list = TestCinema.objects.filter(is_active=True).all() + # 获取提示词版本 + prompt_ver = PromptTemplate.objects.filter( + Q(del_flag=False) & Q(prompt_type='version') & Q(prompt_key='ShowPromptVersion')).first().prompt_val + # 开始处理提示词 # 生成目标日期 show_date = datetime.date.strftime(datetime.date.today() + datetime.timedelta(days=3), '%Y-%m-%d') for cinema in test_cinema_list: @@ -51,7 +55,8 @@ def show_main_process(): result=result, message=message, take_times=int((end - start).seconds), - take_tokens=tokens + take_tokens=tokens, + prompt_version=prompt_ver, ) except Exception as e: print(e)