diff --git a/graphiti_core/graphiti.py b/graphiti_core/graphiti.py index 91cb94cb..7cf9719a 100644 --- a/graphiti_core/graphiti.py +++ b/graphiti_core/graphiti.py @@ -619,22 +619,22 @@ async def search( return edges async def _search( - self, - query: str, - config: SearchConfig, - group_ids: list[str | None] | None = None, - center_node_uuid: str | None = None, + self, + query: str, + config: SearchConfig, + group_ids: list[str | None] | None = None, + center_node_uuid: str | None = None, ) -> SearchResults: return await search( self.driver, self.llm_client.get_embedder(), query, group_ids, config, center_node_uuid ) async def get_nodes_by_query( - self, - query: str, - center_node_uuid: str | None = None, - group_ids: list[str | None] | None = None, - limit: int = DEFAULT_SEARCH_LIMIT, + self, + query: str, + center_node_uuid: str | None = None, + group_ids: list[str | None] | None = None, + limit: int = DEFAULT_SEARCH_LIMIT, ) -> list[EntityNode]: """ Retrieve nodes from the graph database based on a text query. diff --git a/graphiti_core/utils/maintenance/community_operations.py b/graphiti_core/utils/maintenance/community_operations.py index 06fc69e5..53fe6122 100644 --- a/graphiti_core/utils/maintenance/community_operations.py +++ b/graphiti_core/utils/maintenance/community_operations.py @@ -127,7 +127,7 @@ async def build_community( async def build_communities( - driver: AsyncDriver, llm_client: LLMClient + driver: AsyncDriver, llm_client: LLMClient ) -> tuple[list[CommunityNode], list[CommunityEdge]]: projection = await build_community_projection(driver) community_clusters = await get_community_clusters(driver, projection)