mentedb-crewai 0.1.0__py3-none-any.whl

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/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/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.dist-info/METADATA

@@ -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.dist-info/RECORD

@@ -0,0 +1,6 @@
+mentedb_crewai/__init__.py,sha256=B7UF3IRGVmXGac2RHpK9V6wML3mnuFGqMIModPCFYik,144
+mentedb_crewai/memory.py,sha256=FBMbMaXcSE3XUVrpA-HtOJ0Lmr1FmiOaI2b7c-mK-7Y,2283
+mentedb_crewai/tool.py,sha256=SBW2pUossmBR-LzdVN_GotYDAEP5eToOb68FrDn--sI,1024
+mentedb_crewai-0.1.0.dist-info/METADATA,sha256=PaVl8FCkBWckDLNhtBMhA4xGkvxZtz8psGuS-Gr-gIo,222
+mentedb_crewai-0.1.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+mentedb_crewai-0.1.0.dist-info/RECORD,,

mentedb_crewai-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.29.0
+Root-Is-Purelib: true
+Tag: py3-none-any