Cognee Extension

This tutorial covers how to add the Cognee MCP Server as a goose extension to enable knowledge graph memory capabilities, connecting to over 30 data sources for enhanced context and retrieval.

TLDR

Command

uv --directory /path/to/cognee-mcp run python src/server.py

Environment Variables

LLM_API_KEY: <YOUR_OPENAI_API_KEY>

Configuration

info

Note that you'll need uv installed on your system to run this command, as it uses uv.

goose CLI

Install Cognee:

# Clone and install Cognee
git clone https://github.com/topoteretes/cognee
cd cognee/cognee-mcp
uv sync --dev --all-extras --reinstall

# On Linux, install additional dependencies
sudo apt install -y libpq-dev python3-dev

Configure the extension:

1. Run the configure command:

goose configure

2. Choose to add a Command-line Extension.

┌   goose-configure 
│
◇  What would you like to configure?
│  Add Extension 
│
◆  What type of extension would you like to add?
│  ○ Built-in Extension
│  ● Command-line Extension (Run a local command or script)
│  ○ Remote Extension (SSE)
│  ○ Remote Extension (Streaming HTTP)
└

3. Give your extension a name.

┌   goose-configure 
│
◇  What would you like to configure?
│  Add Extension
│
◇  What type of extension would you like to add?
│  Command-line Extension
│
◆  What would you like to call this extension?
│  Cognee
└

4. Enter the command to run when this extension is used.

info

Replace /path/to/cognee-mcp with the actual path to your cloned cognee-mcp directory.

┌   goose-configure 
│
◇  What would you like to configure?
│  Add Extension
│
◇  What type of extension would you like to add?
│  Command-line Extension 
│
◇  What would you like to call this extension?
│  Cognee
│
◆  What command should be run?
│  uv --directory /path/to/cognee-mcp run python src/server.py
└

5. Enter the number of seconds Goose should wait for actions to complete before timing out. Default is 300 seconds.

┌   goose-configure 
│
◇  What would you like to configure?
│  Add Extension
│
◇  What type of extension would you like to add?
│  Command-line Extension
│
◇  What would you like to call this extension?
│  Cognee
│
◇  What command should be run?
│  uv --directory /path/to/cognee-mcp run python src/server.py
│
◆  Please set the timeout for this tool (in secs):
│  300
└

6. Enter a description for this extension.

┌   goose-configure 
│
◇  What would you like to configure?
│  Add Extension
│
◇  What type of extension would you like to add?
│  Command-line Extension
│
◇  What would you like to call this extension?
│  Cognee
│
◇  What command should be run?
│  uv --directory /path/to/cognee-mcp run python src/server.py
│
◇  Please set the timeout for this tool (in secs):
│  300
│
◆  Enter a description for this extension:
│  Knowledge graph memory capabilities with 30+ data source connections
└

7. Add an environment variable for this extension.

info

Obtain an API key for your LLM provider and paste it in. By default, this is an OpenAI API key.

┌   goose-configure 
│
◇  What would you like to configure?
│  Add Extension
│
◇  What type of extension would you like to add?
│  Command-line Extension
│
◇  What would you like to call this extension?
│  Cognee
│
◇  What command should be run?
│  uv --directory /path/to/cognee-mcp run python src/server.py
│
◇  Please set the timeout for this tool (in secs):
│  300
│
◇  Enter a description for this extension:
│  Knowledge graph memory capabilities with 30+ data source connections
│
◆  Would you like to add environment variables?
│  Yes
│
◇  Environment variable name:
│  LLM_API_KEY
│
◇  Environment variable value:
│  ▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪▪
│
◇  Add another environment variable?
│  No
│
└  Added Cognee extension

info

See the Cognee MCP documentation for supported configuration options.

Example Usage

Cognee provides knowledge graph memory capabilities for goose, allowing it to remember and connect information across conversations and documents.

LLM

OpenAI's GPT-4o was used for this task.

goose Prompt

goose, please cognify this information: "I prefer Python for data analysis and use pandas extensively. My current project involves analyzing customer behavior data." Then search for information about my programming preferences.

goose Output

CLI

Tool Calls

─── cognee_cognify | abcdefgh ──────────────────────────

data: I prefer Python for data analysis and use pandas extensively. My current project involves analyzing customer behavior data.

─── cognee_search | abcdefgh ──────────────────────────

search_query: programming preferences Python data analysis

search_type: GRAPH_COMPLETION

I'll help you store that information in the knowledge graph and then search for your programming preferences.

First, let me cognify the information about your preferences:

Cognify Status: Background process launched due to MCP timeout limitations. The information about your Python and pandas preferences for data analysis has been added to the knowledge graph.

Now let me search for information about your programming preferences:

Search Results: Based on the knowledge graph, I found information about your programming preferences:

• Language Preference: You prefer Python for data analysis
• Key Libraries: You use pandas extensively for data manipulation
• Current Focus: Your current project involves analyzing customer behavior data
• Domain: Data science and analytics workflows
• Technical Stack: Python + pandas for data analysis tasks

The knowledge graph has successfully stored and can now retrieve your programming preferences and project context for future conversations.