Jelajahi Sumber

格式化代码

xueyiming 1 hari lalu
induk
melakukan
71798246d9
1 mengubah file dengan 9 tambahan dan 8 penghapusan
  1. 9 8
      applications/utils/task/async_task.py

+ 9 - 8
applications/utils/task/async_task.py

@@ -59,17 +59,18 @@ async def handle_books():
                                 book_id, content_list
                             )
 
-                        # 创建文档 ID
-                        doc_id = f"doc-{uuid.uuid4()}"
-                        chunk_task = ChunkBooksTask(doc_id=doc_id, resource=resource)
-
-                        # 处理分片任务
-                        body = {"book_id": book_id}
-                        await chunk_task.deal(body)  # 异步执行分片任务
-
                 except Exception as e:
+                    await books_mapper.update_book_extract_status(book_id, 99)
                     continue  # 如果提取过程失败,跳过该书籍
 
+                # 创建文档 ID
+                doc_id = f"doc-{uuid.uuid4()}"
+                chunk_task = ChunkBooksTask(doc_id=doc_id, resource=resource)
+
+                # 处理分片任务
+                body = {"book_id": book_id}
+                await chunk_task.deal(body)  # 异步执行分片任务
+
     except Exception as e:
         # 捕获整体异常
         print(f"处理请求失败,错误: {e}")