From a81f1a4551dd97a9cec77f0d93bf16e3011fc699 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Fri, 25 Jul 2025 16:15:30 +0200 Subject: [PATCH] fix: MCP migration fixes --- .gitignore | 1 + cognee-mcp/.python-version | 1 - cognee-mcp/src/server.py | 4 ++-- entrypoint.sh | 6 +++--- 4 files changed, 6 insertions(+), 6 deletions(-) delete mode 100644 cognee-mcp/.python-version diff --git a/.gitignore b/.gitignore index bff09f92a..8441d4b24 100644 --- a/.gitignore +++ b/.gitignore @@ -37,6 +37,7 @@ share/python-wheels/ .installed.cfg *.egg .python-version +cognee-mcp/.python-version MANIFEST # PyInstaller diff --git a/cognee-mcp/.python-version b/cognee-mcp/.python-version deleted file mode 100644 index 9ac380418..000000000 --- a/cognee-mcp/.python-version +++ /dev/null @@ -1 +0,0 @@ -3.11.5 diff --git a/cognee-mcp/src/server.py b/cognee-mcp/src/server.py index c34402c61..fbc103b1a 100755 --- a/cognee-mcp/src/server.py +++ b/cognee-mcp/src/server.py @@ -380,7 +380,7 @@ async def search(search_query: str, search_type: str) -> list: elif ( search_type.upper() == "GRAPH_COMPLETION" or search_type.upper() == "RAG_COMPLETION" ): - return search_results[0] + return str(search_results[0]) elif search_type.upper() == "CHUNKS": return str(search_results) elif search_type.upper() == "INSIGHTS": @@ -787,7 +787,7 @@ async def main(): # Run Alembic migrations from the main cognee directory where alembic.ini is located print("Running database migrations...") migration_result = subprocess.run( - ["alembic", "upgrade", "head"], + ["python", "-m", "alembic", "upgrade", "head"], capture_output=True, text=True, cwd=Path(__file__).resolve().parent.parent.parent, diff --git a/entrypoint.sh b/entrypoint.sh index 965105bfd..bad9b7aa3 100755 --- a/entrypoint.sh +++ b/entrypoint.sh @@ -43,10 +43,10 @@ sleep 2 if [ "$ENVIRONMENT" = "dev" ] || [ "$ENVIRONMENT" = "local" ]; then if [ "$DEBUG" = "true" ]; then echo "Waiting for the debugger to attach..." - debugpy --wait-for-client --listen 0.0.0.0:$DEBUG_PORT -m gunicorn -w 3 -k uvicorn.workers.UvicornWorker -t 30000 --bind=0.0.0.0:$HTTP_PORT --log-level debug --reload cognee.api.client:app + debugpy --wait-for-client --listen 0.0.0.0:$DEBUG_PORT -m gunicorn -w 1 -k uvicorn.workers.UvicornWorker -t 30000 --bind=0.0.0.0:$HTTP_PORT --log-level debug --reload cognee.api.client:app else - gunicorn -w 3 -k uvicorn.workers.UvicornWorker -t 30000 --bind=0.0.0.0:$HTTP_PORT --log-level debug --reload cognee.api.client:app + gunicorn -w 1 -k uvicorn.workers.UvicornWorker -t 30000 --bind=0.0.0.0:$HTTP_PORT --log-level debug --reload cognee.api.client:app fi else - gunicorn -w 3 -k uvicorn.workers.UvicornWorker -t 30000 --bind=0.0.0.0:$HTTP_PORT --log-level error cognee.api.client:app + gunicorn -w 1 -k uvicorn.workers.UvicornWorker -t 30000 --bind=0.0.0.0:$HTTP_PORT --log-level error cognee.api.client:app fi