jibsn commited on
Commit
832ed43
·
verified ·
1 Parent(s): c723b5d

Update utils/arxiv_utils.py

Browse files
Files changed (1) hide show
  1. utils/arxiv_utils.py +2 -1
utils/arxiv_utils.py CHANGED
@@ -333,11 +333,12 @@ async def process_arxiv_generate_subheadings(
333
  logger.info(f"review: {review_arxiv_df}")
334
  results, exceptions = await retry_operation(
335
  generate_subheadings, task,
336
- relevant_papers_df=review_arxiv_df, main_topic = task.direction,
337
  uuid=uuid, customer_name=customer_name, model_name=save_name,
338
  max_retries=max_retries, delay=delay,
339
  chat_func=chat_func
340
  )
 
341
  if results is None: # no valid result after max retries
342
  task.status_string[model_name] = exceptions # store exception strings in status
343
  await upload_task_json_to_minio(task, client)
 
333
  logger.info(f"review: {review_arxiv_df}")
334
  results, exceptions = await retry_operation(
335
  generate_subheadings, task,
336
+ relevant_papers_df=review_arxiv_df, main_topic = task.query,
337
  uuid=uuid, customer_name=customer_name, model_name=save_name,
338
  max_retries=max_retries, delay=delay,
339
  chat_func=chat_func
340
  )
341
+ logger.info(f"results:{results}")
342
  if results is None: # no valid result after max retries
343
  task.status_string[model_name] = exceptions # store exception strings in status
344
  await upload_task_json_to_minio(task, client)