From 9aa36568b9aab12f98eccc26a47b59c4a5eae6ca Mon Sep 17 00:00:00 2001 From: zxstty Date: Sat, 19 Apr 2025 20:16:00 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=B5=81=E7=8A=B6=E6=80=81?= =?UTF-8?q?=E4=BF=AE=E6=94=B9=E7=9A=84bug;=E5=AE=8C=E5=96=84node=E7=9A=84s?= =?UTF-8?q?chema=E8=BD=AC=E6=8D=A2=E4=B8=BAjson=E7=9A=84=E9=80=BB=E8=BE=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/routers/chat.py | 4 +++- apps/scheduler/scheduler/scheduler.py | 3 --- apps/service/flow.py | 8 +++++--- 3 files changed, 8 insertions(+), 7 deletions(-) diff --git a/apps/routers/chat.py b/apps/routers/chat.py index 4dd261e1d..d63efa0f0 100644 --- a/apps/routers/chat.py +++ b/apps/routers/chat.py @@ -24,6 +24,7 @@ from apps.dependency import ( from apps.entities.request_data import RequestData from apps.entities.response_data import ResponseData from apps.manager.blacklist import QuestionBlacklistManager, UserBlacklistManager +from apps.manager.flow import FlowManager from apps.manager.task import TaskManager from apps.scheduler.scheduler import Scheduler from apps.scheduler.scheduler.context import save_data @@ -102,7 +103,8 @@ async def chat_generator(post_body: RequestData, user_sub: str, session_id: str) await save_data(task_id, user_sub, post_body, scheduler.used_docs) yield "data: [DONE]\n\n" - + if post_body.app and post_body.app.flow_id: + await FlowManager.update_flow_debug_by_app_and_flow_id(post_body.app.app_id, post_body.app.flow_id, debug=True) except Exception: logger.exception("[Chat] 生成答案失败") yield "data: [ERROR]\n\n" diff --git a/apps/scheduler/scheduler/scheduler.py b/apps/scheduler/scheduler/scheduler.py index 6ab82513f..982b2fac3 100644 --- a/apps/scheduler/scheduler/scheduler.py +++ b/apps/scheduler/scheduler/scheduler.py @@ -175,7 +175,4 @@ class Scheduler: # 更新Task task = await TaskManager.get_task(task.id) - # 如果状态正常,则更新Flow的debug状态 - if task.state and task.state.status == StepStatus.SUCCESS: - await FlowManager.update_flow_debug_by_app_and_flow_id(app_info.app_id, flow_id, debug=True) return diff --git a/apps/service/flow.py b/apps/service/flow.py index 58c8782ea..9eb6dc2e1 100644 --- a/apps/service/flow.py +++ b/apps/service/flow.py @@ -260,6 +260,8 @@ def generate_from_schema(schema: dict) -> Any: def _generate_example(schema_node: dict) -> Any: # 处理类型为 object 的节点 + if "default" in schema_node: + return schema_node["default"] if schema_node.get("type") == "object": example = {} properties = schema_node.get("properties", {}) @@ -274,15 +276,15 @@ def generate_from_schema(schema: dict) -> Any: # 处理类型为 string 的节点 if schema_node.get("type") == "string": - return schema_node.get("default", "example_string") + return "example_string" # 处理类型为 number 或 integer 的节点 if schema_node.get("type") in ["number", "integer"]: - return schema_node.get("default", 0) + return 0 # 处理类型为 boolean 的节点 if schema_node.get("type") == "boolean": - return schema_node.get("default", False) + return False # 处理其他类型或未定义类型 return None -- Gitee