leocrm/app/plugins/builtins/agent_memory/plugin.py

"""Agent Memory plugin — persistent agent memory with pgvector semantic search."""

from __future__ import annotations

from app.plugins.base import BasePlugin
from app.plugins.manifest import PluginManifest, PluginRouteDef


class AgentMemoryPlugin(BasePlugin):
    """Agent Memory plugin for persistent agent memory with semantic search."""

    manifest = PluginManifest(
        name="agent_memory",
        version="1.0.0",
        display_name="Agent Memory",
        description="Persistent agent memory with pgvector semantic search. Stores facts, context, patterns, and instructions for AI agents.",
        dependencies=["unified_search"],
        routes=[
            PluginRouteDef(
                path="/api/v1/agent-memory",
                module="app.plugins.builtins.agent_memory.routes",
                router_attr="router",
            ),
        ],
        events=[],
        migrations=["0001_initial.sql"],
        permissions=[
            "agent_memory:read",
            "agent_memory:write",
        ],
        is_core=True,
        author="LeoCRM Team",
        min_app_version="1.0.0",
        contract_version="1.0.0",
    )

    def get_entity_models(self) -> dict[str, type]:
        from app.plugins.builtins.agent_memory.models import AgentMemory
        return {"agent_memory": AgentMemory}

    async def on_deactivate(self, db, service_container, event_bus) -> None:
        """Deactivate plugin: unregister contract and event listeners."""
        from app.plugins.builtins.contracts import get_contract_registry
        get_contract_registry().unregister(self.manifest.name)
        await super().on_deactivate(db, service_container, event_bus)