From 51a29a997be718ffd5a29685b481664cbd07af91 Mon Sep 17 00:00:00 2001
From: wxg0103 <727495428@qq.com>
Date: Thu, 27 Feb 2025 15:16:34 +0800
Subject: [PATCH] fix: Knowledge source redirection failed

---
 ui/src/components/ai-chat/KnowledgeSource.vue        | 12 +++++-------
 .../components/ai-chat/component/ParagraphCard.vue   | 10 ++++------
 ui/src/utils/utils.ts                                |  7 +++++++
 3 files changed, 16 insertions(+), 13 deletions(-)

diff --git a/ui/src/components/ai-chat/KnowledgeSource.vue b/ui/src/components/ai-chat/KnowledgeSource.vue
index af99a4775..4e527ae6d 100644
--- a/ui/src/components/ai-chat/KnowledgeSource.vue
+++ b/ui/src/components/ai-chat/KnowledgeSource.vue
@@ -21,11 +21,7 @@
                 </div>
                 <div class="ml-8" v-else>
                   <a
-                    :href="
-                      item.source_url && !item.source_url.endsWith('/')
-                        ? item.source_url + '/'
-                        : item.source_url
-                    "
+                    :href="getNormalizedUrl(item?.source_url)"
                     target="_blank"
                     class="ellipsis"
                     :title="item?.document_name?.trim()"
@@ -43,7 +39,9 @@
 
   <div class="border-t color-secondary flex-between mt-12" style="padding-top: 12px">
     <div>
-      <span class="mr-8"> {{ $t('chat.KnowledgeSource.consume') }}: {{ data?.message_tokens + data?.answer_tokens }} </span>
+      <span class="mr-8">
+        {{ $t('chat.KnowledgeSource.consume') }}: {{ data?.message_tokens + data?.answer_tokens }}
+      </span>
       <span> {{ $t('chat.KnowledgeSource.consumeTime') }}: {{ data?.run_time?.toFixed(2) }} s</span>
     </div>
     <el-button
@@ -66,7 +64,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'
+import { getImgUrl, getNormalizedUrl } from '@/utils/utils'
 const props = defineProps({
   data: {
     type: Object,
diff --git a/ui/src/components/ai-chat/component/ParagraphCard.vue b/ui/src/components/ai-chat/component/ParagraphCard.vue
index 82fbd13cd..d6b7936d9 100644
--- a/ui/src/components/ai-chat/component/ParagraphCard.vue
+++ b/ui/src/components/ai-chat/component/ParagraphCard.vue
@@ -22,11 +22,7 @@
 
           <template v-if="meta?.source_url">
             <a
-              :href="
-                meta?.source_url && !meta?.source_url.endsWith('/')
-                  ? meta?.source_url + '/'
-                  : meta?.source_url
-              "
+              :href="getNormalizedUrl(meta?.source_url)"
               target="_blank"
               class="ellipsis-1 break-all"
               :title="data?.document_name?.trim()"
@@ -54,7 +50,7 @@
   </CardBox>
 </template>
 <script setup lang="ts">
-import { getImgUrl } from '@/utils/utils'
+import { getImgUrl, getNormalizedUrl } from '@/utils/utils'
 import { computed } from 'vue'
 
 const props = defineProps({
@@ -94,6 +90,7 @@ const meta = computed(() => (isMetaObject.value ? props.data.meta : parsedMeta.v
     }
   }
 }
+
 .paragraph-source-card-height {
   height: 260px;
 }
@@ -105,6 +102,7 @@ const meta = computed(() => (isMetaObject.value ? props.data.meta : parsedMeta.v
   .paragraph-source-card {
     .footer-content {
       display: block;
+
       .item {
         max-width: 100%;
       }
diff --git a/ui/src/utils/utils.ts b/ui/src/utils/utils.ts
index 526e09465..473ab28a3 100644
--- a/ui/src/utils/utils.ts
+++ b/ui/src/utils/utils.ts
@@ -106,3 +106,10 @@ export function cutFilename(filename: string, num: number) {
   const suffix = lastIndex === -1 ? '' : filename.substring(lastIndex + 1)
   return filename.substring(0, num - suffix.length - 1) + '.' + suffix
 }
+
+export function getNormalizedUrl(url: string) {
+  if (url && !url.endsWith('/') && !/\.[^/]+$/.test(url)) {
+    return url + '/'
+  }
+  return url
+}