From c66c43e71786aa37b4f93e3a8f362524d46c922d Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Tue, 26 Nov 2024 15:44:11 +0100
Subject: [PATCH] chore: places retrievers under modules directory

---
 cognee/{pipelines => modules/retrieval}/__init__.py           | 0
 .../retrieval}/brute_force_triplet_search.py                  | 0
 cognee/pipelines/retriever/__init__.py                        | 0
 examples/python/dynamic_steps_example.py                      | 4 ++--
 4 files changed, 2 insertions(+), 2 deletions(-)
 rename cognee/{pipelines => modules/retrieval}/__init__.py (100%)
 rename cognee/{pipelines/retriever => modules/retrieval}/brute_force_triplet_search.py (100%)
 delete mode 100644 cognee/pipelines/retriever/__init__.py

diff --git a/cognee/pipelines/__init__.py b/cognee/modules/retrieval/__init__.py
similarity index 100%
rename from cognee/pipelines/__init__.py
rename to cognee/modules/retrieval/__init__.py
diff --git a/cognee/pipelines/retriever/brute_force_triplet_search.py b/cognee/modules/retrieval/brute_force_triplet_search.py
similarity index 100%
rename from cognee/pipelines/retriever/brute_force_triplet_search.py
rename to cognee/modules/retrieval/brute_force_triplet_search.py
diff --git a/cognee/pipelines/retriever/__init__.py b/cognee/pipelines/retriever/__init__.py
deleted file mode 100644
index e69de29bb..000000000
diff --git a/examples/python/dynamic_steps_example.py b/examples/python/dynamic_steps_example.py
index 6b75310cf..68bbb7bce 100644
--- a/examples/python/dynamic_steps_example.py
+++ b/examples/python/dynamic_steps_example.py
@@ -1,6 +1,6 @@
 import cognee
 import asyncio
-from cognee.pipelines.retriever.brute_force_triplet_search import brute_force_triplet_search
+from cognee.modules.retrieval.brute_force_triplet_search import brute_force_triplet_search
 
 job_1 = """
 CV 1: Relevant
@@ -187,7 +187,7 @@ async def main(enable_steps):
 if __name__ == '__main__':
     # Flags to enable/disable steps
 
-    rebuild_kg = False
+    rebuild_kg = True
     retrieve = True
     steps_to_enable = {
         "prune_data": rebuild_kg,