[changes] Modify the execution conditions of the task

api/app/repositories/neo4j/community_repository.py

@@ -21,6 +21,8 @@ from app.repositories.neo4j.cypher_queries import (
     UPDATE_COMMUNITY_METADATA,
     GET_INCOMPLETE_COMMUNITIES,
     GET_INCOMPLETE_COMMUNITIES_WITH_EMBEDDING,
+    CHECK_COMMUNITY_IS_COMPLETE,
+    CHECK_COMMUNITY_IS_COMPLETE_WITH_EMBEDDING,
 )
 
 logger = logging.getLogger(__name__)
@@ -187,6 +189,16 @@ class CommunityRepository:
             logger.error(f"get_incomplete_communities failed: {e}")
             return []
 
+    async def is_community_complete(self, community_id: str, end_user_id: str, check_embedding: bool = False) -> bool:
+        """检查单个社区节点的属性是否完整。"""
+        try:
+            query = CHECK_COMMUNITY_IS_COMPLETE_WITH_EMBEDDING if check_embedding else CHECK_COMMUNITY_IS_COMPLETE
+            result = await self.connector.execute_query(query, community_id=community_id, end_user_id=end_user_id)
+            return result[0]["is_complete"] if result else False
+        except Exception as e:
+            logger.error(f"is_community_complete failed: {e}")
+            return False
+
     async def update_community_metadata(
         self,
         community_id: str,

api/app/repositories/neo4j/cypher_queries.py

@@ -1204,6 +1204,25 @@ RETURN
     startNode(r) = e      AS r_from_e
 """
 
+CHECK_COMMUNITY_IS_COMPLETE = """
+MATCH (c:Community {community_id: $community_id, end_user_id: $end_user_id})
+RETURN (
+    c.name IS NOT NULL AND c.name <> '' AND
+    c.summary IS NOT NULL AND c.summary <> '' AND
+    c.core_entities IS NOT NULL
+) AS is_complete
+"""
+
+CHECK_COMMUNITY_IS_COMPLETE_WITH_EMBEDDING = """
+MATCH (c:Community {community_id: $community_id, end_user_id: $end_user_id})
+RETURN (
+    c.name IS NOT NULL AND c.name <> '' AND
+    c.summary IS NOT NULL AND c.summary <> '' AND
+    c.core_entities IS NOT NULL AND
+    c.summary_embedding IS NOT NULL
+) AS is_complete
+"""
+
 GET_INCOMPLETE_COMMUNITIES = """
 MATCH (c:Community {end_user_id: $end_user_id})
 WHERE c.name IS NULL OR c.summary IS NULL OR c.core_entities IS NULL
@@ -1213,8 +1232,9 @@ RETURN c.community_id AS community_id
 
 GET_INCOMPLETE_COMMUNITIES_WITH_EMBEDDING = """
 MATCH (c:Community {end_user_id: $end_user_id})
-WHERE c.name IS NULL OR c.summary IS NULL OR c.core_entities IS NULL
-   OR c.name = '' OR c.summary = ''
-   OR c.summary_embedding IS NULL
+WHERE c.name IS NULL OR c.name = ''
+   OR c.summary IS NULL OR c.summary = ''
+   OR c.core_entities IS NULL
+   OR (c.summary_embedding IS NULL AND c.summary IS NOT NULL AND c.summary <> '(empty)')
 RETURN c.community_id AS community_id
 """