yujunjun/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2025-11-16 03:13:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

r2

Browse Source
This commit is contained in:
jyong 2025-07-02 18:20:41 +08:00
parent 832bef053f
commit a6ff9b224b
1 changed files with 19 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
api/services/rag_pipeline/rag_pipeline.py
View File

@ -391,7 +391,7 @@ class RagPipelineService:
rag_pipeline_variable = RAGPipelineVariable(**v) rag_pipeline_variable = RAGPipelineVariable(**v)
if rag_pipeline_variable.variable in user_inputs: if rag_pipeline_variable.variable in user_inputs:
rag_pipeline_variables.append( rag_pipeline_variables.append(
RAGPipelineVariableInput( RAGPipelineVariableInput(
variable=rag_pipeline_variable, variable=rag_pipeline_variable,
value=user_inputs[rag_pipeline_variable.variable], value=user_inputs[rag_pipeline_variable.variable],
) )
@ -437,14 +437,14 @@ class RagPipelineService:
rag_pipeline_variables = [] rag_pipeline_variables = []
if published_workflow.rag_pipeline_variables: if published_workflow.rag_pipeline_variables:
for v in published_workflow.rag_pipeline_variables: for v in published_workflow.rag_pipeline_variables:
rag_pipeline_variable = RAGPipelineVariable(**v) rag_pipeline_variable = RAGPipelineVariable(**v)
if rag_pipeline_variable.variable in user_inputs: if rag_pipeline_variable.variable in user_inputs:
rag_pipeline_variables.append( rag_pipeline_variables.append(
RAGPipelineVariableInput( RAGPipelineVariableInput(
variable=rag_pipeline_variable, variable=rag_pipeline_variable,
value=user_inputs[rag_pipeline_variable.variable], value=user_inputs[rag_pipeline_variable.variable],
)
) )
)
workflow_node_execution = self._handle_node_run_result( workflow_node_execution = self._handle_node_run_result(
getter=lambda: WorkflowEntry.single_step_run( getter=lambda: WorkflowEntry.single_step_run(
@ -551,15 +551,17 @@ class RagPipelineService:
yield DatasourceErrorEvent(error=str(e)).model_dump() yield DatasourceErrorEvent(error=str(e)).model_dump()
case DatasourceProviderType.ONLINE_DRIVE: case DatasourceProviderType.ONLINE_DRIVE:
datasource_runtime = cast(OnlineDriveDatasourcePlugin, datasource_runtime) datasource_runtime = cast(OnlineDriveDatasourcePlugin, datasource_runtime)
online_drive_result: Generator[OnlineDriveBrowseFilesResponse, None, None] = datasource_runtime.online_drive_browse_files( online_drive_result: Generator[OnlineDriveBrowseFilesResponse, None, None] = (
user_id=account.id, datasource_runtime.online_drive_browse_files(
request=OnlineDriveBrowseFilesRequest( user_id=account.id,
bucket=user_inputs.get("bucket"), request=OnlineDriveBrowseFilesRequest(
prefix=user_inputs.get("prefix"), bucket=user_inputs.get("bucket"),
max_keys=user_inputs.get("max_keys", 20), prefix=user_inputs.get("prefix"),
start_after=user_inputs.get("start_after"), max_keys=user_inputs.get("max_keys", 20),
), start_after=user_inputs.get("start_after"),
provider_type=datasource_runtime.datasource_provider_type(), ),
provider_type=datasource_runtime.datasource_provider_type(),
)
) )
start_time = time.time() start_time = time.time()
start_event = DatasourceProcessingEvent( start_event = DatasourceProcessingEvent(