From 90d0f12ee933c150947350f51bcb7c217bb50e9e Mon Sep 17 00:00:00 2001
From: twwu <twwu@dify.ai>
Date: Fri, 11 Jul 2025 16:53:03 +0800
Subject: [PATCH] refactor: update file extension handling in ChunkPreview
 component to use dynamic extension retrieval

---
 .../documents/create-from-pipeline/preview/chunk-preview.tsx | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/web/app/components/datasets/documents/create-from-pipeline/preview/chunk-preview.tsx b/web/app/components/datasets/documents/create-from-pipeline/preview/chunk-preview.tsx
index 763b5f38af..09c274e42e 100644
--- a/web/app/components/datasets/documents/create-from-pipeline/preview/chunk-preview.tsx
+++ b/web/app/components/datasets/documents/create-from-pipeline/preview/chunk-preview.tsx
@@ -16,6 +16,7 @@ import Badge from '@/app/components/base/badge'
 import Button from '@/app/components/base/button'
 import type { OnlineDriveFile } from '@/models/pipeline'
 import { DatasourceType } from '@/models/pipeline'
+import { getFileExtension } from '../data-source/online-drive/file-list/list/utils'
 
 type ChunkPreviewProps = {
   dataSourceType: DatasourceType
@@ -122,7 +123,7 @@ const ChunkPreview = ({
                 onlineDriveFiles.map(file => ({
                   id: file.key,
                   name: file.displayName,
-                  extension: 'md',
+                  extension: getFileExtension(previewOnlineDriveFile?.displayName),
                 }))
               }
               onChange={(selected) => {
@@ -134,7 +135,7 @@ const ChunkPreview = ({
                 {
                   id: previewOnlineDriveFile?.key || '',
                   name: previewOnlineDriveFile?.displayName || '',
-                  extension: 'md',
+                  extension: getFileExtension(previewOnlineDriveFile?.displayName),
                 }
               }
             />