From 07e6fd457c4ddbebc1ca18a804db63ad2240b711 Mon Sep 17 00:00:00 2001 From: rogersun Date: Mon, 13 Jul 2026 17:05:13 +0800 Subject: [PATCH] =?UTF-8?q?=E5=8E=BB=E9=99=A4=E5=BD=B1=E7=89=87=E5=90=8D?= =?UTF-8?q?=E7=A7=B0=E5=89=8D=E5=90=8E=E7=9A=84=E4=B9=A6=E5=90=8D=E5=8F=B7?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ai/utils/show_process.py | 88 ++++++++++++++++++++-------------------- 1 file changed, 44 insertions(+), 44 deletions(-) diff --git a/ai/utils/show_process.py b/ai/utils/show_process.py index 0de3e74..24708fc 100644 --- a/ai/utils/show_process.py +++ b/ai/utils/show_process.py @@ -44,50 +44,50 @@ def show_main_process(zz_code=None, day_delta=None): else: target_show = '影厅别名,影厅id,影片别名,本地影片id,语言,放映日期,开始时间,结束时间,片长,场间' prompt = show_ai.general_prompt() - # start = datetime.datetime.now() # 开始计时 - # result, message, tokens = show_ai.get_show_result_ai() - # end = datetime.datetime.now() # 结束计时 - # print('prompt:', prompt) - # print('result:', result) - # print('message:', message) - # print('tokens:', tokens) - # - # # 获取排片数据 - # print('result:', result) - # if model.config_name in ['lmstudio_qwen3.6_35b', 'company_qwen3.5_35b']: - # result_obj = json.loads(result.replace('```json', '').replace('}\n```', '}').strip()) - # else: - # result_obj = json.loads(result.strip()) - # - # # 预测排片数据 - # _show = result_obj['show'].replace('```', '').replace('csv', '').strip() - # - # # 预测销售数据 - # _sales = str(result_obj['income']).strip() - # - # # 处理返回结果 - # try: - # AiShow.objects.create( - # cinema=cinema.name, - # zz_code=cinema.zz_code, - # show_date=show_date, - # is_ai_show=True, - # is_manual_test=False, - # ai_model=model.config_name, - # template=template_show, - # temp_date=template_date, - # target=target_show, - # show=data_cleansing(_show), - # sales=_sales, - # prompt=prompt, - # result=result, - # message=message, - # take_times=int((end - start).seconds), - # take_tokens=tokens, - # prompt_version=prompt_ver, - # ) - # except Exception as e: - # print(e) + start = datetime.datetime.now() # 开始计时 + result, message, tokens = show_ai.get_show_result_ai() + end = datetime.datetime.now() # 结束计时 + print('prompt:', prompt) + print('result:', result) + print('message:', message) + print('tokens:', tokens) + + # 获取排片数据 + print('result:', result) + if model.config_name in ['lmstudio_qwen3.6_35b', 'company_qwen3.5_35b']: + result_obj = json.loads(result.replace('```json', '').replace('}\n```', '}').strip()) + else: + result_obj = json.loads(result.strip()) + + # 预测排片数据 + _show = result_obj['show'].replace('```', '').replace('csv', '').strip() + + # 预测销售数据 + _sales = str(result_obj['income']).strip() + + # 处理返回结果 + try: + AiShow.objects.create( + cinema=cinema.name, + zz_code=cinema.zz_code, + show_date=show_date, + is_ai_show=True, + is_manual_test=False, + ai_model=model.config_name, + template=template_show, + temp_date=template_date, + target=target_show, + show=data_cleansing(_show), + sales=_sales, + prompt=prompt, + result=result, + message=message, + take_times=int((end - start).seconds), + take_tokens=tokens, + prompt_version=prompt_ver, + ) + except Exception as e: + print(e) return True