From 679453372eeb36dc9abbcccf36a044ed0e76724a Mon Sep 17 00:00:00 2001 From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com> Date: Mon, 25 Aug 2025 14:50:33 +0800 Subject: [PATCH] feat: Whether the document support settings of the General Knowledge Base and Feishu Knowledge Base allow downloading (#3924) --- .../reranker_node/impl/base_reranker_node.py | 13 +++++++++++-- .../impl/base_search_knowledge_node.py | 9 ++++++++- .../knowledge-source-component/ParagraphCard.vue | 10 +++++++--- .../component/knowledge-source-component/index.vue | 10 +++++++--- ui/src/locales/lang/en-US/ai-chat.ts | 1 + ui/src/locales/lang/zh-CN/ai-chat.ts | 1 + ui/src/locales/lang/zh-Hant/ai-chat.ts | 1 + 7 files changed, 36 insertions(+), 9 deletions(-) diff --git a/apps/application/flow/step_node/reranker_node/impl/base_reranker_node.py b/apps/application/flow/step_node/reranker_node/impl/base_reranker_node.py index 946399e70..58492d305 100644 --- a/apps/application/flow/step_node/reranker_node/impl/base_reranker_node.py +++ b/apps/application/flow/step_node/reranker_node/impl/base_reranker_node.py @@ -67,6 +67,14 @@ def get_none_result(question): 'result_list': [], 'result': ''}, {}) +def reset_metadata(metadata): + meta = metadata.get('meta') + if isinstance(metadata.get('meta'), dict): + if not meta.get('allow_download', False): + metadata['meta'] = {'allow_download': False} + return metadata + + class BaseRerankerNode(IRerankerNode): def save_context(self, details, workflow_manage): self.context['document_list'] = details.get('document_list', []) @@ -83,8 +91,9 @@ class BaseRerankerNode(IRerankerNode): if len(documents) == 0: return get_none_result(question) top_n = reranker_setting.get('top_n', 3) - self.context['document_list'] = [{'page_content': document.page_content, 'metadata': document.metadata} for - document in documents] + self.context['document_list'] = [ + {'page_content': document.page_content, 'metadata': reset_metadata(document.metadata)} for + document in documents] self.context['question'] = question workspace_id = self.workflow_manage.get_body().get('workspace_id') reranker_model = get_model_instance_by_model_workspace_id(reranker_model_id, diff --git a/apps/application/flow/step_node/search_knowledge_node/impl/base_search_knowledge_node.py b/apps/application/flow/step_node/search_knowledge_node/impl/base_search_knowledge_node.py index ddbe67598..176b7a99e 100644 --- a/apps/application/flow/step_node/search_knowledge_node/impl/base_search_knowledge_node.py +++ b/apps/application/flow/step_node/search_knowledge_node/impl/base_search_knowledge_node.py @@ -46,6 +46,12 @@ def reset_title(title): return f"#### {title}\n" +def reset_meta(meta): + if not meta.get('allow_download', False): + return {'allow_download': False} + return meta + + class BaseSearchKnowledgeNode(ISearchKnowledgeStepNode): def save_context(self, details, workflow_manage): result = details.get('paragraph_list', []) @@ -122,7 +128,8 @@ class BaseSearchKnowledgeNode(ISearchKnowledgeStepNode): 'create_time': paragraph.get('create_time').strftime("%Y-%m-%d %H:%M:%S"), 'id': str(paragraph.get('id')), 'knowledge_id': str(paragraph.get('knowledge_id')), - 'document_id': str(paragraph.get('document_id')) + 'document_id': str(paragraph.get('document_id')), + 'meta': reset_meta(paragraph.get('meta')) } @staticmethod diff --git a/ui/src/components/ai-chat/component/knowledge-source-component/ParagraphCard.vue b/ui/src/components/ai-chat/component/knowledge-source-component/ParagraphCard.vue index e6d3e985e..73ca64060 100644 --- a/ui/src/components/ai-chat/component/knowledge-source-component/ParagraphCard.vue +++ b/ui/src/components/ai-chat/component/knowledge-source-component/ParagraphCard.vue @@ -37,7 +37,7 @@ {{ data?.document_name }} -
+
{{ data?.document_name?.trim() }} @@ -87,8 +87,12 @@ const parsedMeta = computed(() => { }) const meta = computed(() => (isMetaObject.value ? props.data.meta : parsedMeta.value)) -function infoMessage() { - MsgInfo(t('chat.noDocument')) +function infoMessage(data: any) { + if (data?.meta?.allow_download === false) { + MsgInfo(t('chat.noPermissionDownload')) + } else { + MsgInfo(t('chat.noDocument')) + } } 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 784cf2a46..b414ab5b6 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 @@ -45,7 +45,7 @@ {{ item?.document_name }}
-
+
{{ item?.document_name?.trim() }} @@ -167,8 +167,12 @@ const currentComponent = shallowRef(null) const currentChatDetail = ref(null) const dialogType = ref('') -function infoMessage() { - MsgInfo(t('chat.noDocument')) +function infoMessage(data: any) { + if (data?.meta?.allow_download === false) { + MsgInfo(t('chat.noPermissionDownload')) + } else { + MsgInfo(t('chat.noDocument')) + } } function openParagraph(row: any, id?: string) { dialogTitle.value = t('chat.KnowledgeSource.title') diff --git a/ui/src/locales/lang/en-US/ai-chat.ts b/ui/src/locales/lang/en-US/ai-chat.ts index 729089898..4100a079e 100644 --- a/ui/src/locales/lang/en-US/ai-chat.ts +++ b/ui/src/locales/lang/en-US/ai-chat.ts @@ -14,6 +14,7 @@ export default { quote: 'Quote', download: 'Click to Download', noDocument: 'Original Document Not Found', + noPermissionDownload: 'No permission to download', passwordValidator: { title: 'Enter Password to Access', errorMessage1: 'Password cannot be empty', diff --git a/ui/src/locales/lang/zh-CN/ai-chat.ts b/ui/src/locales/lang/zh-CN/ai-chat.ts index 4a024954a..d725a8c46 100644 --- a/ui/src/locales/lang/zh-CN/ai-chat.ts +++ b/ui/src/locales/lang/zh-CN/ai-chat.ts @@ -18,6 +18,7 @@ export default { download: '点击下载文件', transcribing: '转文字中', noDocument: '原文档不存在', + noPermissionDownload: '无权限下载', passwordValidator: { title: '请输入密码打开链接', errorMessage1: '密码不能为空', diff --git a/ui/src/locales/lang/zh-Hant/ai-chat.ts b/ui/src/locales/lang/zh-Hant/ai-chat.ts index 432050783..3b5c430e5 100644 --- a/ui/src/locales/lang/zh-Hant/ai-chat.ts +++ b/ui/src/locales/lang/zh-Hant/ai-chat.ts @@ -14,6 +14,7 @@ export default { quote: '引用', download: '點擊下載文件', noDocument: '原文檔不存在', + noPermissionDownload: '無許可權下載', passwordValidator: { title: '請輸入密碼打開連結', errorMessage1: '密碼不能為空',