Metadata-Version: 2.3
Name: cognee
Version: 0.1.39
Summary: Cognee - is a library for enriching LLM context with a semantic layer for better understanding and reasoning.
License: Apache-2.0
Author: Vasilije Markovic
Requires-Python: >=3.10,<=3.13
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: MacOS :: MacOS X
Classifier: Operating System :: Microsoft :: Windows
Classifier: Operating System :: POSIX :: Linux
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Topic :: Software Development :: Libraries
Provides-Extra: anthropic
Provides-Extra: api
Provides-Extra: chromadb
Provides-Extra: codegraph
Provides-Extra: deepeval
Provides-Extra: docs
Provides-Extra: evals
Provides-Extra: falkordb
Provides-Extra: gemini
Provides-Extra: graphiti
Provides-Extra: groq
Provides-Extra: gui
Provides-Extra: huggingface
Provides-Extra: kuzu
Provides-Extra: langchain
Provides-Extra: llama-index
Provides-Extra: milvus
Provides-Extra: mistral
Provides-Extra: neo4j
Provides-Extra: notebook
Provides-Extra: ollama
Provides-Extra: postgres
Provides-Extra: posthog
Provides-Extra: qdrant
Provides-Extra: weaviate
Requires-Dist: aiofiles (>=23.2.1,<24.0.0)
Requires-Dist: aiohttp (>=3.11.14,<4.0.0)
Requires-Dist: aiosqlite (>=0.20.0,<0.21.0)
Requires-Dist: alembic (>=1.13.3,<2.0.0)
Requires-Dist: anthropic (>=0.26.1,<0.27.0) ; extra == "anthropic"
Requires-Dist: asyncpg (==0.30.0) ; extra == "postgres"
Requires-Dist: boto3 (>=1.26.125,<2.0.0)
Requires-Dist: botocore (>=1.35.54,<2.0.0)
Requires-Dist: chromadb (>=0.6.0,<0.7.0) ; extra == "chromadb"
Requires-Dist: deepeval (>=2.0.1,<3.0.0) ; extra == "deepeval"
Requires-Dist: dlt[sqlalchemy] (>=1.9.0,<2.0.0)
Requires-Dist: falkordb (==1.0.9) ; extra == "falkordb"
Requires-Dist: fastapi (==0.115.7)
Requires-Dist: fastapi-users[sqlalchemy] (==14.0.1)
Requires-Dist: fastembed (<=0.6.0) ; (python_version < "3.13") and (extra == "codegraph")
Requires-Dist: filetype (>=1.2.0,<2.0.0)
Requires-Dist: gdown (>=5.2.0,<6.0.0) ; extra == "evals"
Requires-Dist: google-generativeai (>=0.8.4,<0.9.0) ; extra == "gemini"
Requires-Dist: graphistry (>=0.33.5,<0.34.0)
Requires-Dist: graphiti-core (>=0.7.0,<0.8.0) ; extra == "graphiti"
Requires-Dist: groq (==0.8.0) ; extra == "groq"
Requires-Dist: gunicorn (>=20.1.0,<21.0.0) ; extra == "api"
Requires-Dist: instructor (==1.7.2)
Requires-Dist: jinja2 (>=3.1.3,<4.0.0)
Requires-Dist: kuzu (==0.8.2) ; extra == "kuzu"
Requires-Dist: lancedb (==0.16.0)
Requires-Dist: langchain_text_splitters (==0.3.2) ; extra == "langchain"
Requires-Dist: langfuse (>=2.32.0,<3.0.0)
Requires-Dist: langsmith (==0.2.3) ; extra == "langchain"
Requires-Dist: limits (>=4.4.1,<5.0.0)
Requires-Dist: litellm (>=1.57.4)
Requires-Dist: llama-index-core (>=0.12.11,<0.13.0) ; extra == "llama-index"
Requires-Dist: matplotlib (>=3.8.3,<4.0.0)
Requires-Dist: mistral-common (>=1.5.2,<2.0.0) ; extra == "mistral"
Requires-Dist: neo4j (>=5.20.0,<6.0.0) ; extra == "neo4j"
Requires-Dist: networkx (>=3.2.1,<4.0.0)
Requires-Dist: nltk (==3.9.1)
Requires-Dist: notebook (>=7.1.0,<8.0.0) ; extra == "notebook"
Requires-Dist: numpy (>=1.26.4,<=2.1)
Requires-Dist: openai (>=1.59.4,<2.0.0)
Requires-Dist: owlready2 (>=0.47,<0.48)
Requires-Dist: pandas (==2.2.3)
Requires-Dist: pgvector (>=0.3.5,<0.4.0) ; extra == "postgres"
Requires-Dist: plotly (>=6.0.0,<7.0.0) ; extra == "evals"
Requires-Dist: posthog (>=3.5.0,<4.0.0) ; extra == "posthog"
Requires-Dist: pre-commit (>=4.0.1,<5.0.0)
Requires-Dist: psycopg2 (>=2.9.10,<3.0.0) ; extra == "postgres"
Requires-Dist: pydantic (==2.10.5)
Requires-Dist: pydantic-settings (>=2.2.1,<3.0.0)
Requires-Dist: pymilvus (>=2.5.0,<3.0.0) ; extra == "milvus"
Requires-Dist: pypdf (>=4.1.0,<6.0.0)
Requires-Dist: pyside6 (>=6.8.3,<7.0.0) ; extra == "gui"
Requires-Dist: python-dotenv (==1.0.1)
Requires-Dist: python-multipart (==0.0.20)
Requires-Dist: qasync (>=0.27.1,<0.28.0) ; extra == "gui"
Requires-Dist: qdrant-client (>=1.9.0,<2.0.0) ; extra == "qdrant"
Requires-Dist: s3fs (>=2025.3.2,<2026.0.0)
Requires-Dist: scikit-learn (>=1.6.1,<2.0.0)
Requires-Dist: sentry-sdk[fastapi] (>=2.9.0,<3.0.0)
Requires-Dist: sqlalchemy (==2.0.39)
Requires-Dist: structlog (>=25.2.0,<26.0.0)
Requires-Dist: tiktoken (<=0.9.0)
Requires-Dist: transformers (>=4.46.3,<5.0.0) ; extra == "huggingface" or extra == "ollama" or extra == "codegraph"
Requires-Dist: tree-sitter (>=0.24.0,<0.25.0) ; extra == "codegraph"
Requires-Dist: tree-sitter-python (>=0.23.6,<0.24.0) ; extra == "codegraph"
Requires-Dist: typing_extensions (==4.12.2)
Requires-Dist: unstructured[csv,doc,docx,epub,md,odt,org,ppt,pptx,rst,rtf,tsv,xlsx] (>=0.16.13,<0.17.0) ; extra == "docs"
Requires-Dist: uvicorn (==0.34.0) ; extra == "api"
Requires-Dist: weaviate-client (==4.9.6) ; extra == "weaviate"
Project-URL: Homepage, https://www.cognee.ai
Project-URL: Repository, https://github.com/topoteretes/cognee
Description-Content-Type: text/markdown

<div align="center">
  <a href="https://github.com/topoteretes/cognee">
    <img src="https://raw.githubusercontent.com/topoteretes/cognee/refs/heads/dev/assets/cognee-logo-transparent.png" alt="Cognee Logo" height="60">
  </a>

  <br />

  cognee - Memory for AI Agents in 5 lines of code

  <p align="center">
  <a href="https://www.youtube.com/watch?v=1bezuvLwJmw&t=2s">Demo</a>
  .
  <a href="https://cognee.ai">Learn more</a>
  ·
  <a href="https://discord.gg/NQPKmU5CCg">Join Discord</a>
  </p>


  [![GitHub forks](https://img.shields.io/github/forks/topoteretes/cognee.svg?style=social&label=Fork&maxAge=2592000)](https://GitHub.com/topoteretes/cognee/network/)
  [![GitHub stars](https://img.shields.io/github/stars/topoteretes/cognee.svg?style=social&label=Star&maxAge=2592000)](https://GitHub.com/topoteretes/cognee/stargazers/)
  [![GitHub commits](https://badgen.net/github/commits/topoteretes/cognee)](https://GitHub.com/topoteretes/cognee/commit/)
  [![Github tag](https://badgen.net/github/tag/topoteretes/cognee)](https://github.com/topoteretes/cognee/tags/)
  [![Downloads](https://static.pepy.tech/badge/cognee)](https://pepy.tech/project/cognee)
  [![License](https://img.shields.io/github/license/topoteretes/cognee?colorA=00C586&colorB=000000)](https://github.com/topoteretes/cognee/blob/main/LICENSE)
  [![Contributors](https://img.shields.io/github/contributors/topoteretes/cognee?colorA=00C586&colorB=000000)](https://github.com/topoteretes/cognee/graphs/contributors)

<a href="https://www.producthunt.com/posts/cognee?embed=true&utm_source=badge-top-post-badge&utm_medium=badge&utm_souce=badge-cognee" target="_blank"><img src="https://api.producthunt.com/widgets/embed-image/v1/top-post-badge.svg?post_id=946346&theme=light&period=daily&t=1744472480704" alt="cognee - Memory&#0032;for&#0032;AI&#0032;Agents&#0032;&#0032;in&#0032;5&#0032;lines&#0032;of&#0032;code | Product Hunt" style="width: 250px; height: 54px;" width="250" height="54" /></a>



Build dynamic Agent memory using scalable, modular ECL (Extract, Cognify, Load) pipelines.

More on [use-cases](https://docs.cognee.ai/use-cases) and [evals](https://github.com/topoteretes/cognee/tree/main/evals) 

<div style="text-align: center">
  <img src="https://raw.githubusercontent.com/topoteretes/cognee/refs/heads/main/assets/cognee_benefits.png" alt="Why cognee?" width="50%" />
</div>

</div>




## Features

- Interconnect and retrieve your past conversations, documents, images and audio transcriptions
- Reduce hallucinations, developer effort, and cost.
- Load data to graph and vector databases using only Pydantic
- Manipulate your data while ingesting from 30+ data sources

## Get Started

Get started quickly with a Google Colab  <a href="https://colab.research.google.com/drive/1g-Qnx6l_ecHZi0IOw23rg0qC4TYvEvWZ?usp=sharing">notebook</a>  or  <a href="https://github.com/topoteretes/cognee-starter">starter repo</a>

## Contributing
Your contributions are at the core of making this a true open source project. Any contributions you make are **greatly appreciated**. See [`CONTRIBUTING.md`](CONTRIBUTING.md) for more information.





## 📦 Installation

You can install Cognee using either **pip**, **poetry**, **uv** or any other python package manager.

### With pip

```bash
pip install cognee
```

## 💻 Basic Usage

### Setup

```
import os
os.environ["LLM_API_KEY"] = "YOUR OPENAI_API_KEY"

```

You can also set the variables by creating .env file, using our <a href="https://github.com/topoteretes/cognee/blob/main/.env.template">template.</a>
To use different LLM providers, for more info check out our <a href="https://docs.cognee.ai">documentation</a>


### Simple example

This script will run the default pipeline:

```python
import cognee
import asyncio


async def main():
    # Add text to cognee
    await cognee.add("Natural language processing (NLP) is an interdisciplinary subfield of computer science and information retrieval.")

    # Generate the knowledge graph
    await cognee.cognify()

    # Query the knowledge graph
    results = await cognee.search("Tell me about NLP")

    # Display the results
    for result in results:
        print(result)


if __name__ == '__main__':
    asyncio.run(main())

```
Example output:
```
  Natural Language Processing (NLP) is a cross-disciplinary and interdisciplinary field that involves computer science and information retrieval. It focuses on the interaction between computers and human language, enabling machines to understand and process natural language.
  
```
Graph visualization:
<a href="https://rawcdn.githack.com/topoteretes/cognee/refs/heads/add-visualization-readme/assets/graph_visualization.html"><img src="assets/graph_visualization.png" width="100%" alt="Graph Visualization"></a>
Open in [browser](https://rawcdn.githack.com/topoteretes/cognee/refs/heads/add-visualization-readme/assets/graph_visualization.html).

For more advanced usage, have a look at our <a href="https://docs.cognee.ai"> documentation</a>.


## Understand our architecture

<div style="text-align: center">
  <img src="assets/cognee_diagram.png" alt="cognee concept diagram" width="100%" />
</div>



## Demos

1. What is AI memory:

[Learn about cognee](https://github.com/user-attachments/assets/8b2a0050-5ec4-424c-b417-8269971503f0)

2. Simple GraphRAG demo

[Simple GraphRAG demo](https://github.com/user-attachments/assets/d80b0776-4eb9-4b8e-aa22-3691e2d44b8f)

3. cognee with Ollama

[cognee with local models](https://github.com/user-attachments/assets/8621d3e8-ecb8-4860-afb2-5594f2ee17db)


## Code of Conduct

We are committed to making open source an enjoyable and respectful experience for our community. See <a href="https://github.com/topoteretes/cognee/blob/main/CODE_OF_CONDUCT.md"><code>CODE_OF_CONDUCT</code></a> for more information.

## 💫 Contributors

<a href="https://github.com/topoteretes/cognee/graphs/contributors">
  <img alt="contributors" src="https://contrib.rocks/image?repo=topoteretes/cognee"/>
</a>


## Star History

[![Star History Chart](https://api.star-history.com/svg?repos=topoteretes/cognee&type=Date)](https://star-history.com/#topoteretes/cognee&Date)


