diff --git a/apps/routers/chat.py b/apps/routers/chat.py index 4dd261e1d00ade36740905031a25a71562ff7841..d63efa0f09054875fdd2ceaae9181bac2ec90635 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 6ab82513f4633f10874a4069befe31fe17862cf5..982b2fac37d5e87816bde90d817c57245ca38218 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 58c8782ea1faf97c34b5ae95c05bbafb08bf8794..9eb6dc2e1695e2ebf77943bc5b716b39bb11011c 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