From b40c76d3f3c6d5520943ce863aacee733d6482ba Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Tue, 8 Jul 2025 10:52:27 +0800
Subject: [PATCH] fix: PDF Preview (#3504)

---
 apps/oss/serializers/file.py                           |  2 +-
 .../component/knowledge-source-component/index.vue     | 10 ++++++----
 ui/src/utils/common.ts                                 |  6 ++++++
 3 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/apps/oss/serializers/file.py b/apps/oss/serializers/file.py
index c3584caee..ad083b0a7 100644
--- a/apps/oss/serializers/file.py
+++ b/apps/oss/serializers/file.py
@@ -96,7 +96,7 @@ class FileSerializer(serializers.Serializer):
             content_type = mime_types.get(file_type, 'application/octet-stream')
             headers = {
                 'Content-Type': content_type,
-                'Content-Disposition': f'attachment; filename="{file.file_name}"'
+                'Content-Disposition': f'{"inline" if file_type == "pdf" else "attachment"}; filename="{file.file_name}"'
             }
             return HttpResponse(
                 file.get_bytes(),
diff --git a/ui/src/components/ai-chat/component/knowledge-source-component/index.vue b/ui/src/components/ai-chat/component/knowledge-source-component/index.vue
index a1760ebde..00e268770 100644
--- a/ui/src/components/ai-chat/component/knowledge-source-component/index.vue
+++ b/ui/src/components/ai-chat/component/knowledge-source-component/index.vue
@@ -20,14 +20,14 @@
                   <div
                     class="ml-4 ellipsis-1"
                     :title="item?.document_name"
-                    v-if="!item.source_url"
+                    v-if="showPDF(item)"
                     @click="openParagraphDocument(item)"
                   >
                     <p>{{ item && item?.document_name }}</p>
                   </div>
                   <div class="ml-8" v-else>
                     <a
-                      :href="getNormalizedUrl(item?.source_url)"
+                      :href="getFileUrl(item?.meta?.source_file_id)"
                       target="_blank"
                       class="ellipsis-1"
                       :title="item?.document_name?.trim()"
@@ -89,7 +89,7 @@ import ExecutionDetailContent from './ExecutionDetailContent.vue'
 import ParagraphDocumentContent from './ParagraphDocumentContent.vue'
 import ParagraphSourceContent from './ParagraphSourceContent.vue'
 import { arraySort } from '@/utils/array'
-import { getImgUrl, getNormalizedUrl } from '@/utils/common'
+import { getImgUrl, getFileUrl } from '@/utils/common'
 import { t } from '@/locales'
 const props = defineProps({
   data: {
@@ -107,7 +107,9 @@ const props = defineProps({
 })
 
 const emit = defineEmits(['openExecutionDetail', 'openParagraph', 'openParagraphDocument'])
-
+const showPDF = (item: any) => {
+  return item.document_name.toLocaleLowerCase().endsWith('.pdf') && item.meta?.source_file_id
+}
 const dialogVisible = ref(false)
 const dialogTitle = ref('')
 const currentComponent = shallowRef<any>(null)
diff --git a/ui/src/utils/common.ts b/ui/src/utils/common.ts
index 11ee7ad02..596112a71 100644
--- a/ui/src/utils/common.ts
+++ b/ui/src/utils/common.ts
@@ -143,6 +143,12 @@ export function getNormalizedUrl(url: string) {
   }
   return url
 }
+export function getFileUrl(fileId?: string) {
+  if (fileId) {
+    return `${window.MaxKB.prefix}/oss/file/${fileId}`
+  }
+  return ''
+}
 export const resetUrl = (url: string, defaultUrl?: string) => {
   if (url && url.startsWith('./')) {
     return `${window.MaxKB.prefix}/${url.substring(2)}`