Explorar el Código

Chore: remove dead var in `DocumentBatchIndexingEstimateApi` (#24497)

Co-authored-by: Yongtao Huang <99629139+hyongtao-db@users.noreply.github.com>
tags/1.8.0
Yongtao Huang hace 2 meses
padre
commit
b486d72b8e
No account linked to committer's email address
Se han modificado 1 ficheros con 0 adiciones y 14 borrados
  1. 0
    14
      api/controllers/console/datasets/datasets_document.py

+ 0
- 14
api/controllers/console/datasets/datasets_document.py Ver fichero

@@ -470,25 +470,11 @@ class DocumentBatchIndexingEstimateApi(DocumentResource):
return {"tokens": 0, "total_price": 0, "currency": "USD", "total_segments": 0, "preview": []}, 200
data_process_rule = documents[0].dataset_process_rule
data_process_rule_dict = data_process_rule.to_dict()
info_list = []
extract_settings = []
for document in documents:
if document.indexing_status in {"completed", "error"}:
raise DocumentAlreadyFinishedError()
data_source_info = document.data_source_info_dict
# format document files info
if data_source_info and "upload_file_id" in data_source_info:
file_id = data_source_info["upload_file_id"]
info_list.append(file_id)
# format document notion info
elif (
data_source_info and "notion_workspace_id" in data_source_info and "notion_page_id" in data_source_info
):
pages = []
page = {"page_id": data_source_info["notion_page_id"], "type": data_source_info["type"]}
pages.append(page)
notion_info = {"workspace_id": data_source_info["notion_workspace_id"], "pages": pages}
info_list.append(notion_info)

if document.data_source_type == "upload_file":
file_id = data_source_info["upload_file_id"]

Cargando…
Cancelar
Guardar