From 231e16e656dd49b0b8bb821f8543bebfc7a0d859 Mon Sep 17 00:00:00 2001
From: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
Date: Wed, 13 Mar 2024 17:52:42 +0100
Subject: [PATCH] Add docs

---
 cognee/__init__.py             | 2 +-
 cognee/api/v1/search/search.py | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/cognee/__init__.py b/cognee/__init__.py
index 1720f6f8e..e1533b054 100644
--- a/cognee/__init__.py
+++ b/cognee/__init__.py
@@ -1,4 +1,4 @@
 from .api.v1.add.add import add
 from .api.v1.cognify.cognify import cognify
 from .api.v1.list_datasets.list_datasets import list_datasets
-from .api.v1.search import search
+from .api.v1.search.search import search
diff --git a/cognee/api/v1/search/search.py b/cognee/api/v1/search/search.py
index 7591f10f7..f9d787b94 100644
--- a/cognee/api/v1/search/search.py
+++ b/cognee/api/v1/search/search.py
@@ -8,7 +8,7 @@ from cognee.modules.search.vector.search_similarity import search_similarity
 from cognee.modules.search.graph.search_categories import search_categories
 from cognee.modules.search.graph.search_neighbour import search_neighbour
 from cognee.shared.data_models import GraphDBType
-
+import asyncio
 
 class SearchType(Enum):
     ADJACENT = auto()
@@ -48,7 +48,7 @@ async def search(graph, query_params: Dict[SearchType, Dict[str, Any]]) -> List:
     return results
 
 if __name__ == "__main__":
-    import asyncio
+