From b46421787d53bc1d1f54dcf6163ad0949fce914e Mon Sep 17 00:00:00 2001
From: Wenjie Zhang <wenjie.zhang@stu.jiangnan.edu.cn>
Date: Wed, 19 Mar 2025 01:21:51 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=20traceback?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/core/database.py  | 3 ++-
 src/core/graphbase.py | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/core/database.py b/src/core/database.py
index c7a74f71..ac08b111 100644
--- a/src/core/database.py
+++ b/src/core/database.py
@@ -1,6 +1,7 @@
 import os
 import json
 import time
+import traceback
 
 from src import config
 from src.utils import hashstr, logger
@@ -168,7 +169,7 @@ class DataBaseManager:
                 db.files[idx]["status"] = "done"
 
             except Exception as e:
-                logger.error(f"Failed to add documents to collection {db.metaname}, {e}")
+                logger.error(f"Failed to add documents to collection {db.metaname}, {e}, {traceback.format_exc()}")
                 idx = self.get_idx_by_fileid(db, file_id)
                 db.files[idx]["status"] = "failed"
 
diff --git a/src/core/graphbase.py b/src/core/graphbase.py
index 5ea803cf..15a162eb 100644
--- a/src/core/graphbase.py
+++ b/src/core/graphbase.py
@@ -1,6 +1,7 @@
 import os
 import json
 import warnings
+import traceback
 
 import torch
 from neo4j import GraphDatabase as GD
@@ -445,7 +446,7 @@ class GraphDatabase:
                     session.execute_write(self.set_embedding, node_name, embedding)
                     count += 1
                 except Exception as e:
-                    logger.error(f"为节点 '{node_name}' 添加嵌入向量失败: {e}")
+                    logger.error(f"为节点 '{node_name}' 添加嵌入向量失败: {e}, {traceback.format_exc()}")
 
         return count