From 13551dacf1c9fa1d2be2e7aa92978c36341cdbb3 Mon Sep 17 00:00:00 2001
From: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
Date: Wed, 13 Mar 2024 16:41:43 +0100
Subject: [PATCH] Add docs

---
 cognee/api/v1/search/search.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/cognee/api/v1/search/search.py b/cognee/api/v1/search/search.py
index 5b0719714..7591f10f7 100644
--- a/cognee/api/v1/search/search.py
+++ b/cognee/api/v1/search/search.py
@@ -17,7 +17,7 @@ class SearchType(Enum):
     NEIGHBOR = auto()
 
 
-async def complex_search(graph, query_params: Dict[SearchType, Dict[str, Any]]) -> List:
+async def search(graph, query_params: Dict[SearchType, Dict[str, Any]]) -> List:
     search_functions: Dict[SearchType, Callable] = {
         SearchType.ADJACENT: search_adjacent,
         SearchType.SIMILARITY: search_similarity,
@@ -61,7 +61,7 @@ if __name__ == "__main__":
 
         await graph_client.load_graph_from_file()
         graph = graph_client.graph
-        results = await complex_search(graph, query_params)
+        results = await search(graph, query_params)
         print(results)
 
     asyncio.run(main())