feat(frontend): 添加数据集生成页面的处理进度显示
- 在处理文档片段时添加进度条,提升用户体验 - 优化代码格式,调整缩进和空行
This commit is contained in:
parent
905658073a
commit
3718c75cee
@ -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:
|
||||
|
Loading…
x
Reference in New Issue
Block a user