cognee 0.3.4.dev2__py3-none-any.whl → 0.3.4.dev3__py3-none-any.whl

cognee/tests/unit/modules/users/test_tutorial_notebook_creation.py

@@ -0,0 +1,399 @@
+import json
+import pytest
+from unittest.mock import AsyncMock, patch, MagicMock
+import hashlib
+import time
+from uuid import uuid4
+from sqlalchemy.ext.asyncio import AsyncSession
+from pathlib import Path
+import zipfile
+from cognee.shared.cache import get_tutorial_data_dir
+
+from cognee.modules.notebooks.methods.create_notebook import _create_tutorial_notebook
+from cognee.modules.notebooks.models.Notebook import Notebook
+import cognee
+from cognee.shared.logging_utils import get_logger
+
+logger = get_logger()
+
+
+# Module-level fixtures available to all test classes
+@pytest.fixture
+def mock_session():
+    """Mock database session."""
+    session = AsyncMock(spec=AsyncSession)
+    session.add = MagicMock()
+    session.commit = AsyncMock()
+    return session
+
+
+@pytest.fixture
+def sample_jupyter_notebook():
+    """Sample Jupyter notebook content for testing."""
+    return {
+        "cells": [
+            {
+                "cell_type": "markdown",
+                "metadata": {},
+                "source": ["# Tutorial Introduction\n", "\n", "This is a tutorial notebook."],
+            },
+            {
+                "cell_type": "code",
+                "execution_count": None,
+                "metadata": {},
+                "outputs": [],
+                "source": ["import cognee\n", "print('Hello, Cognee!')"],
+            },
+            {
+                "cell_type": "markdown",
+                "metadata": {},
+                "source": ["## Step 1: Data Ingestion\n", "\n", "Let's add some data."],
+            },
+            {
+                "cell_type": "code",
+                "execution_count": None,
+                "metadata": {},
+                "outputs": [],
+                "source": ["# Add your data here\n", "# await cognee.add('data.txt')"],
+            },
+            {
+                "cell_type": "raw",
+                "metadata": {},
+                "source": ["This is a raw cell that should be skipped"],
+            },
+        ],
+        "metadata": {
+            "kernelspec": {"display_name": "Python 3", "language": "python", "name": "python3"}
+        },
+        "nbformat": 4,
+        "nbformat_minor": 4,
+    }
+
+
+class TestTutorialNotebookCreation:
+    """Test cases for tutorial notebook creation functionality."""
+
+    @pytest.mark.asyncio
+    async def test_notebook_from_ipynb_string_success(self, sample_jupyter_notebook):
+        """Test successful creation of notebook from JSON string."""
+        notebook_json = json.dumps(sample_jupyter_notebook)
+        user_id = uuid4()
+
+        notebook = Notebook.from_ipynb_string(
+            notebook_content=notebook_json, owner_id=user_id, name="String Test Notebook"
+        )
+
+        assert notebook.owner_id == user_id
+        assert notebook.name == "String Test Notebook"
+        assert len(notebook.cells) == 4  # Should skip the raw cell
+        assert notebook.cells[0].type == "markdown"
+        assert notebook.cells[1].type == "code"
+
+    @pytest.mark.asyncio
+    async def test_notebook_cell_name_generation(self, sample_jupyter_notebook):
+        """Test that cell names are generated correctly from markdown headers."""
+        user_id = uuid4()
+        notebook_json = json.dumps(sample_jupyter_notebook)
+
+        notebook = Notebook.from_ipynb_string(notebook_content=notebook_json, owner_id=user_id)
+
+        # Check markdown header extraction
+        assert notebook.cells[0].name == "Tutorial Introduction"
+        assert notebook.cells[2].name == "Step 1: Data Ingestion"
+
+        # Check code cell naming
+        assert notebook.cells[1].name == "Code Cell"
+        assert notebook.cells[3].name == "Code Cell"
+
+    @pytest.mark.asyncio
+    async def test_notebook_from_ipynb_string_with_default_name(self, sample_jupyter_notebook):
+        """Test notebook creation uses kernelspec display_name when no name provided."""
+        user_id = uuid4()
+        notebook_json = json.dumps(sample_jupyter_notebook)
+
+        notebook = Notebook.from_ipynb_string(notebook_content=notebook_json, owner_id=user_id)
+
+        assert notebook.name == "Python 3"  # From kernelspec.display_name
+
+    @pytest.mark.asyncio
+    async def test_notebook_from_ipynb_string_fallback_name(self):
+        """Test fallback naming when kernelspec is missing."""
+        minimal_notebook = {
+            "cells": [{"cell_type": "markdown", "metadata": {}, "source": ["# Test"]}],
+            "metadata": {},  # No kernelspec
+            "nbformat": 4,
+            "nbformat_minor": 4,
+        }
+
+        user_id = uuid4()
+        notebook_json = json.dumps(minimal_notebook)
+
+        notebook = Notebook.from_ipynb_string(notebook_content=notebook_json, owner_id=user_id)
+
+        assert notebook.name == "Imported Notebook"  # Fallback name
+
+    @pytest.mark.asyncio
+    async def test_notebook_from_ipynb_string_invalid_json(self):
+        """Test error handling for invalid JSON."""
+        user_id = uuid4()
+        invalid_json = "{ invalid json content"
+
+        from nbformat.reader import NotJSONError
+
+        with pytest.raises(NotJSONError):
+            Notebook.from_ipynb_string(notebook_content=invalid_json, owner_id=user_id)
+
+    @pytest.mark.asyncio
+    @patch.object(Notebook, "from_ipynb_zip_url")
+    async def test_create_tutorial_notebook_error_propagated(self, mock_from_zip_url, mock_session):
+        """Test that errors are propagated when zip fetch fails."""
+        user_id = uuid4()
+        mock_from_zip_url.side_effect = Exception("Network error")
+
+        # Should raise the exception (not catch it)
+        with pytest.raises(Exception, match="Network error"):
+            await _create_tutorial_notebook(user_id, mock_session)
+
+        # Verify error handling path was taken
+        mock_from_zip_url.assert_called_once()
+        mock_session.add.assert_not_called()
+        mock_session.commit.assert_not_called()
+
+    def test_generate_cell_name_code_cell(self):
+        """Test cell name generation for code cells."""
+        from nbformat.notebooknode import NotebookNode
+
+        mock_cell = NotebookNode(
+            {"cell_type": "code", "source": 'import pandas as pd\nprint("Hello world")'}
+        )
+
+        result = Notebook._generate_cell_name(mock_cell)
+        assert result == "Code Cell"
+
+
+class TestTutorialNotebookZipFunctionality:
+    """Test cases for zip-based tutorial functionality."""
+
+    @pytest.mark.asyncio
+    async def test_notebook_from_ipynb_zip_url_missing_notebook(
+        self,
+    ):
+        """Test error handling when notebook file is missing from zip."""
+        user_id = uuid4()
+
+        with pytest.raises(
+            FileNotFoundError,
+            match="Notebook file 'super_random_tutorial_name.ipynb' not found in zip",
+        ):
+            await Notebook.from_ipynb_zip_url(
+                zip_url="https://github.com/topoteretes/cognee/raw/notebook_tutorial/notebooks/starter_tutorial.zip",
+                owner_id=user_id,
+                notebook_filename="super_random_tutorial_name.ipynb",
+            )
+
+    @pytest.mark.asyncio
+    async def test_notebook_from_ipynb_zip_url_download_failure(self):
+        """Test error handling when zip download fails."""
+        user_id = uuid4()
+        with pytest.raises(RuntimeError, match="Failed to download tutorial zip"):
+            await Notebook.from_ipynb_zip_url(
+                zip_url="https://github.com/topoteretes/cognee/raw/notebook_tutorial/notebooks/nonexistent_tutorial_name.zip",
+                owner_id=user_id,
+            )
+
+    @pytest.mark.asyncio
+    async def test_create_tutorial_notebook_zip_success(self, mock_session):
+        """Test successful tutorial notebook creation with zip."""
+        await cognee.prune.prune_data()
+        await cognee.prune.prune_system(metadata=True)
+
+        user_id = uuid4()
+
+        # Check that tutorial data directory is empty using storage-aware method
+        tutorial_data_dir_path = await get_tutorial_data_dir()
+        tutorial_data_dir = Path(tutorial_data_dir_path)
+        if tutorial_data_dir.exists():
+            assert not any(tutorial_data_dir.iterdir()), "Tutorial data directory should be empty"
+
+        await _create_tutorial_notebook(user_id, mock_session)
+
+        items = list(tutorial_data_dir.iterdir())
+        assert len(items) == 1, "Tutorial data directory should contain exactly one item"
+        assert items[0].is_dir(), "Tutorial data directory item should be a directory"
+
+        # Verify the structure inside the tutorial directory
+        tutorial_dir = items[0]
+
+        # Check for tutorial.ipynb file
+        notebook_file = tutorial_dir / "tutorial.ipynb"
+        assert notebook_file.exists(), f"tutorial.ipynb should exist in {tutorial_dir}"
+        assert notebook_file.is_file(), "tutorial.ipynb should be a file"
+
+        # Check for data subfolder with contents
+        data_folder = tutorial_dir / "data"
+        assert data_folder.exists(), f"data subfolder should exist in {tutorial_dir}"
+        assert data_folder.is_dir(), "data should be a directory"
+
+        data_items = list(data_folder.iterdir())
+        assert len(data_items) > 0, (
+            f"data folder should contain files, but found {len(data_items)} items"
+        )
+
+    @pytest.mark.asyncio
+    async def test_create_tutorial_notebook_with_force_refresh(self, mock_session):
+        """Test tutorial notebook creation with force refresh."""
+        await cognee.prune.prune_data()
+        await cognee.prune.prune_system(metadata=True)
+
+        user_id = uuid4()
+
+        # Check that tutorial data directory is empty using storage-aware method
+        tutorial_data_dir_path = await get_tutorial_data_dir()
+        tutorial_data_dir = Path(tutorial_data_dir_path)
+        if tutorial_data_dir.exists():
+            assert not any(tutorial_data_dir.iterdir()), "Tutorial data directory should be empty"
+
+        # First creation (without force refresh)
+        await _create_tutorial_notebook(user_id, mock_session, force_refresh=False)
+
+        items_first = list(tutorial_data_dir.iterdir())
+        assert len(items_first) == 1, (
+            "Tutorial data directory should contain exactly one item after first creation"
+        )
+        first_dir = items_first[0]
+        assert first_dir.is_dir(), "Tutorial data directory item should be a directory"
+
+        # Verify the structure inside the tutorial directory (first creation)
+        notebook_file = first_dir / "tutorial.ipynb"
+        assert notebook_file.exists(), f"tutorial.ipynb should exist in {first_dir}"
+        assert notebook_file.is_file(), "tutorial.ipynb should be a file"
+
+        data_folder = first_dir / "data"
+        assert data_folder.exists(), f"data subfolder should exist in {first_dir}"
+        assert data_folder.is_dir(), "data should be a directory"
+
+        data_items = list(data_folder.iterdir())
+        assert len(data_items) > 0, (
+            f"data folder should contain files, but found {len(data_items)} items"
+        )
+
+        # Capture metadata from first creation
+
+        first_creation_metadata = {}
+
+        for file_path in first_dir.rglob("*"):
+            if file_path.is_file():
+                relative_path = file_path.relative_to(first_dir)
+                stat = file_path.stat()
+
+                # Store multiple metadata points
+                with open(file_path, "rb") as f:
+                    content = f.read()
+
+                first_creation_metadata[str(relative_path)] = {
+                    "mtime": stat.st_mtime,
+                    "size": stat.st_size,
+                    "hash": hashlib.md5(content).hexdigest(),
+                    "first_bytes": content[:100]
+                    if content
+                    else b"",  # First 100 bytes as fingerprint
+                }
+
+        # Wait a moment to ensure different timestamps
+        time.sleep(0.1)
+
+        # Force refresh - should create new files with different metadata
+        await _create_tutorial_notebook(user_id, mock_session, force_refresh=True)
+
+        items_second = list(tutorial_data_dir.iterdir())
+        assert len(items_second) == 1, (
+            "Tutorial data directory should contain exactly one item after force refresh"
+        )
+        second_dir = items_second[0]
+
+        # Verify the structure is maintained after force refresh
+        notebook_file_second = second_dir / "tutorial.ipynb"
+        assert notebook_file_second.exists(), (
+            f"tutorial.ipynb should exist in {second_dir} after force refresh"
+        )
+        assert notebook_file_second.is_file(), "tutorial.ipynb should be a file after force refresh"
+
+        data_folder_second = second_dir / "data"
+        assert data_folder_second.exists(), (
+            f"data subfolder should exist in {second_dir} after force refresh"
+        )
+        assert data_folder_second.is_dir(), "data should be a directory after force refresh"
+
+        data_items_second = list(data_folder_second.iterdir())
+        assert len(data_items_second) > 0, (
+            f"data folder should still contain files after force refresh, but found {len(data_items_second)} items"
+        )
+
+        # Compare metadata to ensure files are actually different
+        files_with_changed_metadata = 0
+
+        for file_path in second_dir.rglob("*"):
+            if file_path.is_file():
+                relative_path = file_path.relative_to(second_dir)
+                relative_path_str = str(relative_path)
+
+                # File should exist from first creation
+                assert relative_path_str in first_creation_metadata, (
+                    f"File {relative_path_str} missing from first creation"
+                )
+
+                old_metadata = first_creation_metadata[relative_path_str]
+
+                # Get new metadata
+                stat = file_path.stat()
+                with open(file_path, "rb") as f:
+                    new_content = f.read()
+
+                new_metadata = {
+                    "mtime": stat.st_mtime,
+                    "size": stat.st_size,
+                    "hash": hashlib.md5(new_content).hexdigest(),
+                    "first_bytes": new_content[:100] if new_content else b"",
+                }
+
+                # Check if any metadata changed (indicating file was refreshed)
+                metadata_changed = (
+                    new_metadata["mtime"] > old_metadata["mtime"]  # Newer modification time
+                    or new_metadata["hash"] != old_metadata["hash"]  # Different content hash
+                    or new_metadata["size"] != old_metadata["size"]  # Different file size
+                    or new_metadata["first_bytes"]
+                    != old_metadata["first_bytes"]  # Different content
+                )
+
+                if metadata_changed:
+                    files_with_changed_metadata += 1
+
+        # Assert that force refresh actually updated files
+        assert files_with_changed_metadata > 0, (
+            f"Force refresh should have updated at least some files, but all {len(first_creation_metadata)} "
+            f"files appear to have identical metadata. This suggests force refresh didn't work."
+        )
+
+        mock_session.commit.assert_called()
+
+    @pytest.mark.asyncio
+    async def test_tutorial_zip_url_accessibility(self):
+        """Test that the actual tutorial zip URL is accessible (integration test)."""
+        try:
+            import requests
+
+            response = requests.get(
+                "https://github.com/topoteretes/cognee/raw/notebook_tutorial/notebooks/starter_tutorial.zip",
+                timeout=10,
+            )
+            response.raise_for_status()
+
+            # Verify it's a valid zip file by checking headers
+            assert response.headers.get("content-type") in [
+                "application/zip",
+                "application/octet-stream",
+                "application/x-zip-compressed",
+            ] or response.content.startswith(b"PK")  # Zip file signature
+
+        except Exception:
+            pytest.skip("Network request failed or zip not available - skipping integration test")

{cognee-0.3.4.dev2.dist-info → cognee-0.3.4.dev3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cognee
-Version: 0.3.4.dev2
+Version: 0.3.4.dev3
 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
@@ -33,6 +33,7 @@ Requires-Dist: langfuse<3,>=2.32.0
 Requires-Dist: limits<5,>=4.4.1
 Requires-Dist: litellm<2.0.0,>=1.71.0
 Requires-Dist: matplotlib<4,>=3.8.3
+Requires-Dist: nbformat<6.0.0,>=5.7.0
 Requires-Dist: networkx<4,>=3.4.2
 Requires-Dist: nltk<4.0.0,>=3.9.1
 Requires-Dist: numpy<=4.0.0,>=1.26.4

{cognee-0.3.4.dev2.dist-info → cognee-0.3.4.dev3.dist-info}/RECORD

@@ -1,6 +1,6 @@
 cognee/__init__.py,sha256=0KblJJ9tZI6zlsojrK9w-075uDGjDjbCGnB9tt5J0pY,1068
 cognee/__main__.py,sha256=UDwkdKir_2m9z3DtOlWKc5wdBk_8AXGyu9k6PufY9Jg,75
-cognee/base_config.py,sha256=safPCD_1LISzfiIl3i5TG46pXqhUgESwI0-03vPYrok,1540
+cognee/base_config.py,sha256=s5syv0WwaLJzoBKqImxNAbzxwUaziZ9wCPkCIft4xyE,2451
 cognee/context_global_variables.py,sha256=X0PDjVwwB8qaDTGtdjHRzKvUo9fv4ygmxLz6g0FM9bc,2679
 cognee/get_token.py,sha256=6qrXc5P0zal6zo90gVYlZHv5g5bpic9i_J_2mAkNri0,634
 cognee/low_level.py,sha256=ffJMBQDYwXiYk_clHCCy7N7zIjJqhUUulNJ8ot4Xx0E,131
@@ -45,7 +45,7 @@ cognee/api/v1/notebooks/routers/get_notebooks_router.py,sha256=m8OH3Kw1UHF8aTP4y
 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
-cognee/api/v1/prune/prune.py,sha256=rmH2i2k1STqxt4bsgCipqR-vgdUEgowfTzh3EgfaGsY,492
+cognee/api/v1/prune/prune.py,sha256=e5Wavom90OqBoehBA4The-AUw6sCtXV2sJ1v2pdssFk,511
 cognee/api/v1/responses/__init__.py,sha256=HMs5gmses4IFGqDnadqWCkGnxIfWTRUUbJk2MolxiE4,101
 cognee/api/v1/responses/default_tools.py,sha256=6_cR9KOdN8IHMKMSMu9sT5uDL8qZzApO6TrWfkyuHrw,2309
 cognee/api/v1/responses/dispatch_function.py,sha256=LJoV618_IIqVb5JEmQKEQl6LsEgAl75P5Ejfqw0vRsU,3689
@@ -60,7 +60,7 @@ cognee/api/v1/search/routers/get_search_router.py,sha256=-5GLgHipflEblYAwl3uiPAZ
 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
-cognee/api/v1/sync/sync.py,sha256=zzCVJD1AvcSXtNsgLJr1iPMRxY6vRxGdkt7sVdJ8W2c,33905
+cognee/api/v1/sync/sync.py,sha256=N-bRZJQuYeK2tffaVgRNf0hjAWFNWSFILPfuXLyvIgo,34582
 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
@@ -158,7 +158,7 @@ cognee/infrastructure/databases/graph/use_graph_adapter.py,sha256=a_T2NIhSw-cxn9
 cognee/infrastructure/databases/graph/kuzu/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cognee/infrastructure/databases/graph/kuzu/adapter.py,sha256=bZav3kZwge6PEMDrgYXb3M4z8gGUuV4qIw7cv0mVIQA,68094
 cognee/infrastructure/databases/graph/kuzu/kuzu_migrate.py,sha256=SpNuvw2f8wOFVm6BXOVsiQs_5n3SZMKz4IhuuHvH2_U,10542
-cognee/infrastructure/databases/graph/kuzu/remote_kuzu_adapter.py,sha256=j0GOZl2yTvB_9JbulPec9gfQLoiTH-t932-uW3Ufr_0,7324
+cognee/infrastructure/databases/graph/kuzu/remote_kuzu_adapter.py,sha256=lZbpkPBpqS3XbhD3v3sV1dYBolGb4iEV_1ePpl52Z6g,7517
 cognee/infrastructure/databases/graph/kuzu/show_remote_kuzu_stats.py,sha256=l5TQUORnoCusIrPNbTuNDzVvUUAMjQNak-9I8KT7N3I,1044
 cognee/infrastructure/databases/graph/memgraph/memgraph_adapter.py,sha256=eoOpDEn6lw6rVXxj00Ek3lkLjxLDLtTmQbjezMYf850,34376
 cognee/infrastructure/databases/graph/neo4j_driver/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -198,7 +198,7 @@ cognee/infrastructure/databases/vector/chromadb/__init__.py,sha256=47DEQpj8HBSa-
 cognee/infrastructure/databases/vector/embeddings/EmbeddingEngine.py,sha256=boNJ55dxJQ_ImW1_DDjToQa0Hos9mkeRYwfCI7UPLn0,983
 cognee/infrastructure/databases/vector/embeddings/FastembedEmbeddingEngine.py,sha256=_R3yIuDaMN2lz9JhMy6SNpZeeCRZxHA9hmSB3gOxKkA,3823
 cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py,sha256=XUZnVftE57qWlAebr99aOEg-FynMKB7IS-kmBBT8E5Y,7544
-cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py,sha256=uR9ItOYN0ySsnPrmHGaoLGjiKJcFF-88KMwbtH6j0DU,4173
+cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py,sha256=nBA9XaKkOvDRg_ud9oFejD6FpCH0TTEVQbTxBHaRFaY,4358
 cognee/infrastructure/databases/vector/embeddings/__init__.py,sha256=Akv-ShdXjHw-BE00Gw55GgGxIMr0SZ9FHi3RlpsJmiE,55
 cognee/infrastructure/databases/vector/embeddings/config.py,sha256=s9acnhn1DLFggCNJMVcN9AxruMf3J00O_R--JVGqMNs,2221
 cognee/infrastructure/databases/vector/embeddings/embedding_rate_limiter.py,sha256=TyCoo_SipQ6JNy5eqXY2shrZnhb2JVjt9xOsJltOCdw,17598
@@ -227,9 +227,9 @@ cognee/infrastructure/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
 cognee/infrastructure/files/__init__.py,sha256=j1cLINAadEfM0yDe1Rq-klWu25jRHvEFJsaU5kqfElc,69
 cognee/infrastructure/files/exceptions.py,sha256=bXRG_AIp_bQ3uVNsyiIHoT2i43Vt8Bsk0o4nQxmhLKo,388
 cognee/infrastructure/files/storage/FileBufferedReader.py,sha256=j0NGfk4GVgCPvWHWePHmJ8cJ_Bu4AgK2BetnGX2aB0o,360
-cognee/infrastructure/files/storage/LocalFileStorage.py,sha256=Bd6go8qC9bXBRUT-oLo-PQmxI5I9UiIeeBqxU66BhO4,9835
-cognee/infrastructure/files/storage/S3FileStorage.py,sha256=Nxph8kOV_rafCB7wrfgjb4uwqWC8_wu_DosuzCKqnWM,7876
-cognee/infrastructure/files/storage/StorageManager.py,sha256=kYlKxEUO_21vSBcSeO0OlR9QdkPWmZcm0vuNR7Oo75A,4946
+cognee/infrastructure/files/storage/LocalFileStorage.py,sha256=QGbttdjXeJkfIjhBElh7KJEytVyJFwDxSz2AzUFqEFM,11771
+cognee/infrastructure/files/storage/S3FileStorage.py,sha256=2ZKUFzIyOSyM7XAZD6hs13dj-74ylR7zL6Eptp0hSbY,9904
+cognee/infrastructure/files/storage/StorageManager.py,sha256=IAReStvUA5cQIaxTamSp92yFvvnaPHIpUkRZNCCiWNM,5643
 cognee/infrastructure/files/storage/__init__.py,sha256=M4QOn-jddfTWzaeZ6UxD4GMScf-DJpwDaNv80Olk0pM,141
 cognee/infrastructure/files/storage/config.py,sha256=uoI3--AEM_s82hlu8FA4anDE2WzR4Zx9bvS9BEj-CXs,107
 cognee/infrastructure/files/storage/get_file_storage.py,sha256=yf91iWBtWGAlhQ1TjR0wrs7nduytb-bLJuFL4gv7uVM,797
@@ -377,13 +377,13 @@ cognee/modules/cloud/exceptions/CloudApiKeyMissingError.py,sha256=kFmTLGSxFTfiZS
 cognee/modules/cloud/exceptions/CloudConnectionError.py,sha256=FBy-CEwitOxnQnRXlV5tT1FrKT8MzfpperivP6Pw0C0,527
 cognee/modules/cloud/exceptions/__init__.py,sha256=HpgBbUFJsSAl6-xWN6wF6c5iMdp2Gm34Nr_ThX7Mljw,116
 cognee/modules/cloud/operations/__init__.py,sha256=sg1lEtgTnNobMLqzfaMbpX4xpepeYQnEh-biEfX41WQ,41
-cognee/modules/cloud/operations/check_api_key.py,sha256=KkAC4egW7ZGgrWu852oGOwkCPzhXq4DpwNCQf1M_04I,824
+cognee/modules/cloud/operations/check_api_key.py,sha256=qfu1WijhoNTlNRqxI1Q1MV_hrO9EDroVsShBLfr_MiU,1009
 cognee/modules/cognify/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cognee/modules/cognify/config.py,sha256=Neh-uOnFr8WgOe3GO7NFAX65E0ZwEiSD0JoLsKEsEhg,683
 cognee/modules/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cognee/modules/data/deletion/__init__.py,sha256=bEOcuz1MPPd7Cqqwd7B4B22Xa81oYdUOTombamtto8g,74
 cognee/modules/data/deletion/prune_data.py,sha256=Aus7o3PDJPEaTh2u0yCHkT92rEesS0NRu12vwBE2D_s,278
-cognee/modules/data/deletion/prune_system.py,sha256=4o5yb5jrs6UR09-iEINB_eGOA8_0ZgUl0ZDOYYkuj4A,600
+cognee/modules/data/deletion/prune_system.py,sha256=5uR75qLvy__5CX5aQZpQRfC_W9BiT7pPXE-DX3BojBI,701
 cognee/modules/data/exceptions/__init__.py,sha256=8eKBra_q0CcaShB6jca-EM1gLTMRXxc9iWAJCyoFqXs,266
 cognee/modules/data/exceptions/exceptions.py,sha256=fYdHFXwOGQAyTcJGbvrtZjTGb4ipnwh2FK5CPHxiZlE,1720
 cognee/modules/data/methods/__init__.py,sha256=o_H5bkTu626ttisr2rZzdFFepCR9RVr6lpF_GUCR7_8,869
@@ -481,12 +481,12 @@ cognee/modules/memify/memify.py,sha256=EFPaFpUm6gCz4pV3E49gUeeiuRBf28EXSeytd-aPx
 cognee/modules/metrics/operations/__init__.py,sha256=MZ3xbVdfEKqfLct8WnbyFVyZmkBfl-77GoBQgktilUM,63
 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_notebook.py,sha256=3gOVaEE7WOAZ6pLx1TvsjFtE1Eo3vKdA9YHQuwAc0qQ,1700
 cognee/modules/notebooks/methods/delete_notebook.py,sha256=BKxoRlPzkwXvTYh5WcF-zo_iVmaXqEiptS42JwB0KQU,309
 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=MnZbfh-WfEfH3ImNvyQNhDeNwpYeS7p8FPVwnmBvZVg,361
-cognee/modules/notebooks/models/Notebook.py,sha256=Jth47QxJQ2-VGPyIcS0ul3bS8bgGrk9vCGoJVagxanw,1690
+cognee/modules/notebooks/models/Notebook.py,sha256=jr25KxLuf-P679e4TrjIQNLPjv0W0MBeV_8YHRfwljw,9740
 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=17hMEQC3LZTfPvbRUrPN9SzDeJPWSTq_BAhtwRZiqT8,1338
@@ -625,7 +625,7 @@ cognee/modules/users/exceptions/__init__.py,sha256=wDaNSgJtRXA8xMw3qhPX3FS8dFOz4
 cognee/modules/users/exceptions/exceptions.py,sha256=J4xBeR0VtKMpr8sQwHD2BOWSzBfPdUn700Z0bFCBxwM,1639
 cognee/modules/users/methods/__init__.py,sha256=1J0g3-WracmU-1wruczdFirpJuRmJdd-MNtMQl8tiSY,350
 cognee/modules/users/methods/create_default_user.py,sha256=MRdbYof0NJZARO6t3wUr0hw6Kk0lJ_aE3kOvGy9HbW8,555
-cognee/modules/users/methods/create_user.py,sha256=IJCw95rA-TlWUQkihX1YxJ7TUfDNolUjaYdFsRTQbSA,3686
+cognee/modules/users/methods/create_user.py,sha256=2xA-6YngXQJVwkkatuNP3wse29EuZMy84Ul1TjTZwYY,2788
 cognee/modules/users/methods/delete_user.py,sha256=B7NJz_ar4jLhfr2QbNsruUg-LNSoT6MYYG6OF72GNcg,758
 cognee/modules/users/methods/get_authenticated_user.py,sha256=V0WdCsUK7RAMfH46sthdyOZbTtoQ-0AHP0fh868NK7A,1554
 cognee/modules/users/methods/get_default_user.py,sha256=UHxy5htJypsIQv37KrMtODiQrc-Cjp9ojzU1o8ThDRo,1771
@@ -672,10 +672,11 @@ cognee/shared/GithubClassification.py,sha256=3B-CghZ6F3hDrwoKBtJ83Zr0AyLcnLQZn6O
 cognee/shared/GithubTopology.py,sha256=eYFF4oBjqRPMnH2ufu3bo2_HGElXQTZ0aH4vQD9-l2c,975
 cognee/shared/SourceCodeGraph.py,sha256=9j3vIiatAP6tEA_0CPQ__y2huTFhvuSNKcFhGYeAm_4,2088
 cognee/shared/__init__.py,sha256=2V1IwCRt4hY5KIsnK9oMj8-MaiiK-i3QD18FJYlIjP4,154
+cognee/shared/cache.py,sha256=VZEA-nEXkdkF0nRAYp_Fe1ntoL7OjyFFbyXassaJL-c,13923
 cognee/shared/data_models.py,sha256=T_Vsjy7Gvf_3_NUxS-7tSc-nRyOmLGFAaSO2w7Ns6J8,10631
 cognee/shared/encode_uuid.py,sha256=-EVtObzdnhSXKb5gz-SH1abicaaeiJ2rqcrLMr8V_cs,366
 cognee/shared/logging_utils.py,sha256=xzUo8mMfD_8t8mcJyENKxWvrtd5UGFJMltw_LW5RA2s,20368
-cognee/shared/utils.py,sha256=1pNTlMcrfchiAobyXdLsu6M1wm2nXPMg0qsN5yK4XqU,55185
+cognee/shared/utils.py,sha256=lvhS4_IMyVIUWdNC1SYtatSP9m0tuh_unZn1YBg36lQ,55534
 cognee/shared/exceptions/__init__.py,sha256=NNi2QcqIxj8tZNuQP5FC-kJCw1e5jcRaoQAmkQEpyjM,179
 cognee/shared/exceptions/exceptions.py,sha256=-sCKV-NReNDLSa98iZqT22s2rUfk0geMGJVsYRyzy3Q,361
 cognee/tasks/chunk_naive_llm_classifier/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -728,7 +729,7 @@ cognee/tasks/ingestion/get_dlt_destination.py,sha256=vzky_-TFlnaREbdbndAkXwwlb-0
 cognee/tasks/ingestion/ingest_data.py,sha256=AWdQ5YKzYZB4nkra1Zm5SJ0F-bXOcsnLdpp83eAX68w,8280
 cognee/tasks/ingestion/migrate_relational_database.py,sha256=dbHu7TAMjq9zH69bT9q1ziQwXCoZ-VvJY5uY5aer0nc,10260
 cognee/tasks/ingestion/resolve_data_directories.py,sha256=Q8wDNY62S9Rutj5EHYe7nB51p6BonUIsm7E_ZBy7WpY,3251
-cognee/tasks/ingestion/save_data_item_to_storage.py,sha256=zJXRqe0icmSh0tL8qQ5-snMAOeifN5bD_Oqz6_YNg0g,3424
+cognee/tasks/ingestion/save_data_item_to_storage.py,sha256=NHkKUPEnoyoplOO5yMcKM-L_P6_ww-2zmGs0mkd23yE,3457
 cognee/tasks/ingestion/transform_data.py,sha256=cbI1FX86-Ft3pGRRHedjarXst9TR7O9ce39bYRhbf2Y,1506
 cognee/tasks/ingestion/exceptions/__init__.py,sha256=GhFSvM6ChVsm16b-zaCkbluH9zCX72Wy-QLyMXyEAe4,240
 cognee/tasks/ingestion/exceptions/exceptions.py,sha256=3wgLiK4G77nMc95-STtqs3LPWcRtzH_bDamikG4uyD0,385
@@ -865,6 +866,7 @@ cognee/tests/unit/modules/retrieval/temporal_retriever_test.py,sha256=bvGvJgq9JF
 cognee/tests/unit/modules/retriever/test_description_to_codepart_search.py,sha256=oayCbXQtvmTnlgOuR67w_r278TGMEv-puaTR_jI6weo,4164
 cognee/tests/unit/modules/users/__init__.py,sha256=SkGMpbXemeX0834-eUj14VuNlZgtOGMxk0C-r-jSsnU,37
 cognee/tests/unit/modules/users/test_conditional_authentication.py,sha256=KfPWLMbPpbm3stg7LxIaK5cWVPZVEhMaxPX5s6grLw8,11120
+cognee/tests/unit/modules/users/test_tutorial_notebook_creation.py,sha256=M_NXBSLr-U7MqKtHiERMRzI7pWRm0CywZYzFUwKYV0w,16028
 cognee/tests/unit/modules/visualization/visualization_test.py,sha256=JuNsyqAbEWgO5QZP1lCE0_MDQDJ75WhXLXPyat0mhVw,929
 cognee/tests/unit/processing/chunks/chunk_by_paragraph_2_test.py,sha256=KGZZq1cMgn-im9hgfeNW74zobSLSOwuzQKzwNBmW6sM,2487
 cognee/tests/unit/processing/chunks/chunk_by_paragraph_test.py,sha256=GRLoeusPg-jzaSLXhZv0zNpmt8gd5vNrCYCAzR5W1TE,2658
@@ -889,9 +891,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.4.dev2.dist-info/METADATA,sha256=bPHsZZiZEo6spQLzpKDqBj8Gf_yyfvAjwavk6lRram4,15097
-cognee-0.3.4.dev2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-cognee-0.3.4.dev2.dist-info/entry_points.txt,sha256=GCCTsNg8gzOJkolq7dR7OK1VlIAO202dGDnMI8nm8oQ,55
-cognee-0.3.4.dev2.dist-info/licenses/LICENSE,sha256=pHHjSQj1DD8SDppW88MMs04TPk7eAanL1c5xj8NY7NQ,11344
-cognee-0.3.4.dev2.dist-info/licenses/NOTICE.md,sha256=6L3saP3kSpcingOxDh-SGjMS8GY79Rlh2dBNLaO0o5c,339
-cognee-0.3.4.dev2.dist-info/RECORD,,
+cognee-0.3.4.dev3.dist-info/METADATA,sha256=LmGc7VycRZe5MVrtkla4l4YVK62TUhY5E6vjAOZe0oI,15135
+cognee-0.3.4.dev3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+cognee-0.3.4.dev3.dist-info/entry_points.txt,sha256=GCCTsNg8gzOJkolq7dR7OK1VlIAO202dGDnMI8nm8oQ,55
+cognee-0.3.4.dev3.dist-info/licenses/LICENSE,sha256=pHHjSQj1DD8SDppW88MMs04TPk7eAanL1c5xj8NY7NQ,11344
+cognee-0.3.4.dev3.dist-info/licenses/NOTICE.md,sha256=6L3saP3kSpcingOxDh-SGjMS8GY79Rlh2dBNLaO0o5c,339
+cognee-0.3.4.dev3.dist-info/RECORD,,