opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-27 02:36:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(credential-icon, create-from-pipeline, test-run): improve component structure and enhance data handling for online drive files

Browse Source
This commit is contained in:
twwu 2025-08-19 15:24:14 +08:00
parent 95f60d89ab
commit be045a68ee
3 changed files with 22 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
web/app/components/datasets/common/credential-icon.tsx
View File

@ -26,13 +26,17 @@ export const CredentialIcon: React.FC<CredentialIconProps> = ({
if (avatar_url && avatar_url !== 'default') { if (avatar_url && avatar_url !== 'default') {
return ( return (
<img <div
src={avatar_url} className='flex shrink-0 items-center justify-center overflow-hidden rounded-md border border-divider-regular'
alt={`${name} logo`} style={{ width: `${size}px`, height: `${size}px` }}
width={size} >
height={size} <img
className={cn('shrink-0 rounded-md border border-divider-regular object-contain', className)} src={avatar_url}
/> width={size}
height={size}
className={cn('shrink-0 object-contain', className)}
/>
</div>
) )
} }

11
web/app/components/datasets/documents/create-from-pipeline/index.tsx
View File

@ -209,10 +209,11 @@ const CreateFormPipeline = () => {
} }
if (datasourceType === DatasourceType.onlineDrive) { if (datasourceType === DatasourceType.onlineDrive) {
const { bucket } = dataSourceStore.getState() const { bucket } = dataSourceStore.getState()
const { id } = previewOnlineDriveFileRef.current! const { id, type } = previewOnlineDriveFileRef.current!
datasourceInfoList.push({ datasourceInfoList.push({
bucket, bucket,
id, id,
type,
credential_id: currentCredentialId, credential_id: currentCredentialId,
}) })
} }
@ -233,7 +234,7 @@ const CreateFormPipeline = () => {
const handleProcess = useCallback(async (data: Record<string, any>) => { const handleProcess = useCallback(async (data: Record<string, any>) => {
if (!datasource) if (!datasource)
return return
const { bucket, currentCredentialId } = dataSourceStore.getState() const { bucket, currentCredentialId, fileList: onlineDriveFileList } = dataSourceStore.getState()
const datasourceInfoList: Record<string, any>[] = [] const datasourceInfoList: Record<string, any>[] = []
if (datasourceType === DatasourceType.localFile) { if (datasourceType === DatasourceType.localFile) {
fileList.forEach((file) => { fileList.forEach((file) => {
@ -273,10 +274,12 @@ const CreateFormPipeline = () => {
} }
if (datasourceType === DatasourceType.onlineDrive) { if (datasourceType === DatasourceType.onlineDrive) {
if (datasourceType === DatasourceType.onlineDrive) { if (datasourceType === DatasourceType.onlineDrive) {
selectedFileIds.forEach((key) => { selectedFileIds.forEach((id) => {
const file = onlineDriveFileList.find(file => file.id === id)
datasourceInfoList.push({ datasourceInfoList.push({
bucket, bucket,
key, id: file?.id,
type: file?.type,
credential_id: currentCredentialId, credential_id: currentCredentialId,
}) })
}) })

6
web/app/components/rag-pipeline/components/panel/test-run/index.tsx
View File

@ -107,10 +107,12 @@ const TestRunPanel = () => {
}) })
} }
if (datasourceType === DatasourceType.onlineDrive) { if (datasourceType === DatasourceType.onlineDrive) {
const { bucket } = dataSourceStore.getState() const { bucket, fileList } = dataSourceStore.getState()
const file = fileList.find(file => file.id === selectedFileIds[0])
datasourceInfoList.push({ datasourceInfoList.push({
bucket, bucket,
id: selectedFileIds[0], id: file?.id,
type: file?.type,
credential_id: credentialId, credential_id: credentialId,
}) })
} }