carry/gzhu-biyesheji
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(frontend): 添加数据集生成页面的处理进度显示

Browse Source 
- 在处理文档片段时添加进度条，提升用户体验
- 优化代码格式，调整缩进和空行
This commit is contained in:
carry 2025-04-22 00:14:16 +08:00
parent 905658073a
commit 3718c75cee
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
frontend/dataset_generate_page.py
View File

@ -135,8 +135,6 @@ def dataset_generate_page():
prompt = prompt.partial(**variables_dict) prompt = prompt.partial(**variables_dict)
dataset = Dataset( dataset = Dataset(
name=dataset_name, name=dataset_name,
model_id=[api_provider.model_id], model_id=[api_provider.model_id],
@ -144,7 +142,9 @@ def dataset_generate_page():
dataset_items=[] 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, request = LLMRequest(api_provider=api_provider,
prompt=prompt.format(document_slice=document_slice), prompt=prompt.format(document_slice=document_slice),
format=generate_json_example(DatasetItem)) format=generate_json_example(DatasetItem))