diff --git a/cognee/modules/retrieval/code_graph_retrieval.py b/cognee/modules/retrieval/code_graph_retrieval.py index 0647a7ce3..8328aaf83 100644 --- a/cognee/modules/retrieval/code_graph_retrieval.py +++ b/cognee/modules/retrieval/code_graph_retrieval.py @@ -33,10 +33,10 @@ async def code_graph_retrieval(query: str) -> dict[str, str]: ] return [ - { - "name": file_path, - "description": file_path, - "content": source_code, - } - for file_path, source_code in retrieved_files.items() - ] + { + "name": file_path, + "description": file_path, + "content": source_code, + } + for file_path, source_code in retrieved_files.items() + ]