feat: Add context evaluation to eval framework [COG-1366]

cognee/api/v1/cognify/routers/get_code_pipeline_router.py

@@ -4,7 +4,7 @@
 from fastapi.responses import JSONResponse
 from cognee.api.DTO import InDTO
 from cognee.api.v1.cognify.code_graph_pipeline import run_code_graph_pipeline
-from cognee.modules.retrieval import code_graph_retrieval
+from cognee.modules.retrieval.base_retriever import CodeRetriever
 from cognee.modules.storage.utils import JSONEncoder
 
 
@@ -43,7 +43,8 @@ async def code_pipeline_retrieve(payload: CodePipelineRetrievePayloadDTO):
                 else payload.full_input
             )
 
-            retrieved_files = await code_graph_retrieval(query)
+            retriever = CodeRetriever()
+            retrieved_files = await retriever.get_context(query)
 
             return json.dumps(retrieved_files, cls=JSONEncoder)
         except Exception as error:

cognee/modules/retrieval/__init__.py

@@ -1 +1 @@
-from cognee.modules.retrieval.utils.code_graph_retrieval import code_graph_retrieval
+from cognee.modules.retrieval.code_retriever import CodeRetriever

cognee/modules/retrieval/base_retriever.py

@@ -1,5 +1,5 @@
 from abc import ABC, abstractmethod
-from typing import Any, Optional
+from typing import Any, Optional, Callable
 
 
 class BaseRetriever(ABC):
@@ -14,3 +14,8 @@ async def get_context(self, query: str) -> Any:
     async def get_completion(self, query: str, context: Optional[Any] = None) -> Any:
         """Generates a response using the query and optional context."""
         pass
+
+    @classmethod
+    def as_search(cls) -> Callable:
+        """Creates a search function from the retriever class."""
+        return lambda query: cls().get_completion(query)