使用runtime控制是否请求ai的逻辑

ai/utils/show_process.py

@@ -39,55 +39,56 @@ def show_main_process(zz_code=None, day_delta=None):
         show_ai = ShowAI(cinema, show_date)
         # template_show, template_date = get_template_show(cinema, show_date)
         template_show, template_date = get_template_show_tomorrow(cinema)
-        if get_runtime_val('real_target') == 1:
+        if str(get_runtime_val('real_target')) == '1':
             target_show = get_target_show(cinema, show_date)
         else:
             target_show = '影厅别名,影厅id,影片别名,本地影片id,语言,放映日期,开始时间,结束时间,片长,场间'
         prompt = show_ai.general_prompt()
-        start = datetime.datetime.now()  # 开始计时
+        if str(get_runtime_val('test')) == '0':
-        result, message, tokens = show_ai.get_show_result_ai()
+            start = datetime.datetime.now()  # 开始计时
-        end = datetime.datetime.now()  # 结束计时
+            result, message, tokens = show_ai.get_show_result_ai()
-        print('prompt:', prompt)
+            end = datetime.datetime.now()  # 结束计时
-        print('result:', result)
+            print('prompt:', prompt)
-        print('message:', message)
+            print('result:', result)
-        print('tokens:', tokens)
+            print('message:', message)
+            print('tokens:', tokens)
 
-        # 获取排片数据
+            # 获取排片数据
-        print('result:', result)
+            print('result:', result)
-        if model.config_name in ['lmstudio_qwen3.6_35b', 'company_qwen3.5_35b']:
+            if model.config_name in ['lmstudio_qwen3.6_35b', 'company_qwen3.5_35b']:
-            result_obj = json.loads(result.replace('```json', '').replace('}\n```', '}').strip())
+                result_obj = json.loads(result.replace('```json', '').replace('}\n```', '}').strip())
-        else:
+            else:
-            result_obj = json.loads(result.strip())
+                result_obj = json.loads(result.strip())
 
-        # 预测排片数据
+            # 预测排片数据
-        _show = result_obj['show'].replace('```', '').replace('csv', '').strip()
+            _show = result_obj['show'].replace('```', '').replace('csv', '').strip()
 
-        # 预测销售数据
+            # 预测销售数据
-        _sales = str(result_obj['income']).strip()
+            _sales = str(result_obj['income']).strip()
 
-        # 处理返回结果
+            # 处理返回结果
-        try:
+            try:
-            AiShow.objects.create(
+                AiShow.objects.create(
-                cinema=cinema.name,
+                    cinema=cinema.name,
-                zz_code=cinema.zz_code,
+                    zz_code=cinema.zz_code,
-                show_date=show_date,
+                    show_date=show_date,
-                is_ai_show=True,
+                    is_ai_show=True,
-                is_manual_test=False,
+                    is_manual_test=False,
-                ai_model=model.config_name,
+                    ai_model=model.config_name,
-                template=template_show,
+                    template=template_show,
-                temp_date=template_date,
+                    temp_date=template_date,
-                target=target_show,
+                    target=target_show,
-                show=data_cleansing(_show),
+                    show=data_cleansing(_show),
-                sales=_sales,
+                    sales=_sales,
-                prompt=prompt,
+                    prompt=prompt,
-                result=result,
+                    result=result,
-                message=message,
+                    message=message,
-                take_times=int((end - start).seconds),
+                    take_times=int((end - start).seconds),
-                take_tokens=tokens,
+                    take_tokens=tokens,
-                prompt_version=prompt_ver,
+                    prompt_version=prompt_ver,
-            )
+                )
-        except Exception as e:
+            except Exception as e:
-            print(e)
+                print(e)
     return True