From bac9c097079651856e93306c7b34826849ec2858 Mon Sep 17 00:00:00 2001 From: prestonrasmussen Date: Wed, 18 Sep 2024 12:24:40 -0400 Subject: [PATCH] rebase --- graphiti_core/graphiti.py | 20 +++++++++---------- .../utils/maintenance/community_operations.py | 2 +- 2 files changed, 11 insertions(+), 11 deletions(-) diff --git a/graphiti_core/graphiti.py b/graphiti_core/graphiti.py index 91cb94c..7cf9719 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 06fc69e..53fe612 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)