From 527788d5186927e75ee86384c177a12b8407b4c7 Mon Sep 17 00:00:00 2001 From: zy123 <646228430@qq.com> Date: Wed, 19 Feb 2025 16:35:51 +0800 Subject: [PATCH] =?UTF-8?q?2.19=20=E5=88=A4=E6=96=AD=E6=98=AF=E5=90=A6?= =?UTF-8?q?=E6=8B=9B=E6=A0=87=E6=96=87=E4=BB=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- flask_app/routes/judge_zbfile.py | 2 +- flask_app/routes/判断是否是招标文件.py | 6 +++--- flask_app/start_up.py | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/flask_app/routes/judge_zbfile.py b/flask_app/routes/judge_zbfile.py index b34df9b..ba43a31 100644 --- a/flask_app/routes/judge_zbfile.py +++ b/flask_app/routes/judge_zbfile.py @@ -36,7 +36,7 @@ def judge_zbfile() -> Any: #判断是否是招标文件 # ) result = pool.apply( judge_zbfile_exec, # 你的实际执行函数 - args=(file_path,logger,) + args=(file_path,) ) return result def wrapper() -> None: diff --git a/flask_app/routes/判断是否是招标文件.py b/flask_app/routes/判断是否是招标文件.py index 8437fb7..ef1aec6 100644 --- a/flask_app/routes/判断是否是招标文件.py +++ b/flask_app/routes/判断是否是招标文件.py @@ -4,7 +4,7 @@ from docx import Document import fitz from flask_app.general.llm.通义千问long import upload_file, qianwen_long -def judge_zbfile_exec(file_path,logger): +def judge_zbfile_exec(file_path): """ 判断文件是否属于招标文件,并返回结果。 """ @@ -45,7 +45,7 @@ def judge_zbfile_exec(file_path,logger): print("DOCX 文件读取失败,可能为乱码文件") return False # 解析失败直接返回 False pre_endtime=time.time() - logger.info(f"judge_zbfile_exec预处理耗时:{pre_endtime - start_time:.2f} 秒") + print(f"judge_zbfile_exec预处理耗时:{pre_endtime - start_time:.2f} 秒") # 使用大模型进行判断 user_query = """该文件是否属于招标文件?如果是的话,请返回'是',如果不是的话,返回'否'。请不要返回其他解释或内容。 以下是常见的招标文件类型: @@ -59,7 +59,7 @@ def judge_zbfile_exec(file_path,logger): file_id = upload_file(file_path) model_res = qianwen_long(file_id, user_query) end_time = time.time() - logger.info(f"judge_zbfile_exec实际耗时:{end_time - start_time:.2f} 秒") + print(f"judge_zbfile_exec实际耗时:{end_time - start_time:.2f} 秒") print(f"判断是否属于招标文件:{model_res} 实际耗时:{end_time - start_time:.2f} 秒") return '否' not in model_res diff --git a/flask_app/start_up.py b/flask_app/start_up.py index 75ae428..b00343d 100644 --- a/flask_app/start_up.py +++ b/flask_app/start_up.py @@ -16,7 +16,7 @@ from flask_app.routes.test_preprocess import test_process_bp def create_app(): # 创建全局日志记录器 app = Flask(__name__) - app.process_pool = Pool(processes=3, maxtasksperchild=1) + app.process_pool = Pool(processes=10, maxtasksperchild=3) app.global_logger = create_logger_main('model_log') # 全局日志记录器 # 注册蓝图 app.register_blueprint(get_deviation_bp)