Agent Zero
108 lines
3.9 KiB
Python
108 lines
3.9 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 (
|
|
FrontendMenuItem,
|
|
FrontendPageRoute,
|
|
MiniAppContribution,
|
|
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"],
|
|
miniapps=[
|
|
MiniAppContribution(
|
|
app_id="graph_overview",
|
|
name="Wissens-Graph",
|
|
icon="Share2",
|
|
description="Beziehungsübersicht des Knowledge-Graphs.",
|
|
permission="graph:read",
|
|
settings_schema={
|
|
"fields": [
|
|
{"name": "max_items", "label": "Max. Beziehungen", "type": "number", "default": 6},
|
|
]
|
|
},
|
|
col_span=2,
|
|
row_span=1,
|
|
hosts=["chat", "dashboard", "window"],
|
|
component="@/components/dashboard/GraphOverviewWidget",
|
|
order=90,
|
|
),
|
|
],
|
|
menu_items=[
|
|
FrontendMenuItem(
|
|
label_key="nav.graphRag",
|
|
label="Wissens-Graph",
|
|
path="/graph-rag",
|
|
icon="Share2",
|
|
order=87,
|
|
permission="graph:read",
|
|
),
|
|
],
|
|
page_routes=[
|
|
FrontendPageRoute(
|
|
path="/graph-rag",
|
|
component="@/pages/GraphRag",
|
|
protected=True,
|
|
permission="graph:read",
|
|
),
|
|
],
|
|
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."""
|
|
# Base class first (event subscriptions etc.), then own registrations
|
|
# so a failure in our code cannot leave the base half-initialized (ARCH-037).
|
|
await super().on_activate(db, service_container, event_bus)
|
|
|
|
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")
|
|
|
|
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)
|