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: