From 3cbcfc1425b69153167e817a29e6a55e79a54932 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Mon, 9 Dec 2024 15:12:20 +0100
Subject: [PATCH] fix: turns off raising duplicated edges unit test as we have
 these in our current codegraph generation

---
 .../retrieval/description_to_codepart_search.py     |  6 +++---
 .../tests/unit/modules/graph/cognee_graph_test.py   | 13 -------------
 2 files changed, 3 insertions(+), 16 deletions(-)

diff --git a/cognee/modules/retrieval/description_to_codepart_search.py b/cognee/modules/retrieval/description_to_codepart_search.py
index c3ad477fd..9a68e77fc 100644
--- a/cognee/modules/retrieval/description_to_codepart_search.py
+++ b/cognee/modules/retrieval/description_to_codepart_search.py
@@ -1,7 +1,7 @@
 import asyncio
 import logging
 
-from typing import Set
+from typing import Set, List
 from cognee.infrastructure.databases.graph import get_graph_engine
 from cognee.infrastructure.databases.vector import get_vector_engine
 from cognee.modules.graph.cognee_graph.CogneeGraph import CogneeGraph
@@ -10,7 +10,7 @@
 from cognee.shared.utils import send_telemetry
 
 
-async def code_description_to_code_part_search(query: str, user: User = None, top_k = 1) -> list:
+async def code_description_to_code_part_search(query: str, user: User = None, top_k = 2) -> list:
     if user is None:
         user = await get_default_user()
 
@@ -26,7 +26,7 @@ async def code_description_to_code_part(
     query: str,
     user: User,
     top_k: int
-) -> Set[str]:
+) -> List[str]:
     """
     Maps a code description query to relevant code parts using a CodeGraph pipeline.
 
diff --git a/cognee/tests/unit/modules/graph/cognee_graph_test.py b/cognee/tests/unit/modules/graph/cognee_graph_test.py
index 6f6165202..6888648c3 100644
--- a/cognee/tests/unit/modules/graph/cognee_graph_test.py
+++ b/cognee/tests/unit/modules/graph/cognee_graph_test.py
@@ -42,19 +42,6 @@ def test_add_edge_success(setup_graph):
     assert edge in node2.skeleton_edges
 
 
-def test_add_duplicate_edge(setup_graph):
-    """Test adding a duplicate edge raises an exception."""
-    graph = setup_graph
-    node1 = Node("node1")
-    node2 = Node("node2")
-    graph.add_node(node1)
-    graph.add_node(node2)
-    edge = Edge(node1, node2)
-    graph.add_edge(edge)
-    with pytest.raises(EntityAlreadyExistsError, match="Edge .* already exists in the graph."):
-        graph.add_edge(edge)
-
-
 def test_get_node_success(setup_graph):
     """Test retrieving an existing node."""
     graph = setup_graph