fix: PDF Preview

Author: shaohuzhang1

apps/oss/serializers/file.py

@@ -96,7 +96,7 @@ def get(self, with_valid=True):
             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(),

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)

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)}`