cognee

Cognee

Author: topoteretes
Version: 0.0.2
Type: tool

Description

This is an AI memory plugin for Dify. Cognee is an open-source knowledge engine that lets you ingest data in any format or structure and continuously learns to provide the right context for AI agents. It combines vector search, graph databases and cognitive science approaches to make your documents both searchable by meaning and connected by relationships as they change and evolve.

Setup

Get your API key and base URL from your Cognee Cloud dashboard.
Install the plugin in your Dify workspace.
Configure it with your Base URL (e.g. ) and API Key.

Tools

Tool	Purpose
Create Dataset	Create a named dataset to hold your data
Add Data	Ingest text into a dataset
Add File	Upload files (PDF, DOCX, TXT, etc.) into a dataset
Cognify	Process a dataset into a searchable knowledge engine
Search	Query the knowledge engine
Get Datasets	List all available datasets
Get Dataset Data	List all data items in a dataset
Delete Dataset	Permanently delete a dataset
Delete Data	Delete a specific data item from a dataset

Create Dataset

Create a new dataset or return an existing one with the same name. Use this before ingesting data if you want to set up the dataset explicitly.

Parameter	Required	Description
Dataset Name	yes	Name for the new dataset

Returns: ,

Add Data

Ingest text content into a dataset. Multiple text items can be separated by newlines.

Parameter	Required	Description
Text Data	yes	Text content to add (newline-separated for multiple items)
Dataset Name	no	Target dataset name (either this or Dataset ID required)
Dataset ID	no	Target dataset UUID (either this or Dataset Name required)
Node Set	no	Comma-separated node set names for graph organization

Returns: , , ,

Add File

Upload files into a dataset. Accepts documents, images, and other file types supported by Cognee. Files are passed as variables from chat input or workflow start nodes.

Parameter	Required	Description
Files	yes	Files to upload (variable from chat or workflow)
Dataset Name	no	Target dataset name (either this or Dataset ID required)
Dataset ID	no	Target dataset UUID (either this or Dataset Name required)
Node Set	no	Comma-separated node set names for graph organization

Returns: , ,

Cognify

Transform ingested data into a knowledge engine. This runs a multi-step pipeline: classifying documents, extracting text chunks, identifying entities and relationships via LLM, generating summaries, and embedding everything into vector and graph stores.

This is a long-running operation — processing time depends on data volume.

Parameter	Required	Description
Datasets	no	Comma-separated dataset names (either this or Dataset IDs required)
Dataset IDs	no	Comma-separated dataset UUIDs (either this or Datasets required)
Custom Prompt	no	Custom prompt for entity extraction and graph generation
Ontology Key	no	Comma-separated keys referencing previously uploaded ontology files

Returns:

Search

Search within the cognee memory using one of 14 search strategies. Each strategy is suited to different use cases.

Parameter	Required	Default	Description
Query	yes		Natural language search query
Search Type	yes		Search strategy (see table below)
Datasets	no		Comma-separated dataset names to search
Dataset IDs	no		Comma-separated dataset UUIDs to search
System Prompt	no		Custom system prompt for completion-type searches
Node Name	no		Comma-separated node set names to filter results
Top K	no	10	Maximum number of results
Only Context	no	false	Return raw retrieval context instead of LLM completion
Verbose	no	false	Include additional details in the response

Read more about the search types in cognee docs: link

Returns: ,

Get Datasets

List all datasets accessible to the authenticated user. No parameters required.

Returns: ,

Get Dataset Data

List all data items stored in a specific dataset.

Parameter	Required	Description
Dataset ID	yes	UUID of the dataset to inspect

Returns: ,

Delete Dataset

Permanently delete a dataset and all its associated data.

Parameter	Required	Description
Dataset ID	yes	UUID of the dataset to delete

Returns: ,

Delete Data

Delete a specific data item from a dataset.

Parameter	Required	Description
Dataset ID	yes	UUID of the dataset
Data ID	yes	UUID of the data item to delete

Returns: , ,

Typical Workflow

Create Dataset — set up a named dataset.
Add Data or Add File — ingest text or documents into the dataset.
Cognify — build the knowledge engine from the ingested data.
Search — search within cognee memory
Get Datasets / Get Dataset Data — inspect what's available.
Delete Data / Delete Dataset — clean up when needed.

Cognee

Description

Setup

Tools

Create Dataset

Add Data

Add File

Cognify

Search

Get Datasets

Get Dataset Data

Delete Dataset

Delete Data

Typical Workflow

Links