diff --git a/ecc/app/graphrag/util.py b/ecc/app/graphrag/util.py index 2fad066..4f4b204 100644 --- a/ecc/app/graphrag/util.py +++ b/ecc/app/graphrag/util.py @@ -615,11 +615,20 @@ async def check_vertex_has_desc(conn, i: int): params={"iter": i}, ) except Exception as e: - logger.error(f"Check Vert Desc err:\n{e}") + logger.error( + f"Check Vert Desc err: {e!r}\n{traceback.format_exc()}" + ) + return False - res = resp[0]["all_have_desc"] - logger.info(res) + try: + res = resp[0]["all_have_desc"] + except (IndexError, KeyError, TypeError) as e: + logger.error( + f"Check Vert Desc unexpected response {resp!r}: {e!r}" + ) + return False + logger.info(res) return res async def check_embedding_rebuilt(conn, v_type: str): @@ -632,11 +641,20 @@ async def check_embedding_rebuilt(conn, v_type: str): } ) except Exception as e: - logger.error(f"Check embedding rebuilt err:\n{e}") + logger.error( + f"Check embedding rebuilt err: {e!r}\n{traceback.format_exc()}" + ) + return False - res = resp[0]["all_have_embedding"] - logger.info(resp) + try: + res = resp[0]["all_have_embedding"] + except (IndexError, KeyError, TypeError) as e: + logger.error( + f"Check embedding rebuilt unexpected response {resp!r}: {e!r}" + ) + return False + logger.info(resp) return res