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

Merge branch 'main' into feat/r2

Browse Source
This commit is contained in:
Yeuoly
2025-04-27 14:31:19 +08:00
parent d4007ae073 abafa68647
commit c2d02f8f4d
874 changed files with 31114 additions and 19811 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
api/core/datasource/datasource_engine.py
View File

@ -39,8 +39,9 @@ class DatasourceEngine:
"""
try:
# hit the callback handler
workflow_tool_callback.on_datasource_start(datasource_name=datasource.entity.identity.name,
datasource_inputs=datasource_parameters)
workflow_tool_callback.on_datasource_start(
datasource_name=datasource.entity.identity.name, datasource_inputs=datasource_parameters
)
if datasource.runtime and datasource.runtime.runtime_parameters:
datasource_parameters = {**datasource.runtime.runtime_parameters, **datasource_parameters}
@ -86,7 +87,6 @@ class DatasourceEngine:
workflow_tool_callback.on_tool_error(e)
raise e
@staticmethod
def _convert_datasource_response_to_str(datasource_response: list[DatasourceInvokeMessage]) -> str:
"""
@ -101,7 +101,10 @@ class DatasourceEngine:
f"result link: {cast(DatasourceInvokeMessage.TextMessage, response.message).text}."
+ " please tell user to check it."
)
elif response.type in {DatasourceInvokeMessage.MessageType.IMAGE_LINK, DatasourceInvokeMessage.MessageType.IMAGE}:
elif response.type in {
DatasourceInvokeMessage.MessageType.IMAGE_LINK,
DatasourceInvokeMessage.MessageType.IMAGE,
}:
result += (
"image has been created and sent to user already, "
+ "you do not need to create it, just tell the user to check it now."
@ -123,7 +126,10 @@ class DatasourceEngine:
Extract datasource response binary
"""
for response in datasource_response:
if response.type in {DatasourceInvokeMessage.MessageType.IMAGE_LINK, DatasourceInvokeMessage.MessageType.IMAGE}:
if response.type in {
DatasourceInvokeMessage.MessageType.IMAGE_LINK,
DatasourceInvokeMessage.MessageType.IMAGE,
}:
mimetype = None
if not response.meta:
raise ValueError("missing meta data")