feat: Add ONLINE_DRIVE_OUTPUT and integrate into DataSource components for online drive support

This commit is contained in:
twwu
2025-07-04 18:04:47 +08:00
parent e67a19b26b
commit 0d9991ec88
4 changed files with 70 additions and 0 deletions

View File

@ -7,6 +7,7 @@ import {
COMMON_OUTPUT,
LOCAL_FILE_OUTPUT,
ONLINE_DOCUMENT_OUTPUT,
ONLINE_DRIVE_OUTPUT,
WEBSITE_CRAWL_OUTPUT,
} from './constants'
import { VarType as VarKindType } from '@/app/components/workflow/nodes/tool/types'
@ -62,6 +63,7 @@ const nodeDefault: NodeDefault<DataSourceNodeType> = {
const isLocalFile = provider_type === DataSourceClassification.localFile
const isWebsiteCrawl = provider_type === DataSourceClassification.websiteCrawl
const isOnlineDocument = provider_type === DataSourceClassification.onlineDocument
const isOnlineDrive = provider_type === DataSourceClassification.onlineDrive
return [
...COMMON_OUTPUT.map(item => ({ variable: item.name, type: item.type })),
...(
@ -79,6 +81,11 @@ const nodeDefault: NodeDefault<DataSourceNodeType> = {
? ONLINE_DOCUMENT_OUTPUT.map(item => ({ variable: item.name, type: item.type }))
: []
),
...(
isOnlineDrive
? ONLINE_DRIVE_OUTPUT.map(item => ({ variable: item.name, type: item.type }))
: []
),
...ragVars,
]
},