jandres/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-01-08 07:14:14 +00:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

add recommended rag plugin endpoint

Browse Source
This commit is contained in:
jyong
2025-09-01 20:41:14 +08:00
parent 633de2cb47
commit 826b9d5b21
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

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

2
api/factories/file_factory.py
View File

@@ -355,7 +355,7 @@ def _build_from_datasource_file(
tenant_id=tenant_id,
filename=datasource_file.name,
type=file_type,
transfer_method=transfer_method,
transfer_method=FileTransferMethod.TOOL_FILE,
remote_url=datasource_file.source_url,
related_id=datasource_file.id,
extension=extension,