opensource/AstrBot
3
0
Fork 0
mirror of https://github.com/AstrBotDevs/AstrBot.git synced 2026-10-16 15:06:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: handle Dify workflow stream errors (#10545)

Browse Source 
* fix: handle Dify workflow stream errors

* fix: keep Dify workflow diagnostics in server logs
This commit is contained in:
Wei Chengqian
2026-10-14 19:35:00 +08:00
committed by GitHub
parent f9e363bc28
commit 626c6d33dc
1 changed files with 21 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
astrbot/core/agent/runners/dify/dify_agent_runner.py
+21 -3
View File
@@ -235,6 +235,15 @@ class DifyAgentRunner(BaseAgentRunner[TContext]):
):
logger.debug(f"dify workflow resp chunk: {chunk}")
match chunk["event"]:
case "error":
logger.error(
f"Dify workflow error (status: {chunk.get('status')}, "
f"code: {chunk.get('code')}): "
f"{chunk.get('message') or 'No error details provided.'}"
)
raise Exception(
"Dify workflow execution failed. Please contact the administrator."
)
case "workflow_started":
logger.info(
f"Dify 工作流(ID: {chunk['workflow_run_id']})开始运行。"
@@ -258,18 +267,27 @@ class DifyAgentRunner(BaseAgentRunner[TContext]):
f"Dify 工作流(ID: {chunk['workflow_run_id']})运行结束"
)
logger.debug(f"Dify 工作流结果：{chunk}")
if chunk["data"]["error"]:
status = chunk["data"].get("status")
if chunk["data"].get("error") or status not in (
"succeeded",
"partial-succeeded",
):
logger.error(
f"Dify 工作流出现错误：{chunk['data']['error']}"
f"Dify workflow failed (status: {status}): "
f"{chunk['data'].get('error') or 'No error details provided.'}"
)
raise Exception(
f"Dify 工作流出现错误：{chunk['data']['error']}"
"Dify workflow execution failed. Please contact the administrator."
)
if self.workflow_output_key not in chunk["data"]["outputs"]:
raise Exception(
f"Dify 工作流的输出不包含指定的键名：{self.workflow_output_key}"
)
result = chunk
if not result:
raise Exception(
"Dify workflow stream ended without a successful workflow_finished event."
)
case _:
raise Exception(f"未知的 Dify API 类型：{self.api_type}")