Update utils/arxiv_utils.py
Browse files- 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.
|
| 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)
|