cognee 0.2.0.dev0__tar.gz → 0.2.1.dev1__tar.gz

cognee-0.2.1.dev1/.env.template

@@ -0,0 +1,210 @@
+###############################################################################
+# NOTE: With default settings Cognee only needs an OpenAI LLM_API_KEY to be set.
+#       The rest of the settings don't have to be set.
+#       Default relational database: SQLite
+#       Default vector database   : LanceDB
+#       Default graph database    : Kuzu
+#
+#       These default databases are all file-based, so no extra setup is needed
+#       for local use.
+###############################################################################
+
+################################################################################
+#  🧠 LLM Settings
+################################################################################
+
+LLM_API_KEY="your_api_key"
+LLM_MODEL="openai/gpt-4o-mini"
+LLM_PROVIDER="openai"
+LLM_ENDPOINT=""
+LLM_API_VERSION=""
+LLM_MAX_TOKENS="16384"
+
+EMBEDDING_PROVIDER="openai"
+EMBEDDING_MODEL="openai/text-embedding-3-large"
+EMBEDDING_ENDPOINT=""
+EMBEDDING_API_VERSION=""
+EMBEDDING_DIMENSIONS=3072
+EMBEDDING_MAX_TOKENS=8191
+# If embedding key is not provided same key set for LLM_API_KEY will be used
+#EMBEDDING_API_KEY="your_api_key"
+
+################################################################################
+# 🗄️ Relational database settings
+################################################################################
+
+DB_PROVIDER="sqlite"
+DB_NAME=cognee_db
+
+# -- To switch to Postgres / PGVector, uncomment and fill these: -------------
+#DB_PROVIDER=postgres
+#DB_NAME=cognee_db
+# To use Postgres with the Cognee backend in Docker compose use the following instead: DB_HOST=host.docker.internal
+#DB_HOST=127.0.0.1
+#DB_PORT=5432
+#DB_USERNAME=cognee
+#DB_PASSWORD=cognee
+
+################################################################################
+# 🕸️ Graph Database settings
+################################################################################
+
+# Default (local file-based)
+GRAPH_DATABASE_PROVIDER="kuzu"
+
+# -- To switch to Remote Kuzu uncomment and fill these: -------------------------------------------------------------
+#GRAPH_DATABASE_PROVIDER="kuzu"
+#GRAPH_DATABASE_PROVIDER="kuzu-remote"
+#GRAPH_DATABASE_URL="http://localhost:8000"
+#GRAPH_DATABASE_USERNAME=XXX
+#GRAPH_DATABASE_PASSWORD=YYY
+
+# -- To switch to Neo4j uncomment and fill these: -------------------------------------------------------------------
+#GRAPH_DATABASE_PROVIDER="neo4j"
+#GRAPH_DATABASE_URL=bolt://localhost:7687
+#GRAPH_DATABASE_USERNAME=neo4j
+#GRAPH_DATABASE_PASSWORD=localneo4j
+
+################################################################################
+#  📐 Vector Database settings
+################################################################################
+
+# Supported providers: pgvector | qdrant | weaviate | milvus | lancedb | chromadb
+VECTOR_DB_PROVIDER="lancedb"
+# Not needed if a cloud vector database is not used
+VECTOR_DB_URL=
+VECTOR_DB_KEY=
+
+################################################################################
+#  📂 ROOT DIRECTORY IF USING COGNEE LIB INSIDE A DOCKER
+################################################################################
+# Set up the Cognee system directory. Cognee will store system files and databases here.
+DATA_ROOT_DIRECTORY='/cognee_data/data'
+SYSTEM_ROOT_DIRECTORY='/cognee_data/system'
+
+
+################################################################################
+#  🔄  MIGRATION (RELATIONAL → GRAPH) SETTINGS
+################################################################################
+
+MIGRATION_DB_PATH="/path/to/migration/directory"
+MIGRATION_DB_NAME="migration_database.sqlite"
+MIGRATION_DB_PROVIDER="sqlite"
+
+# -- Postgres-specific migration params --------------------------------------
+# MIGRATION_DB_USERNAME=cognee
+# MIGRATION_DB_PASSWORD=cognee
+# MIGRATION_DB_HOST="127.0.0.1"
+# MIGRATION_DB_PORT=5432
+
+################################################################################
+# 🔒 Security Settings
+################################################################################
+
+# When set to false don't allow adding of local system files to Cognee. Should be set to False when Cognee is used as a backend.
+ACCEPT_LOCAL_FILE_PATH=True
+
+# When set to false don't allow HTTP requests to be sent from Cognee.
+# This protects against Server Side Request Forgery when proper infrastructure is not in place.
+ALLOW_HTTP_REQUESTS=True
+
+# Set this variable to True to enforce usage of backend access control for Cognee
+# Note: This is only currently supported by the following databases:
+#       Relational: SQLite, Postgres
+#       Vector: LanceDB
+#       Graph: KuzuDB
+#
+# It enforces LanceDB and KuzuDB use and uses them to create databases per Cognee user + dataset
+ENABLE_BACKEND_ACCESS_CONTROL=False
+
+################################################################################
+#  🛠️ DEV Settings
+################################################################################
+
+ENV="local"
+
+TOKENIZERS_PARALLELISM="false"
+
+# LITELLM Logging Level. Set to quiet down logging
+LITELLM_LOG="ERROR"
+
+# Set this environment variable to disable sending telemetry data
+# TELEMETRY_DISABLED=1
+
+# Default User Configuration
+# DEFAULT_USER_EMAIL=""
+# DEFAULT_USER_PASSWORD=""
+
+------------------------------- END OF POSSIBLE SETTINGS -------------------------------
+
+
+###############################################################################
+# 🧪  EXAMPLE OVERRIDES (commented out)
+###############################################################################
+# The blocks below show how to configure alternative providers.
+# Uncomment + fill values to switch.
+
+########## Azure OpenAI #######################################################
+#LLM_MODEL="azure/gpt-4o-mini"
+#LLM_ENDPOINT="https://DNS.azure.com/openai/deployments/gpt-4o-mini"
+#LLM_API_KEY="<<TALK TO YOUR AZURE GUY"
+#LLM_API_VERSION="2024-12-01-preview"
+
+## llm api version might not be relevant
+#LLM_MAX_TOKENS="16384"
+
+#EMBEDDING_MODEL="azure/text-embedding-3-large"
+#EMBEDDING_ENDPOINT="https://DNS.openai.azure.com/openai/deployments/text-embedding-3-large"
+#EMBEDDING_API_KEY="<<TALK TO YOUR AZURE GUY>"
+#EMBEDDING_API_VERSION="2024-12-01-preview"
+#EMBEDDING_DIMENSIONS=3072
+#EMBEDDING_MAX_TOKENS=8191
+
+########## Local LLM via Ollama ###############################################
+
+#LLM_API_KEY ="ollama"
+#LLM_MODEL="llama3.1:8b"
+#LLM_PROVIDER="ollama"
+#LLM_ENDPOINT="http://localhost:11434/v1"
+#EMBEDDING_PROVIDER="ollama"
+#EMBEDDING_MODEL="avr/sfr-embedding-mistral:latest"
+#EMBEDDING_ENDPOINT="http://localhost:11434/api/embeddings"
+#EMBEDDING_DIMENSIONS=4096
+#HUGGINGFACE_TOKENIZER="Salesforce/SFR-Embedding-Mistral"
+
+########## OpenRouter (also free) #########################################################
+
+#LLM_API_KEY="<<go-get-one-yourself"
+#LLM_PROVIDER="custom"
+#LLM_MODEL="openrouter/google/gemini-2.0-flash-lite-preview-02-05:free"
+#LLM_ENDPOINT="https://openrouter.ai/api/v1"
+
+########## DeepInfra ##########################################################
+
+#LLM_API_KEY="<<>>"
+#LLM_PROVIDER="custom"
+#LLM_MODEL="deepinfra/meta-llama/Meta-Llama-3-8B-Instruct"
+#LLM_ENDPOINT="https://api.deepinfra.com/v1/openai"
+
+#EMBEDDING_PROVIDER="openai"
+#EMBEDDING_API_KEY="<<>>"
+#EMBEDDING_MODEL="deepinfra/BAAI/bge-base-en-v1.5"
+#EMBEDDING_ENDPOINT=""
+#EMBEDDING_API_VERSION=""
+#EMBEDDING_DIMENSIONS=3072
+#EMBEDDING_MAX_TOKENS=8191
+
+
+########## Release Test ###############################################
+
+#LLM_API_KEY="..."
+
+#OPENAI_API_KEY="..."
+
+#MIGRATION_DB_PATH="~/Downloads/"
+#MIGRATION_DB_NAME="Chinook_Sqlite.sqlite"
+#MIGRATION_DB_PROVIDER="sqlite"
+
+#GRAPH_DATABASE_URL="bolt://54.246.89.112:7687"
+#GRAPH_DATABASE_USERNAME="neo4j"
+#GRAPH_DATABASE_PASSWORD="pleaseletmein"

cognee-0.2.1.dev1/.gitguardian.yml

@@ -0,0 +1,26 @@
+# .gitguardian.yml
+version: v1
+
+secret-scan:
+  # Ignore specific files
+  excluded-paths:
+    - '.env.template'
+    - '.github/workflows/*.yml'
+    - 'examples/**'
+    - 'tests/**'
+
+  # Ignore specific patterns
+  excluded-detectors:
+    - 'Generic Password'
+    - 'Generic High Entropy Secret'
+
+  # Ignore by commit (if needed)
+  excluded-commits:
+    - '782bbb4'
+
+  # Custom rules for template files
+  paths-ignore:
+    - path: '.env.template'
+      comment: 'Template file with placeholder values'
+    - path: '.github/workflows/search_db_tests.yml'
+      comment: 'Test workflow with test credentials'

{cognee-0.2.0.dev0/cognee-starter-kit → cognee-0.2.1.dev1}/.gitignore

@@ -36,6 +36,7 @@ share/python-wheels/
 *.egg-info/
 .installed.cfg
 *.egg
+.python-version
 MANIFEST
 
 # PyInstaller

{cognee-0.2.0.dev0 → cognee-0.2.1.dev1}/Dockerfile

@@ -31,7 +31,7 @@ COPY README.md pyproject.toml uv.lock entrypoint.sh ./
 
 # Install the project's dependencies using the lockfile and settings
 RUN --mount=type=cache,target=/root/.cache/uv \
-    uv sync --extra debug --extra api --extra postgres --extra weaviate --extra qdrant --extra neo4j --extra kuzu --extra llama-index --extra gemini --extra ollama --extra mistral --extra groq --extra anthropic --frozen --no-install-project --no-dev --no-editable
+    uv sync --extra debug --extra api --extra postgres --extra weaviate --extra qdrant --extra neo4j --extra llama-index --extra gemini --extra ollama --extra mistral --extra groq --extra anthropic --frozen --no-install-project --no-dev --no-editable
 
 # Copy Alembic configuration
 COPY alembic.ini /app/alembic.ini
@@ -41,7 +41,7 @@ COPY alembic/ /app/alembic
 # Installing separately from its dependencies allows optimal layer caching
 COPY ./cognee /app/cognee
 RUN --mount=type=cache,target=/root/.cache/uv \
-uv sync --extra debug --extra api --extra postgres --extra weaviate --extra qdrant --extra neo4j --extra kuzu --extra llama-index --extra gemini --extra ollama --extra mistral --extra groq --extra anthropic --frozen --no-dev --no-editable
+uv sync --extra debug --extra api --extra postgres --extra weaviate --extra qdrant --extra neo4j --extra llama-index --extra gemini --extra ollama --extra mistral --extra groq --extra anthropic --frozen --no-dev --no-editable
 
 FROM python:3.12-slim-bookworm
 

{cognee-0.2.0.dev0 → cognee-0.2.1.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cognee
-Version: 0.2.0.dev0
+Version: 0.2.1.dev1
 Summary: Cognee - is a library for enriching LLM context with a semantic layer for better understanding and reasoning.
 Project-URL: Homepage, https://www.cognee.ai
 Project-URL: Repository, https://github.com/topoteretes/cognee
@@ -16,49 +16,51 @@ Classifier: Operating System :: Microsoft :: Windows
 Classifier: Operating System :: POSIX :: Linux
 Classifier: Topic :: Software Development :: Libraries
 Requires-Python: <=3.13,>=3.10
-Requires-Dist: aiofiles>=23.2.1
-Requires-Dist: aiohttp>=3.11.14
-Requires-Dist: aiosqlite<0.21,>=0.20.0
+Requires-Dist: aiofiles<24.0.0,>=23.2.1
+Requires-Dist: aiohttp<4.0.0,>=3.11.14
+Requires-Dist: aiosqlite<1.0.0,>=0.20.0
 Requires-Dist: alembic<2,>=1.13.3
 Requires-Dist: dlt[sqlalchemy]<2,>=1.9.0
-Requires-Dist: fastapi-users[sqlalchemy]==14.0.1
-Requires-Dist: fastapi==0.115.7
-Requires-Dist: filetype>=1.2.0
+Requires-Dist: fastapi-users[sqlalchemy]<15.0.0,>=14.0.1
+Requires-Dist: fastapi<1.0.0,>=0.115.7
+Requires-Dist: filetype<2.0.0,>=1.2.0
 Requires-Dist: graphistry<0.34,>=0.33.5
-Requires-Dist: instructor>=1.7.2
+Requires-Dist: instructor<2.0.0,>=1.9.1
 Requires-Dist: jinja2<4,>=3.1.3
-Requires-Dist: lancedb==0.21.0
+Requires-Dist: kuzu==0.11.0
+Requires-Dist: lancedb<1.0.0,>=0.24.0
 Requires-Dist: langfuse<3,>=2.32.0
 Requires-Dist: limits<5,>=4.4.1
-Requires-Dist: litellm>=1.57.4
+Requires-Dist: litellm<1.71.0,>=1.57.4
 Requires-Dist: matplotlib<4,>=3.8.3
 Requires-Dist: networkx<4,>=3.4.2
-Requires-Dist: nltk==3.9.1
-Requires-Dist: numpy<=2.1,>=1.26.4
-Requires-Dist: onnxruntime<=1.21.1
-Requires-Dist: openai<2,>=1.59.4
-Requires-Dist: pandas>=2.2.2
+Requires-Dist: nltk<4.0.0,>=3.9.1
+Requires-Dist: numpy<=2.3.1,>=1.26.4
+Requires-Dist: onnxruntime<2.0.0,>=1.0.0
+Requires-Dist: openai<2,>=1.80.1
+Requires-Dist: pandas<3.0.0,>=2.2.2
 Requires-Dist: pre-commit<5,>=4.0.1
 Requires-Dist: pydantic-settings<3,>=2.2.1
-Requires-Dist: pydantic==2.10.5
-Requires-Dist: pylance==0.22.0
+Requires-Dist: pydantic<3.0.0,>=2.10.5
+Requires-Dist: pylance<1.0.0,>=0.22.0
+Requires-Dist: pympler<2.0.0,>=1.1
 Requires-Dist: pypdf<6.0.0,>=4.1.0
-Requires-Dist: python-dotenv>=1.0.1
-Requires-Dist: python-multipart==0.0.20
+Requires-Dist: python-dotenv<2.0.0,>=1.0.1
+Requires-Dist: python-multipart<1.0.0,>=0.0.20
 Requires-Dist: rdflib<7.2.0,>=7.1.4
+Requires-Dist: s3fs[boto3]==2025.3.2
 Requires-Dist: scikit-learn<2,>=1.6.1
 Requires-Dist: sentry-sdk[fastapi]<3,>=2.9.0
-Requires-Dist: sqlalchemy==2.0.39
+Requires-Dist: sqlalchemy<3.0.0,>=2.0.39
 Requires-Dist: structlog<26,>=25.2.0
-Requires-Dist: tiktoken<=0.9.0
-Requires-Dist: typing-extensions==4.12.2
+Requires-Dist: tiktoken<1.0.0,>=0.8.0
+Requires-Dist: typing-extensions<5.0.0,>=4.12.2
 Provides-Extra: anthropic
 Requires-Dist: anthropic<0.27,>=0.26.1; extra == 'anthropic'
 Provides-Extra: api
-Requires-Dist: gunicorn<21,>=20.1.0; extra == 'api'
-Requires-Dist: kuzu==0.9.0; extra == 'api'
-Requires-Dist: uvicorn==0.34.0; extra == 'api'
-Requires-Dist: websockets>=15.0.1; extra == 'api'
+Requires-Dist: gunicorn<24,>=20.1.0; extra == 'api'
+Requires-Dist: uvicorn<1.0.0,>=0.34.0; extra == 'api'
+Requires-Dist: websockets<16.0.0,>=15.0.1; extra == 'api'
 Provides-Extra: aws
 Requires-Dist: s3fs[boto3]==2025.3.2; extra == 'aws'
 Provides-Extra: chromadb
@@ -70,7 +72,7 @@ Requires-Dist: transformers<5,>=4.46.3; extra == 'codegraph'
 Requires-Dist: tree-sitter-python<0.24,>=0.23.6; extra == 'codegraph'
 Requires-Dist: tree-sitter<0.25,>=0.24.0; extra == 'codegraph'
 Provides-Extra: debug
-Requires-Dist: debugpy==1.8.9; extra == 'debug'
+Requires-Dist: debugpy<2.0.0,>=1.8.9; extra == 'debug'
 Provides-Extra: deepeval
 Requires-Dist: deepeval<3,>=2.0.1; extra == 'deepeval'
 Provides-Extra: dev
@@ -84,33 +86,33 @@ Requires-Dist: mypy<2,>=1.7.1; extra == 'dev'
 Requires-Dist: notebook<8,>=7.1.0; extra == 'dev'
 Requires-Dist: pylint<4,>=3.0.3; extra == 'dev'
 Requires-Dist: pytest-asyncio<0.22,>=0.21.1; extra == 'dev'
-Requires-Dist: pytest-cov>=6.1.1; extra == 'dev'
+Requires-Dist: pytest-cov<7.0.0,>=6.1.1; extra == 'dev'
 Requires-Dist: pytest<8,>=7.4.0; extra == 'dev'
 Requires-Dist: ruff<1.0.0,>=0.9.2; extra == 'dev'
-Requires-Dist: tweepy==4.14.0; extra == 'dev'
+Requires-Dist: tweepy<5.0.0,>=4.14.0; extra == 'dev'
+Provides-Extra: distributed
+Requires-Dist: modal<2.0.0,>=1.0.5; extra == 'distributed'
 Provides-Extra: docs
-Requires-Dist: unstructured[csv,doc,docx,epub,md,odt,org,ppt,pptx,rst,rtf,tsv,xlsx]<18,>=0.16.13; extra == 'docs'
+Requires-Dist: unstructured[csv,doc,docx,epub,md,odt,org,ppt,pptx,rst,rtf,tsv,xlsx]<19,>=0.18.1; extra == 'docs'
 Provides-Extra: evals
 Requires-Dist: gdown<6,>=5.2.0; extra == 'evals'
 Requires-Dist: plotly<7,>=6.0.0; extra == 'evals'
 Provides-Extra: falkordb
-Requires-Dist: falkordb==1.0.9; extra == 'falkordb'
+Requires-Dist: falkordb<2.0.0,>=1.0.9; extra == 'falkordb'
 Provides-Extra: gemini
 Requires-Dist: google-generativeai<0.9,>=0.8.4; extra == 'gemini'
 Provides-Extra: graphiti
 Requires-Dist: graphiti-core<0.8,>=0.7.0; extra == 'graphiti'
 Provides-Extra: groq
-Requires-Dist: groq==0.8.0; extra == 'groq'
+Requires-Dist: groq<1.0.0,>=0.8.0; extra == 'groq'
 Provides-Extra: gui
 Requires-Dist: pyside6<7,>=6.8.3; extra == 'gui'
 Requires-Dist: qasync<0.28,>=0.27.1; extra == 'gui'
 Provides-Extra: huggingface
 Requires-Dist: transformers<5,>=4.46.3; extra == 'huggingface'
-Provides-Extra: kuzu
-Requires-Dist: kuzu==0.9.0; extra == 'kuzu'
 Provides-Extra: langchain
-Requires-Dist: langchain-text-splitters==0.3.2; extra == 'langchain'
-Requires-Dist: langsmith==0.2.3; extra == 'langchain'
+Requires-Dist: langchain-text-splitters<1.0.0,>=0.3.2; extra == 'langchain'
+Requires-Dist: langsmith<1.0.0,>=0.2.3; extra == 'langchain'
 Provides-Extra: llama-index
 Requires-Dist: llama-index-core<0.13,>=0.12.11; extra == 'llama-index'
 Provides-Extra: milvus
@@ -118,26 +120,26 @@ Requires-Dist: pymilvus<3,>=2.5.0; extra == 'milvus'
 Provides-Extra: mistral
 Requires-Dist: mistral-common<2,>=1.5.2; extra == 'mistral'
 Provides-Extra: neo4j
-Requires-Dist: neo4j<6,>=5.20.0; extra == 'neo4j'
+Requires-Dist: neo4j<6,>=5.28.0; extra == 'neo4j'
 Provides-Extra: notebook
 Requires-Dist: notebook<8,>=7.1.0; extra == 'notebook'
 Provides-Extra: ollama
 Requires-Dist: transformers<5,>=4.46.3; extra == 'ollama'
 Provides-Extra: postgres
-Requires-Dist: asyncpg==0.30.0; extra == 'postgres'
+Requires-Dist: asyncpg<1.0.0,>=0.30.0; extra == 'postgres'
 Requires-Dist: pgvector<0.4,>=0.3.5; extra == 'postgres'
 Requires-Dist: psycopg2<3,>=2.9.10; extra == 'postgres'
 Provides-Extra: postgres-binary
-Requires-Dist: asyncpg==0.30.0; extra == 'postgres-binary'
+Requires-Dist: asyncpg<1.0.0,>=0.30.0; extra == 'postgres-binary'
 Requires-Dist: pgvector<0.4,>=0.3.5; extra == 'postgres-binary'
-Requires-Dist: psycopg2-binary==2.9.10; extra == 'postgres-binary'
+Requires-Dist: psycopg2-binary<3.0.0,>=2.9.10; extra == 'postgres-binary'
 Requires-Dist: psycopg2<3,>=2.9.10; extra == 'postgres-binary'
 Provides-Extra: posthog
 Requires-Dist: posthog<4,>=3.5.0; extra == 'posthog'
 Provides-Extra: qdrant
 Requires-Dist: qdrant-client<2,>=1.14.2; extra == 'qdrant'
 Provides-Extra: weaviate
-Requires-Dist: weaviate-client==4.9.6; extra == 'weaviate'
+Requires-Dist: weaviate-client<5.0.0,>=4.9.6; extra == 'weaviate'
 Description-Content-Type: text/markdown
 
 <div align="center">
@@ -157,6 +159,10 @@ Description-Content-Type: text/markdown
   <a href="https://discord.gg/NQPKmU5CCg">Join Discord</a>
   ·
   <a href="https://www.reddit.com/r/AIMemory/">Join r/AIMemory</a>
+  .
+  <a href="https://www.docs.cognee.ai">Docs</a>
+  .
+  <a href="https://github.com/topoteretes/cognee-community">cognee community repo</a>
   </p>
 
 
@@ -167,11 +173,21 @@ Description-Content-Type: text/markdown
   [![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://github.com/sponsors/topoteretes"><img src="https://img.shields.io/badge/Sponsor-❤️-ff69b4.svg" alt="Sponsor"></a>
+
+<p>
+  <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" style="display:inline-block; margin-right:10px;">
+    <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" width="250" height="54" />
+  </a>
+
+  <a href="https://trendshift.io/repositories/13955" target="_blank" style="display:inline-block;">
+    <img src="https://trendshift.io/api/badge/repositories/13955" alt="topoteretes%2Fcognee | Trendshift" width="250" height="55" />
+  </a>
+</p>
 
-<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>
 
-<a href="https://trendshift.io/repositories/13955" target="_blank"><img src="https://trendshift.io/api/badge/repositories/13955" alt="topoteretes%2Fcognee | Trendshift" style="width: 250px; height: 55px;" width="250" height="55"/></a>
 
+**🚀 We are launching Cognee SaaS: Sign up [here](https://www.cognee.ai/waitlist) for the hosted beta!**
 
 Build dynamic memory for Agents and replace RAG using scalable, modular ECL (Extract, Cognify, Load) pipelines.
 

{cognee-0.2.0.dev0 → cognee-0.2.1.dev1}/README.md

@@ -15,6 +15,10 @@
   <a href="https://discord.gg/NQPKmU5CCg">Join Discord</a>
   ·
   <a href="https://www.reddit.com/r/AIMemory/">Join r/AIMemory</a>
+  .
+  <a href="https://www.docs.cognee.ai">Docs</a>
+  .
+  <a href="https://github.com/topoteretes/cognee-community">cognee community repo</a>
   </p>
 
 
@@ -25,11 +29,21 @@
   [![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://github.com/sponsors/topoteretes"><img src="https://img.shields.io/badge/Sponsor-❤️-ff69b4.svg" alt="Sponsor"></a>
+
+<p>
+  <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" style="display:inline-block; margin-right:10px;">
+    <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" width="250" height="54" />
+  </a>
+
+  <a href="https://trendshift.io/repositories/13955" target="_blank" style="display:inline-block;">
+    <img src="https://trendshift.io/api/badge/repositories/13955" alt="topoteretes%2Fcognee | Trendshift" width="250" height="55" />
+  </a>
+</p>
 
-<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>
 
-<a href="https://trendshift.io/repositories/13955" target="_blank"><img src="https://trendshift.io/api/badge/repositories/13955" alt="topoteretes%2Fcognee | Trendshift" style="width: 250px; height: 55px;" width="250" height="55"/></a>
 
+**🚀 We are launching Cognee SaaS: Sign up [here](https://www.cognee.ai/waitlist) for the hosted beta!**
 
 Build dynamic memory for Agents and replace RAG using scalable, modular ECL (Extract, Cognify, Load) pipelines.
 

{cognee-0.2.0.dev0 → cognee-0.2.1.dev1}/alembic.ini

@@ -102,7 +102,7 @@ handlers =
 qualname = sqlalchemy.engine
 
 [logger_alembic]
-level = INFO
+level = WARN
 handlers =
 qualname = alembic
 

{cognee-0.2.0.dev0 → cognee-0.2.1.dev1}/cognee/api/client.py

@@ -12,6 +12,7 @@ from fastapi.encoders import jsonable_encoder
 from fastapi.responses import JSONResponse, Response
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.exceptions import RequestValidationError
+from fastapi.openapi.utils import get_openapi
 
 from cognee.exceptions import CogneeApiError
 from cognee.shared.logging_utils import get_logger, setup_logging
@@ -75,6 +76,36 @@ app.add_middleware(
 )
 
 
+def custom_openapi():
+    if app.openapi_schema:
+        return app.openapi_schema
+
+    openapi_schema = get_openapi(
+        title="Cognee API",
+        version="1.0.0",
+        description="Cognee API with Bearer token and Cookie auth",
+        routes=app.routes,
+    )
+
+    openapi_schema["components"]["securitySchemes"] = {
+        "BearerAuth": {"type": "http", "scheme": "bearer"},
+        "CookieAuth": {
+            "type": "apiKey",
+            "in": "cookie",
+            "name": os.getenv("AUTH_TOKEN_COOKIE_NAME", "auth_token"),
+        },
+    }
+
+    openapi_schema["security"] = [{"BearerAuth": []}, {"CookieAuth": []}]
+
+    app.openapi_schema = openapi_schema
+
+    return app.openapi_schema
+
+
+app.openapi = custom_openapi
+
+
 @app.exception_handler(RequestValidationError)
 async def request_validation_exception_handler(request: Request, exc: RequestValidationError):
     if request.url.path == "/api/v1/auth/login":
@@ -108,6 +139,22 @@ async def exception_handler(_: Request, exc: CogneeApiError) -> JSONResponse:
     return JSONResponse(status_code=status_code, content={"detail": detail["message"]})
 
 
+@app.get("/")
+async def root():
+    """
+    Root endpoint that returns a welcome message.
+    """
+    return {"message": "Hello, World, I am alive!"}
+
+
+@app.get("/health")
+def health_check():
+    """
+    Health check endpoint that returns the server status.
+    """
+    return Response(status_code=200)
+
+
 app.include_router(get_auth_router(), prefix="/api/v1/auth", tags=["auth"])
 
 app.include_router(
@@ -128,11 +175,11 @@ app.include_router(
     tags=["auth"],
 )
 
-app.include_router(
-    get_users_router(),
-    prefix="/api/v1/users",
-    tags=["users"],
-)
+app.include_router(get_add_router(), prefix="/api/v1/add", tags=["add"])
+
+app.include_router(get_cognify_router(), prefix="/api/v1/cognify", tags=["cognify"])
+
+app.include_router(get_search_router(), prefix="/api/v1/search", tags=["search"])
 
 app.include_router(
     get_permissions_router(),
@@ -140,31 +187,8 @@ app.include_router(
     tags=["permissions"],
 )
 
-
-@app.get("/")
-async def root():
-    """
-    Root endpoint that returns a welcome message.
-    """
-    return {"message": "Hello, World, I am alive!"}
-
-
-@app.get("/health")
-def health_check():
-    """
-    Health check endpoint that returns the server status.
-    """
-    return Response(status_code=200)
-
-
 app.include_router(get_datasets_router(), prefix="/api/v1/datasets", tags=["datasets"])
 
-app.include_router(get_add_router(), prefix="/api/v1/add", tags=["add"])
-
-app.include_router(get_cognify_router(), prefix="/api/v1/cognify", tags=["cognify"])
-
-app.include_router(get_search_router(), prefix="/api/v1/search", tags=["search"])
-
 app.include_router(get_settings_router(), prefix="/api/v1/settings", tags=["settings"])
 
 app.include_router(get_visualize_router(), prefix="/api/v1/visualize", tags=["visualize"])
@@ -177,6 +201,12 @@ codegraph_routes = get_code_pipeline_router()
 if codegraph_routes:
     app.include_router(codegraph_routes, prefix="/api/v1/code-pipeline", tags=["code-pipeline"])
 
+app.include_router(
+    get_users_router(),
+    prefix="/api/v1/users",
+    tags=["users"],
+)
+
 
 def start_api_server(host: str = "0.0.0.0", port: int = 8000):
     """
@@ -197,4 +227,7 @@ def start_api_server(host: str = "0.0.0.0", port: int = 8000):
 
 if __name__ == "__main__":
     logger = setup_logging()
-    start_api_server()
+
+    start_api_server(
+        host=os.getenv("HTTP_API_HOST", "0.0.0.0"), port=int(os.getenv("HTTP_API_PORT", 8000))
+    )