From 5f136657391c21565565b3a06c8946ed60eb896f Mon Sep 17 00:00:00 2001 From: zy123 <646228430@qq.com> Date: Mon, 17 Feb 2025 17:42:48 +0800 Subject: [PATCH] =?UTF-8?q?2.17=20=E5=B7=B2=E8=A7=A3=E5=86=B3=E5=86=85?= =?UTF-8?q?=E5=AD=98=E6=B3=84=E6=BC=8F=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- flask_app/货物标/技术参数要求提取.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/flask_app/货物标/技术参数要求提取.py b/flask_app/货物标/技术参数要求提取.py index 9a682a1..747864c 100644 --- a/flask_app/货物标/技术参数要求提取.py +++ b/flask_app/货物标/技术参数要求提取.py @@ -550,7 +550,9 @@ def get_technical_requirements(invalid_path, processed_filepath, model_type=1): processed_data = truncate_system_keys(preprocessed_data) # 限制深度 key_paths, grouped_paths, good_list, data_copy = generate_key_paths( processed_data) # 提取需要采购的货物清单 key_list:交通监控视频子系统.高清视频抓拍像机 ... grouped_paths是同一系统下同时有'交换机-1'和'交换机-2',提取'交换机' ,输出eg:{'交通标志.标志牌铝板', '交通信号灯.交换机'} - # if len(good_list)>100 and model_type==1: #并发特别高(len(good_list)),目前是对每个货物都开一个线程获取结果,对于较多的货物,tokens会比较贵,以后可以考虑qianwen-long, 目前qianwen-plus:0.0008/ktokens long:0.0005/ktokens 差不多价格,暂不考虑 + if len(good_list)>300: + return {"采购需求": {}} + # elif len(good_list)>100 and model_type==1: #并发特别高(len(good_list)),目前是对每个货物都开一个线程获取结果,对于较多的货物,tokens会比较贵,以后可以考虑qianwen-long, 目前qianwen-plus:0.0008/ktokens long:0.0005/ktokens 差不多价格,暂不考虑 # model_type=2 # file_id=upload_file(processed_filepath) modified_data = rename_keys(data_copy)