Skip to content

Commit 4610be9

Browse files
committed
fix: Download template error
--bug=1051908 --user=王孝刚 【知识库】导出的文档打开后里面为错误日志,无文档内容,导出的压缩包不能解压 https://www.tapd.cn/57709429/s/1650087
1 parent 5ebcad7 commit 4610be9

13 files changed

+53
-8
lines changed

apps/dataset/serializers/document_serializers.py

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@
2323
from django.db.models import QuerySet, Count
2424
from django.db.models.functions import Substr, Reverse
2525
from django.http import HttpResponse
26-
from django.utils.translation import gettext_lazy as _, gettext
2726
from drf_yasg import openapi
2827
from openpyxl.cell.cell import ILLEGAL_CHARACTERS_RE
2928
from rest_framework import serializers
@@ -65,6 +64,8 @@
6564
embedding_by_document_list
6665
from setting.models import Model
6766
from smartdoc.conf import PROJECT_DIR
67+
from django.utils.translation import gettext_lazy as _, gettext, to_locale
68+
from django.utils.translation import get_language
6869

6970
parse_qa_handle_list = [XlsParseQAHandle(), CsvParseQAHandle(), XlsxParseQAHandle(), ZipParseQAHandle()]
7071
parse_table_handle_list = [CsvSplitTableHandle(), XlsSplitTableHandle(), XlsxSplitTableHandle()]
@@ -240,15 +241,15 @@ def get_request_params_api():
240241
def export(self, with_valid=True):
241242
if with_valid:
242243
self.is_valid(raise_exception=True)
243-
244+
language = get_language()
244245
if self.data.get('type') == 'csv':
245-
file = open(os.path.join(PROJECT_DIR, "apps", "dataset", 'template', 'csv_template.csv'), "rb")
246+
file = open(os.path.join(PROJECT_DIR, "apps", "dataset", 'template', f'csv_template_{to_locale(language)}.csv'), "rb")
246247
content = file.read()
247248
file.close()
248249
return HttpResponse(content, status=200, headers={'Content-Type': 'text/cxv',
249250
'Content-Disposition': 'attachment; filename="csv_template.csv"'})
250251
elif self.data.get('type') == 'excel':
251-
file = open(os.path.join(PROJECT_DIR, "apps", "dataset", 'template', 'excel_template.xlsx'), "rb")
252+
file = open(os.path.join(PROJECT_DIR, "apps", "dataset", 'template', f'csv_template_{to_locale(language)}.xlsx'), "rb")
252253
content = file.read()
253254
file.close()
254255
return HttpResponse(content, status=200, headers={'Content-Type': 'application/vnd.ms-excel',
@@ -257,16 +258,18 @@ def export(self, with_valid=True):
257258
def table_export(self, with_valid=True):
258259
if with_valid:
259260
self.is_valid(raise_exception=True)
260-
261+
language = get_language()
261262
if self.data.get('type') == 'csv':
262-
file = open(os.path.join(PROJECT_DIR, "apps", "dataset", 'template', _('MaxKB table template.csv')),
263-
"rb")
263+
file = open(
264+
os.path.join(PROJECT_DIR, "apps", "dataset", 'template', f'table_template_{to_locale(language)}.csv'),
265+
"rb")
264266
content = file.read()
265267
file.close()
266268
return HttpResponse(content, status=200, headers={'Content-Type': 'text/cxv',
267269
'Content-Disposition': 'attachment; filename="csv_template.csv"'})
268270
elif self.data.get('type') == 'excel':
269-
file = open(os.path.join(PROJECT_DIR, "apps", "dataset", 'template', _('MaxKB table template.xlsx')),
271+
file = open(os.path.join(PROJECT_DIR, "apps", "dataset", 'template',
272+
f'table_template_{to_locale(language)}.xlsx'),
270273
"rb")
271274
content = file.read()
272275
file.close()
Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
Section title (optional), section content (required, question answer, no more than 4096 characters)), question (optional, one per line in the cell)
2+
MaxKB product introduction, "MaxKB is a knowledge base question and answer system based on the LLM large language model. MaxKB = Max Knowledge Base aims to become the most powerful brain of the enterprise.
3+
Out-of-the-box: supports direct uploading of documents, automatic crawling of online documents, automatic text splitting and vectorization, and a good intelligent Q&A interactive experience;
4+
Seamless embedding: supports rapid embedding into third-party business systems with zero coding;
5+
Multi-model support: Supports docking with mainstream large models, including Ollama local private large models (such as Llama 2, Llama 3, qwen), Tongyi Qianwen, OpenAI, Azure OpenAI, Kimi, Zhipu AI, iFlytek Spark and Baidu Qianfan Large models etc. ","What is MaxKB?
6+
MaxKB product introduction
7+
Large language model supported by MaxKB
8+
MaxKB advantages"
Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
分段標題(選填),分段內容(必填,問題答案,最長不超過4096個字元)),問題(選填,單元格內一行一個)
2+
MaxKB產品介紹,"MaxKB 是一款基於 LLM 大語言模型的知識庫問答系統。MaxKB = Max Knowledge Base,旨在成為企業的最強大大腦。
3+
開箱即用:支援直接上傳文檔、自動爬取線上文檔,支援文字自動分割、向量化,智慧問答互動體驗好;
4+
無縫嵌入:支援零編碼快速嵌入到第三方業務系統;
5+
多模型支援:支持對接主流的大模型,包括Ollama 本地私有大模型(如Llama 2、Llama 3、qwen)、通義千問、OpenAI、Azure OpenAI、Kimi、智譜AI、訊飛星火和百度千帆大模型等。 ","MaxKB是什麼?
6+
MaxKB產品介紹
7+
MaxKB支援的大語言模型
8+
MaxKB優勢"
10.5 KB
Binary file not shown.
File renamed without changes.
10.6 KB
Binary file not shown.
Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
Position, reimbursement type, first-tier city reimbursement standard (yuan), second-tier city reimbursement standard (yuan), third-tier city reimbursement standard (yuan)
2+
Ordinary employees, accommodation expenses, 500, 400, 300
3+
Department head, accommodation fee, 600, 500, 400
4+
Department director, accommodation fee, 700, 600, 500
5+
Regional general manager, accommodation fee, 800, 700, 600
6+
Ordinary employees, food expenses, 50, 40, 30
7+
Department head, food expenses, 50, 40, 30
8+
Department director, food expenses, 50, 40, 30
9+
Regional general manager, food expenses, 50, 40, 30
10+
Ordinary employees, transportation expenses, 50, 40, 30
11+
Department head, transportation expenses, 50, 40, 30
12+
Department director, transportation expenses, 50, 40, 30
13+
Regional general manager, transportation expenses, 50, 40, 30
9.83 KB
Binary file not shown.
File renamed without changes.

0 commit comments

Comments
 (0)