.github/workflows/release.yml

@@ -10,6 +10,10 @@ on:
           - dev
           - main
         description: Dev or Main release
+      test_mode:
+        required: true
+        type: boolean
+        description: Aka Dry Run. If true, it won't affect public indices or repositories
 
 jobs:
   release-github:
@@ -31,6 +35,8 @@ jobs:
 
       - name: Create and push git tag
         id: create_tag
+        env: 
+          TEST_MODE: ${{ inputs.test_mode }}
         run: |
           VERSION="$(uv version --short)"
           TAG="v${VERSION}"
@@ -43,9 +49,12 @@ jobs:
           echo "tag=${TAG}" >> "$GITHUB_OUTPUT"
           echo "version=${VERSION}" >> "$GITHUB_OUTPUT"
 
-          git tag "${TAG}"
+          if [ "$TEST_MODE" = "false" ]; then
-          git push origin "${TAG}"
+            git tag "${TAG}"
-          
+            git push origin "${TAG}"
+          else 
+            echo "Test mode is enabled. Skipping tag creation and push."
+          fi
 
       - name: Create GitHub Release
         uses: softprops/action-gh-release@v2
@@ -80,7 +89,14 @@ jobs:
       - name: Build distributions
         run: uv build
       
+      - name: Publish ${{ inputs.flavour }} release to TestPyPI
+        if: ${{ inputs.test_mode }}
+        env:
+          UV_PUBLISH_TOKEN: ${{ secrets.TEST_PYPI_TOKEN }}
+        run: uv publish --publish-url https://test.pypi.org/legacy/
+      
       - name: Publish ${{ inputs.flavour }} release to PyPI
+        if: ${{ !inputs.test_mode }}
         env:
           UV_PUBLISH_TOKEN: ${{ secrets.PYPI_TOKEN }}
         run: uv publish
@@ -113,7 +129,7 @@ jobs:
         with:
           context: .
           platforms: linux/amd64,linux/arm64
-          push: true
+          push: ${{ !inputs.test_mode }}
           tags: cognee/cognee:${{ needs.release-github.outputs.version }}
           labels: |
             version=${{ needs.release-github.outputs.version }}
@@ -127,7 +143,7 @@ jobs:
         with:
           context: .
           platforms: linux/amd64,linux/arm64
-          push: true
+          push: ${{ !inputs.test_mode }}
           tags: | 
             cognee/cognee:${{ needs.release-github.outputs.version }}
             cognee/cognee:latest

CONTRIBUTING.md

@@ -71,7 +71,7 @@ git clone https://github.com/<your-github-username>/cognee.git
 cd cognee
 ```
 In case you are working on Vector and Graph Adapters
-1. Fork the [**cognee-community**](https://github.com/topoteretes/cognee-community) repository
+1. Fork the [**cognee**](https://github.com/topoteretes/cognee-community) repository
 2. Clone your fork:
 ```shell
 git clone https://github.com/<your-github-username>/cognee-community.git
@@ -97,21 +97,6 @@ git checkout -b feature/your-feature-name
 python cognee/cognee/tests/test_library.py
 ```
 
-### Running Simple Example
-
-Change .env.example into .env and provide your OPENAI_API_KEY as LLM_API_KEY
-
-Make sure to run ```shell uv sync ``` in the root cloned folder or set up a virtual environment to run cognee
-
-```shell
-python cognee/cognee/examples/python/simple_example.py
-```
-or 
-
-```shell
-uv run python cognee/cognee/examples/python/simple_example.py
-```
-
 ## 4. 📤 Submitting Changes
 
 1. Install ruff on your system

README.md

@@ -66,10 +66,13 @@ Use your data to build personalized and dynamic memory for AI Agents. Cognee let
 ## About Cognee
 
 Cognee is an open-source tool and platform that transforms your raw data into persistent and dynamic AI memory for Agents. It combines vector search with graph databases to make your documents both searchable by meaning and connected by relationships. 
-Cognee offers default memory creation and search which we describe bellow. But with Cognee you can build your own!
 
+You can use Cognee in two ways:
 
-### Cognee Open Source:
+1. [Self-host Cognee Open Source](https://docs.cognee.ai/getting-started/installation), which stores all data locally by default.
+2. [Connect to Cognee Cloud](https://platform.cognee.ai/), and get the same OSS stack on managed infrastructure for easier development and productionization. 
+
+### Cognee Open Source (self-hosted):
 
 - Interconnects any type of data — including past conversations, files, images, and audio transcriptions
 - Replaces traditional RAG systems with a unified memory layer built on graphs and vectors
@@ -77,6 +80,11 @@ Cognee offers default memory creation and search which we describe bellow. But w
 - Provides Pythonic data pipelines for ingestion from 30+ data sources
 - Offers high customizability through user-defined tasks, modular pipelines, and built-in search endpoints
 
+### Cognee Cloud (managed):
+- Hosted web UI dashboard 
+- Automatic version updates 
+- Resource usage analytics
+- GDPR compliant, enterprise-grade security
 
 ## Basic Usage & Feature Guide
 
@@ -118,7 +126,6 @@ Now, run a minimal pipeline:
 ```python
 import cognee
 import asyncio
-from pprint import pprint
 
 
 async def main():
@@ -136,7 +143,7 @@ async def main():
 
     # Display the results
     for result in results:
-        pprint(result)
+        print(result)
 
 
 if __name__ == '__main__':

cognee-frontend/package-lock.json

@@ -12,7 +12,7 @@
         "classnames": "^2.5.1",
         "culori": "^4.0.1",
         "d3-force-3d": "^3.0.6",
-        "next": "16.1.1",
+        "next": "16.0.4",
         "react": "^19.2.0",
         "react-dom": "^19.2.0",
         "react-force-graph-2d": "^1.27.1",
@@ -96,6 +96,7 @@
       "integrity": "sha512-e7jT4DxYvIDLk1ZHmU/m/mB19rex9sv0c2ftBtjSBv+kVM/902eh0fINUzD7UwLLNR+jU585GxUJ8/EBfAM5fw==",
       "dev": true,
       "license": "MIT",
+      "peer": true,
       "dependencies": {
         "@babel/code-frame": "^7.27.1",
         "@babel/generator": "^7.28.5",
@@ -1073,9 +1074,9 @@
       }
     },
     "node_modules/@next/env": {
-      "version": "16.1.1",
+      "version": "16.0.4",
-      "resolved": "https://registry.npmjs.org/@next/env/-/env-16.1.1.tgz",
+      "resolved": "https://registry.npmjs.org/@next/env/-/env-16.0.4.tgz",
-      "integrity": "sha512-3oxyM97Sr2PqiVyMyrZUtrtM3jqqFxOQJVuKclDsgj/L728iZt/GyslkN4NwarledZATCenbk4Offjk1hQmaAA==",
+      "integrity": "sha512-FDPaVoB1kYhtOz6Le0Jn2QV7RZJ3Ngxzqri7YX4yu3Ini+l5lciR7nA9eNDpKTmDm7LWZtxSju+/CQnwRBn2pA==",
       "license": "MIT"
     },
     "node_modules/@next/eslint-plugin-next": {
@@ -1089,9 +1090,9 @@
       }
     },
     "node_modules/@next/swc-darwin-arm64": {
-      "version": "16.1.1",
+      "version": "16.0.4",
-      "resolved": "https://registry.npmjs.org/@next/swc-darwin-arm64/-/swc-darwin-arm64-16.1.1.tgz",
+      "resolved": "https://registry.npmjs.org/@next/swc-darwin-arm64/-/swc-darwin-arm64-16.0.4.tgz",
-      "integrity": "sha512-JS3m42ifsVSJjSTzh27nW+Igfha3NdBOFScr9C80hHGrWx55pTrVL23RJbqir7k7/15SKlrLHhh/MQzqBBYrQA==",
+      "integrity": "sha512-TN0cfB4HT2YyEio9fLwZY33J+s+vMIgC84gQCOLZOYusW7ptgjIn8RwxQt0BUpoo9XRRVVWEHLld0uhyux1ZcA==",
       "cpu": [
         "arm64"
       ],
@@ -1105,9 +1106,9 @@
       }
     },
     "node_modules/@next/swc-darwin-x64": {
-      "version": "16.1.1",
+      "version": "16.0.4",
-      "resolved": "https://registry.npmjs.org/@next/swc-darwin-x64/-/swc-darwin-x64-16.1.1.tgz",
+      "resolved": "https://registry.npmjs.org/@next/swc-darwin-x64/-/swc-darwin-x64-16.0.4.tgz",
-      "integrity": "sha512-hbyKtrDGUkgkyQi1m1IyD3q4I/3m9ngr+V93z4oKHrPcmxwNL5iMWORvLSGAf2YujL+6HxgVvZuCYZfLfb4bGw==",
+      "integrity": "sha512-XsfI23jvimCaA7e+9f3yMCoVjrny2D11G6H8NCcgv+Ina/TQhKPXB9P4q0WjTuEoyZmcNvPdrZ+XtTh3uPfH7Q==",
       "cpu": [
         "x64"
       ],
@@ -1121,9 +1122,9 @@
       }
     },
     "node_modules/@next/swc-linux-arm64-gnu": {
-      "version": "16.1.1",
+      "version": "16.0.4",
-      "resolved": "https://registry.npmjs.org/@next/swc-linux-arm64-gnu/-/swc-linux-arm64-gnu-16.1.1.tgz",
+      "resolved": "https://registry.npmjs.org/@next/swc-linux-arm64-gnu/-/swc-linux-arm64-gnu-16.0.4.tgz",
-      "integrity": "sha512-/fvHet+EYckFvRLQ0jPHJCUI5/B56+2DpI1xDSvi80r/3Ez+Eaa2Yq4tJcRTaB1kqj/HrYKn8Yplm9bNoMJpwQ==",
+      "integrity": "sha512-uo8X7qHDy4YdJUhaoJDMAbL8VT5Ed3lijip2DdBHIB4tfKAvB1XBih6INH2L4qIi4jA0Qq1J0ErxcOocBmUSwg==",
       "cpu": [
         "arm64"
       ],
@@ -1137,9 +1138,9 @@
       }
     },
     "node_modules/@next/swc-linux-arm64-musl": {
-      "version": "16.1.1",
+      "version": "16.0.4",
-      "resolved": "https://registry.npmjs.org/@next/swc-linux-arm64-musl/-/swc-linux-arm64-musl-16.1.1.tgz",
+      "resolved": "https://registry.npmjs.org/@next/swc-linux-arm64-musl/-/swc-linux-arm64-musl-16.0.4.tgz",
-      "integrity": "sha512-MFHrgL4TXNQbBPzkKKur4Fb5ICEJa87HM7fczFs2+HWblM7mMLdco3dvyTI+QmLBU9xgns/EeeINSZD6Ar+oLg==",
+      "integrity": "sha512-pvR/AjNIAxsIz0PCNcZYpH+WmNIKNLcL4XYEfo+ArDi7GsxKWFO5BvVBLXbhti8Coyv3DE983NsitzUsGH5yTw==",
       "cpu": [
         "arm64"
       ],
@@ -1153,9 +1154,9 @@
       }
     },
     "node_modules/@next/swc-linux-x64-gnu": {
-      "version": "16.1.1",
+      "version": "16.0.4",
-      "resolved": "https://registry.npmjs.org/@next/swc-linux-x64-gnu/-/swc-linux-x64-gnu-16.1.1.tgz",
+      "resolved": "https://registry.npmjs.org/@next/swc-linux-x64-gnu/-/swc-linux-x64-gnu-16.0.4.tgz",
-      "integrity": "sha512-20bYDfgOQAPUkkKBnyP9PTuHiJGM7HzNBbuqmD0jiFVZ0aOldz+VnJhbxzjcSabYsnNjMPsE0cyzEudpYxsrUQ==",
+      "integrity": "sha512-2hebpsd5MRRtgqmT7Jj/Wze+wG+ZEXUK2KFFL4IlZ0amEEFADo4ywsifJNeFTQGsamH3/aXkKWymDvgEi+pc2Q==",
       "cpu": [
         "x64"
       ],
@@ -1169,9 +1170,9 @@
       }
     },
     "node_modules/@next/swc-linux-x64-musl": {
-      "version": "16.1.1",
+      "version": "16.0.4",
-      "resolved": "https://registry.npmjs.org/@next/swc-linux-x64-musl/-/swc-linux-x64-musl-16.1.1.tgz",
+      "resolved": "https://registry.npmjs.org/@next/swc-linux-x64-musl/-/swc-linux-x64-musl-16.0.4.tgz",
-      "integrity": "sha512-9pRbK3M4asAHQRkwaXwu601oPZHghuSC8IXNENgbBSyImHv/zY4K5udBusgdHkvJ/Tcr96jJwQYOll0qU8+fPA==",
+      "integrity": "sha512-pzRXf0LZZ8zMljH78j8SeLncg9ifIOp3ugAFka+Bq8qMzw6hPXOc7wydY7ardIELlczzzreahyTpwsim/WL3Sg==",
       "cpu": [
         "x64"
       ],
@@ -1185,9 +1186,9 @@
       }
     },
     "node_modules/@next/swc-win32-arm64-msvc": {
-      "version": "16.1.1",
+      "version": "16.0.4",
-      "resolved": "https://registry.npmjs.org/@next/swc-win32-arm64-msvc/-/swc-win32-arm64-msvc-16.1.1.tgz",
+      "resolved": "https://registry.npmjs.org/@next/swc-win32-arm64-msvc/-/swc-win32-arm64-msvc-16.0.4.tgz",
-      "integrity": "sha512-bdfQkggaLgnmYrFkSQfsHfOhk/mCYmjnrbRCGgkMcoOBZ4n+TRRSLmT/CU5SATzlBJ9TpioUyBW/vWFXTqQRiA==",
+      "integrity": "sha512-7G/yJVzum52B5HOqqbQYX9bJHkN+c4YyZ2AIvEssMHQlbAWOn3iIJjD4sM6ihWsBxuljiTKJovEYlD1K8lCUHw==",
       "cpu": [
         "arm64"
       ],
@@ -1201,9 +1202,9 @@
       }
     },
     "node_modules/@next/swc-win32-x64-msvc": {
-      "version": "16.1.1",
+      "version": "16.0.4",
-      "resolved": "https://registry.npmjs.org/@next/swc-win32-x64-msvc/-/swc-win32-x64-msvc-16.1.1.tgz",
+      "resolved": "https://registry.npmjs.org/@next/swc-win32-x64-msvc/-/swc-win32-x64-msvc-16.0.4.tgz",
-      "integrity": "sha512-Ncwbw2WJ57Al5OX0k4chM68DKhEPlrXBaSXDCi2kPi5f4d8b3ejr3RRJGfKBLrn2YJL5ezNS7w2TZLHSti8CMw==",
+      "integrity": "sha512-0Vy4g8SSeVkuU89g2OFHqGKM4rxsQtihGfenjx2tRckPrge5+gtFnRWGAAwvGXr0ty3twQvcnYjEyOrLHJ4JWA==",
       "cpu": [
         "x64"
       ],
@@ -1512,66 +1513,6 @@
         "node": ">=14.0.0"
       }
     },
-    "node_modules/@tailwindcss/oxide-wasm32-wasi/node_modules/@emnapi/core": {
-      "version": "1.6.0",
-      "dev": true,
-      "inBundle": true,
-      "license": "MIT",
-      "optional": true,
-      "dependencies": {
-        "@emnapi/wasi-threads": "1.1.0",
-        "tslib": "^2.4.0"
-      }
-    },
-    "node_modules/@tailwindcss/oxide-wasm32-wasi/node_modules/@emnapi/runtime": {
-      "version": "1.6.0",
-      "dev": true,
-      "inBundle": true,
-      "license": "MIT",
-      "optional": true,
-      "dependencies": {
-        "tslib": "^2.4.0"
-      }
-    },
-    "node_modules/@tailwindcss/oxide-wasm32-wasi/node_modules/@emnapi/wasi-threads": {
-      "version": "1.1.0",
-      "dev": true,
-      "inBundle": true,
-      "license": "MIT",
-      "optional": true,
-      "dependencies": {
-        "tslib": "^2.4.0"
-      }
-    },
-    "node_modules/@tailwindcss/oxide-wasm32-wasi/node_modules/@napi-rs/wasm-runtime": {
-      "version": "1.0.7",
-      "dev": true,
-      "inBundle": true,
-      "license": "MIT",
-      "optional": true,
-      "dependencies": {
-        "@emnapi/core": "^1.5.0",
-        "@emnapi/runtime": "^1.5.0",
-        "@tybys/wasm-util": "^0.10.1"
-      }
-    },
-    "node_modules/@tailwindcss/oxide-wasm32-wasi/node_modules/@tybys/wasm-util": {
-      "version": "0.10.1",
-      "dev": true,
-      "inBundle": true,
-      "license": "MIT",
-      "optional": true,
-      "dependencies": {
-        "tslib": "^2.4.0"
-      }
-    },
-    "node_modules/@tailwindcss/oxide-wasm32-wasi/node_modules/tslib": {
-      "version": "2.8.1",
-      "dev": true,
-      "inBundle": true,
-      "license": "0BSD",
-      "optional": true
-    },
     "node_modules/@tailwindcss/oxide-win32-arm64-msvc": {
       "version": "4.1.17",
       "resolved": "https://registry.npmjs.org/@tailwindcss/oxide-win32-arm64-msvc/-/oxide-win32-arm64-msvc-4.1.17.tgz",
@@ -1681,6 +1622,7 @@
       "integrity": "sha512-MWtvHrGZLFttgeEj28VXHxpmwYbor/ATPYbBfSFZEIRK0ecCFLl2Qo55z52Hss+UV9CRN7trSeq1zbgx7YDWWg==",
       "dev": true,
       "license": "MIT",
+      "peer": true,
       "dependencies": {
         "csstype": "^3.2.2"
       }
@@ -1748,6 +1690,7 @@
       "integrity": "sha512-jCzKdm/QK0Kg4V4IK/oMlRZlY+QOcdjv89U2NgKHZk1CYTj82/RVSx1mV/0gqCVMJ/DA+Zf/S4NBWNF8GQ+eqQ==",
       "dev": true,
       "license": "MIT",
+      "peer": true,
       "dependencies": {
         "@typescript-eslint/scope-manager": "8.48.0",
         "@typescript-eslint/types": "8.48.0",
@@ -2256,6 +2199,7 @@
       "integrity": "sha512-NZyJarBfL7nWwIq+FDL6Zp/yHEhePMNnnJ0y3qfieCrmNvYct8uvtiV41UvlSe6apAfk0fY1FbWx+NwfmpvtTg==",
       "dev": true,
       "license": "MIT",
+      "peer": true,
       "bin": {
         "acorn": "bin/acorn"
       },
@@ -2547,6 +2491,7 @@
       "version": "2.8.31",
       "resolved": "https://registry.npmjs.org/baseline-browser-mapping/-/baseline-browser-mapping-2.8.31.tgz",
       "integrity": "sha512-a28v2eWrrRWPpJSzxc+mKwm0ZtVx/G8SepdQZDArnXYU/XS+IF6mp8aB/4E+hH1tyGCoDo3KlUCdlSxGDsRkAw==",
+      "dev": true,
       "license": "Apache-2.0",
       "bin": {
         "baseline-browser-mapping": "dist/cli.js"
@@ -2606,6 +2551,7 @@
         }
       ],
       "license": "MIT",
+      "peer": true,
       "dependencies": {
         "baseline-browser-mapping": "^2.8.25",
         "caniuse-lite": "^1.0.30001754",
@@ -2950,6 +2896,7 @@
       "resolved": "https://registry.npmjs.org/d3-selection/-/d3-selection-3.0.0.tgz",
       "integrity": "sha512-fmTRWbNMmsmWq6xJV8D19U/gw/bwrHfNXxrIN+HfZgnzqTHp9jOmKMhsTUjXOJnZOdZY9Q28y4yebKzqDKlxlQ==",
       "license": "ISC",
+      "peer": true,
       "engines": {
         "node": ">=12"
       }
@@ -3425,6 +3372,7 @@
       "integrity": "sha512-BhHmn2yNOFA9H9JmmIVKJmd288g9hrVRDkdoIgRCRuSySRUHH7r/DI6aAXW9T1WwUuY3DFgrcaqB+deURBLR5g==",
       "dev": true,
       "license": "MIT",
+      "peer": true,
       "dependencies": {
         "@eslint-community/eslint-utils": "^4.8.0",
         "@eslint-community/regexpp": "^4.12.1",
@@ -5463,14 +5411,14 @@
       "license": "MIT"
     },
     "node_modules/next": {
-      "version": "16.1.1",
+      "version": "16.0.4",
-      "resolved": "https://registry.npmjs.org/next/-/next-16.1.1.tgz",
+      "resolved": "https://registry.npmjs.org/next/-/next-16.0.4.tgz",
-      "integrity": "sha512-QI+T7xrxt1pF6SQ/JYFz95ro/mg/1Znk5vBebsWwbpejj1T0A23hO7GYEaVac9QUOT2BIMiuzm0L99ooq7k0/w==",
+      "integrity": "sha512-vICcxKusY8qW7QFOzTvnRL1ejz2ClTqDKtm1AcUjm2mPv/lVAdgpGNsftsPRIDJOXOjRQO68i1dM8Lp8GZnqoA==",
       "license": "MIT",
+      "peer": true,
       "dependencies": {
-        "@next/env": "16.1.1",
+        "@next/env": "16.0.4",
         "@swc/helpers": "0.5.15",
-        "baseline-browser-mapping": "^2.8.3",
         "caniuse-lite": "^1.0.30001579",
         "postcss": "8.4.31",
         "styled-jsx": "5.1.6"
@@ -5482,14 +5430,14 @@
         "node": ">=20.9.0"
       },
       "optionalDependencies": {
-        "@next/swc-darwin-arm64": "16.1.1",
+        "@next/swc-darwin-arm64": "16.0.4",
-        "@next/swc-darwin-x64": "16.1.1",
+        "@next/swc-darwin-x64": "16.0.4",
-        "@next/swc-linux-arm64-gnu": "16.1.1",
+        "@next/swc-linux-arm64-gnu": "16.0.4",
-        "@next/swc-linux-arm64-musl": "16.1.1",
+        "@next/swc-linux-arm64-musl": "16.0.4",
-        "@next/swc-linux-x64-gnu": "16.1.1",
+        "@next/swc-linux-x64-gnu": "16.0.4",
-        "@next/swc-linux-x64-musl": "16.1.1",
+        "@next/swc-linux-x64-musl": "16.0.4",
-        "@next/swc-win32-arm64-msvc": "16.1.1",
+        "@next/swc-win32-arm64-msvc": "16.0.4",
-        "@next/swc-win32-x64-msvc": "16.1.1",
+        "@next/swc-win32-x64-msvc": "16.0.4",
         "sharp": "^0.34.4"
       },
       "peerDependencies": {
@@ -5861,9 +5809,9 @@
       }
     },
     "node_modules/preact": {
-      "version": "10.28.2",
+      "version": "10.27.2",
-      "resolved": "https://registry.npmjs.org/preact/-/preact-10.28.2.tgz",
+      "resolved": "https://registry.npmjs.org/preact/-/preact-10.27.2.tgz",
-      "integrity": "sha512-lbteaWGzGHdlIuiJ0l2Jq454m6kcpI1zNje6d8MlGAFlYvP2GO4ibnat7P74Esfz4sPTdM6UxtTwh/d3pwM9JA==",
+      "integrity": "sha512-5SYSgFKSyhCbk6SrXyMpqjb5+MQBgfvEKE/OC+PujcY34sOpqtr+0AZQtPYx5IA6VxynQ7rUPCtKzyovpj9Bpg==",
       "license": "MIT",
       "funding": {
         "type": "opencollective",
@@ -5927,6 +5875,7 @@
       "resolved": "https://registry.npmjs.org/react/-/react-19.2.0.tgz",
       "integrity": "sha512-tmbWg6W31tQLeB5cdIBOicJDJRR2KzXsV7uSK9iNfLWQ5bIZfxuPEHp7M8wiHyHnn0DD1i7w3Zmin0FtkrwoCQ==",
       "license": "MIT",
+      "peer": true,
       "engines": {
         "node": ">=0.10.0"
       }
@@ -5936,6 +5885,7 @@
       "resolved": "https://registry.npmjs.org/react-dom/-/react-dom-19.2.0.tgz",
       "integrity": "sha512-UlbRu4cAiGaIewkPyiRGJk0imDN2T3JjieT6spoL2UeSf5od4n5LB/mQ4ejmxhCFT1tYe8IvaFulzynWovsEFQ==",
       "license": "MIT",
+      "peer": true,
       "dependencies": {
         "scheduler": "^0.27.0"
       },
@@ -6674,6 +6624,7 @@
       "integrity": "sha512-5gTmgEY/sqK6gFXLIsQNH19lWb4ebPDLA4SdLP7dsWkIXHWlG66oPuVvXSGFPppYZz8ZDZq0dYYrbHfBCVUb1Q==",
       "dev": true,
       "license": "MIT",
+      "peer": true,
       "engines": {
         "node": ">=12"
       },
@@ -6836,6 +6787,7 @@
       "integrity": "sha512-jl1vZzPDinLr9eUt3J/t7V6FgNEw9QjvBPdysz9KfQDD41fQrC2Y4vKQdiaUpFT4bXlb1RHhLpp8wtm6M5TgSw==",
       "dev": true,
       "license": "Apache-2.0",
+      "peer": true,
       "bin": {
         "tsc": "bin/tsc",
         "tsserver": "bin/tsserver"
@@ -7133,6 +7085,7 @@
       "integrity": "sha512-AvvthqfqrAhNH9dnfmrfKzX5upOdjUVJYFqNSlkmGf64gRaTzlPwz99IHYnVs28qYAybvAlBV+H7pn0saFY4Ig==",
       "dev": true,
       "license": "MIT",
+      "peer": true,
       "funding": {
         "url": "https://github.com/sponsors/colinhacks"
       }

cognee-frontend/package.json

@@ -13,7 +13,7 @@
     "classnames": "^2.5.1",
     "culori": "^4.0.1",
     "d3-force-3d": "^3.0.6",
-    "next": "16.1.1",
+    "next": "16.0.4",
     "react": "^19.2.0",
     "react-dom": "^19.2.0",
     "react-force-graph-2d": "^1.27.1",

cognee-mcp/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "cognee-mcp"
-version = "0.5.0"
+version = "0.4.0"
 description = "Cognee MCP server"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -9,7 +9,7 @@ dependencies = [
     # For local cognee repo usage remove comment bellow and add absolute path to cognee. Then run `uv sync --reinstall` in the mcp folder on local cognee changes.
     #"cognee[postgres,codegraph,gemini,huggingface,docs,neo4j] @ file:/Users/igorilic/Desktop/cognee",
     # TODO: Remove gemini from optional dependecnies for new Cognee version after 0.3.4
-    "cognee[postgres,docs,neo4j]==0.5.0",
+    "cognee[postgres,docs,neo4j]==0.3.7",
     "fastmcp>=2.10.0,<3.0.0",
     "mcp>=1.12.0,<2.0.0",
     "uv>=0.6.3,<1.0.0",

cognee-mcp/src/cognee_client.py

@@ -151,7 +151,7 @@ class CogneeClient:
         query_type: str,
         datasets: Optional[List[str]] = None,
         system_prompt: Optional[str] = None,
-        top_k: int = 5,
+        top_k: int = 10,
     ) -> Any:
         """
         Search the knowledge graph.
@@ -192,7 +192,7 @@ class CogneeClient:
 
             with redirect_stdout(sys.stderr):
                 results = await self.cognee.search(
-                    query_type=SearchType[query_type.upper()], query_text=query_text, top_k=top_k
+                    query_type=SearchType[query_type.upper()], query_text=query_text
                 )
                 return results
 

cognee-mcp/src/server.py

@@ -316,7 +316,7 @@ async def save_interaction(data: str) -> list:
 
 
 @mcp.tool()
-async def search(search_query: str, search_type: str, top_k: int = 10) -> list:
+async def search(search_query: str, search_type: str) -> list:
     """
     Search and query the knowledge graph for insights, information, and connections.
 
@@ -389,13 +389,6 @@ async def search(search_query: str, search_type: str, top_k: int = 10) -> list:
 
         The search_type is case-insensitive and will be converted to uppercase.
 
-    top_k : int, optional
-        Maximum number of results to return (default: 10).
-        Controls the amount of context retrieved from the knowledge graph.
-        - Lower values (3-5): Faster, more focused results
-        - Higher values (10-20): More comprehensive, but slower and more context-heavy
-        Helps manage response size and context window usage in MCP clients.
-
     Returns
     -------
     list
@@ -432,32 +425,13 @@ async def search(search_query: str, search_type: str, top_k: int = 10) -> list:
 
     """
 
-    async def search_task(search_query: str, search_type: str, top_k: int) -> str:
+    async def search_task(search_query: str, search_type: str) -> str:
-        """
+        """Search the knowledge graph"""
-        Internal task to execute knowledge graph search with result formatting.
-
-        Handles the actual search execution and formats results appropriately
-        for MCP clients based on the search type and execution mode (API vs direct).
-
-        Parameters
-        ----------
-        search_query : str
-            The search query in natural language
-        search_type : str
-            Type of search to perform (GRAPH_COMPLETION, CHUNKS, etc.)
-        top_k : int
-            Maximum number of results to return
-
-        Returns
-        -------
-        str
-            Formatted search results as a string, with format depending on search_type
-        """
         # NOTE: MCP uses stdout to communicate, we must redirect all output
         #       going to stdout ( like the print function ) to stderr.
         with redirect_stdout(sys.stderr):
             search_results = await cognee_client.search(
-                query_text=search_query, query_type=search_type, top_k=top_k
+                query_text=search_query, query_type=search_type
             )
 
             # Handle different result formats based on API vs direct mode
@@ -491,7 +465,7 @@ async def search(search_query: str, search_type: str, top_k: int = 10) -> list:
                 else:
                     return str(search_results)
 
-    search_results = await search_task(search_query, search_type, top_k)
+    search_results = await search_task(search_query, search_type)
     return [types.TextContent(type="text", text=search_results)]
 
 

cognee/api/v1/search/search.py

@@ -33,7 +33,6 @@ async def search(
     session_id: Optional[str] = None,
     wide_search_top_k: Optional[int] = 100,
     triplet_distance_penalty: Optional[float] = 3.5,
-    verbose: bool = False,
 ) -> Union[List[SearchResult], CombinedSearchResult]:
     """
     Search and query the knowledge graph for insights, information, and connections.
@@ -124,8 +123,6 @@ async def search(
 
         session_id: Optional session identifier for caching Q&A interactions. Defaults to 'default_session' if None.
 
-        verbose: If True, returns detailed result information including graph representation (when possible).
-
     Returns:
         list: Search results in format determined by query_type:
 
@@ -207,7 +204,6 @@ async def search(
         session_id=session_id,
         wide_search_top_k=wide_search_top_k,
         triplet_distance_penalty=triplet_distance_penalty,
-        verbose=verbose,
     )
 
     return filtered_search_results

cognee/modules/engine/operations/setup.py

@@ -15,9 +15,3 @@ async def setup():
     """
     await create_relational_db_and_tables()
     await create_pgvector_db_and_tables()
-
-
-if __name__ == "__main__":
-    import asyncio
-
-    asyncio.run(setup())

cognee/modules/search/methods/search.py

@@ -49,7 +49,6 @@ async def search(
     session_id: Optional[str] = None,
     wide_search_top_k: Optional[int] = 100,
     triplet_distance_penalty: Optional[float] = 3.5,
-    verbose: bool = False,
 ) -> Union[CombinedSearchResult, List[SearchResult]]:
     """
 
@@ -141,7 +140,6 @@ async def search(
     )
 
     if use_combined_context:
-        # Note: combined context search must always be verbose and return a CombinedSearchResult with graphs info
         prepared_search_results = await prepare_search_result(
             search_results[0] if isinstance(search_results, list) else search_results
         )
@@ -175,30 +173,25 @@ async def search(
                 datasets = prepared_search_results["datasets"]
 
                 if only_context:
-                    search_result_dict = {
+                    return_value.append(
-                        "search_result": [context] if context else None,
+                        {
-                        "dataset_id": datasets[0].id,
+                            "search_result": [context] if context else None,
-                        "dataset_name": datasets[0].name,
+                            "dataset_id": datasets[0].id,
-                        "dataset_tenant_id": datasets[0].tenant_id,
+                            "dataset_name": datasets[0].name,
-                    }
+                            "dataset_tenant_id": datasets[0].tenant_id,
-                    if verbose:
+                            "graphs": graphs,
-                        # Include graphs only in verbose mode
+                        }
-                        search_result_dict["graphs"] = graphs
+                    )
-
-                    return_value.append(search_result_dict)
                 else:
-                    search_result_dict = {
+                    return_value.append(
-                        "search_result": [result] if result else None,
+                        {
-                        "dataset_id": datasets[0].id,
+                            "search_result": [result] if result else None,
-                        "dataset_name": datasets[0].name,
+                            "dataset_id": datasets[0].id,
-                        "dataset_tenant_id": datasets[0].tenant_id,
+                            "dataset_name": datasets[0].name,
-                    }
+                            "dataset_tenant_id": datasets[0].tenant_id,
-                    if verbose:
+                            "graphs": graphs,
-                        # Include graphs only in verbose mode
+                        }
-                        search_result_dict["graphs"] = graphs
+                    )
-
-                    return_value.append(search_result_dict)
-
             return return_value
         else:
             return_value = []

cognee/modules/visualization/cognee_network_visualization.py

@@ -92,7 +92,7 @@ async def cognee_network_visualization(graph_data, destination_file_path: str =
         }
         links_list.append(link_data)
 
-    html_template = r"""
+    html_template = """
     <!DOCTYPE html>
     <html>
     <head>

cognee/tests/unit/modules/search/test_search.py

@@ -1,100 +0,0 @@
-import types
-from uuid import uuid4
-
-import pytest
-
-from cognee.modules.search.types import SearchType
-
-
-def _make_user(user_id: str = "u1", tenant_id=None):
-    return types.SimpleNamespace(id=user_id, tenant_id=tenant_id)
-
-
-def _make_dataset(*, name="ds", tenant_id="t1", dataset_id=None, owner_id=None):
-    return types.SimpleNamespace(
-        id=dataset_id or uuid4(),
-        name=name,
-        tenant_id=tenant_id,
-        owner_id=owner_id or uuid4(),
-    )
-
-
-@pytest.fixture
-def search_mod():
-    import importlib
-
-    return importlib.import_module("cognee.modules.search.methods.search")
-
-
-@pytest.fixture(autouse=True)
-def _patch_side_effect_boundaries(monkeypatch, search_mod):
-    """
-    Keep production logic; patch only unavoidable side-effect boundaries.
-    """
-
-    async def dummy_log_query(_query_text, _query_type, _user_id):
-        return types.SimpleNamespace(id="qid-1")
-
-    async def dummy_log_result(*_args, **_kwargs):
-        return None
-
-    async def dummy_prepare_search_result(search_result):
-        if isinstance(search_result, tuple) and len(search_result) == 3:
-            result, context, datasets = search_result
-            return {"result": result, "context": context, "graphs": {}, "datasets": datasets}
-        return {"result": None, "context": None, "graphs": {}, "datasets": []}
-
-    monkeypatch.setattr(search_mod, "send_telemetry", lambda *a, **k: None)
-    monkeypatch.setattr(search_mod, "log_query", dummy_log_query)
-    monkeypatch.setattr(search_mod, "log_result", dummy_log_result)
-    monkeypatch.setattr(search_mod, "prepare_search_result", dummy_prepare_search_result)
-
-    yield
-
-
-@pytest.mark.asyncio
-async def test_search_access_control_returns_dataset_shaped_dicts(monkeypatch, search_mod):
-    user = _make_user()
-    ds = _make_dataset(name="ds1", tenant_id="t1")
-
-    async def dummy_authorized_search(**kwargs):
-        assert kwargs["dataset_ids"] == [ds.id]
-        return [("r", ["ctx"], [ds])]
-
-    monkeypatch.setattr(search_mod, "backend_access_control_enabled", lambda: True)
-    monkeypatch.setattr(search_mod, "authorized_search", dummy_authorized_search)
-
-    out_non_verbose = await search_mod.search(
-        query_text="q",
-        query_type=SearchType.CHUNKS,
-        dataset_ids=[ds.id],
-        user=user,
-        verbose=False,
-    )
-
-    assert out_non_verbose == [
-        {
-            "search_result": ["r"],
-            "dataset_id": ds.id,
-            "dataset_name": "ds1",
-            "dataset_tenant_id": "t1",
-        }
-    ]
-
-    out_verbose = await search_mod.search(
-        query_text="q",
-        query_type=SearchType.CHUNKS,
-        dataset_ids=[ds.id],
-        user=user,
-        verbose=True,
-    )
-
-    assert out_verbose == [
-        {
-            "search_result": ["r"],
-            "dataset_id": ds.id,
-            "dataset_name": "ds1",
-            "dataset_tenant_id": "t1",
-            "graphs": {},
-        }
-    ]

entrypoint.sh

@@ -20,30 +20,20 @@ echo "HTTP port: $HTTP_PORT"
 # smooth redeployments and container restarts while maintaining data integrity.
 echo "Running database migrations..."
 
-set +e # Disable exit on error to handle specific migration errors
 MIGRATION_OUTPUT=$(alembic upgrade head)
 MIGRATION_EXIT_CODE=$?
-set -e
 
 if [[ $MIGRATION_EXIT_CODE -ne 0 ]]; then
     if [[ "$MIGRATION_OUTPUT" == *"UserAlreadyExists"* ]] || [[ "$MIGRATION_OUTPUT" == *"User default_user@example.com already exists"* ]]; then
         echo "Warning: Default user already exists, continuing startup..."
     else
-        echo "Migration failed with unexpected error. Trying to run Cognee without migrations."
+        echo "Migration failed with unexpected error."
-
+        exit 1
-        echo "Initializing database tables..."
-        python /app/cognee/modules/engine/operations/setup.py
-        INIT_EXIT_CODE=$?
-
-        if [[ $INIT_EXIT_CODE -ne 0 ]]; then
-            echo "Database initialization failed!"
-            exit 1
-        fi
     fi
-else
-    echo "Database migrations done."
 fi
 
+echo "Database migrations done."
+
 echo "Starting server..."
 
 # Add startup delay to ensure DB is ready

examples/python/cognee_simple_document_demo.py

@@ -1,8 +1,7 @@
 import asyncio
 import cognee
-import os
 
-from pprint import pprint
+import os
 
 # By default cognee uses OpenAI's gpt-5-mini LLM model
 # Provide your OpenAI LLM API KEY
@@ -25,13 +24,13 @@ async def cognee_demo():
 
     # Query Cognee for information from provided document
     answer = await cognee.search("List me all the important characters in Alice in Wonderland.")
-    pprint(answer)
+    print(answer)
 
     answer = await cognee.search("How did Alice end up in Wonderland?")
-    pprint(answer)
+    print(answer)
 
     answer = await cognee.search("Tell me about Alice's personality.")
-    pprint(answer)
+    print(answer)
 
 
 # Cognee is an async library, it has to be called in an async context

examples/python/dynamic_steps_example.py

@@ -1,5 +1,4 @@
 import asyncio
-from pprint import pprint
 
 import cognee
 from cognee.api.v1.search import SearchType
@@ -188,7 +187,7 @@ async def main(enable_steps):
         search_results = await cognee.search(
             query_type=SearchType.GRAPH_COMPLETION, query_text="Who has experience in design tools?"
         )
-        pprint(search_results)
+        print(search_results)
 
 
 if __name__ == "__main__":

examples/python/multimedia_example.py

@@ -1,8 +1,6 @@
 import os
 import asyncio
 import pathlib
-from pprint import pprint
-
 from cognee.shared.logging_utils import setup_logging, ERROR
 
 import cognee
@@ -44,7 +42,7 @@ async def main():
 
     # Display search results
     for result_text in search_results:
-        pprint(result_text)
+        print(result_text)
 
 
 if __name__ == "__main__":

examples/python/ontology_demo_example.py

@@ -1,6 +1,5 @@
 import asyncio
 import os
-from pprint import pprint
 
 import cognee
 from cognee.api.v1.search import SearchType
@@ -78,7 +77,7 @@ async def main():
         query_type=SearchType.GRAPH_COMPLETION,
         query_text="What are the exact cars and their types produced by Audi?",
     )
-    pprint(search_results)
+    print(search_results)
 
     await visualize_graph()
 

examples/python/permissions_example.py

@@ -1,7 +1,6 @@
 import os
 import cognee
 import pathlib
-from pprint import pprint
 
 from cognee.modules.users.exceptions import PermissionDeniedError
 from cognee.modules.users.tenants.methods import select_tenant
@@ -87,7 +86,7 @@ async def main():
     )
     print("\nSearch results as user_1 on dataset owned by user_1:")
     for result in search_results:
-        pprint(result)
+        print(f"{result}\n")
 
     # But user_1 cant read the dataset owned by user_2 (QUANTUM dataset)
     print("\nSearch result as user_1 on the dataset owned by user_2:")
@@ -135,7 +134,7 @@ async def main():
         dataset_ids=[quantum_dataset_id],
     )
     for result in search_results:
-        pprint(result)
+        print(f"{result}\n")
 
     # If we'd like for user_1 to add new documents to the QUANTUM dataset owned by user_2, user_1 would have to get
     # "write" access permission, which user_1 currently does not have
@@ -218,7 +217,7 @@ async def main():
         dataset_ids=[quantum_cognee_lab_dataset_id],
     )
     for result in search_results:
-        pprint(result)
+        print(f"{result}\n")
 
     # Note: All of these function calls and permission system is available through our backend endpoints as well
 

examples/python/run_custom_pipeline_example.py

@@ -1,6 +1,4 @@
 import asyncio
-from pprint import pprint
-
 import cognee
 from cognee.modules.engine.operations.setup import setup
 from cognee.modules.users.methods import get_default_user
@@ -73,7 +71,7 @@ async def main():
     print("Search results:")
     # Display results
     for result_text in search_results:
-        pprint(result_text)
+        print(result_text)
 
 
 if __name__ == "__main__":

examples/python/simple_example.py

@@ -1,6 +1,4 @@
 import asyncio
-from pprint import pprint
-
 import cognee
 from cognee.shared.logging_utils import setup_logging, ERROR
 from cognee.api.v1.search import SearchType
@@ -56,7 +54,7 @@ async def main():
     print("Search results:")
     # Display results
     for result_text in search_results:
-        pprint(result_text)
+        print(result_text)
 
 
 if __name__ == "__main__":

examples/python/temporal_example.py

@@ -1,5 +1,4 @@
 import asyncio
-from pprint import pprint
 import cognee
 from cognee.shared.logging_utils import setup_logging, INFO
 from cognee.api.v1.search import SearchType
@@ -88,8 +87,7 @@ async def main():
             top_k=15,
         )
         print(f"Query: {query_text}")
-        print("Results:")
+        print(f"Results: {search_results}\n")
-        pprint(search_results)
 
 
 if __name__ == "__main__":

examples/python/triplet_embeddings_example.py

@@ -1,5 +1,4 @@
 import asyncio
-from pprint import pprint
 
 import cognee
 from cognee.memify_pipelines.create_triplet_embeddings import create_triplet_embeddings
@@ -66,7 +65,7 @@ async def main():
         query_type=SearchType.TRIPLET_COMPLETION,
         query_text="What are the models produced by Volkswagen based on the context?",
     )
-    pprint(search_results)
+    print(search_results)
 
 
 if __name__ == "__main__":

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "cognee"
 
-version = "0.5.1"
+version = "0.5.0.dev1"
 description = "Cognee - is a library for enriching LLM context with a semantic layer for better understanding and reasoning."
 authors = [
     { name = "Vasilije Markovic" },

uv.lock

@@ -946,7 +946,7 @@ wheels = [
 
 [[package]]
 name = "cognee"
-version = "0.5.1"
+version = "0.5.0.dev1"
 source = { editable = "." }
 dependencies = [
     { name = "aiofiles" },