youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-02-26 04:27:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

add recommended rag plugin endpoint

Browse Source
This commit is contained in:
jyong
2025-09-01 20:12:27 +08:00
parent d94e03c72b
commit 633de2cb47
1 changed files with 6 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
api/core/workflow/nodes/datasource/datasource_node.py
View File

@ -376,10 +376,7 @@ class DatasourceNode(Node):
assert isinstance(message.message, DatasourceMessage.TextMessage)
url = message.message.text
if message.meta:
transfer_method = message.meta.get("transfer_method", FileTransferMethod.DATASOURCE_FILE)
else:
transfer_method = FileTransferMethod.DATASOURCE_FILE
transfer_method = FileTransferMethod.TOOL_FILE
datasource_file_id = str(url).split("/")[-1].split(".")[0]
@ -390,7 +387,7 @@ class DatasourceNode(Node):
raise ToolFileError(f"Tool file {datasource_file_id} does not exist")
mapping = {
"datasource_file_id": datasource_file_id,
"id": datasource_file_id,
"type": file_factory.get_file_type_by_mime_type(datasource_file.mime_type),
"transfer_method": transfer_method,
"url": url,
@ -413,8 +410,8 @@ class DatasourceNode(Node):
raise ToolFileError(f"datasource file {datasource_file_id} not exists")
mapping = {
"datasource_file_id": datasource_file_id,
"transfer_method": FileTransferMethod.DATASOURCE_FILE,
"id": datasource_file_id,
"transfer_method": FileTransferMethod.TOOL_FILE,
}
files.append(
@ -516,10 +513,7 @@ class DatasourceNode(Node):
assert isinstance(message.message, DatasourceMessage.TextMessage)
url = message.message.text
if message.meta:
transfer_method = message.meta.get("transfer_method", FileTransferMethod.DATASOURCE_FILE)
else:
transfer_method = FileTransferMethod.DATASOURCE_FILE
transfer_method = FileTransferMethod.TOOL_FILE
datasource_file_id = str(url).split("/")[-1].split(".")[0]
@ -530,7 +524,7 @@ class DatasourceNode(Node):
raise ToolFileError(f"Tool file {datasource_file_id} does not exist")
mapping = {
"datasource_file_id": datasource_file_id,
"id": datasource_file_id,
"type": file_factory.get_file_type_by_mime_type(datasource_file.mime_type),
"transfer_method": transfer_method,
"url": url,