From 3718c75cee7fee0612e81fe6114d973a9c8708ca Mon Sep 17 00:00:00 2001 From: carry <2641257231@qq.com> Date: Tue, 22 Apr 2025 00:14:16 +0800 Subject: [PATCH] =?UTF-8?q?feat(frontend):=20=E6=B7=BB=E5=8A=A0=E6=95=B0?= =?UTF-8?q?=E6=8D=AE=E9=9B=86=E7=94=9F=E6=88=90=E9=A1=B5=E9=9D=A2=E7=9A=84?= =?UTF-8?q?=E5=A4=84=E7=90=86=E8=BF=9B=E5=BA=A6=E6=98=BE=E7=A4=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 在处理文档片段时添加进度条，提升用户体验 - 优化代码格式，调整缩进和空行 --- frontend/dataset_generate_page.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/frontend/dataset_generate_page.py b/frontend/dataset_generate_page.py index 536b66f..d87899b 100644 --- a/frontend/dataset_generate_page.py +++ b/frontend/dataset_generate_page.py @@ -135,8 +135,6 @@ def dataset_generate_page(): prompt = prompt.partial(**variables_dict) - - dataset = Dataset( name=dataset_name, model_id=[api_provider.model_id], @@ -144,10 +142,12 @@ def dataset_generate_page(): dataset_items=[] ) - for document_slice in document_slice_list: + total_slices = len(document_slice_list) + for i, document_slice in enumerate(document_slice_list): + progress((i + 1) / total_slices, desc=f"处理文档片段 {i + 1}/{total_slices}") request = LLMRequest(api_provider=api_provider, - prompt=prompt.format(document_slice=document_slice), - format=generate_json_example(DatasetItem)) + prompt=prompt.format(document_slice=document_slice), + format=generate_json_example(DatasetItem)) call_openai_api(request, rounds) for resp in request.response: