diff --git a/ui/src/components/ai-chat/KnowledgeSource.vue b/ui/src/components/ai-chat/KnowledgeSource.vue
index 4de9ce1c..1c5369a3 100644
--- a/ui/src/components/ai-chat/KnowledgeSource.vue
+++ b/ui/src/components/ai-chat/KnowledgeSource.vue
@@ -8,30 +8,31 @@
     >
   </div>
   <div class="mt-8" v-if="!isWorkFlow(props.type)">
-    <el-space wrap>
-      <div v-for="(paragraph, index) in uniqueParagraphList" :key="index">
-        <el-icon class="mr-4" :size="25">
-          <img :src="getIconPath(paragraph.document_name)" style="width: 90%" alt="" />
-        </el-icon>
-        <span
-          v-if="!paragraph.source_url"
-          class="ellipsis"
-          :title="paragraph?.document_name?.trim()"
-        >
-          {{ paragraph?.document_name }}
-        </span>
-        <a
-          v-else
-          @click="openLink(paragraph.source_url)"
-          class="ellipsis"
-          :title="paragraph?.document_name?.trim()"
-        >
-          <span :title="paragraph?.document_name?.trim()">
-            {{ paragraph?.document_name }}
-          </span>
-        </a>
-      </div>
-    </el-space>
+    <el-row :gutter="8" v-if="uniqueParagraphList?.length">
+      <template v-for="(item, index) in uniqueParagraphList" :key="index">
+        <el-col :span="12" class="mb-8">
+          <el-card shadow="never" class="file-List-card" data-width="40">
+            <div class="flex-between">
+              <div class="flex">
+                <img :src="getImgUrl(item && item?.document_name)" alt="" width="20" />
+                <div class="ml-4" v-if="!item.source_url">
+                  <p>{{ item && item?.document_name }}</p>
+                </div>
+                <div class="ml-8" v-else>
+                  <a
+                    @click="openLink(item.source_url)"
+                    class="ellipsis"
+                    :title="item?.document_name?.trim()"
+                  >
+                    <span :title="item?.document_name?.trim()">{{ item?.document_name }}</span>
+                  </a>
+                </div>
+              </div>
+            </div>
+          </el-card>
+        </el-col>
+      </template>
+    </el-row>
   </div>
 
   <div class="border-t color-secondary flex-between mt-12" style="padding-top: 12px">
@@ -59,7 +60,7 @@ import { computed, ref } from 'vue'
 import ParagraphSourceDialog from './ParagraphSourceDialog.vue'
 import ExecutionDetailDialog from './ExecutionDetailDialog.vue'
 import { isWorkFlow } from '@/utils/application'
-
+import { getImgUrl } from '@/utils/utils'
 const props = defineProps({
   data: {
     type: Object,
@@ -70,15 +71,6 @@ const props = defineProps({
     default: ''
   }
 })
-const iconMap: { [key: string]: string } = {
-  doc: '../../assets/doc-icon.svg',
-  docx: '../../assets/docx-icon.svg',
-  pdf: '../../assets/pdf-icon.svg',
-  md: '../../assets/md-icon.svg',
-  txt: '../../assets/txt-icon.svg',
-  xls: '../../assets/xls-icon.svg',
-  xlsx: '../../assets/xlsx-icon.svg'
-}
 
 const ParagraphSourceDialogRef = ref()
 const ExecutionDetailDialogRef = ref()
@@ -106,15 +98,16 @@ const uniqueParagraphList = computed(() => {
     }) || []
   )
 })
-
-function getIconPath(documentName: string) {
-  const extension = documentName.split('.').pop()?.toLowerCase()
-  if (!documentName || !extension) return new URL(`${iconMap['doc']}`, import.meta.url).href
-  if (iconMap && extension && iconMap[extension]) {
-    return new URL(`${iconMap[extension]}`, import.meta.url).href
-  }
-  return new URL(`${iconMap['doc']}`, import.meta.url).href
+function fileType(name: string) {
+  const suffix = name.split('.')
+  return suffix[suffix.length - 1]
 }
+const typeList: any = {
+  txt: ['txt', 'pdf', 'docx', 'csv', 'md', 'html'],
+  table: ['xlsx', 'xls', 'csv'],
+  QA: ['xlsx', 'csv', 'xls']
+}
+
 function openLink(url: string) {
   // 如果url不是以/结尾，加上/
   if (url && !url.endsWith('/')) {
diff --git a/ui/src/components/ai-chat/component/ParagraphCard.vue b/ui/src/components/ai-chat/component/ParagraphCard.vue
index 1a566c14..4fb785e6 100644
--- a/ui/src/components/ai-chat/component/ParagraphCard.vue
+++ b/ui/src/components/ai-chat/component/ParagraphCard.vue
@@ -18,45 +18,8 @@
     <template #footer>
       <div class="footer-content flex-between">
         <el-text class="flex align-center" style="width: 70%">
-          <el-icon class="mr-4" :size="25">
-            <img
-              src="@/assets/doc-icon.svg"
-              style="width: 90%"
-              alt=""
-              v-if="data?.document_name?.includes('doc')"
-            />
-            <img
-              src="@/assets/docx-icon.svg"
-              style="width: 90%"
-              alt=""
-              v-else-if="data?.document_name?.includes('docx')"
-            />
-            <img
-              src="@/assets/pdf-icon.svg"
-              style="width: 90%"
-              alt=""
-              v-else-if="data?.document_name?.includes('pdf')"
-            />
-            <img
-              src="@/assets/md-icon.svg"
-              style="width: 90%"
-              alt=""
-              v-else-if="data?.document_name?.includes('md')"
-            />
-            <img
-              src="@/assets/xls-icon.svg"
-              style="width: 90%"
-              alt=""
-              v-else-if="data?.document_name?.includes('xls')"
-            />
-            <img
-              src="@/assets/txt-icon.svg"
-              style="width: 90%"
-              alt=""
-              v-else-if="data?.document_name?.includes('txt')"
-            />
-            <img src="@/assets/doc-icon.svg" style="width: 90%" alt="" v-else />
-          </el-icon>
+          <img :src="getImgUrl(data?.document_name?.trim())" alt="" width="20" class="mr-4" />
+
           <span class="ellipsis" :title="data?.document_name?.trim()">
             {{ data?.document_name.trim() }}</span
           >
@@ -73,6 +36,7 @@
   </CardBox>
 </template>
 <script setup lang="ts">
+import { getImgUrl } from '@/utils/utils'
 const props = defineProps({
   data: {
     type: Object,
@@ -83,23 +47,6 @@ const props = defineProps({
     default: 0
   }
 })
-const iconMap: { [key: string]: string } = {
-  doc: '../../assets/doc-icon.svg',
-  docx: '../../assets/docx-icon.svg',
-  pdf: '../../assets/pdf-icon.svg',
-  md: '../../assets/md-icon.svg',
-  txt: '../../assets/txt-icon.svg',
-  xls: '../../assets/xls-icon.svg',
-  xlsx: '../../assets/xlsx-icon.svg'
-}
-function getIconPath(documentName: string) {
-  const extension = documentName.split('.').pop()?.toLowerCase()
-  if (!documentName || !extension) return new URL(`${iconMap['doc']}`, import.meta.url).href
-  if (iconMap && extension && iconMap[extension]) {
-    return new URL(`${iconMap[extension]}`, import.meta.url).href
-  }
-  return new URL(`${iconMap['doc']}`, import.meta.url).href
-}
 </script>
 <style lang="scss" scoped>
 .paragraph-source-card-height {