diff --git a/ai/utils/show_process.py b/ai/utils/show_process.py index 99d5844..aa9f5fc 100644 --- a/ai/utils/show_process.py +++ b/ai/utils/show_process.py @@ -2,6 +2,7 @@ from ai.models import * from ai.utils.show_prompt import * import datetime from django_redis import get_redis_connection +import re # ai排片主流程 def show_main_process(): @@ -31,8 +32,8 @@ def show_main_process(): _show = _show.strip() print('_show:', _show) # 获取销售额 - _sales = next((s for s in result.split('\n') if s.startswith('预估销售数据')), '').replace('预估销售数据:', '') - _sales = _sales.replace('约', '').replace('元', '') + _sales = next((s for s in result.split('\n') if re.search('预估销售数据', s)), '') + _sales = str(re.findall(r'\d+\.?\d*', _sales)[0]) print('_sales:', _sales) # 处理返回结果 try: