From bc797495deaaa704c13fb0bafc28f9f123789fed Mon Sep 17 00:00:00 2001 From: Wenruli Date: Wed, 6 Aug 2025 14:08:51 +0800 Subject: [PATCH] =?UTF-8?q?feat:=E7=81=B5=E6=80=9D=E6=89=A7=E8=A1=8C?= =?UTF-8?q?=E4=B8=AD=E9=97=B4=E6=96=87=E4=BB=B6=E5=B1=95=E7=A4=BA=E5=8A=9F?= =?UTF-8?q?=E8=83=BD=E5=A4=8D=E7=94=A8=E4=B8=8A=E4=BC=A0=E8=BF=87=E7=9A=84?= =?UTF-8?q?=E6=96=87=E4=BB=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/bisheng/linsight/task_exec.py | 1 + src/backend/bisheng/linsight/utils.py | 17 +++++++++++++++++ 2 files changed, 18 insertions(+) diff --git a/src/backend/bisheng/linsight/task_exec.py b/src/backend/bisheng/linsight/task_exec.py index 82988db99..29ae69b1d 100644 --- a/src/backend/bisheng/linsight/task_exec.py +++ b/src/backend/bisheng/linsight/task_exec.py @@ -56,6 +56,7 @@ class LinsightWorkflowTask: self._final_result: Optional[TaskEnd] = None self.file_dir: Optional[str] = None self.session_version_id: Optional[str] = None + self.step_event_extra_files: List[Dict] = [] # 用于存储步骤事件额外处理的文件信息 # ==================== 资源管理 ==================== diff --git a/src/backend/bisheng/linsight/utils.py b/src/backend/bisheng/linsight/utils.py index 38219304a..be8145f60 100644 --- a/src/backend/bisheng/linsight/utils.py +++ b/src/backend/bisheng/linsight/utils.py @@ -232,6 +232,19 @@ async def handle_step_event_extra(event: ExecStep, task_exec_obj) -> ExecStep: file_md5 = await util.async_calculate_md5(file_path) + # 判断文件是否已经上传过 + step_event_extra_files = task_exec_obj.step_event_extra_files + if step_event_extra_files: + existing_file = next((f for f in step_event_extra_files if f["file_md5"] == file_md5), None) + if existing_file: + logger.debug(f"步骤事件额外处理，文件已存在: {existing_file['file_name']}, file_md5: {file_md5}") + event.extra_info["file_info"] = { + "file_name": file_name, + "file_md5": existing_file["file_md5"], + "file_url": existing_file["file_url"] + } + return event + object_name = f"linsight/step_event/{task_exec_obj.session_version_id}/{uuid.uuid4().hex[:8]}.{file_name.split('.')[-1]}" logger.debug(f"步骤事件额外处理，上传文件到MinIO: {object_name}") @@ -241,12 +254,16 @@ async def handle_step_event_extra(event: ExecStep, task_exec_obj) -> ExecStep: object_name=object_name, file_path=file_path ) + event.extra_info["file_info"] = { "file_name": file_name, "file_md5": file_md5, "file_url": object_name } + # 添加到步骤事件额外文件列表 + task_exec_obj.step_event_extra_files.append(event.extra_info["file_info"]) + except Exception as e: logger.error(f"步骤事件额外处理异常: {e}") # 发生异常时，返回原始事件，不做任何修改