diff --git a/apps/manager/flow.py b/apps/manager/flow.py index 26a19670e988c62e09a528fff48d4efcb5a5d994..9677eb68763820ad04e4e35c8cc607f5c3b4f95e 100644 --- a/apps/manager/flow.py +++ b/apps/manager/flow.py @@ -23,9 +23,9 @@ from apps.manager.node import NodeManager from apps.models.mongo import MongoDB from apps.scheduler.pool.loader.app import AppLoader from apps.scheduler.pool.loader.flow import FlowLoader -from apps.manager.node import NodeManager from apps.utils.flow import generate_from_schema + class FlowManager: """Flow相关操作""" @@ -482,14 +482,14 @@ class FlowManager: for flows in metadata.flows: if flows.id == flow_id: flows.debug = debug - app_loader = AppLoader() - await app_loader.save(metadata, app_id) flow_loader = FlowLoader() flow = await flow_loader.load(app_id, flow_id) if flow is None: return False flow.debug = debug await flow_loader.save(app_id=app_id,flow_id=flow_id,flow=flow) + app_loader = AppLoader() + await app_loader.save(metadata, app_id) return True except Exception as e: LOGGER.error(f"Update flow debug from app pool failed: {e!s}") diff --git a/apps/routers/mock.py b/apps/routers/mock.py index 72af6eb3cee2aadd6ec44241cd89320dc79688ee..9081486052c384d7d06a44a6717b84972a89f740 100644 --- a/apps/routers/mock.py +++ b/apps/routers/mock.py @@ -3,6 +3,7 @@ Copyright (c) Huawei Technologies Co., Ltd. 2023-2024. All rights reserved. """ +import copy import json import random import time @@ -383,8 +384,11 @@ async def mock_data( sample_output["content"] = {"message": ""} if "content" in sample_output and isinstance(sample_output["content"], dict): for key, value in sample_output["content"].items(): - params[key] = value + params[key] = copy.deepcopy(value) time.sleep(sample_output["metadata"]["time_cost"]) + if sample_output["flow"]["stepName"] == "知识库": + for i in range(len(sample_output["content"]["chunk_list"])): + sample_output["content"]["chunk_list"][i] = sample_output["content"]["chunk_list"][i][:100] + "..." yield "data: " + json.dumps(sample_output, ensure_ascii=False) + "\n\n" now_flow_item = edge.edge_to