添加 traceback

This commit is contained in:
Wenjie Zhang 2025-03-19 01:21:51 +08:00
parent 5c59741848
commit b46421787d
2 changed files with 4 additions and 2 deletions

View File

@ -1,6 +1,7 @@
import os import os
import json import json
import time import time
import traceback
from src import config from src import config
from src.utils import hashstr, logger from src.utils import hashstr, logger
@ -168,7 +169,7 @@ class DataBaseManager:
db.files[idx]["status"] = "done" db.files[idx]["status"] = "done"
except Exception as e: 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) idx = self.get_idx_by_fileid(db, file_id)
db.files[idx]["status"] = "failed" db.files[idx]["status"] = "failed"

View File

@ -1,6 +1,7 @@
import os import os
import json import json
import warnings import warnings
import traceback
import torch import torch
from neo4j import GraphDatabase as GD from neo4j import GraphDatabase as GD
@ -445,7 +446,7 @@ class GraphDatabase:
session.execute_write(self.set_embedding, node_name, embedding) session.execute_write(self.set_embedding, node_name, embedding)
count += 1 count += 1
except Exception as e: except Exception as e:
logger.error(f"为节点 '{node_name}' 添加嵌入向量失败: {e}") logger.error(f"为节点 '{node_name}' 添加嵌入向量失败: {e}, {traceback.format_exc()}")
return count return count