Unverified Commit e0532151 authored by Jyong's avatar Jyong Committed by GitHub

fix document estimate parameter (#1019)

Co-authored-by: 's avatarjyong <jyong@dify.ai>
parent 13482b0f
...@@ -416,7 +416,8 @@ class DocumentIndexingEstimateApi(DocumentResource): ...@@ -416,7 +416,8 @@ class DocumentIndexingEstimateApi(DocumentResource):
try: try:
response = indexing_runner.file_indexing_estimate(current_user.current_tenant_id, [file], response = indexing_runner.file_indexing_estimate(current_user.current_tenant_id, [file],
data_process_rule_dict, None, dataset_id) data_process_rule_dict, None,
'English', dataset_id)
except LLMBadRequestError: except LLMBadRequestError:
raise ProviderNotInitializeError( raise ProviderNotInitializeError(
f"No Embedding Model available. Please configure a valid provider " f"No Embedding Model available. Please configure a valid provider "
...@@ -485,7 +486,8 @@ class DocumentBatchIndexingEstimateApi(DocumentResource): ...@@ -485,7 +486,8 @@ class DocumentBatchIndexingEstimateApi(DocumentResource):
indexing_runner = IndexingRunner() indexing_runner = IndexingRunner()
try: try:
response = indexing_runner.file_indexing_estimate(current_user.current_tenant_id, file_details, response = indexing_runner.file_indexing_estimate(current_user.current_tenant_id, file_details,
data_process_rule_dict, None, dataset_id) data_process_rule_dict, None,
'English', dataset_id)
except LLMBadRequestError: except LLMBadRequestError:
raise ProviderNotInitializeError( raise ProviderNotInitializeError(
f"No Embedding Model available. Please configure a valid provider " f"No Embedding Model available. Please configure a valid provider "
...@@ -499,7 +501,7 @@ class DocumentBatchIndexingEstimateApi(DocumentResource): ...@@ -499,7 +501,7 @@ class DocumentBatchIndexingEstimateApi(DocumentResource):
response = indexing_runner.notion_indexing_estimate(current_user.current_tenant_id, response = indexing_runner.notion_indexing_estimate(current_user.current_tenant_id,
info_list, info_list,
data_process_rule_dict, data_process_rule_dict,
None, dataset_id) None, 'English', dataset_id)
except LLMBadRequestError: except LLMBadRequestError:
raise ProviderNotInitializeError( raise ProviderNotInitializeError(
f"No Embedding Model available. Please configure a valid provider " f"No Embedding Model available. Please configure a valid provider "
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment