mentedb-crewai 0.1.0__tar.gz

mentedb_crewai-0.1.0/.gitignore

@@ -0,0 +1,9 @@
+target/
+*.so
+*.dylib
+*.pyd
+*.egg-info/
+__pycache__/
+dist/
+build/
+.eggs/

mentedb_crewai-0.1.0/PKG-INFO

@@ -0,0 +1,8 @@
+Metadata-Version: 2.4
+Name: mentedb-crewai
+Version: 0.1.0
+Summary: MenteDB integration for CrewAI and AutoGen
+License-Expression: Apache-2.0
+Requires-Python: >=3.9
+Requires-Dist: crewai>=0.30
+Requires-Dist: mentedb>=0.7.0

mentedb_crewai-0.1.0/README.md

@@ -0,0 +1,102 @@
+# mentedb-crewai
+
+MenteDB integration for CrewAI and AutoGen. Gives your multi agent teams persistent, cognitive memory that survives across sessions and tracks reasoning quality over time.
+
+## Installation
+
+```bash
+pip install mentedb-crewai
+```
+
+## Components
+
+### MenteDBCrewMemory
+
+A memory backend for CrewAI agents that stores and retrieves memories through MenteDB. Each agent can have its own memory space, or agents on the same team can share a common space for collaborative recall.
+
+```python
+from crewai import Agent, Crew, Task
+from mentedb_crewai import MenteDBCrewMemory
+
+memory = MenteDBCrewMemory(
+    data_dir="./crew-memory",
+    space="research-team",
+)
+
+researcher = Agent(
+    role="Senior Researcher",
+    goal="Find comprehensive information on the topic",
+    memory=memory,
+)
+
+writer = Agent(
+    role="Technical Writer",
+    goal="Write clear documentation from research",
+    memory=MenteDBCrewMemory(
+        data_dir="./crew-memory",
+        space="research-team",
+        agent_name="writer",
+    ),
+)
+
+task = Task(
+    description="Research and document the latest trends in vector databases",
+    agent=researcher,
+)
+
+crew = Crew(agents=[researcher, writer], tasks=[task])
+crew.kickoff()
+```
+
+Memory is stored persistently, so the next time the crew runs it can recall findings from previous sessions.
+
+### MenteDBTool
+
+A tool that lets agents query MenteDB directly using MQL (Mente Query Language). Attach it to any agent so the agent can search and retrieve from the team's knowledge base during task execution.
+
+```python
+from crewai import Agent
+from mentedb_crewai import MenteDBTool
+
+memory_tool = MenteDBTool(data_dir="./crew-memory")
+
+agent = Agent(
+    role="Analyst",
+    goal="Analyze data using historical context",
+    tools=[memory_tool],
+)
+```
+
+The agent can then invoke the tool during its reasoning to recall past decisions, find related context, or check what the team already knows about a topic.
+
+## Usage with AutoGen
+
+MenteDB works with AutoGen agents through the same memory and tool interfaces. Use `MenteDBCrewMemory` to give AutoGen agents persistent memory, or `MenteDBTool` to let them query the knowledge base.
+
+```python
+from mentedb_crewai import MenteDBCrewMemory, MenteDBTool
+
+memory = MenteDBCrewMemory(data_dir="./autogen-memory", space="dev-team")
+tool = MenteDBTool(data_dir="./autogen-memory")
+
+# Store context from agent work
+memory.store("The team decided to use PostgreSQL for the main database.")
+
+# Search for relevant memories
+results = memory.search("database decision", k=5)
+
+# Get assembled context within a token budget
+context = memory.get_context("Plan the database migration", token_budget=2000)
+```
+
+## Configuration
+
+| Parameter | Default | Description |
+|---|---|---|
+| `data_dir` | `./mentedb-data` | Path to the MenteDB data directory |
+| `space` | `default` | Memory space name for agent isolation or sharing |
+| `agent_name` | `None` | Optional agent name for per agent memory scoping |
+
+## License
+
+Apache 2.0

mentedb_crewai-0.1.0/mentedb_crewai/__init__.py

@@ -0,0 +1,4 @@
+from mentedb_crewai.memory import MenteDBCrewMemory
+from mentedb_crewai.tool import MenteDBTool
+
+__all__ = ["MenteDBCrewMemory", "MenteDBTool"]

mentedb_crewai-0.1.0/mentedb_crewai/memory.py

@@ -0,0 +1,67 @@
+"""CrewAI memory backend powered by MenteDB."""
+from typing import Any, Dict, List, Optional
+
+from mentedb import MenteDB
+
+
+class MenteDBCrewMemory:
+    """Drop-in memory backend for CrewAI agents.
+
+    Gives CrewAI agents persistent, cognitive memory with:
+    - Cross-session recall
+    - Multi-agent memory isolation via spaces
+    - Semantic search over past context
+
+    Usage:
+        from crewai import Agent
+        from mentedb_crewai import MenteDBCrewMemory
+
+        memory = MenteDBCrewMemory(
+            data_dir="./crew-memory",
+            space="research-team",
+        )
+
+        agent = Agent(
+            role="Researcher",
+            memory=memory,
+        )
+    """
+
+    def __init__(self, data_dir: str = "./mentedb-data", space: str = "default",
+                 agent_name: Optional[str] = None):
+        self.data_dir = data_dir
+        self.space = space
+        self.agent_name = agent_name
+        self._db = MenteDB(data_dir)
+
+    def store(self, content: str, metadata: Optional[Dict[str, Any]] = None) -> str:
+        """Store a memory from the agent's work."""
+        tags = [self.space]
+        if self.agent_name:
+            tags.append(self.agent_name)
+        if metadata and "tags" in metadata:
+            tags.extend(metadata["tags"])
+        memory_type = (metadata or {}).get("memory_type", "episodic")
+        return self._db.store(content, memory_type=memory_type, tags=tags)
+
+    def search(self, query: str, k: int = 5) -> List[Dict[str, Any]]:
+        """Search for relevant memories."""
+        results = self._db.search_text(query, k=k, tags=[self.space])
+        docs = []
+        for r in results:
+            mem = self._db.get_memory(r.id)
+            if mem:
+                docs.append({
+                    "content": mem["content"],
+                    "score": r.score,
+                    "id": r.id,
+                    "memory_type": mem.get("memory_type", ""),
+                })
+        return docs
+
+    def get_context(self, task_description: str, token_budget: int = 2000) -> str:
+        """Get assembled context for a task using MQL recall."""
+        result = self._db.recall(
+            f"RECALL memories WHERE content ~> \"{task_description}\" LIMIT 10"
+        )
+        return result.text if result else ""

mentedb_crewai-0.1.0/mentedb_crewai/tool.py

@@ -0,0 +1,35 @@
+"""CrewAI tool for querying MenteDB with MQL."""
+
+from mentedb import MenteDB
+
+
+class MenteDBTool:
+    """A CrewAI tool that lets agents query their memory database directly.
+
+    Usage:
+        from crewai import Agent
+        from mentedb_crewai import MenteDBTool
+
+        memory_tool = MenteDBTool(data_dir="./crew-memory")
+
+        agent = Agent(
+            role="Researcher",
+            tools=[memory_tool],
+        )
+    """
+
+    name: str = "mentedb_query"
+    description: str = (
+        "Query the team's knowledge database using MQL (Mente Query Language). "
+        "Use for recalling past decisions, finding related context, or checking "
+        "what the team knows about a topic."
+    )
+
+    def __init__(self, data_dir: str = "./mentedb-data"):
+        self.data_dir = data_dir
+        self._db = MenteDB(data_dir)
+
+    def run(self, query: str) -> str:
+        """Execute an MQL query and return assembled context."""
+        result = self._db.recall(query)
+        return result.text if result else ""

mentedb_crewai-0.1.0/pyproject.toml

@@ -0,0 +1,14 @@
+[project]
+name = "mentedb-crewai"
+version = "0.1.0"
+description = "MenteDB integration for CrewAI and AutoGen"
+license = "Apache-2.0"
+requires-python = ">=3.9"
+dependencies = [
+    "mentedb>=0.7.0",
+    "crewai>=0.30",
+]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"