Update logger message
This commit is contained in:
@@ -1318,6 +1318,18 @@ class PGGraphStorage(BaseGraphStorage):
|
|||||||
retry=retry_if_exception_type((PGGraphQueryException,)),
|
retry=retry_if_exception_type((PGGraphQueryException,)),
|
||||||
)
|
)
|
||||||
async def upsert_node(self, node_id: str, node_data: dict[str, str]) -> None:
|
async def upsert_node(self, node_id: str, node_data: dict[str, str]) -> None:
|
||||||
|
"""
|
||||||
|
Upsert a node in the Neo4j database.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
node_id: The unique identifier for the node (used as label)
|
||||||
|
node_data: Dictionary of node properties
|
||||||
|
"""
|
||||||
|
if "entity_id" not in node_data:
|
||||||
|
raise ValueError(
|
||||||
|
"PostgreSQL: node properties must contain an 'entity_id' field"
|
||||||
|
)
|
||||||
|
|
||||||
label = node_id.strip('"')
|
label = node_id.strip('"')
|
||||||
properties = self._format_properties(node_data)
|
properties = self._format_properties(node_data)
|
||||||
|
|
||||||
@@ -1334,8 +1346,8 @@ class PGGraphStorage(BaseGraphStorage):
|
|||||||
try:
|
try:
|
||||||
await self._query(query, readonly=False, upsert=True)
|
await self._query(query, readonly=False, upsert=True)
|
||||||
|
|
||||||
except Exception as e:
|
except Exception:
|
||||||
logger.error("POSTGRES, Error during upsert: {%s}", e)
|
logger.error("POSTGRES, upsert_node error on node_id: {node_id}")
|
||||||
raise
|
raise
|
||||||
|
|
||||||
@retry(
|
@retry(
|
||||||
@@ -1375,8 +1387,10 @@ class PGGraphStorage(BaseGraphStorage):
|
|||||||
try:
|
try:
|
||||||
await self._query(query, readonly=False, upsert=True)
|
await self._query(query, readonly=False, upsert=True)
|
||||||
|
|
||||||
except Exception as e:
|
except Exception:
|
||||||
logger.error("Error during edge upsert: {%s}", e)
|
logger.error(
|
||||||
|
"POSTGRES, upsert_edge error on edge: {source_node_id} - {target_node_id}"
|
||||||
|
)
|
||||||
raise
|
raise
|
||||||
|
|
||||||
async def _node2vec_embed(self):
|
async def _node2vec_embed(self):
|
||||||
|
Reference in New Issue
Block a user