diff --git a/cognee-mcp/Dockerfile b/cognee-mcp/Dockerfile
index ae5621f19..4196fbede 100644
--- a/cognee-mcp/Dockerfile
+++ b/cognee-mcp/Dockerfile
@@ -51,7 +51,7 @@ RUN apt-get update && apt-get install -y \
 
 WORKDIR /app
 
-COPY --from=uv /root/.local /root/.local
+COPY --from=uv /usr/local /usr/local
 COPY --from=uv /app /app
 
 RUN chmod +x /app/entrypoint.sh
diff --git a/cognee-mcp/entrypoint.sh b/cognee-mcp/entrypoint.sh
index 2660accb7..47988baf7 100644
--- a/cognee-mcp/entrypoint.sh
+++ b/cognee-mcp/entrypoint.sh
@@ -48,27 +48,27 @@ if [ "$ENVIRONMENT" = "dev" ] || [ "$ENVIRONMENT" = "local" ]; then
     if [ "$DEBUG" = "true" ]; then
         echo "Waiting for the debugger to attach..."
         if [ "$TRANSPORT_MODE" = "sse" ]; then
-            exec python -m debugpy --wait-for-client --listen 0.0.0.0:$DEBUG_PORT -m cognee --transport sse
+            exec python -m debugpy --wait-for-client --listen 0.0.0.0:$DEBUG_PORT -m cognee --transport sse --no-migration
         elif [ "$TRANSPORT_MODE" = "http" ]; then
-            exec python -m debugpy --wait-for-client --listen 0.0.0.0:$DEBUG_PORT -m cognee --transport http --host 0.0.0.0 --port $HTTP_PORT
+            exec python -m debugpy --wait-for-client --listen 0.0.0.0:$DEBUG_PORT -m cognee --transport http --host 0.0.0.0 --port $HTTP_PORT --no-migration
         else
-            exec python -m debugpy --wait-for-client --listen 0.0.0.0:$DEBUG_PORT -m cognee --transport stdio
+            exec python -m debugpy --wait-for-client --listen 0.0.0.0:$DEBUG_PORT -m cognee --transport stdio --no-migration
         fi
     else
         if [ "$TRANSPORT_MODE" = "sse" ]; then
-            exec cognee --transport sse
+            exec cognee --transport sse --no-migration
         elif [ "$TRANSPORT_MODE" = "http" ]; then
-            exec cognee --transport http --host 0.0.0.0 --port $HTTP_PORT
+            exec cognee --transport http --host 0.0.0.0 --port $HTTP_PORT --no-migration
         else
-            exec cognee --transport stdio
+            exec cognee --transport stdio --no-migration
         fi
     fi
 else
     if [ "$TRANSPORT_MODE" = "sse" ]; then
-        exec cognee --transport sse
+        exec cognee --transport sse --no-migration
     elif [ "$TRANSPORT_MODE" = "http" ]; then
-        exec cognee --transport http --host 0.0.0.0 --port $HTTP_PORT
+        exec cognee --transport http --host 0.0.0.0 --port $HTTP_PORT --no-migration
     else
-        exec cognee --transport stdio
+        exec cognee --transport stdio --no-migration
     fi
 fi
diff --git a/cognee-mcp/pyproject.toml b/cognee-mcp/pyproject.toml
index 385971261..7dd9ad8e2 100644
--- a/cognee-mcp/pyproject.toml
+++ b/cognee-mcp/pyproject.toml
@@ -8,7 +8,7 @@ requires-python = ">=3.10"
 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/vasilije/Projects/tiktok/cognee",
-    "cognee[postgres,codegraph,gemini,huggingface,docs,neo4j]>=0.2.0,<1.0.0",
+    "cognee[postgres,codegraph,gemini,huggingface,docs,neo4j]==0.2.1",
     "fastmcp>=2.10.0,<3.0.0",
     "mcp>=1.12.0,<2.0.0",
     "uv>=0.6.3,<1.0.0",
diff --git a/cognee-mcp/src/server.py b/cognee-mcp/src/server.py
index 726b7fcc5..58ad13752 100755
--- a/cognee-mcp/src/server.py
+++ b/cognee-mcp/src/server.py
@@ -938,30 +938,38 @@ async def main():
         help="Log level for the HTTP server (default: info)",
     )
 
-    args = parser.parse_args()
-
-    # Run Alembic migrations from the main cognee directory where alembic.ini is located
-    print("Running database migrations...")
-    migration_result = subprocess.run(
-        ["python", "-m", "alembic", "upgrade", "head"],
-        capture_output=True,
-        text=True,
-        cwd=Path(__file__).resolve().parent.parent.parent,
+    parser.add_argument(
+        "--no-migration",
+        default=False,
+        action="store_true",
+        help="Argument stops database migration from being attempted",
     )
 
-    if migration_result.returncode != 0:
-        migration_output = migration_result.stderr + migration_result.stdout
-        # Check for the expected UserAlreadyExists error (which is not critical)
-        if (
-            "UserAlreadyExists" in migration_output
-            or "User default_user@example.com already exists" in migration_output
-        ):
-            print("Warning: Default user already exists, continuing startup...")
-        else:
-            print(f"Migration failed with unexpected error: {migration_output}")
-            sys.exit(1)
+    args = parser.parse_args()
 
-    print("Database migrations done.")
+    if not args.no_migration:
+        # Run Alembic migrations from the main cognee directory where alembic.ini is located
+        logger.info("Running database migrations...")
+        migration_result = subprocess.run(
+            ["python", "-m", "alembic", "upgrade", "head"],
+            capture_output=True,
+            text=True,
+            cwd=Path(__file__).resolve().parent.parent.parent,
+        )
+
+        if migration_result.returncode != 0:
+            migration_output = migration_result.stderr + migration_result.stdout
+            # Check for the expected UserAlreadyExists error (which is not critical)
+            if (
+                "UserAlreadyExists" in migration_output
+                or "User default_user@example.com already exists" in migration_output
+            ):
+                logger.warning("Warning: Default user already exists, continuing startup...")
+            else:
+                logger.error(f"Migration failed with unexpected error: {migration_output}")
+                sys.exit(1)
+
+        logger.info("Database migrations done.")
 
     logger.info(f"Starting MCP server with transport: {args.transport}")
     if args.transport == "stdio":
diff --git a/cognee-mcp/uv.lock b/cognee-mcp/uv.lock
index 0e27391a3..4e725d4d7 100644
--- a/cognee-mcp/uv.lock
+++ b/cognee-mcp/uv.lock
@@ -621,8 +621,8 @@ wheels = [
 
 [[package]]
 name = "cognee"
-version = "0.2.1.dev7"
-source = { directory = "../" }
+version = "0.2.1"
+source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "aiofiles" },
     { name = "aiohttp" },
@@ -664,6 +664,10 @@ dependencies = [
     { name = "tiktoken" },
     { name = "typing-extensions" },
 ]
+sdist = { url = "https://files.pythonhosted.org/packages/41/46/e7df1faebc92fa31ef8e33faf81feb435782727a789de5532d178e047224/cognee-0.2.1.tar.gz", hash = "sha256:bf5208383fc841981641c040e5b6588e58111af4d771f9eab6552f441e6a8e6c", size = 15497626, upload-time = "2025-07-25T15:53:57.009Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/b3/0e/b705c6eeb538dcdd8fbbb331be25fe8e0bbc1af7d76e61566ec9845b29d3/cognee-0.2.1-py3-none-any.whl", hash = "sha256:6e9d437e0c58a16233841ebf19b1a3d8b67da069460a4f08d0c0e00301b1d36d", size = 1019851, upload-time = "2025-07-25T15:53:53.488Z" },
+]
 
 [package.optional-dependencies]
 codegraph = [
@@ -690,102 +694,6 @@ postgres = [
     { name = "psycopg2" },
 ]
 
-[package.metadata]
-requires-dist = [
-    { name = "aiofiles", specifier = ">=23.2.1,<24.0.0" },
-    { name = "aiohttp", specifier = ">=3.11.14,<4.0.0" },
-    { name = "aiosqlite", specifier = ">=0.20.0,<1.0.0" },
-    { name = "alembic", specifier = ">=1.13.3,<2" },
-    { name = "anthropic", marker = "extra == 'anthropic'", specifier = ">=0.26.1,<0.27" },
-    { name = "asyncpg", marker = "extra == 'postgres'", specifier = ">=0.30.0,<1.0.0" },
-    { name = "asyncpg", marker = "extra == 'postgres-binary'", specifier = ">=0.30.0,<1.0.0" },
-    { name = "chromadb", marker = "extra == 'chromadb'", specifier = ">=0.3.0,<0.7" },
-    { name = "coverage", marker = "extra == 'dev'", specifier = ">=7.3.2,<8" },
-    { name = "debugpy", marker = "extra == 'debug'", specifier = ">=1.8.9,<2.0.0" },
-    { name = "deepeval", marker = "extra == 'deepeval'", specifier = ">=2.0.1,<3" },
-    { name = "deptry", marker = "extra == 'dev'", specifier = ">=0.20.0,<0.21" },
-    { name = "dlt", extras = ["sqlalchemy"], specifier = ">=1.9.0,<2" },
-    { name = "falkordb", marker = "extra == 'falkordb'", specifier = ">=1.0.9,<2.0.0" },
-    { name = "fastapi", specifier = ">=0.115.7,<1.0.0" },
-    { name = "fastapi-users", extras = ["sqlalchemy"], specifier = ">=14.0.1,<15.0.0" },
-    { name = "fastembed", marker = "python_full_version < '3.13' and extra == 'codegraph'", specifier = "<=0.6.0" },
-    { name = "filetype", specifier = ">=1.2.0,<2.0.0" },
-    { name = "gdown", marker = "extra == 'evals'", specifier = ">=5.2.0,<6" },
-    { name = "gitpython", marker = "extra == 'dev'", specifier = ">=3.1.43,<4" },
-    { name = "google-generativeai", marker = "extra == 'gemini'", specifier = ">=0.8.4,<0.9" },
-    { name = "graphiti-core", marker = "extra == 'graphiti'", specifier = ">=0.7.0,<0.8" },
-    { name = "groq", marker = "extra == 'groq'", specifier = ">=0.8.0,<1.0.0" },
-    { name = "gunicorn", marker = "extra == 'api'", specifier = ">=20.1.0,<24" },
-    { name = "instructor", specifier = ">=1.9.1,<2.0.0" },
-    { name = "jinja2", specifier = ">=3.1.3,<4" },
-    { name = "kuzu", specifier = "==0.11.0" },
-    { name = "lancedb", specifier = ">=0.24.0,<1.0.0" },
-    { name = "langchain-text-splitters", marker = "extra == 'langchain'", specifier = ">=0.3.2,<1.0.0" },
-    { name = "langfuse", specifier = ">=2.32.0,<3" },
-    { name = "langsmith", marker = "extra == 'langchain'", specifier = ">=0.2.3,<1.0.0" },
-    { name = "limits", specifier = ">=4.4.1,<5" },
-    { name = "litellm", specifier = ">=1.57.4,<1.71.0" },
-    { name = "llama-index-core", marker = "extra == 'llama-index'", specifier = ">=0.12.11,<0.13" },
-    { name = "matplotlib", specifier = ">=3.8.3,<4" },
-    { name = "mistral-common", marker = "extra == 'mistral'", specifier = ">=1.5.2,<2" },
-    { name = "mkdocs-material", marker = "extra == 'dev'", specifier = ">=9.5.42,<10" },
-    { name = "mkdocs-minify-plugin", marker = "extra == 'dev'", specifier = ">=0.8.0,<0.9" },
-    { name = "mkdocstrings", extras = ["python"], marker = "extra == 'dev'", specifier = ">=0.26.2,<0.27" },
-    { name = "modal", marker = "extra == 'distributed'", specifier = ">=1.0.5,<2.0.0" },
-    { name = "mypy", marker = "extra == 'dev'", specifier = ">=1.7.1,<2" },
-    { name = "neo4j", marker = "extra == 'neo4j'", specifier = ">=5.28.0,<6" },
-    { name = "networkx", specifier = ">=3.4.2,<4" },
-    { name = "nltk", specifier = ">=3.9.1,<4.0.0" },
-    { name = "notebook", marker = "extra == 'dev'", specifier = ">=7.1.0,<8" },
-    { name = "notebook", marker = "extra == 'notebook'", specifier = ">=7.1.0,<8" },
-    { name = "numpy", specifier = ">=1.26.4,<=4.0.0" },
-    { name = "onnxruntime", specifier = ">=1.0.0,<2.0.0" },
-    { name = "openai", specifier = ">=1.80.1,<2" },
-    { name = "pandas", specifier = ">=2.2.2,<3.0.0" },
-    { name = "pgvector", marker = "extra == 'postgres'", specifier = ">=0.3.5,<0.4" },
-    { name = "pgvector", marker = "extra == 'postgres-binary'", specifier = ">=0.3.5,<0.4" },
-    { name = "plotly", marker = "extra == 'evals'", specifier = ">=6.0.0,<7" },
-    { name = "posthog", marker = "extra == 'posthog'", specifier = ">=3.5.0,<4" },
-    { name = "pre-commit", specifier = ">=4.0.1,<5" },
-    { name = "psycopg2", marker = "extra == 'postgres'", specifier = ">=2.9.10,<3" },
-    { name = "psycopg2-binary", marker = "extra == 'postgres-binary'", specifier = ">=2.9.10,<3.0.0" },
-    { name = "pydantic", specifier = ">=2.10.5,<3.0.0" },
-    { name = "pydantic-settings", specifier = ">=2.2.1,<3" },
-    { name = "pylance", specifier = ">=0.22.0,<1.0.0" },
-    { name = "pylint", marker = "extra == 'dev'", specifier = ">=3.0.3,<4" },
-    { name = "pympler", specifier = ">=1.1,<2.0.0" },
-    { name = "pypdf", specifier = ">=4.1.0,<6.0.0" },
-    { name = "pypika", marker = "extra == 'chromadb'", specifier = "==0.48.8" },
-    { name = "pyside6", marker = "extra == 'gui'", specifier = ">=6.8.3,<7" },
-    { name = "pytest", marker = "extra == 'dev'", specifier = ">=7.4.0,<8" },
-    { name = "pytest-asyncio", marker = "extra == 'dev'", specifier = ">=0.21.1,<0.22" },
-    { name = "pytest-cov", marker = "extra == 'dev'", specifier = ">=6.1.1,<7.0.0" },
-    { name = "python-dotenv", specifier = ">=1.0.1,<2.0.0" },
-    { name = "python-multipart", specifier = ">=0.0.20,<1.0.0" },
-    { name = "qasync", marker = "extra == 'gui'", specifier = ">=0.27.1,<0.28" },
-    { name = "qdrant-client", marker = "extra == 'qdrant'", specifier = ">=1.14.2,<2" },
-    { name = "rdflib", specifier = ">=7.1.4,<7.2.0" },
-    { name = "ruff", marker = "extra == 'dev'", specifier = ">=0.9.2,<1.0.0" },
-    { name = "s3fs", extras = ["boto3"], specifier = "==2025.3.2" },
-    { name = "s3fs", extras = ["boto3"], marker = "extra == 'aws'", specifier = "==2025.3.2" },
-    { name = "scikit-learn", specifier = ">=1.6.1,<2" },
-    { name = "sentry-sdk", extras = ["fastapi"], specifier = ">=2.9.0,<3" },
-    { name = "sqlalchemy", specifier = ">=2.0.39,<3.0.0" },
-    { name = "structlog", specifier = ">=25.2.0,<26" },
-    { name = "tiktoken", specifier = ">=0.8.0,<1.0.0" },
-    { name = "transformers", marker = "extra == 'codegraph'", specifier = ">=4.46.3,<5" },
-    { name = "transformers", marker = "extra == 'huggingface'", specifier = ">=4.46.3,<5" },
-    { name = "transformers", marker = "extra == 'ollama'", specifier = ">=4.46.3,<5" },
-    { name = "tree-sitter", marker = "extra == 'codegraph'", specifier = ">=0.24.0,<0.25" },
-    { name = "tree-sitter-python", marker = "extra == 'codegraph'", specifier = ">=0.23.6,<0.24" },
-    { name = "tweepy", marker = "extra == 'dev'", specifier = ">=4.14.0,<5.0.0" },
-    { name = "typing-extensions", specifier = ">=4.12.2,<5.0.0" },
-    { name = "unstructured", extras = ["csv", "doc", "docx", "epub", "md", "odt", "org", "ppt", "pptx", "rst", "rtf", "tsv", "xlsx"], marker = "extra == 'docs'", specifier = ">=0.18.1,<19" },
-    { name = "uvicorn", marker = "extra == 'api'", specifier = ">=0.34.0,<1.0.0" },
-    { name = "websockets", marker = "extra == 'api'", specifier = ">=15.0.1,<16.0.0" },
-]
-provides-extras = ["api", "distributed", "qdrant", "neo4j", "postgres", "postgres-binary", "notebook", "langchain", "llama-index", "gemini", "huggingface", "ollama", "mistral", "anthropic", "deepeval", "posthog", "falkordb", "groq", "chromadb", "docs", "codegraph", "evals", "gui", "graphiti", "aws", "dev", "debug"]
-
 [[package]]
 name = "cognee-mcp"
 version = "0.4.0"
@@ -804,7 +712,7 @@ dev = [
 
 [package.metadata]
 requires-dist = [
-    { name = "cognee", extras = ["postgres", "codegraph", "gemini", "huggingface", "docs", "neo4j"], directory = "../" },
+    { name = "cognee", extras = ["postgres", "codegraph", "gemini", "huggingface", "docs", "neo4j"], specifier = "==0.2.1" },
     { name = "fastmcp", specifier = ">=2.10.0,<3.0.0" },
     { name = "mcp", specifier = ">=1.12.0,<2.0.0" },
     { name = "uv", specifier = ">=0.6.3,<1.0.0" },