- Remove useless _label_exists method

This commit is contained in:
yangdx
2025-03-07 16:43:18 +08:00
parent 17ccab8db2
commit 6e3b23069c

View File

@@ -164,23 +164,13 @@ class Neo4JStorage(BaseGraphStorage):
# Noe4J handles persistence automatically # Noe4J handles persistence automatically
pass pass
async def _label_exists(self, label: str) -> bool:
"""Check if a label exists in the Neo4j database."""
query = "CALL db.labels() YIELD label RETURN label"
try:
async with self._driver.session(database=self._DATABASE) as session:
result = await session.run(query)
labels = [record["label"] for record in await result.data()]
return label in labels
except Exception as e:
logger.error(f"Error checking label existence: {e}")
return False
async def _ensure_label(self, label: str) -> str: async def _ensure_label(self, label: str) -> str:
"""Ensure a label exists by validating it.""" """Ensure a label is valid
Args:
label: The label to validate
"""
clean_label = label.strip('"') clean_label = label.strip('"')
if not await self._label_exists(clean_label):
logger.warning(f"Label '{clean_label}' does not exist in Neo4j")
return clean_label return clean_label
async def has_node(self, node_id: str) -> bool: async def has_node(self, node_id: str) -> bool:
@@ -290,7 +280,7 @@ class Neo4JStorage(BaseGraphStorage):
if record: if record:
try: try:
result = dict(record["edge_properties"]) result = dict(record["edge_properties"])
logger.info(f"Result: {result}") logger.debug(f"Result: {result}")
# Ensure required keys exist with defaults # Ensure required keys exist with defaults
required_keys = { required_keys = {
"weight": 0.0, "weight": 0.0,