diff --git a/cognee-mcp/pyproject.toml b/cognee-mcp/pyproject.toml index e1a3a092c..bf9ecd51f 100644 --- a/cognee-mcp/pyproject.toml +++ b/cognee-mcp/pyproject.toml @@ -36,4 +36,5 @@ dev = [ allow-direct-references = true [project.scripts] -cognee-mcp = "src:main" +cognee = "src:main" +cognee-mcp = "src:main_mcp" \ No newline at end of file diff --git a/cognee-mcp/src/__init__.py b/cognee-mcp/src/__init__.py index 1939efe9f..77d325085 100644 --- a/cognee-mcp/src/__init__.py +++ b/cognee-mcp/src/__init__.py @@ -1,8 +1,28 @@ from .server import main as server_main +import warnings +import sys def main(): - """Main entry point for the package.""" + """Deprecated main entry point for the package.""" + import asyncio + + warnings.warn( + "The 'cognee' command for cognee-mcp is deprecated and will be removed in a future version. " + "Please use 'cognee-mcp' instead to avoid conflicts with the main cognee library.", + DeprecationWarning, + stacklevel=2, + ) + + print("⚠️ DEPRECATION WARNING: Use 'cognee-mcp' command instead of 'cognee'", file=sys.stderr) + print(" This avoids conflicts with the main cognee library.", file=sys.stderr) + print(file=sys.stderr) + + asyncio.run(server_main()) + + +def main_mcp(): + """Clean main entry point for cognee-mcp command.""" import asyncio asyncio.run(server_main()) diff --git a/cognee/api/v1/ui/ui.py b/cognee/api/v1/ui/ui.py index f22a1f00b..f4df0edde 100644 --- a/cognee/api/v1/ui/ui.py +++ b/cognee/api/v1/ui/ui.py @@ -516,7 +516,7 @@ def start_ui( env_file, "-e", "TRANSPORT_MODE=sse", - "cognee/cognee-mcp:daulet-dev", + "cognee/cognee-mcp:main", ], stdout=subprocess.PIPE, stderr=subprocess.PIPE,