rebase

getzep · Sep 18, 2024 · bac9c09 · bac9c09
1 parent 5316bc4
commit bac9c09
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 11 deletions.
diff --git 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
@@ -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)