cognee 0.3.1__py3-none-any.whl → 0.3.3__py3-none-any.whl

cognee/api/v1/notebooks/routers/get_notebooks_router.py

@@ -31,7 +31,8 @@ def get_notebooks_router():
 
     @router.get("")
     async def get_notebooks_endpoint(user: User = Depends(get_authenticated_user)):
-        return await get_notebooks(user.id)
+        async with get_async_session() as session:
+            return await get_notebooks(user.id, session)
 
     @router.post("")
     async def create_notebook_endpoint(

cognee/api/v1/search/routers/get_search_router.py

@@ -1,12 +1,12 @@
 from uuid import UUID
-from typing import Optional
+from typing import Optional, Union, List, Any
 from datetime import datetime
 from pydantic import Field
 from fastapi import Depends, APIRouter
 from fastapi.responses import JSONResponse
 from fastapi.encoders import jsonable_encoder
 
-from cognee.modules.search.types import SearchType
+from cognee.modules.search.types import SearchType, SearchResult, CombinedSearchResult
 from cognee.api.DTO import InDTO, OutDTO
 from cognee.modules.users.exceptions.exceptions import PermissionDeniedError
 from cognee.modules.users.models import User
@@ -73,7 +73,7 @@ def get_search_router() -> APIRouter:
         except Exception as error:
             return JSONResponse(status_code=500, content={"error": str(error)})
 
-    @router.post("", response_model=list)
+    @router.post("", response_model=Union[List[SearchResult], CombinedSearchResult, List])
     async def search(payload: SearchPayloadDTO, user: User = Depends(get_authenticated_user)):
         """
         Search for nodes in the graph database.

cognee/api/v1/ui/ui.py

@@ -7,7 +7,7 @@ import webbrowser
 import zipfile
 import requests
 from pathlib import Path
-from typing import Optional, Tuple
+from typing import Callable, Optional, Tuple
 import tempfile
 import shutil
 
@@ -326,38 +326,93 @@ def prompt_user_for_download() -> bool:
 
 
 def start_ui(
+    pid_callback: Callable[[int], None],
     host: str = "localhost",
     port: int = 3000,
     open_browser: bool = True,
     auto_download: bool = False,
+    start_backend: bool = False,
+    backend_host: str = "localhost",
+    backend_port: int = 8000,
 ) -> Optional[subprocess.Popen]:
     """
-    Start the cognee frontend UI server.
+    Start the cognee frontend UI server, optionally with the backend API server.
 
     This function will:
-    1. Find the cognee-frontend directory (development) or download it (pip install)
-    2. Check if Node.js and npm are available (for development mode)
-    3. Install dependencies if needed (development mode)
-    4. Start the appropriate server
-    5. Optionally open the browser
+    1. Optionally start the cognee backend API server
+    2. Find the cognee-frontend directory (development) or download it (pip install)
+    3. Check if Node.js and npm are available (for development mode)
+    4. Install dependencies if needed (development mode)
+    5. Start the frontend server
+    6. Optionally open the browser
 
     Args:
-        host: Host to bind the server to (default: localhost)
-        port: Port to run the server on (default: 3000)
+        pid_callback: Callback to notify with PID of each spawned process
+        host: Host to bind the frontend server to (default: localhost)
+        port: Port to run the frontend server on (default: 3000)
         open_browser: Whether to open the browser automatically (default: True)
         auto_download: If True, download frontend without prompting (default: False)
+        start_backend: If True, also start the cognee API backend server (default: False)
+        backend_host: Host to bind the backend server to (default: localhost)
+        backend_port: Port to run the backend server on (default: 8000)
 
     Returns:
-        subprocess.Popen object representing the running server, or None if failed
+        subprocess.Popen object representing the running frontend server, or None if failed
+        Note: If backend is started, it runs in a separate process that will be cleaned up
+        when the frontend process is terminated.
 
     Example:
         >>> import cognee
+        >>> # Start just the frontend
         >>> server = cognee.start_ui()
+        >>>
+        >>> # Start both frontend and backend
+        >>> server = cognee.start_ui(start_backend=True)
         >>> # UI will be available at http://localhost:3000
-        >>> # To stop the server later:
+        >>> # API will be available at http://localhost:8000
+        >>> # To stop both servers later:
         >>> server.terminate()
     """
     logger.info("Starting cognee UI...")
+    backend_process = None
+
+    # Start backend server if requested
+    if start_backend:
+        logger.info("Starting cognee backend API server...")
+        try:
+            import sys
+
+            backend_process = subprocess.Popen(
+                [
+                    sys.executable,
+                    "-m",
+                    "uvicorn",
+                    "cognee.api.client:app",
+                    "--host",
+                    backend_host,
+                    "--port",
+                    str(backend_port),
+                ],
+                # Inherit stdout/stderr from parent process to show logs
+                stdout=None,
+                stderr=None,
+                preexec_fn=os.setsid if hasattr(os, "setsid") else None,
+            )
+
+            pid_callback(backend_process.pid)
+
+            # Give the backend a moment to start
+            time.sleep(2)
+
+            if backend_process.poll() is not None:
+                logger.error("Backend server failed to start - process exited early")
+                return None
+
+            logger.info(f"✓ Backend API started at http://{backend_host}:{backend_port}")
+
+        except Exception as e:
+            logger.error(f"Failed to start backend server: {str(e)}")
+            return None
 
     # Find frontend directory
     frontend_path = find_frontend_path()
@@ -406,7 +461,7 @@ def start_ui(
     logger.info("This may take a moment to compile and start...")
 
     try:
-        # Use process group to ensure all child processes get terminated together
+        # Create frontend in its own process group for clean termination
         process = subprocess.Popen(
             ["npm", "run", "dev"],
             cwd=frontend_path,
@@ -414,11 +469,11 @@ def start_ui(
             stdout=subprocess.PIPE,
             stderr=subprocess.PIPE,
             text=True,
-            preexec_fn=os.setsid
-            if hasattr(os, "setsid")
-            else None,  # Create new process group on Unix
+            preexec_fn=os.setsid if hasattr(os, "setsid") else None,
         )
 
+        pid_callback(process.pid)
+
         # Give it a moment to start up
         time.sleep(3)
 
@@ -447,16 +502,32 @@ def start_ui(
         logger.info(f"✓ Open your browser to: http://{host}:{port}")
         logger.info("✓ The UI will be available once Next.js finishes compiling")
 
+        # Store backend process reference in the frontend process for cleanup
+        if backend_process:
+            process._cognee_backend_process = backend_process
+
         return process
 
     except Exception as e:
         logger.error(f"Failed to start frontend server: {str(e)}")
+        # Clean up backend process if it was started
+        if backend_process:
+            logger.info("Cleaning up backend process due to frontend failure...")
+            try:
+                backend_process.terminate()
+                backend_process.wait(timeout=5)
+            except (subprocess.TimeoutExpired, OSError, ProcessLookupError):
+                try:
+                    backend_process.kill()
+                    backend_process.wait()
+                except (OSError, ProcessLookupError):
+                    pass
         return None
 
 
 def stop_ui(process: subprocess.Popen) -> bool:
     """
-    Stop a running UI server process and all its children.
+    Stop a running UI server process and backend process (if started), along with all their children.
 
     Args:
         process: The subprocess.Popen object returned by start_ui()
@@ -467,7 +538,29 @@ def stop_ui(process: subprocess.Popen) -> bool:
     if not process:
         return False
 
+    success = True
+
     try:
+        # First, stop the backend process if it exists
+        backend_process = getattr(process, "_cognee_backend_process", None)
+        if backend_process:
+            logger.info("Stopping backend server...")
+            try:
+                backend_process.terminate()
+                try:
+                    backend_process.wait(timeout=5)
+                    logger.info("Backend server stopped gracefully")
+                except subprocess.TimeoutExpired:
+                    logger.warning("Backend didn't terminate gracefully, forcing kill")
+                    backend_process.kill()
+                    backend_process.wait()
+                    logger.info("Backend server stopped")
+            except Exception as e:
+                logger.error(f"Error stopping backend server: {str(e)}")
+                success = False
+
+        # Now stop the frontend process
+        logger.info("Stopping frontend server...")
         # Try to terminate the process group (includes child processes like Next.js)
         if hasattr(os, "killpg"):
             try:
@@ -484,9 +577,9 @@ def stop_ui(process: subprocess.Popen) -> bool:
 
         try:
             process.wait(timeout=10)
-            logger.info("UI server stopped gracefully")
+            logger.info("Frontend server stopped gracefully")
         except subprocess.TimeoutExpired:
-            logger.warning("Process didn't terminate gracefully, forcing kill")
+            logger.warning("Frontend didn't terminate gracefully, forcing kill")
 
             # Force kill the process group
             if hasattr(os, "killpg"):
@@ -502,11 +595,13 @@ def stop_ui(process: subprocess.Popen) -> bool:
 
             process.wait()
 
-        logger.info("UI server stopped")
-        return True
+        if success:
+            logger.info("UI servers stopped successfully")
+
+        return success
 
     except Exception as e:
-        logger.error(f"Error stopping UI server: {str(e)}")
+        logger.error(f"Error stopping UI servers: {str(e)}")
         return False
 
 

cognee/cli/_cognee.py

@@ -174,30 +174,23 @@ def main() -> int:
 
     # Handle UI flag
     if hasattr(args, "start_ui") and args.start_ui:
-        server_process = None
+        spawned_pids = []
 
         def signal_handler(signum, frame):
             """Handle Ctrl+C and other termination signals"""
-            nonlocal server_process
+            nonlocal spawned_pids
             fmt.echo("\nShutting down UI server...")
-            if server_process:
+
+            for pid in spawned_pids:
                 try:
-                    # Try graceful termination first
-                    server_process.terminate()
-                    try:
-                        server_process.wait(timeout=5)
-                        fmt.success("UI server stopped gracefully.")
-                    except subprocess.TimeoutExpired:
-                        # If graceful termination fails, force kill
-                        fmt.echo("Force stopping UI server...")
-                        server_process.kill()
-                        server_process.wait()
-                        fmt.success("UI server stopped.")
-                except Exception as e:
-                    fmt.warning(f"Error stopping server: {e}")
+                    pgid = os.getpgid(pid)
+                    os.killpg(pgid, signal.SIGTERM)
+                    fmt.success(f"✓ Process group {pgid} (PID {pid}) terminated.")
+                except (OSError, ProcessLookupError) as e:
+                    fmt.warning(f"Could not terminate process {pid}: {e}")
+
             sys.exit(0)
 
-        # Set up signal handlers
         signal.signal(signal.SIGINT, signal_handler)  # Ctrl+C
         signal.signal(signal.SIGTERM, signal_handler)  # Termination request
 
@@ -205,11 +198,25 @@ def main() -> int:
             from cognee import start_ui
 
             fmt.echo("Starting cognee UI...")
-            server_process = start_ui(host="localhost", port=3000, open_browser=True)
+
+            # Callback to capture PIDs of all spawned processes
+            def pid_callback(pid):
+                nonlocal spawned_pids
+                spawned_pids.append(pid)
+
+            server_process = start_ui(
+                host="localhost",
+                port=3000,
+                open_browser=True,
+                start_backend=True,
+                auto_download=True,
+                pid_callback=pid_callback,
+            )
 
             if server_process:
                 fmt.success("UI server started successfully!")
                 fmt.echo("The interface is available at: http://localhost:3000")
+                fmt.echo("The API backend is available at: http://localhost:8000")
                 fmt.note("Press Ctrl+C to stop the server...")
 
                 try:
@@ -225,10 +232,12 @@ def main() -> int:
                 return 0
             else:
                 fmt.error("Failed to start UI server. Check the logs above for details.")
+                signal_handler(signal.SIGTERM, None)
                 return 1
 
         except Exception as ex:
             fmt.error(f"Error starting UI: {str(ex)}")
+            signal_handler(signal.SIGTERM, None)
             if debug.is_debug_enabled():
                 raise ex
             return 1

cognee/infrastructure/databases/relational/sqlalchemy/SqlAlchemyAdapter.py

@@ -58,8 +58,8 @@ class SQLAlchemyAdapter:
         else:
             self.engine = create_async_engine(
                 connection_string,
-                pool_size=5,
-                max_overflow=10,
+                pool_size=20,
+                max_overflow=20,
                 pool_recycle=280,
                 pool_pre_ping=True,
                 pool_timeout=280,

cognee/infrastructure/databases/vector/config.py

@@ -39,7 +39,7 @@ class VectorConfig(BaseSettings):
             values.vector_db_url = ensure_absolute_path(
                 values.vector_db_url,
             )
-        else:
+        elif not values.vector_db_url:
             # Default path
             databases_directory_path = os.path.join(base_config.system_root_directory, "databases")
             values.vector_db_url = os.path.join(databases_directory_path, "cognee.lancedb")

cognee/infrastructure/utils/run_async.py

@@ -1,13 +1,18 @@
 import asyncio
 from functools import partial
+import inspect
 
 
 async def run_async(func, *args, loop=None, executor=None, **kwargs):
     if loop is None:
         try:
-            running_loop = asyncio.get_running_loop()
+            loop = asyncio.get_running_loop()
         except RuntimeError:
-            running_loop = asyncio.get_event_loop()
+            loop = asyncio.get_event_loop()
 
-    pfunc = partial(func, *args, **kwargs)
-    return await running_loop.run_in_executor(executor, pfunc)
+    if "loop" in inspect.signature(func).parameters:
+        pfunc = partial(func, *args, loop=loop, **kwargs)
+    else:
+        pfunc = partial(func, *args, **kwargs)
+
+    return await loop.run_in_executor(executor, pfunc)

cognee/infrastructure/utils/run_sync.py

@@ -2,16 +2,17 @@ import asyncio
 import threading
 
 
-def run_sync(coro, timeout=None):
+def run_sync(coro, running_loop=None, timeout=None):
     result = None
     exception = None
 
     def runner():
-        nonlocal result, exception
+        nonlocal result, exception, running_loop
 
         try:
             try:
-                running_loop = asyncio.get_running_loop()
+                if not running_loop:
+                    running_loop = asyncio.get_running_loop()
 
                 result = asyncio.run_coroutine_threadsafe(coro, running_loop).result(timeout)
             except RuntimeError:

cognee/modules/notebooks/methods/create_tutorial_notebook.py

@@ -0,0 +1,92 @@
+
+from uuid import UUID, uuid4
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from ..models import NotebookCell
+from .create_notebook import create_notebook
+
+
+async def create_tutorial_notebook(user_id: UUID, session: AsyncSession):
+    await create_notebook(
+        user_id=user_id,
+        notebook_name="Welcome to cognee 🧠",
+        cells=[
+            NotebookCell(
+                id=uuid4(),
+                name="Welcome",
+                content="Cognee is your toolkit for turning text into a structured knowledge graph, optionally enhanced by ontologies, and then querying it with advanced retrieval techniques. This notebook will guide you through a simple example.",
+                type="markdown",
+            ),
+            NotebookCell(
+                id=uuid4(),
+                name="Example",
+                content="",
+                type="markdown",
+            ),
+        ],
+        deletable=False,
+        session=session,
+    )
+
+cell_content = [
+"""
+# Using Cognee with Python Development Data
+
+Unite authoritative Python practice (Guido van Rossum's own contributions!), normative guidance (Zen/PEP 8), and your lived context (rules + conversations) into one *AI memory* that produces answers that are relevant, explainable, and consistent.
+""",
+
+"""
+## What You'll Learn
+
+In this comprehensive tutorial, you'll discover how to transform scattered development data into an intelligent knowledge system that enhances your coding workflow. By the end, you'll have:
+- Connected disparate data sources (Guido's CPython contributions, mypy development, PEP discussions, your Python projects) into a unified AI memory graph
+- Built an memory layer that understands Python design philosophy, best practice coding patterns, and your preferences and experience
+- Learn how to use intelligent search capabilities that combine the diverse context
+- Integrated everything with your coding environment through MCP (Model Context Protocol)
+
+This tutorial demonstrates the power of **knowledge graphs** and **retrieval-augmented generation (RAG)** for software development, showing you how to build systems that learn from Python's creator and improve your own Python development.
+""",
+
+"""
+## Cognee and its core operations
+
+Before we dive in, let's understand the core Cognee operations we'll be working with:
+- `cognee.add()` - Ingests raw data (files, text, APIs) into the system
+- `cognee.cognify()` - Processes and structures data into a knowledge graph using AI
+- `cognee.search()` - Queries the knowledge graph with natural language or Cypher
+- `cognee.memify()` - Cognee's \"secret sauce\" that infers implicit connections and rules from your data
+""",
+
+"""
+## Data used in this tutorial
+
+Cognee can ingest many types of sources. In this tutorial, we use a small, concrete set of files that cover different perspectives:
+- `guido_contributions.json` — Authoritative exemplars. Real PRs and commits from Guido van Rossum (mypy, CPython). These show how Python's creator solved problems and provide concrete anchors for patterns.
+- `pep_style_guide.md` — Norms. Encodes community style and typing conventions (PEP 8 and related). Ensures that search results and inferred rules align with widely accepted standards.
+- `zen_principles.md` — Philosophy. The Zen of Python. Grounds design trade-offs (simplicity, explicitness, readability) beyond syntax or mechanics.
+- `my_developer_rules.md` — Local constraints. Your house rules, conventions, and project-specific requirements (scope, privacy, Spec.md). Keeps recommendations relevant to your actual workflow.
+- `copilot_conversations.json` — Personal history. Transcripts of real assistant conversations, including your questions, code snippets, and discussion topics. Captures "how you code" and connects it to "how Guido codes."
+""",
+
+"""
+# Preliminaries
+
+To strike the balanace between speed, cost, anc quality, we recommend using OpenAI's `4o-mini` model; make sure your `.env` file contains this line:
+`
+LLM_MODEL="gpt-4o-mini"
+`
+""",
+
+"""
+import cognee
+
+result = await cognee.add(
+    "file://data/guido_contributions.json",
+    node_set=["guido_data"]
+)
+
+await cognee.cognify(temporal_cognify=True)
+
+results = await cognee.search("Show me commits")
+"""
+]

cognee/modules/notebooks/methods/get_notebook.py

@@ -1,6 +1,6 @@
 from uuid import UUID
 from typing import Optional
-from sqlalchemy import select
+from sqlalchemy import and_, select
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from cognee.infrastructure.databases.relational import with_async_session
@@ -15,7 +15,7 @@ async def get_notebook(
     session: AsyncSession,
 ) -> Optional[Notebook]:
     result = await session.execute(
-        select(Notebook).where(Notebook.owner_id == user_id and Notebook.id == notebook_id)
+        select(Notebook).where(and_(Notebook.owner_id == user_id, Notebook.id == notebook_id))
     )
 
     return result.scalar()

cognee/modules/notebooks/methods/update_notebook.py

@@ -1,4 +1,3 @@
-from typing import Callable, AsyncContextManager
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from cognee.infrastructure.databases.relational import with_async_session

cognee/modules/notebooks/operations/run_in_local_sandbox.py

@@ -5,16 +5,18 @@ import traceback
 
 def wrap_in_async_handler(user_code: str) -> str:
     return (
-        "from cognee.infrastructure.utils.run_sync import run_sync\n\n"
-        "async def __user_main__():\n"
+        "import asyncio\n"
+        + "asyncio.set_event_loop(running_loop)\n\n"
+        + "from cognee.infrastructure.utils.run_sync import run_sync\n\n"
+        + "async def __user_main__():\n"
         + "\n".join("    " + line for line in user_code.strip().split("\n"))
         + "\n"
-        "    globals().update(locals())\n\n"
-        "run_sync(__user_main__())\n"
+        + "    globals().update(locals())\n\n"
+        + "run_sync(__user_main__(), running_loop)\n"
     )
 
 
-def run_in_local_sandbox(code, environment=None):
+def run_in_local_sandbox(code, environment=None, loop=None):
     environment = environment or {}
     code = wrap_in_async_handler(code.replace("\xa0", "\n"))
 
@@ -31,6 +33,7 @@ def run_in_local_sandbox(code, environment=None):
         printOutput.append(output)
 
     environment["print"] = customPrintFunction
+    environment["running_loop"] = loop
 
     try:
         exec(code, environment)

cognee/modules/retrieval/graph_completion_context_extension_retriever.py

@@ -48,7 +48,7 @@ class GraphCompletionContextExtensionRetriever(GraphCompletionRetriever):
         query: str,
         context: Optional[List[Edge]] = None,
         context_extension_rounds=4,
-    ) -> str:
+    ) -> List[str]:
         """
         Extends the context for a given query by retrieving related triplets and generating new
         completions based on them.
@@ -128,4 +128,4 @@ class GraphCompletionContextExtensionRetriever(GraphCompletionRetriever):
                 question=query, answer=completion, context=context_text, triplets=triplets
             )
 
-        return completion
+        return [completion]

cognee/modules/retrieval/graph_completion_cot_retriever.py

@@ -58,7 +58,7 @@ class GraphCompletionCotRetriever(GraphCompletionRetriever):
         query: str,
         context: Optional[List[Edge]] = None,
         max_iter=4,
-    ) -> str:
+    ) -> List[str]:
         """
         Generate completion responses based on a user query and contextual information.
 
@@ -138,4 +138,4 @@ class GraphCompletionCotRetriever(GraphCompletionRetriever):
                 question=query, answer=completion, context=context_text, triplets=triplets
             )
 
-        return completion
+        return [completion]

cognee/modules/retrieval/graph_completion_retriever.py

@@ -135,7 +135,7 @@ class GraphCompletionRetriever(BaseGraphRetriever):
         self,
         query: str,
         context: Optional[List[Edge]] = None,
-    ) -> Any:
+    ) -> List[str]:
         """
         Generates a completion using graph connections context based on a query.
 
@@ -171,7 +171,7 @@ class GraphCompletionRetriever(BaseGraphRetriever):
                 question=query, answer=completion, context=context_text, triplets=triplets
             )
 
-        return completion
+        return [completion]
 
     async def save_qa(self, question: str, answer: str, context: str, triplets: List) -> None:
         """

cognee/modules/retrieval/insights_retriever.py

@@ -96,17 +96,18 @@ class InsightsRetriever(BaseGraphRetriever):
                 unique_node_connections_map[unique_id] = True
                 unique_node_connections.append(node_connection)
 
-        return [
-            Edge(
-                node1=Node(node_id=connection[0]["id"], attributes=connection[0]),
-                node2=Node(node_id=connection[2]["id"], attributes=connection[2]),
-                attributes={
-                    **connection[1],
-                    "relationship_type": connection[1]["relationship_name"],
-                },
-            )
-            for connection in unique_node_connections
-        ]
+        return unique_node_connections
+        # return [
+        #     Edge(
+        #         node1=Node(node_id=connection[0]["id"], attributes=connection[0]),
+        #         node2=Node(node_id=connection[2]["id"], attributes=connection[2]),
+        #         attributes={
+        #             **connection[1],
+        #             "relationship_type": connection[1]["relationship_name"],
+        #         },
+        #     )
+        #     for connection in unique_node_connections
+        # ]
 
     async def get_completion(self, query: str, context: Optional[Any] = None) -> Any:
         """

cognee/modules/retrieval/temporal_retriever.py

@@ -136,7 +136,7 @@ class TemporalRetriever(GraphCompletionRetriever):
 
         return self.descriptions_to_string(top_k_events)
 
-    async def get_completion(self, query: str, context: Optional[str] = None) -> str:
+    async def get_completion(self, query: str, context: Optional[str] = None) -> List[str]:
         """Generates a response using the query and optional context."""
         if not context:
             context = await self.get_context(query=query)
@@ -149,4 +149,4 @@ class TemporalRetriever(GraphCompletionRetriever):
                 system_prompt_path=self.system_prompt_path,
             )
 
-        return completion
+        return [completion]

cognee/modules/retrieval/user_qa_feedback.py

@@ -1,4 +1,4 @@
-from typing import Any, Optional, List
+from typing import Optional, List
 
 from uuid import NAMESPACE_OID, uuid5, UUID
 from cognee.infrastructure.databases.graph import get_graph_engine

cognee/modules/search/methods/search.py

@@ -132,14 +132,37 @@ async def search(
             ],
         )
     else:
-        return [
-            SearchResult(
-                search_result=result,
-                dataset_id=datasets[min(index, len(datasets) - 1)].id if datasets else None,
-                dataset_name=datasets[min(index, len(datasets) - 1)].name if datasets else None,
-            )
-            for index, (result, _, datasets) in enumerate(search_results)
-        ]
+        # This is for maintaining backwards compatibility
+        if os.getenv("ENABLE_BACKEND_ACCESS_CONTROL", "false").lower() == "true":
+            return_value = []
+            for search_result in search_results:
+                result, context, datasets = search_result
+                return_value.append(
+                    {
+                        "search_result": result,
+                        "dataset_id": datasets[0].id,
+                        "dataset_name": datasets[0].name,
+                    }
+                )
+            return return_value
+        else:
+            return_value = []
+            for search_result in search_results:
+                result, context, datasets = search_result
+                return_value.append(result)
+            # For maintaining backwards compatibility
+            if len(return_value) == 1 and isinstance(return_value[0], list):
+                return return_value[0]
+            else:
+                return return_value
+        # return [
+        #     SearchResult(
+        #         search_result=result,
+        #         dataset_id=datasets[min(index, len(datasets) - 1)].id if datasets else None,
+        #         dataset_name=datasets[min(index, len(datasets) - 1)].name if datasets else None,
+        #     )
+        #     for index, (result, _, datasets) in enumerate(search_results)
+        # ]
 
 
 async def authorized_search(

cognee/modules/search/utils/prepare_search_result.py

@@ -6,7 +6,7 @@ from cognee.modules.search.utils.transform_context_to_graph import transform_con
 
 
 async def prepare_search_result(search_result):
-    result, context, datasets = search_result
+    results, context, datasets = search_result
 
     graphs = None
     result_graph = None
@@ -30,11 +30,11 @@ async def prepare_search_result(search_result):
             "*": "\n".join(cast(List[str], context)),
         }
 
-    if isinstance(result, List) and len(result) > 0 and isinstance(result[0], Edge):
-        result_graph = transform_context_to_graph(result)
+    if isinstance(results, List) and len(results) > 0 and isinstance(results[0], Edge):
+        result_graph = transform_context_to_graph(results)
 
     return {
-        "result": result_graph or result,
+        "result": result_graph or results[0] if len(results) == 1 else results,
         "graphs": graphs,
         "context": context_texts,
         "datasets": datasets,

cognee/tests/test_permissions.py

@@ -79,7 +79,7 @@ async def main():
     print("\n\nExtracted sentences are:\n")
     for result in search_results:
         print(f"{result}\n")
-    assert search_results[0].dataset_name == "NLP", (
+    assert search_results[0]["dataset_name"] == "NLP", (
         f"Dict must contain dataset name 'NLP': {search_results[0]}"
     )
 
@@ -93,7 +93,7 @@ async def main():
     print("\n\nExtracted sentences are:\n")
     for result in search_results:
         print(f"{result}\n")
-    assert search_results[0].dataset_name == "QUANTUM", (
+    assert search_results[0]["dataset_name"] == "QUANTUM", (
         f"Dict must contain dataset name 'QUANTUM': {search_results[0]}"
     )
 
@@ -170,7 +170,7 @@ async def main():
     for result in search_results:
         print(f"{result}\n")
 
-    assert search_results[0].dataset_name == "QUANTUM", (
+    assert search_results[0]["dataset_name"] == "QUANTUM", (
         f"Dict must contain dataset name 'QUANTUM': {search_results[0]}"
     )
 

cognee/tests/test_relational_db_migration.py

@@ -45,15 +45,13 @@ async def relational_db_migration():
     await migrate_relational_database(graph_engine, schema=schema)
 
     # 1. Search the graph
-    search_results: List[SearchResult] = await cognee.search(
+    search_results = await cognee.search(
         query_type=SearchType.GRAPH_COMPLETION, query_text="Tell me about the artist AC/DC"
-    )  # type: ignore
+    )
     print("Search results:", search_results)
 
     # 2. Assert that the search results contain "AC/DC"
-    assert any("AC/DC" in r.search_result for r in search_results), (
-        "AC/DC not found in search results!"
-    )
+    assert any("AC/DC" in r for r in search_results), "AC/DC not found in search results!"
 
     migration_db_provider = migration_engine.engine.dialect.name
     if migration_db_provider == "postgresql":

cognee/tests/test_search_db.py

@@ -144,13 +144,16 @@ async def main():
         ("GRAPH_COMPLETION_CONTEXT_EXTENSION", completion_ext),
         ("GRAPH_SUMMARY_COMPLETION", completion_sum),
     ]:
-        for search_result in search_results:
-            completion = search_result.search_result
-            assert isinstance(completion, str), f"{name}: should return a string"
-            assert completion.strip(), f"{name}: string should not be empty"
-            assert "netherlands" in completion.lower(), (
-                f"{name}: expected 'netherlands' in result, got: {completion!r}"
-            )
+        assert isinstance(search_results, list), f"{name}: should return a list"
+        assert len(search_results) == 1, (
+            f"{name}: expected single-element list, got {len(search_results)}"
+        )
+        text = search_results[0]
+        assert isinstance(text, str), f"{name}: element should be a string"
+        assert text.strip(), f"{name}: string should not be empty"
+        assert "netherlands" in text.lower(), (
+            f"{name}: expected 'netherlands' in result, got: {text!r}"
+        )
 
     graph_engine = await get_graph_engine()
     graph = await graph_engine.get_graph_data()

cognee/tests/unit/modules/retrieval/graph_completion_retriever_context_extension_test.py

@@ -59,8 +59,10 @@ class TestGraphCompletionWithContextExtensionRetriever:
 
         answer = await retriever.get_completion("Who works at Canva?")
 
-        assert isinstance(answer, str), f"Expected string, got {type(answer).__name__}"
-        assert answer.strip(), "Answer must contain only non-empty strings"
+        assert isinstance(answer, list), f"Expected list, got {type(answer).__name__}"
+        assert all(isinstance(item, str) and item.strip() for item in answer), (
+            "Answer must contain only non-empty strings"
+        )
 
     @pytest.mark.asyncio
     async def test_graph_completion_extension_context_complex(self):
@@ -140,8 +142,10 @@ class TestGraphCompletionWithContextExtensionRetriever:
 
         answer = await retriever.get_completion("Who works at Figma?")
 
-        assert isinstance(answer, str), f"Expected string, got {type(answer).__name__}"
-        assert answer.strip(), "Answer must contain only non-empty strings"
+        assert isinstance(answer, list), f"Expected list, got {type(answer).__name__}"
+        assert all(isinstance(item, str) and item.strip() for item in answer), (
+            "Answer must contain only non-empty strings"
+        )
 
     @pytest.mark.asyncio
     async def test_get_graph_completion_extension_context_on_empty_graph(self):
@@ -171,5 +175,7 @@ class TestGraphCompletionWithContextExtensionRetriever:
 
         answer = await retriever.get_completion("Who works at Figma?")
 
-        assert isinstance(answer, str), f"Expected string, got {type(answer).__name__}"
-        assert answer.strip(), "Answer must contain only non-empty strings"
+        assert isinstance(answer, list), f"Expected list, got {type(answer).__name__}"
+        assert all(isinstance(item, str) and item.strip() for item in answer), (
+            "Answer must contain only non-empty strings"
+        )

cognee/tests/unit/modules/retrieval/graph_completion_retriever_cot_test.py

@@ -55,8 +55,10 @@ class TestGraphCompletionCoTRetriever:
 
         answer = await retriever.get_completion("Who works at Canva?")
 
-        assert isinstance(answer, str), f"Expected string, got {type(answer).__name__}"
-        assert answer.strip(), "Answer must contain only non-empty strings"
+        assert isinstance(answer, list), f"Expected list, got {type(answer).__name__}"
+        assert all(isinstance(item, str) and item.strip() for item in answer), (
+            "Answer must contain only non-empty strings"
+        )
 
     @pytest.mark.asyncio
     async def test_graph_completion_cot_context_complex(self):
@@ -133,8 +135,10 @@ class TestGraphCompletionCoTRetriever:
 
         answer = await retriever.get_completion("Who works at Figma?")
 
-        assert isinstance(answer, str), f"Expected string, got {type(answer).__name__}"
-        assert answer.strip(), "Answer must contain only non-empty strings"
+        assert isinstance(answer, list), f"Expected list, got {type(answer).__name__}"
+        assert all(isinstance(item, str) and item.strip() for item in answer), (
+            "Answer must contain only non-empty strings"
+        )
 
     @pytest.mark.asyncio
     async def test_get_graph_completion_cot_context_on_empty_graph(self):
@@ -164,5 +168,7 @@ class TestGraphCompletionCoTRetriever:
 
         answer = await retriever.get_completion("Who works at Figma?")
 
-        assert isinstance(answer, str), f"Expected string, got {type(answer).__name__}"
-        assert answer.strip(), "Answer must contain only non-empty strings"
+        assert isinstance(answer, list), f"Expected list, got {type(answer).__name__}"
+        assert all(isinstance(item, str) and item.strip() for item in answer), (
+            "Answer must contain only non-empty strings"
+        )

cognee/tests/unit/modules/retrieval/insights_retriever_test.py

@@ -82,7 +82,7 @@ class TestInsightsRetriever:
 
         context = await retriever.get_context("Mike")
 
-        assert context[0].node1.attributes["name"] == "Mike Broski", "Failed to get Mike Broski"
+        assert context[0][0]["name"] == "Mike Broski", "Failed to get Mike Broski"
 
     @pytest.mark.asyncio
     async def test_insights_context_complex(self):
@@ -222,9 +222,7 @@ class TestInsightsRetriever:
 
         context = await retriever.get_context("Christina")
 
-        assert context[0].node1.attributes["name"] == "Christina Mayer", (
-            "Failed to get Christina Mayer"
-        )
+        assert context[0][0]["name"] == "Christina Mayer", "Failed to get Christina Mayer"
 
     @pytest.mark.asyncio
     async def test_insights_context_on_empty_graph(self):

{cognee-0.3.1.dist-info → cognee-0.3.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cognee
-Version: 0.3.1
+Version: 0.3.3
 Summary: Cognee - is a library for enriching LLM context with a semantic layer for better understanding and reasoning.
 Project-URL: Homepage, https://www.cognee.ai
 Project-URL: Repository, https://github.com/topoteretes/cognee

{cognee-0.3.1.dist-info → cognee-0.3.3.dist-info}/RECORD

@@ -41,7 +41,7 @@ cognee/api/v1/memify/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 cognee/api/v1/memify/routers/__init__.py,sha256=Uv25PVGhfjnNi1NYWOmOLIlzaeTlyMYF9m7BEfdu45Q,49
 cognee/api/v1/memify/routers/get_memify_router.py,sha256=C1Cjt9D5TxhqBPmXZGNrCS4lJqPVXIJYgxZFtWVjZNs,4599
 cognee/api/v1/notebooks/routers/__init__.py,sha256=TvQz6caluaMoXNvjbE1p_C8savypgs8rAyP5lQ8jlpc,55
-cognee/api/v1/notebooks/routers/get_notebooks_router.py,sha256=YFxvs3WR5RCjB-Rk4uJ8yZtlLU0vl0AcOuRNFrb-i6U,3420
+cognee/api/v1/notebooks/routers/get_notebooks_router.py,sha256=m8OH3Kw1UHF8aTP4yNuSpv7gNThE4HxmLIrUnvECYGA,3484
 cognee/api/v1/permissions/routers/__init__.py,sha256=ljE3YnrzlMcVfThmkR5GSIxkm7sQVyibaLNtYQL4HO0,59
 cognee/api/v1/permissions/routers/get_permissions_router.py,sha256=tqd-J__UBlstTWnQocesdjVM9JnYO5rtJhhFj-Zv1_o,8316
 cognee/api/v1/prune/__init__.py,sha256=FEr5tTlX7wf3X4aFff6NPlVhNrPyqx7RBoJ71bJN1cY,25
@@ -56,7 +56,7 @@ cognee/api/v1/responses/routers/get_responses_router.py,sha256=ggbLhY9IXaInCgIs5
 cognee/api/v1/search/__init__.py,sha256=Sqw60DcOj4Bnvt-EWFknT31sPcvROIRKCWLr5pbkFr4,39
 cognee/api/v1/search/search.py,sha256=YQicNVi9q4FteAmt_EtY75I_EuNZ9ZjGE73wg-NcDwY,8824
 cognee/api/v1/search/routers/__init__.py,sha256=6RebeLX_2NTRxIMPH_mGuLztPxnGnMJK1y_O93CtRm8,49
-cognee/api/v1/search/routers/get_search_router.py,sha256=30DpoGn1ceSBaKOH1jLXodQJZCwBwSZ_OZgrKkJzGs0,6149
+cognee/api/v1/search/routers/get_search_router.py,sha256=-5GLgHipflEblYAwl3uiPAZ2i3TgrLEjDuiO_cCqcB8,6252
 cognee/api/v1/settings/routers/__init__.py,sha256=wj_UYAXNMPCkn6Mo1YB01dCBiV9DQwTIf6OWjnGRpf8,53
 cognee/api/v1/settings/routers/get_settings_router.py,sha256=EKVj2kw5MDKZcxAIAyi7ltz7wD6Hfs5feGrkd9R_vCA,3195
 cognee/api/v1/sync/__init__.py,sha256=hx2Af6GtX8soyHiYpWieWpAglLD05_7BK7PgdBqGbVE,313
@@ -64,7 +64,7 @@ cognee/api/v1/sync/sync.py,sha256=zzCVJD1AvcSXtNsgLJr1iPMRxY6vRxGdkt7sVdJ8W2c,33
 cognee/api/v1/sync/routers/__init__.py,sha256=hZArat9DDyzBll8qej0_o16QhtQRciTB37b5rc3ckGM,76
 cognee/api/v1/sync/routers/get_sync_router.py,sha256=7fD0QL0IIjyg9VBadNcLD7G7rypy_1glyWv8HVHBrao,9703
 cognee/api/v1/ui/__init__.py,sha256=SKfmAWokGT3_ZGqDkEtQihrvXCog6WTP3UdZrD20DBc,38
-cognee/api/v1/ui/ui.py,sha256=AgH7js_34rjW7Z_19Xv-oKBA1TBH5-t2beVtIzoAMAY,19215
+cognee/api/v1/ui/ui.py,sha256=CTgEmVrpeG174jTalgc8F_4hO6LsBXtTFkczMaLlNvc,23137
 cognee/api/v1/users/__init__.py,sha256=TMOZ_3puQxVqVIjWNA0yb16Tpp8yoNKAfwxIxoFpgus,37
 cognee/api/v1/users/create_user.py,sha256=PRuc7aUhOpyb-g5nUGDKSegp3cxkZy5TDeX1sxX6jjM,324
 cognee/api/v1/users/routers/__init__.py,sha256=_m3tyK2deFQCBjx6p-0t23e7qnnhAyx-2PBM7Wc6E7A,314
@@ -78,7 +78,7 @@ cognee/api/v1/visualize/__init__.py,sha256=TBk58R8cza6Qx7IP2r9RvAtE8Fmoo9vOh9VjC
 cognee/api/v1/visualize/start_visualization_server.py,sha256=3esCKYYmBx9Sb2H5JWrliT47qNyt_rGrv1OvR0LJVAg,440
 cognee/api/v1/visualize/visualize.py,sha256=xKhh1N-doIgFcnq9Tz1acwrS4fOqBFZlgif4prMBqP4,1077
 cognee/cli/__init__.py,sha256=MaKUkdFaETdbuMFoV02V8BZNuYr7tZQJKt6y25CaUhk,243
-cognee/cli/_cognee.py,sha256=iupokPsdRXVQYzcUBzt-GDmn_E-Ts6enoCAhm89lNCY,8739
+cognee/cli/_cognee.py,sha256=v7GeAGLXaVkkkjUpbmo4Ya3ff-cycE297F41Sjx0kNA,8878
 cognee/cli/config.py,sha256=8XhUqpkmNNzCFbnIpRvNQIO2Hvw0OD44zWYM0eADozA,998
 cognee/cli/debug.py,sha256=-u3REG2xloCFLwOWQ3wVM7RpZRn06QlnfDyCRoxrrek,444
 cognee/cli/echo.py,sha256=3G4qYcYn1cShTeIKaZMPD_TgoS7LBqyUnMnTFaj5dUE,1128
@@ -181,12 +181,12 @@ cognee/infrastructure/databases/relational/get_async_session.py,sha256=qfiXSsTAA
 cognee/infrastructure/databases/relational/get_migration_relational_engine.py,sha256=5RtH281iIQo3vqgwmKT0nuiJp9jNd7vw6xRUjc5xIDM,1070
 cognee/infrastructure/databases/relational/get_relational_engine.py,sha256=De51ieg9eFhRLX08k9oNc-oszvt_9J5DHebqI1qI8_U,741
 cognee/infrastructure/databases/relational/with_async_session.py,sha256=UgQeJOvgeM6yhyNDwWdGULtTjZosTnjDlr267Losnfs,803
-cognee/infrastructure/databases/relational/sqlalchemy/SqlAlchemyAdapter.py,sha256=uuX2tFPWueX0etooLFFr1PvaPQSdFtZRyCnRYSau20Q,27539
+cognee/infrastructure/databases/relational/sqlalchemy/SqlAlchemyAdapter.py,sha256=j4mnqNJAO-U-Qfveam6NgjIH5lt7WjSMLVlemBrdpYU,27540
 cognee/infrastructure/databases/relational/sqlalchemy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cognee/infrastructure/databases/utils/__init__.py,sha256=4C0ncZG-O6bOFJpKgscCHu6D5vodLWRIKpe-WT4Ijbs,75
 cognee/infrastructure/databases/utils/get_or_create_dataset_database.py,sha256=wn7pRgeX-BU0L191_6pgT9P54uhVQlGMPqxQdvIlv4Y,2101
 cognee/infrastructure/databases/vector/__init__.py,sha256=7MdGJ3Mxdh2RyDq39rcjD99liIa-yGXxDUzq--1qQZs,291
-cognee/infrastructure/databases/vector/config.py,sha256=cY833pGsse4_dBmacNXmsdNZJQrSWPevKcGW1f_klYU,2927
+cognee/infrastructure/databases/vector/config.py,sha256=4HOmqZOEfVNmAhjxRNePMU9haTVeR35R2XbhPTcMqFg,2952
 cognee/infrastructure/databases/vector/create_vector_engine.py,sha256=ECtICkIW5QM_lX9465ZTxVXC5MCRo_h219q3GyFXxpc,4716
 cognee/infrastructure/databases/vector/get_vector_engine.py,sha256=y4TMWJ6B6DxwKF9PMfjB6WqujPnVhf0oR2j35Q-KhvA,272
 cognee/infrastructure/databases/vector/supported_databases.py,sha256=0UIYcQ15p7-rq5y_2A-E9ydcXyP6frdg8T5e5ECDDMI,25
@@ -365,8 +365,8 @@ cognee/infrastructure/loaders/external/pypdf_loader.py,sha256=nFa_h3LURBPoguRIID
 cognee/infrastructure/loaders/external/unstructured_loader.py,sha256=XCRVHwpM5XmcjRmL4Pr9ELzBU_qYDPhX_Ahn5K8w0AU,4603
 cognee/infrastructure/loaders/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cognee/infrastructure/utils/calculate_backoff.py,sha256=O6h4MCe357BKaECmLZPLGYpffrMol65LwQCklBj4sh4,935
-cognee/infrastructure/utils/run_async.py,sha256=3J0OGzh3HLO6wHQN-rjEnGitVD_mbs4AO6VFgZ47eQE,393
-cognee/infrastructure/utils/run_sync.py,sha256=wLhXUdopsEaIRU7CrzcfPdj1KiRBjCf83HjqvSsace8,726
+cognee/infrastructure/utils/run_async.py,sha256=gZY8ZLG_86O9YVK8hciduIoDONHaEEnGOILh3EeD9LA,510
+cognee/infrastructure/utils/run_sync.py,sha256=9pAXc-EmjtV03exnUMOVSC-IJq_KCslX05z62MHQjlQ,800
 cognee/modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cognee/modules/chunking/Chunker.py,sha256=KezN4WBiV0KNJtx6daMg4g1-a-_oJxn_l_iQT94T1lQ,343
 cognee/modules/chunking/LangchainChunker.py,sha256=Yo9Jza-t3x3V8I8PWbxUu48vlVVdvJKxwzL2gManwDc,2351
@@ -482,14 +482,15 @@ cognee/modules/metrics/operations/__init__.py,sha256=MZ3xbVdfEKqfLct8WnbyFVyZmkB
 cognee/modules/metrics/operations/get_pipeline_run_metrics.py,sha256=upIWnzKeJT1_XbL_ABdGxW-Ai7mO3AqMK35BNmItIQQ,2434
 cognee/modules/notebooks/methods/__init__.py,sha256=IhY4fUVPJbuvS83QESsWzjZRC6oC1I-kJi5gr3kPTLk,215
 cognee/modules/notebooks/methods/create_notebook.py,sha256=S41H3Rha0pj9dEKFy1nBG9atTGHhUdOmDZgr0ckUA6M,633
+cognee/modules/notebooks/methods/create_tutorial_notebook.py,sha256=8YPoDcMUZSNhEWSKxUcPOM61y0St2Z1Y-PC1HFRmlbk,4248
 cognee/modules/notebooks/methods/delete_notebook.py,sha256=BKxoRlPzkwXvTYh5WcF-zo_iVmaXqEiptS42JwB0KQU,309
-cognee/modules/notebooks/methods/get_notebook.py,sha256=O-iWX4sElOn_5EpI9_WCwdvbfPRgVQVGBev1U4tI8AA,545
+cognee/modules/notebooks/methods/get_notebook.py,sha256=IP4imsdt9X6GYd6i6WF6PlVhotGNH0i7XZpPqbtqMwo,554
 cognee/modules/notebooks/methods/get_notebooks.py,sha256=ee40ALHvebVORuwZVkQ271qAj260rrYy6eVGxAmfo8c,483
-cognee/modules/notebooks/methods/update_notebook.py,sha256=L-WgIxEr_uPClRZQZtnBEV9iT2C7aWqs0FuSW-F5qqk,410
+cognee/modules/notebooks/methods/update_notebook.py,sha256=MnZbfh-WfEfH3ImNvyQNhDeNwpYeS7p8FPVwnmBvZVg,361
 cognee/modules/notebooks/models/Notebook.py,sha256=Jth47QxJQ2-VGPyIcS0ul3bS8bgGrk9vCGoJVagxanw,1690
 cognee/modules/notebooks/models/__init__.py,sha256=jldsDjwRvFMreGpe4wxxr5TlFXTZuU7rbsRkGQvTO5s,45
 cognee/modules/notebooks/operations/__init__.py,sha256=VR_2w_d0lEiJ5Xw7_mboo2qWUv0umrR_Bp58MaMoE6w,55
-cognee/modules/notebooks/operations/run_in_local_sandbox.py,sha256=0Au8-bDy7S-c1eNLKInQI5HV7u3bhl7Lvvtt79c5J4Q,1186
+cognee/modules/notebooks/operations/run_in_local_sandbox.py,sha256=17hMEQC3LZTfPvbRUrPN9SzDeJPWSTq_BAhtwRZiqT8,1338
 cognee/modules/observability/get_observe.py,sha256=chRw4jmpmrwEvDecF9sgApm23IOzVgCbwkKEAyz1_AI,264
 cognee/modules/observability/observers.py,sha256=SKQSWWyGDG0QY2_bqsFgfpLUb7OUL4WFf8tDZYe5JMM,157
 cognee/modules/ontology/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -549,15 +550,15 @@ cognee/modules/retrieval/code_retriever.py,sha256=cnOjgfCATzz0-XZGFrIIkuVZLc6HBh
 cognee/modules/retrieval/coding_rules_retriever.py,sha256=3GU259jTbGLqmp_A8sUdE4fyf0td06SKuxBJVW-npIQ,1134
 cognee/modules/retrieval/completion_retriever.py,sha256=Lw5sxN_UrtmWSOtcSS7Yj50Gw9p4nNBmW3dr2kV9JJ0,3754
 cognee/modules/retrieval/cypher_search_retriever.py,sha256=_3rZJ23hSZpDa8kVyOSWN3fwjMI_aLF2m5p-FtBek8k,2440
-cognee/modules/retrieval/graph_completion_context_extension_retriever.py,sha256=e5F12b5TSMZK7eHoaKlC3GSe69MHh80s9acc5nDcY6Y,4524
-cognee/modules/retrieval/graph_completion_cot_retriever.py,sha256=p4yl2w4NysShX9S2IaS1t62DYaSoEALZ0VaihkmmPiY,6132
-cognee/modules/retrieval/graph_completion_retriever.py,sha256=LJ9gOQkUteMboUz6GdTrWDlYevKgoDMHNBdHgxfW5Eo,8814
+cognee/modules/retrieval/graph_completion_context_extension_retriever.py,sha256=-6yN8gpRlDue8d28rk-Ly-gq0T8BW-i1-Jgbp1x-Zsg,4532
+cognee/modules/retrieval/graph_completion_cot_retriever.py,sha256=JU-FkikaU68v8fT8VAmG6jojwhwroKYW2RUxdlJ1R-k,6140
+cognee/modules/retrieval/graph_completion_retriever.py,sha256=VnrFD4xUQewIO83mfmIUcPLA_HBGdUlDVRyA2Pm4ARo,8822
 cognee/modules/retrieval/graph_summary_completion_retriever.py,sha256=3AMisk3fObk2Vh1heY4veHkDjLsHgSSUc_ChZseJUYw,2456
-cognee/modules/retrieval/insights_retriever.py,sha256=CYlLrRZrVh50fbRJLAihtSt-8licb_lnNJee5vmqCA4,4892
+cognee/modules/retrieval/insights_retriever.py,sha256=1pcYd34EfKk85MSPFQ8b-ZbSARmnauks8TxXfNOxvOw,4953
 cognee/modules/retrieval/natural_language_retriever.py,sha256=zJz35zRmBP8-pRlkoxxSxn3-jtG2lUW0xcu58bq9Ebs,5761
 cognee/modules/retrieval/summaries_retriever.py,sha256=joXYphypACm2JiCjbC8nBS61m1q2oYkzyIt9bdgALNw,3384
-cognee/modules/retrieval/temporal_retriever.py,sha256=SXYvlph0QbhelaYc2lyZxYC550L7CgU6L_mR3kMAASM,5677
-cognee/modules/retrieval/user_qa_feedback.py,sha256=WSMPg6WjteR-XgK0vK9f_bkZ_o0JMPb4XZ9OAcFyz9E,3371
+cognee/modules/retrieval/temporal_retriever.py,sha256=EUEYN94LpoWfbPjsToe_pC3rFsUUTIPA5K6wNjv8Nds,5685
+cognee/modules/retrieval/user_qa_feedback.py,sha256=-VEOsE_t0FiTy00OpOMWAYv12YSLPieAcMsu82vm7h4,3366
 cognee/modules/retrieval/context_providers/DummyContextProvider.py,sha256=9GsvINc7ekRyRWO5IefFGyytRYqsSlhpwAOw6Q691cA,419
 cognee/modules/retrieval/context_providers/SummarizedTripletSearchContextProvider.py,sha256=ypO6yWLxvmRsj_5dyYdvXTbztJmB_ioLrgyG6bF5WGA,894
 cognee/modules/retrieval/context_providers/TripletSearchContextProvider.py,sha256=8PzksHAtRw7tZarP3nZuxhi0cd1EYEDHOT4Q74mNEvc,3656
@@ -578,7 +579,7 @@ cognee/modules/search/exceptions/exceptions.py,sha256=Zc5Y0M-r-UnSSlpKzHKBplfjZ-
 cognee/modules/search/methods/__init__.py,sha256=jGfRvNwM5yIzj025gaVhcx7nCupRSXbUUnFjYVjL_Js,27
 cognee/modules/search/methods/get_search_type_tools.py,sha256=wXxOZx3uEnMhRhUO2HGswQ5iVbWvjUj17UT_qdJg6Oo,6837
 cognee/modules/search/methods/no_access_control_search.py,sha256=R08aMgaB8AkD0_XVaX15qLyC9KJ3fSVFv9zeZwuyez4,1566
-cognee/modules/search/methods/search.py,sha256=RhdNNpky3HEb_RDTTgQouXdc1lmU5vGV2uiQ8harz1Y,11455
+cognee/modules/search/methods/search.py,sha256=Akqf4a913_nG56TMxTKU65kOwL0tWURDLHEXlwcgV1c,12459
 cognee/modules/search/models/Query.py,sha256=9WcF5Z1oCFtA4O-7An37eNAPX3iyygO4B5NSwhx7iIg,558
 cognee/modules/search/models/Result.py,sha256=U7QtoNzAtZnUDwGWhjVfcalHQd4daKtYYvJz2BeWQ4w,564
 cognee/modules/search/operations/__init__.py,sha256=AwJl6v9BTpocoefEZLk-flo1EtydYb46NSUoNFHkhX0,156
@@ -592,7 +593,7 @@ cognee/modules/search/types/SearchResult.py,sha256=blEean6PRFKcDRQugsojZPfH-Wohx
 cognee/modules/search/types/SearchType.py,sha256=-lT4bLKKunV4cL4FfF3tjNbdN7X4AsRMLpTkReNwXZM,594
 cognee/modules/search/types/__init__.py,sha256=8k6OjVrL70W1Jh-ClTbG2ETYIhOtSk3tfqjzYgEdPzA,117
 cognee/modules/search/utils/__init__.py,sha256=86mRtCN-B5-2NNChdQoU5x8_8hqTczGZjBoKVE9O7hA,124
-cognee/modules/search/utils/prepare_search_result.py,sha256=nfK8aqR2tRL_SYHqtkK1ssG8Ws_oflEDZZAEvQmu5F4,1293
+cognee/modules/search/utils/prepare_search_result.py,sha256=FTM-tVlprL8EswIcwOy8jO1bRmKG61GZqFfM8FNJUJg,1336
 cognee/modules/search/utils/transform_context_to_graph.py,sha256=rUQeEH-Z-GqAzAZTCetRVpwgrOHlNe3mUBRLwRb0478,1238
 cognee/modules/settings/__init__.py,sha256=_SZQgCQnnnIHLJuKOMO9uWzXNBQxwYHHMUSBp0qa2uQ,210
 cognee/modules/settings/get_current_settings.py,sha256=R2lOusG5Q2PMa2-2vDndh3Lm7nXyZVkdzTV7vQHT81Y,1642
@@ -781,19 +782,25 @@ cognee/tests/test_neptune_analytics_graph.py,sha256=bZqPNk8ag_tilpRobK5RJVwTS473
 cognee/tests/test_neptune_analytics_hybrid.py,sha256=Q9mCGGqroLnHrRo3kHdhkMZnlNtvCshRG1BgU81voBc,6222
 cognee/tests/test_neptune_analytics_vector.py,sha256=h_Ofp4ZAdyGpCWzuQyoXmLO5lOycNLtliIFvJt7nXHg,8652
 cognee/tests/test_parallel_databases.py,sha256=Hhm4zh-luaXKmy7mjEHq3VkMppt6QaJ3IB2IRUVkwSk,1997
-cognee/tests/test_permissions.py,sha256=PFHSUOUC-zMiK9N8JOzO81fZy4q6CdSiALRaOcR9dj8,11500
+cognee/tests/test_permissions.py,sha256=h2Gyug-1DI8YycYMBhfEY0XdZbG3qt7ubiK5x7EJCVc,11509
 cognee/tests/test_pgvector.py,sha256=ZAaeWcnNBSYuyciYPBnzJSrGkuIjmKYWoNu3Jj7cPOM,9568
-cognee/tests/test_relational_db_migration.py,sha256=jFkiLVAEpJtI8iGeR7WWRRP7Zi5TFeIn4VLiqfhl2Ms,8761
+cognee/tests/test_relational_db_migration.py,sha256=QUgS40w3ZDO3fwvM0x0b1U0SxfFVJ3J6UEK5C48GVHA,8695
 cognee/tests/test_remote_kuzu.py,sha256=2GG05MtGuhOo6ST82OxjdVDetBS0GWHvKKmmmEtQO2U,7245
 cognee/tests/test_remote_kuzu_stress.py,sha256=5vgnu4Uz_NoKKqFZJeVceHwb2zNhvdTVBgpN3NjhfAE,5304
 cognee/tests/test_s3.py,sha256=rY2UDK15cdyywlyVrR8N2DRtVXWYIW5REaaz99gaQeE,2694
 cognee/tests/test_s3_file_storage.py,sha256=62tvIFyh_uTP0TFF9Ck4Y-sxWPW-cwJKYEJUJI1atPI,5654
-cognee/tests/test_search_db.py,sha256=7d72uJvERbp5NyIj5ouXBx7Uf_38qPrU4ms8IsnKSx8,13312
+cognee/tests/test_search_db.py,sha256=4GpLx8ZJoMjkp-XqQ-LCrkf3NhAM4j_rMmlOFgmDO-A,13420
 cognee/tests/test_starter_pipelines.py,sha256=X1J8RDD0bFMKnRETyi5nyaF4TYdmUIu0EuD3WQwShNs,2475
 cognee/tests/test_telemetry.py,sha256=FIneuVofSKWFYqxNC88sT_P5GPzgfjVyqDCf2TYBE2E,4130
 cognee/tests/test_temporal_graph.py,sha256=G0PyzuvIYylwFT-3eZSzjtBik9O1g75sGLj3QK9RYTA,12624
-cognee/tests/integration/cli/__init__.py,sha256=xYkvpZkxv_HRWmX71pGM3NUw2KKkDQIM-V6Ehxu-f0I,39
-cognee/tests/integration/cli/test_cli_integration.py,sha256=3hdz1DoGeidJInqbCy1YQte6J0QeQG1_WKGs9utjAFg,11560
+cognee/tests/cli_tests/cli_integration_tests/__init__.py,sha256=xYkvpZkxv_HRWmX71pGM3NUw2KKkDQIM-V6Ehxu-f0I,39
+cognee/tests/cli_tests/cli_integration_tests/test_cli_integration.py,sha256=3hdz1DoGeidJInqbCy1YQte6J0QeQG1_WKGs9utjAFg,11560
+cognee/tests/cli_tests/cli_unit_tests/__init__.py,sha256=U069aFvdwfKPd6YsR_FJML5LRphHHF5wx9mwug1hRh4,32
+cognee/tests/cli_tests/cli_unit_tests/test_cli_commands.py,sha256=5a3vPiSFmKumq6sTfdfMyeUpJGjbZ6_5zX4TUcV0ZJQ,17625
+cognee/tests/cli_tests/cli_unit_tests/test_cli_edge_cases.py,sha256=PyFCnClvbXG1GaiS16qwcuyXXDJ4sRyBCKV5WHrOUxk,23501
+cognee/tests/cli_tests/cli_unit_tests/test_cli_main.py,sha256=Gsj2zYlVL80iU9EjRj4Q4QzgsYuIngUvDbA9suV99oA,6098
+cognee/tests/cli_tests/cli_unit_tests/test_cli_runner.py,sha256=WZ8oZIlc_JintDq_cnEg9tmLEMZMGFPQGhU7Y_7sfgs,1497
+cognee/tests/cli_tests/cli_unit_tests/test_cli_utils.py,sha256=Flej8LNYRXNkWd2tq8elMm8MkqbhCUb8RtXaPzfNYm4,4323
 cognee/tests/integration/documents/AudioDocument_test.py,sha256=0mJnlWRc7gWqOxAUfdSSIxntcUrzkPXhlsd-MFsiRoM,2790
 cognee/tests/integration/documents/ImageDocument_test.py,sha256=vrb3uti0RF6a336LLI95i8fso3hOFw9AFe1NxPnOf6k,2802
 cognee/tests/integration/documents/PdfDocument_test.py,sha256=IY0Cck8J2gEyuJHPK0HODPbZPIXQ799KhWrgkjn5feM,1798
@@ -821,12 +828,6 @@ cognee/tests/test_data/text_to_speech.mp3,sha256=h0xuFwn_ddt-q2AeBu_BdLmMJUc4QtE
 cognee/tests/test_data/text_to_speech_copy.mp3,sha256=h0xuFwn_ddt-q2AeBu_BdLmMJUc4QtEKWdBQ9ydGYXI,28173
 cognee/tests/unit/api/__init__.py,sha256=tKoksC3QC3r43L7MDdEdjE2A34r8iOD1YPv8mT-iZzk,29
 cognee/tests/unit/api/test_conditional_authentication_endpoints.py,sha256=t5HX6s8D-5pFANy9IJEtY5ht_GhlJSZK_KkpqVj8ZdI,9349
-cognee/tests/unit/cli/__init__.py,sha256=U069aFvdwfKPd6YsR_FJML5LRphHHF5wx9mwug1hRh4,32
-cognee/tests/unit/cli/test_cli_commands.py,sha256=5a3vPiSFmKumq6sTfdfMyeUpJGjbZ6_5zX4TUcV0ZJQ,17625
-cognee/tests/unit/cli/test_cli_edge_cases.py,sha256=PyFCnClvbXG1GaiS16qwcuyXXDJ4sRyBCKV5WHrOUxk,23501
-cognee/tests/unit/cli/test_cli_main.py,sha256=Gsj2zYlVL80iU9EjRj4Q4QzgsYuIngUvDbA9suV99oA,6098
-cognee/tests/unit/cli/test_cli_runner.py,sha256=WZ8oZIlc_JintDq_cnEg9tmLEMZMGFPQGhU7Y_7sfgs,1497
-cognee/tests/unit/cli/test_cli_utils.py,sha256=Flej8LNYRXNkWd2tq8elMm8MkqbhCUb8RtXaPzfNYm4,4323
 cognee/tests/unit/entity_extraction/regex_entity_extraction_test.py,sha256=3zNvSI56FBltg_lda06n93l2vl702i5O1ewoQXoo50E,10234
 cognee/tests/unit/eval_framework/answer_generation_test.py,sha256=TVrAJneOiTSztq7J6poo4GGPsow3MWnBtpBwPkDHq08,1309
 cognee/tests/unit/eval_framework/benchmark_adapters_test.py,sha256=yXmr5089j1KB5lrLs4v17JXPuUk2iwXJRJGOb_wdnqk,3382
@@ -854,10 +855,10 @@ cognee/tests/unit/modules/pipelines/run_task_from_queue_test.py,sha256=X2clLQYoP
 cognee/tests/unit/modules/pipelines/run_tasks_test.py,sha256=IJ_2NBOizC-PtW4c1asYZB-SI85dQswB0Lt5e_n-5zI,1399
 cognee/tests/unit/modules/pipelines/run_tasks_with_context_test.py,sha256=Bi5XgQWfrgCgTtRu1nrUAqraDYHUzILleOka5fpTsKE,1058
 cognee/tests/unit/modules/retrieval/chunks_retriever_test.py,sha256=jHsvi1Y-bsOVdFrGIfGaTXV4UvwI4KzQF_hV0i3oy2I,6341
-cognee/tests/unit/modules/retrieval/graph_completion_retriever_context_extension_test.py,sha256=Mj2FTslG5g5mfK5jlGQ8MlwdUVq6ENpJKdwBpuOQsHI,6693
-cognee/tests/unit/modules/retrieval/graph_completion_retriever_cot_test.py,sha256=_rDUE0kP0ewwuIH8iU_Jb7TLCikFhWA8DlBaiSgy3BM,6469
+cognee/tests/unit/modules/retrieval/graph_completion_retriever_context_extension_test.py,sha256=nVKITq5N0cOLCjvJvS6e9vAdbwFcNWxe11O1F7rEzok,6906
+cognee/tests/unit/modules/retrieval/graph_completion_retriever_cot_test.py,sha256=n5tMlsRzTsrMWpdqWS7nKOITwRnv3wNMMKFnXHTsdtU,6682
 cognee/tests/unit/modules/retrieval/graph_completion_retriever_test.py,sha256=T-WXkG9opjtlxY8R6CXcHazc2xouPOdUR9bumxxzFsY,8986
-cognee/tests/unit/modules/retrieval/insights_retriever_test.py,sha256=wt-JLTsRdi3A8kqEvXoQQZA5gZAauASjWPzYkYK_DQk,8644
+cognee/tests/unit/modules/retrieval/insights_retriever_test.py,sha256=xkbxlNiHY6evVbBYMncllXDNs3nNC_jZeYP47oT8vG0,8592
 cognee/tests/unit/modules/retrieval/rag_completion_retriever_test.py,sha256=RQi5EAjB5ffiIcsFmfg3Hao3AVPOkTEL72xnXIxbTKM,6551
 cognee/tests/unit/modules/retrieval/summaries_retriever_test.py,sha256=IfhDyVuKUrjCEy22-Mva9w7li2mtPZT9FlNIFvpFMKw,4950
 cognee/tests/unit/modules/retrieval/temporal_retriever_test.py,sha256=bvGvJgq9JF8nxnFvBlSo2qOBc0FKjCe006Io5HityAo,7672
@@ -888,9 +889,9 @@ distributed/tasks/queued_add_edges.py,sha256=kz1DHE05y-kNHORQJjYWHUi6Q1QWUp_v3Dl
 distributed/tasks/queued_add_nodes.py,sha256=aqK4Ij--ADwUWknxYpiwbYrpa6CcvFfqHWbUZW4Kh3A,452
 distributed/workers/data_point_saving_worker.py,sha256=jFmA0-P_0Ru2IUDrSug0wML-5goAKrGtlBm5BA5Ryw4,3229
 distributed/workers/graph_saving_worker.py,sha256=oUYl99CdhlrPAIsUOHbHnS3d4XhGoV0_OIbCO8wYzRg,3648
-cognee-0.3.1.dist-info/METADATA,sha256=FiH8UcAy24p0Fj79WChu6lznpJ4Ikn2Wa7RXn4yK7SQ,14753
-cognee-0.3.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-cognee-0.3.1.dist-info/entry_points.txt,sha256=4Fe5PRV0e3j5MFUo7kYyRFa3MhMNbOu69pGBazTxPps,51
-cognee-0.3.1.dist-info/licenses/LICENSE,sha256=pHHjSQj1DD8SDppW88MMs04TPk7eAanL1c5xj8NY7NQ,11344
-cognee-0.3.1.dist-info/licenses/NOTICE.md,sha256=6L3saP3kSpcingOxDh-SGjMS8GY79Rlh2dBNLaO0o5c,339
-cognee-0.3.1.dist-info/RECORD,,
+cognee-0.3.3.dist-info/METADATA,sha256=MofBzxb-pUo59hyKjasnooG9SDbuVPsvy5UK6sjXluA,14753
+cognee-0.3.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+cognee-0.3.3.dist-info/entry_points.txt,sha256=GCCTsNg8gzOJkolq7dR7OK1VlIAO202dGDnMI8nm8oQ,55
+cognee-0.3.3.dist-info/licenses/LICENSE,sha256=pHHjSQj1DD8SDppW88MMs04TPk7eAanL1c5xj8NY7NQ,11344
+cognee-0.3.3.dist-info/licenses/NOTICE.md,sha256=6L3saP3kSpcingOxDh-SGjMS8GY79Rlh2dBNLaO0o5c,339
+cognee-0.3.3.dist-info/RECORD,,

cognee-0.3.3.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+cognee-cli = cognee.cli._cognee:main

cognee-0.3.1.dist-info/entry_points.txt

@@ -1,2 +0,0 @@
-[console_scripts]
-cognee = cognee.cli._cognee:main