datasource

This commit is contained in:
zxhlyh
2025-06-06 10:54:44 +08:00
parent cf2ef93ad5
commit 8e4165defe
5 changed files with 92 additions and 10 deletions

View File

@ -0,0 +1,37 @@
import type {
InputVar,
ToolWithProvider,
} from '../types'
import type { DataSourceNodeType } from '../nodes/data-source/types'
import { CollectionType } from '@/app/components/tools/types'
import { toolParametersToFormSchemas } from '@/app/components/tools/utils/to-form-schema'
export const getDataSourceCheckParams = (
toolData: DataSourceNodeType,
dataSourceList: ToolWithProvider[],
language: string,
) => {
const { plugin_id, provider_type, datasource_name } = toolData
const isBuiltIn = provider_type === CollectionType.builtIn
const currentDataSource = dataSourceList.find(item => item.plugin_id === plugin_id)
const currentDataSourceItem = currentDataSource?.tools.find(tool => tool.name === datasource_name)
const formSchemas = currentDataSourceItem ? toolParametersToFormSchemas(currentDataSourceItem.parameters) : []
return {
dataSourceInputsSchema: (() => {
const formInputs: InputVar[] = []
formSchemas.forEach((item: any) => {
formInputs.push({
label: item.label[language] || item.label.en_US,
variable: item.variable,
type: item.type,
required: item.required,
hide: item.hide,
})
})
return formInputs
})(),
notAuthed: isBuiltIn && !!currentDataSource?.allow_delete && !currentDataSource?.is_authorized,
language,
}
}

View File

@ -7,3 +7,4 @@ export * from './tool'
export * from './workflow'
export * from './variable'
export * from './gen-node-meta-data'
export * from './data-source'