From b80b775dd58ce29d5e93c40149996b6d5596aeb6 Mon Sep 17 00:00:00 2001
From: Masaki Yatsu <yatsu@yatsu.info>
Date: Sun, 30 Nov 2025 16:22:50 +0900
Subject: [PATCH] feat(jupyterhub): install falkordb and cognee packages to
 kernels

---
 jupyterhub/images/datastack-cuda-notebook/Dockerfile | 5 +++++
 jupyterhub/images/datastack-notebook/Dockerfile      | 5 +++++
 jupyterhub/jupyterhub-values.gomplate.yaml           | 8 ++++++++
 jupyterhub/justfile                                  | 2 +-
 4 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/jupyterhub/images/datastack-cuda-notebook/Dockerfile b/jupyterhub/images/datastack-cuda-notebook/Dockerfile
index b6426f7..8f982ad 100644
--- a/jupyterhub/images/datastack-cuda-notebook/Dockerfile
+++ b/jupyterhub/images/datastack-cuda-notebook/Dockerfile
@@ -147,7 +147,12 @@ ARG pip_repository_url="https://pypi.org/simple/"
 
 RUN --mount=type=cache,target=/home/${NB_USER}/.cache/pip pip install -i "${pip_repository_url}" \
     agno \
+    cognee \
+    cognee-integration-langgraph \
+    cognee-community-hybrid-adapter-falkor \
+    cognee-community-vector-adapter-qdrant \
     dagster-dlt \
+    falkordb \
     fastembed \
     feature-engine \
     kreuzberg \
diff --git a/jupyterhub/images/datastack-notebook/Dockerfile b/jupyterhub/images/datastack-notebook/Dockerfile
index c83357d..29accd0 100644
--- a/jupyterhub/images/datastack-notebook/Dockerfile
+++ b/jupyterhub/images/datastack-notebook/Dockerfile
@@ -146,7 +146,12 @@ ARG pip_repository_url="https://pypi.org/simple/"
 
 RUN --mount=type=cache,target=/home/${NB_USER}/.cache/pip pip install -i "${pip_repository_url}" \
     agno \
+    cognee \
+    cognee-integration-langgraph \
+    cognee-community-hybrid-adapter-falkor \
+    cognee-community-vector-adapter-qdrant \
     dagster-dlt \
+    falkordb \
     fastembed \
     feature-engine \
     kreuzberg \
diff --git a/jupyterhub/jupyterhub-values.gomplate.yaml b/jupyterhub/jupyterhub-values.gomplate.yaml
index a3dbd61..8ca1f4a 100644
--- a/jupyterhub/jupyterhub-values.gomplate.yaml
+++ b/jupyterhub/jupyterhub-values.gomplate.yaml
@@ -316,6 +316,14 @@ singleuser:
         ports:
           - port: 8181
             protocol: TCP
+      # Allow communication with FalkorDB
+      - to:
+          - namespaceSelector:
+              matchLabels:
+                kubernetes.io/metadata.name: falkordb
+        ports:
+          - port: 6379
+            protocol: TCP
       # Allow DNS resolution
       - to:
         - ipBlock:
diff --git a/jupyterhub/justfile b/jupyterhub/justfile
index 70eb329..f346d8c 100644
--- a/jupyterhub/justfile
+++ b/jupyterhub/justfile
@@ -9,7 +9,7 @@ export JUPYTERHUB_NFS_PV_ENABLED := env("JUPYTERHUB_NFS_PV_ENABLED", "")
 export JUPYTERHUB_STORAGE_CLASS := env("JUPYTERHUB_STORAGE_CLASS", "")
 export JUPYTERHUB_VAULT_INTEGRATION_ENABLED := env("JUPYTERHUB_VAULT_INTEGRATION_ENABLED", "")
 export JUPYTERHUB_AIRFLOW_DAGS_PERSISTENCE_ENABLED := env("JUPYTERHUB_AIRFLOW_DAGS_PERSISTENCE_ENABLED", "")
-export JUPYTER_PYTHON_KERNEL_TAG := env("JUPYTER_PYTHON_KERNEL_TAG", "python-3.12-51")
+export JUPYTER_PYTHON_KERNEL_TAG := env("JUPYTER_PYTHON_KERNEL_TAG", "python-3.12-52")
 export KERNEL_IMAGE_BUUN_STACK_REPOSITORY := env("KERNEL_IMAGE_BUUN_STACK_REPOSITORY", "buun-stack-notebook")
 export KERNEL_IMAGE_BUUN_STACK_CUDA_REPOSITORY := env("KERNEL_IMAGE_BUUN_STACK_CUDA_REPOSITORY", "buun-stack-cuda-notebook")
 export JUPYTER_PROFILE_MINIMAL_ENABLED := env("JUPYTER_PROFILE_MINIMAL_ENABLED", "false")