ngskcloud
/
gcgj-dify-1.7.0
17
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

r2

Browse Source
feat/r2
jyong 11 months ago
parent 832bef053f
commit a6ff9b224b
1 changed files with 19 additions and 17 deletions
Show Stats Download Patch File Download Diff File

4
api/services/rag_pipeline/rag_pipeline.py
Unescape Escape View File

@ -551,7 +551,8 @@ 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] = (
datasource_runtime.online_drive_browse_files(
user_id=account.id, user_id=account.id,
request=OnlineDriveBrowseFilesRequest( request=OnlineDriveBrowseFilesRequest(
bucket=user_inputs.get("bucket"), bucket=user_inputs.get("bucket"),
@ -561,6 +562,7 @@ class RagPipelineService:
), ),
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(
total=0, total=0,

Write Preview
Loading…
Cancel
Save