zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: Dialogue log export, question time does not match on the page (#2396)

Browse Source
This commit is contained in:
shaohuzhang1 2025-02-25 16:17:41 +08:00 committed by GitHub
parent 5da758e8a1
commit 218a247684
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
apps/application/serializers/chat_serializers.py
View File

@ -13,7 +13,7 @@ import uuid
from functools import reduce from functools import reduce
from io import BytesIO from io import BytesIO
from typing import Dict from typing import Dict
import pytz
import openpyxl import openpyxl
from django.core import validators from django.core import validators
from django.core.cache import caches from django.core.cache import caches
@ -46,6 +46,7 @@ from embedding.task import embedding_by_paragraph, embedding_by_paragraph_list
from setting.models import Model from setting.models import Model
from setting.models_provider import get_model_credential from setting.models_provider import get_model_credential
from smartdoc.conf import PROJECT_DIR from smartdoc.conf import PROJECT_DIR
from smartdoc.settings import TIME_ZONE
chat_cache = caches['chat_cache'] chat_cache = caches['chat_cache']
@ -208,7 +209,6 @@ class ChatSerializers(serializers.Serializer):
[])) for [])) for
key, node in search_dataset_node_list]) key, node in search_dataset_node_list])
improve_paragraph_list = row.get('improve_paragraph_list') improve_paragraph_list = row.get('improve_paragraph_list')
vote_status_map = {'-1': '未投票', '0': '赞同', '1': '反对'} vote_status_map = {'-1': '未投票', '0': '赞同', '1': '反对'}
return [str(row.get('chat_id')), row.get('abstract'), row.get('problem_text'), padding_problem_text, return [str(row.get('chat_id')), row.get('abstract'), row.get('problem_text'), padding_problem_text,
row.get('answer_text'), vote_status_map.get(row.get('vote_status')), reference_paragraph_len, row.get('answer_text'), vote_status_map.get(row.get('vote_status')), reference_paragraph_len,
@ -217,7 +217,7 @@ class ChatSerializers(serializers.Serializer):
f"{improve_paragraph_list[index].get('title')}\n{improve_paragraph_list[index].get('content')}" f"{improve_paragraph_list[index].get('title')}\n{improve_paragraph_list[index].get('content')}"
for index in range(len(improve_paragraph_list))]), for index in range(len(improve_paragraph_list))]),
row.get('message_tokens') + row.get('answer_tokens'), row.get('run_time'), row.get('message_tokens') + row.get('answer_tokens'), row.get('run_time'),
str(row.get('create_time').strftime('%Y-%m-%d %H:%M:%S') str(row.get('create_time').astimezone(pytz.timezone(TIME_ZONE)).strftime('%Y-%m-%d %H:%M:%S')
)] )]
def export(self, data, with_valid=True): def export(self, data, with_valid=True):
@ -256,6 +256,10 @@ class ChatSerializers(serializers.Serializer):
cell = worksheet.cell(row=row_idx, column=col_idx) cell = worksheet.cell(row=row_idx, column=col_idx)
if isinstance(value, str): if isinstance(value, str):
value = re.sub(ILLEGAL_CHARACTERS_RE, '', value) value = re.sub(ILLEGAL_CHARACTERS_RE, '', value)
if isinstance(value, datetime.datetime):
eastern = pytz.timezone(TIME_ZONE)
c = datetime.timezone(eastern._utcoffset)
value = value.astimezone(c)
cell.value = value cell.value = value
output = BytesIO() output = BytesIO()