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)