From 59105a4511ed2f134797719fadd3de1155e5aa52 Mon Sep 17 00:00:00 2001 From: xiaohan Date: Thu, 15 Aug 2024 14:15:27 +0800 Subject: [PATCH] =?UTF-8?q?=E5=8A=A0=E5=85=A5=E5=BC=82=E5=B8=B8=E5=A4=84?= =?UTF-8?q?=E7=90=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- knowledge_graph_maker/graph_maker.py | 21 ++++++++++++--------- 1 file changed, 12 insertions(+), 9 deletions(-) diff --git a/knowledge_graph_maker/graph_maker.py b/knowledge_graph_maker/graph_maker.py index 6c58474..7383239 100644 --- a/knowledge_graph_maker/graph_maker.py +++ b/knowledge_graph_maker/graph_maker.py @@ -153,13 +153,16 @@ def from_documents( graph: List[Edge] = [] for index, doc in enumerate(docs): ## order defines the chronology or the order in which the documents should in interpretted. - order = getattr(doc, order_attribute) if order_attribute else index - green_logger.info(f"Document: {index+1}") - subgraph = self.from_document(doc, order) - graph = [*graph, *subgraph] - if delay_s_between > 0: - green_logger.info( - f"Waiting for {delay_s_between}s before the next request ... " - ) - time.sleep(delay_s_between) + try: + order = getattr(doc, order_attribute) if order_attribute else index + green_logger.info(f"Document: {index + 1}") + subgraph = self.from_document(doc, order) + graph = [*graph, *subgraph] + if delay_s_between > 0: + green_logger.info( + f"Waiting for {delay_s_between}s before the next request ... " + ) + time.sleep(delay_s_between) + except Exception: + green_logger.error(f"error:{e},doc:{doc}") return graph