youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-03 17:08:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: handle none value in tool node

Browse Source
This commit is contained in:
Novice
2025-06-20 10:34:00 +08:00
parent 58faef8b5b
commit c1884c2e40
2 changed files with 27 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/core/tools/mcp_tool/tool.py
View File

@ -1,3 +1,4 @@
import base64
from collections.abc import Generator
from typing import Any, Optional
@ -48,7 +49,9 @@ class MCPTool(Tool):
if isinstance(content, TextContent):
yield self.create_text_message(content.text)
elif isinstance(content, ImageContent):
yield self.create_image_message(content.data)
yield self.create_blob_message(
blob=base64.b64decode(content.data), meta={"mime_type": content.mimeType}
)
def fork_tool_runtime(self, runtime: ToolRuntime) -> "MCPTool":
return MCPTool(