Agent Zero
000c969b13
Check Cross-Plugin Imports / check (push) Has been cancelled
5.5 Plugin-Marketplace: - New plugin: marketplace/ (models, routes, services, schemas, config) - MarketplaceListing model (global, no tenant_id) - Ed25519 signature verification via PluginSignature - Endpoints: list, detail, install, verify, categories - Config: MARKETPLACE_SERVER_URL setting 5.6 Agent Memory (persistent): - New plugin: agent_memory/ (models, routes, services, schemas) - AgentMemory model with embedding vector(768) + HNSW index - store_memory() with auto-embedding - retrieve_relevant_memories() with pgvector cosine similarity - Semantic search endpoint 5.7 GraphRAG: - New plugin: graph_rag/ (models, routes, services, provider, schemas) - EntityRelationship model (source/target type+id, relationship_type, metadata) - BFS graph traversal (bidirectional, configurable depth) - GraphRAGSearchProvider registered in unified_search 5.8 Subagents / Multi-Agent: - AgentCoordinator class (create_subtask, wait_for_subtask, aggregate, cancel) - AgentSubtask model + migration 0002_agent_subtasks.sql - 6 new API endpoints for subtask management - Tools registered in AI tool registry 5.9 External Agent API: - external_api.py: POST /run, GET /status, POST /stream (SSE) - Bearer API token authentication - Rate limiting: 10 req/min per token - ExternalAgentRequest/Response schemas 3 new plugins registered in main.py and __init__.py All files py_compile clean
53 lines
2.2 KiB
Python
53 lines
2.2 KiB
Python
"""EntityRelationship model for GraphRAG — stores relationships between entities."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import uuid
|
|
from typing import Any
|
|
|
|
from sqlalchemy import Index, String
|
|
from sqlalchemy.dialects.postgresql import JSONB
|
|
from sqlalchemy.dialects.postgresql import UUID as PGUUID
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from app.core.db import Base, TenantMixin
|
|
from app.models.owned_mixin import OwnedMixin
|
|
|
|
|
|
class EntityRelationship(Base, TenantMixin, OwnedMixin):
|
|
"""Directed relationship between two entities in the knowledge graph.
|
|
|
|
Stores typed relationships (e.g., 'works_for', 'has_email', 'related_to')
|
|
between any two entities in the system. Used for GraphRAG traversal.
|
|
"""
|
|
|
|
__tablename__ = "entity_relationships"
|
|
__table_args__ = (
|
|
Index("ix_entity_relationships_source", "tenant_id", "source_type", "source_id"),
|
|
Index("ix_entity_relationships_target", "tenant_id", "target_type", "target_id"),
|
|
Index("ix_entity_relationships_type", "tenant_id", "relationship_type"),
|
|
Index("ix_entity_relationships_source_target", "tenant_id", "source_type", "source_id", "target_type", "target_id"),
|
|
)
|
|
|
|
id: Mapped[uuid.UUID] = mapped_column(
|
|
PGUUID(as_uuid=True), primary_key=True, default=uuid.uuid4
|
|
)
|
|
source_type: Mapped[str] = mapped_column(
|
|
String(50), nullable=False, comment="Entity type of the source (e.g. 'contact', 'company', 'file')"
|
|
)
|
|
source_id: Mapped[uuid.UUID] = mapped_column(
|
|
PGUUID(as_uuid=True), nullable=False, comment="UUID of the source entity"
|
|
)
|
|
target_type: Mapped[str] = mapped_column(
|
|
String(50), nullable=False, comment="Entity type of the target (e.g. 'contact', 'email', 'task')"
|
|
)
|
|
target_id: Mapped[uuid.UUID] = mapped_column(
|
|
PGUUID(as_uuid=True), nullable=False, comment="UUID of the target entity"
|
|
)
|
|
relationship_type: Mapped[str] = mapped_column(
|
|
String(50), nullable=False, comment="Type of relationship (e.g. 'works_for', 'has_email', 'related_to')"
|
|
)
|
|
metadata: Mapped[dict[str, Any] | None] = mapped_column(
|
|
JSONB, nullable=True, default=dict, comment="Arbitrary metadata about the relationship"
|
|
)
|