agentgraph/shared/__init__.py

"""
Shared Components

Common utilities and models used across pipeline stages:
- Data models (Entity, Relation, KnowledgeGraph)
- Knowledge graph utilities
- Cross-stage utilities
"""

# Import models from reference-based schema (primary schema used by production methods)
from .models.reference_based.entity import Entity
from .models.reference_based.relation import Relation
from .models.reference_based.report import KnowledgeGraph
from .models.reference_based.content_reference import ContentReference
from .models.reference_based.failure import Failure

# Also make direct-based models available
from .models.direct_based import models as direct_models

__all__ = [
    # Reference-based models (primary)
    'Entity',
    'Relation', 
    'KnowledgeGraph',
    'ContentReference',
    'Failure',
    
    # Direct-based models (alternative)
    'direct_models',
]