CrewAI

pip install cognee-integration-crewai

export LLM_API_KEY="your-openai-api-key-here"   # or OPENAI_API_KEY

import asyncio
from dotenv import load_dotenv
import cognee
from crewai import Agent
from cognee_integration_crewai import add_tool, search_tool

load_dotenv()

async def main():
    # Optional: start from a clean knowledge base
    await cognee.prune.prune_data()
    await cognee.prune.prune_system(metadata=True)

    agent = Agent(
        role="Research Analyst",
        goal="Find and analyze information using the knowledge base",
        backstory="You are an expert analyst with access to a comprehensive knowledge base.",
        tools=[add_tool, search_tool],
        verbose=True,
    )

    # Store information
    response = agent.kickoff(
        "Remember that our company signed a contract with HealthBridge Systems "
        "in the healthcare industry, starting Feb 2023, ending Jan 2026, worth £2.4M"
    )
    print(response.raw)

    # Query the stored information
    response = agent.kickoff("What contracts do we have in the healthcare industry?")
    print(response.raw)

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

from cognee_integration_crewai import get_sessionized_cognee_tools

user1_add, user1_search = get_sessionized_cognee_tools("user-123")
user2_add, user2_search = get_sessionized_cognee_tools("user-456")

agent1 = Agent(
    role="Assistant",
    goal="Help user 1",
    backstory="You are a helpful assistant.",
    tools=[user1_add, user1_search],
)