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

- 在处理文档片段时添加进度条,提升用户体验
- 优化代码格式,调整缩进和空行
This commit is contained in:
carry 2025-04-22 00:14:16 +08:00
parent 905658073a
commit 3718c75cee

View File

@ -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: