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

cognee/modules/notebooks/methods/create_notebook.py

@@ -6,6 +6,8 @@ from cognee.infrastructure.databases.relational import with_async_session
 
 from ..models.Notebook import Notebook, NotebookCell
 
+TUTORIAL_NOTEBOOK_NAME = "Python Development with Cognee Tutorial 🧠"
+
 
 async def _create_tutorial_notebook(
     user_id: UUID, session: AsyncSession, force_refresh: bool = False
@@ -24,7 +26,7 @@ async def _create_tutorial_notebook(
             zip_url=TUTORIAL_ZIP_URL,
             owner_id=user_id,
             notebook_filename="tutorial.ipynb",
-            name="Python Development with Cognee Tutorial 🧠",
+            name=TUTORIAL_NOTEBOOK_NAME,
             deletable=False,
             force=force_refresh,
         )

cognee/modules/notebooks/methods/get_notebooks.py

@@ -1,11 +1,16 @@
 from uuid import UUID
 from typing import List
-from sqlalchemy import select
+from sqlalchemy import select, and_
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from cognee.infrastructure.databases.relational import with_async_session
 
 from ..models.Notebook import Notebook
+from .create_notebook import _create_tutorial_notebook, TUTORIAL_NOTEBOOK_NAME
+
+from cognee.shared.logging_utils import get_logger
+
+logger = get_logger()
 
 
 @with_async_session
@@ -13,6 +18,27 @@ async def get_notebooks(
     user_id: UUID,
     session: AsyncSession,
 ) -> List[Notebook]:
+    # Check if tutorial notebook already exists for this user
+    tutorial_query = select(Notebook).where(
+        and_(
+            Notebook.owner_id == user_id,
+            Notebook.name == TUTORIAL_NOTEBOOK_NAME,
+            ~Notebook.deletable,
+        )
+    )
+    tutorial_result = await session.execute(tutorial_query)
+    tutorial_notebook = tutorial_result.scalar_one_or_none()
+
+    # If tutorial notebook doesn't exist, create it
+    if tutorial_notebook is None:
+        logger.info(f"Tutorial notebook not found for user {user_id}, creating it")
+        try:
+            await _create_tutorial_notebook(user_id, session, force_refresh=False)
+        except Exception as e:
+            # Log the error but continue to return existing notebooks
+            logger.error(f"Failed to create tutorial notebook for user {user_id}: {e}")
+
+    # Get all notebooks for the user
     result = await session.execute(select(Notebook).where(Notebook.owner_id == user_id))
 
     return list(result.scalars().all())

cognee/modules/users/methods/create_user.py

@@ -61,8 +61,6 @@ async def create_user(
                     if auto_login:
                         await session.refresh(user)
 
-                    await _create_tutorial_notebook(user.id, session)
-
                     return user
     except UserAlreadyExists as error:
         print(f"User {email} already exists")

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cognee
-Version: 0.3.4.dev3
+Version: 0.3.4.dev4
 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.4.dev3.dist-info → cognee-0.3.4.dev4.dist-info}/RECORD

@@ -481,10 +481,10 @@ 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=3gOVaEE7WOAZ6pLx1TvsjFtE1Eo3vKdA9YHQuwAc0qQ,1700
+cognee/modules/notebooks/methods/create_notebook.py,sha256=K-lWLWNG-Lh-aZYT6KBlCu2X6Ff3IW3okKLowupLJ04,1749
 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/get_notebooks.py,sha256=vxvhjiUh2quisDPv92N711ioODpCuSNyfH4jxbYF9GA,1523
 cognee/modules/notebooks/methods/update_notebook.py,sha256=MnZbfh-WfEfH3ImNvyQNhDeNwpYeS7p8FPVwnmBvZVg,361
 cognee/modules/notebooks/models/Notebook.py,sha256=jr25KxLuf-P679e4TrjIQNLPjv0W0MBeV_8YHRfwljw,9740
 cognee/modules/notebooks/models/__init__.py,sha256=jldsDjwRvFMreGpe4wxxr5TlFXTZuU7rbsRkGQvTO5s,45
@@ -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=2xA-6YngXQJVwkkatuNP3wse29EuZMy84Ul1TjTZwYY,2788
+cognee/modules/users/methods/create_user.py,sha256=kRMjd6WrjgwnO3zw9nlQIW6gswsQc6idusCmUwUzJYo,2717
 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
@@ -891,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.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,,
+cognee-0.3.4.dev4.dist-info/METADATA,sha256=pdKJriwhqmBgPKKCPeChX6EC0uxxMxeNGF7cLOEZoxI,15135
+cognee-0.3.4.dev4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+cognee-0.3.4.dev4.dist-info/entry_points.txt,sha256=GCCTsNg8gzOJkolq7dR7OK1VlIAO202dGDnMI8nm8oQ,55
+cognee-0.3.4.dev4.dist-info/licenses/LICENSE,sha256=pHHjSQj1DD8SDppW88MMs04TPk7eAanL1c5xj8NY7NQ,11344
+cognee-0.3.4.dev4.dist-info/licenses/NOTICE.md,sha256=6L3saP3kSpcingOxDh-SGjMS8GY79Rlh2dBNLaO0o5c,339
+cognee-0.3.4.dev4.dist-info/RECORD,,