From 08b07d77727c23184c756475f472375db29c43eb Mon Sep 17 00:00:00 2001 From: Shreyas Pimpalgaonkar Date: Tue, 22 Oct 2024 18:49:29 -0700 Subject: [PATCH] Update postgres.py --- py/core/providers/kg/postgres.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/py/core/providers/kg/postgres.py b/py/core/providers/kg/postgres.py index 8a10f59fc..d32a805d4 100644 --- a/py/core/providers/kg/postgres.py +++ b/py/core/providers/kg/postgres.py @@ -748,7 +748,7 @@ async def get_community_details( ) -> Tuple[int, List[Dict[str, Any]], List[Dict[str, Any]]]: QUERY = f""" - SELECT level FROM {self._get_table_name("community")} WHERE cluster = $1 AND collection_id = $2 + SELECT level FROM {self._get_table_name("community_info")} WHERE cluster = $1 AND collection_id = $2 LIMIT 1 """ level = ( @@ -770,7 +770,7 @@ async def get_community_details( QUERY = f""" WITH node_triple_ids AS ( SELECT node, triple_ids - FROM {self._get_table_name("community")} + FROM {self._get_table_name("community_info")} WHERE cluster = $1 AND collection_id = $2 ) SELECT DISTINCT @@ -871,7 +871,7 @@ async def delete_graph_for_collection( ) for query in DELETE_QUERIES: - if "community" in query or "entity_collection" in query: + if "community" in query or "collection_entity" in query: await self.execute_query(query, [collection_id]) else: await self.execute_query(query, [document_ids])