agentgraph/shared/models/__init__.py

"""
Models Package for Knowledge Extraction

This package contains schema models organized by type:
- reference_based: Models that use content references and line numbers
- direct_based: Models that use direct text content
"""

# Import commonly used models for backward compatibility
from .reference_based import Entity, Relation, KnowledgeGraph, ContentReference, Failure

__all__ = [
    "Entity",
    "Relation", 
    "KnowledgeGraph",
    "ContentReference",
    "Failure"
]