Agent Zero
abbe7a18fc
- P0: hooks.py 3-tuple fix, trigger_dispatcher Contract, contacts/plugin unregister_actions_by_owner - P0: 5 test files — check_permission mocks removed, hardcoded DB credential → env var - P1: attachment_service DmsFile via Contract helper, restore_registry/history_hooks dedup - P1: mail/plugin restore unregister, mcp_client datetime.now(UTC), saved_views/filters patterns - P1: ProtectedRoute fail-closed, 13 test assertion fixes (bcrypt, DB-URLs, SECRET_KEYs) - P2: deprecated notifications → post_system_message (3 files), forgejo Base, report_generator lazy import - P2: webhooks permissions, deps.py/roles.py plugin perms removed, import_export default - P2: address/tags/entity_links patterns removed, worker.py Contract-Umgehungen fixed - P2: 28 frontend TODOs (hardcoded constants, deprecated notification API) - P3: dead code, duplicates, deprecated imports, private attr, __import__ inline - P3: 8 frontend TODOs (LucideIcons, inline styles, XSS, i18n) - ruff: 838 → 0 (612 auto-fix + 246 manual + 27 F821 regression fix) - F821: 30 → 0 (AutomationDefinition, DmsFile, user_id, Path, Any, String) - Contract-Umgehungen: 2 neue gefunden (worker.py:169, worker.py:280) und gefixt
63 lines
2.4 KiB
Python
63 lines
2.4 KiB
Python
"""GraphRAG plugin — entity relationship graph with BFS traversal and search provider."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from app.plugins.base import BasePlugin
|
|
from app.plugins.manifest import PluginManifest, PluginRouteDef
|
|
|
|
|
|
class GraphRAGPlugin(BasePlugin):
|
|
"""GraphRAG plugin for entity relationship graph with BFS traversal."""
|
|
|
|
manifest = PluginManifest(
|
|
name="graph_rag",
|
|
version="1.0.0",
|
|
display_name="GraphRAG",
|
|
description="Entity relationship graph with BFS traversal and unified search integration. Stores typed relationships between entities for knowledge graph exploration.",
|
|
dependencies=["unified_search"],
|
|
routes=[
|
|
PluginRouteDef(
|
|
path="/api/v1/graph",
|
|
module="app.plugins.builtins.graph_rag.routes",
|
|
router_attr="router",
|
|
),
|
|
],
|
|
events=[],
|
|
migrations=["0001_initial.sql"],
|
|
permissions=[
|
|
"graph:read",
|
|
"graph: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.graph_rag.models import EntityRelationship
|
|
return {"entity_relationship": EntityRelationship}
|
|
|
|
async def on_activate(self, db, service_container, event_bus) -> None:
|
|
"""Activate plugin: register GraphRAG search provider."""
|
|
from app.plugins.builtins.graph_rag.provider import GraphRAGSearchProvider
|
|
from app.plugins.builtins.unified_search.contracts import get_search_registry
|
|
|
|
registry = get_search_registry()
|
|
try:
|
|
registry.register(GraphRAGSearchProvider())
|
|
except Exception:
|
|
import logging
|
|
logging.getLogger(__name__).exception("Failed to register GraphRAGSearchProvider")
|
|
|
|
await super().on_activate(db, service_container, event_bus)
|
|
|
|
async def on_deactivate(self, db, service_container, event_bus) -> None:
|
|
"""Deactivate plugin: unregister search provider and contract."""
|
|
from app.plugins.builtins.unified_search.contracts import get_search_registry
|
|
get_search_registry().unregister("graph_relationship")
|
|
|
|
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)
|