merge migration

This commit is contained in:
jyong
2024-08-21 16:25:18 +08:00
parent e7762b731c
commit 067b956b2c
2 changed files with 46 additions and 11 deletions

View File

@ -59,9 +59,11 @@ def external_document_indexing_task(dataset_id: str, api_template_id: str, data_
# do http request
response = ExternalDatasetService.process_external_api(settings, headers, process_parameter, files)
if response.status_code != 200:
logging.info(click.style('Processed external dataset: {} failed, status code: {}'.format(dataset.id, response.status_code), fg='red'))
return
job_id = response.json().get('job_id')
if job_id:
# save job_id to dataset
dataset.job_id = job_id
db.session.commit()
end_at = time.perf_counter()
logging.info(