浏览代码

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 2 个月前
父节点
当前提交
b486d72b8e
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 0 次插入14 次删除
  1. 0
    14
      api/controllers/console/datasets/datasets_document.py

+ 0
- 14
api/controllers/console/datasets/datasets_document.py 查看文件

@@ -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"]

正在加载...
取消
保存