diff --git a/cognee/api/v1/search/search_v2.py b/cognee/api/v1/search/search_v2.py
index 862b6ede0..4166fd3f3 100644
--- a/cognee/api/v1/search/search_v2.py
+++ b/cognee/api/v1/search/search_v2.py
@@ -67,7 +67,7 @@ async def specific_search(query_type: SearchType, query: str, user) -> list:
         SearchType.INSIGHTS: query_graph_connections,
         SearchType.CHUNKS: query_chunks,
         SearchType.COMPLETION: query_completion,
-        SearchType.GRAPH_COMPLETION: graph_query_completion
+        SearchType.GRAPH_COMPLETION: graph_query_completion,
     }
 
     search_task = search_tasks.get(query_type)
diff --git a/cognee/tasks/completion/graph_query_completion.py b/cognee/tasks/completion/graph_query_completion.py
index ffa11e9dd..d579aea4d 100644
--- a/cognee/tasks/completion/graph_query_completion.py
+++ b/cognee/tasks/completion/graph_query_completion.py
@@ -8,13 +8,14 @@ from cognee.modules.retrieval.brute_force_triplet_search import brute_force_trip
 def retrieved_edges_to_string(retrieved_edges):
     edge_strings = []
     for edge in retrieved_edges:
-        node1_string = edge.node1.attributes['text'] or edge.node1.attributes.get('name')
-        node2_string = edge.node2.attributes['text'] or edge.node2.attributes.get('name')
-        edge_string = edge.attributes['relationship_type']
+        node1_string = edge.node1.attributes["text"] or edge.node1.attributes.get("name")
+        node2_string = edge.node2.attributes["text"] or edge.node2.attributes.get("name")
+        edge_string = edge.attributes["relationship_type"]
         edge_str = f"{node1_string} -- {edge_string} -- {node2_string}"
         edge_strings.append(edge_str)
     return "\n---\n".join(edge_strings)
 
+
 async def graph_query_completion(query: str) -> list:
     """
     Parameters:
diff --git a/examples/python/dynamic_steps_example.py b/examples/python/dynamic_steps_example.py
index 04db3d7f2..11596a5e2 100644
--- a/examples/python/dynamic_steps_example.py
+++ b/examples/python/dynamic_steps_example.py
@@ -186,11 +186,12 @@ async def main(enable_steps):
     # Step 4: Query insights
     if enable_steps.get("retriever"):
         search_results = await cognee.search(
-            SearchType.GRAPH_COMPLETION, query_text='Who has experience in design tools?'
+            SearchType.GRAPH_COMPLETION, query_text="Who has experience in design tools?"
         )
         print(search_results)
 
-if __name__ == '__main__':
+
+if __name__ == "__main__":
     setup_logging(logging.INFO)
 
     rebuild_kg = True