增加提示词版本

main
rogersun 2 weeks ago
parent 793f818a45
commit a9d2cdb4cb
  1. 18
      ai/migrations/0007_aishow_prompt_version.py
  2. 1
      ai/models.py
  3. 7
      ai/utils/show_process.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),
),
]

@ -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

@ -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)

Loading…
Cancel
Save