src/core/__init__.py

"""
HITL-KG Core Module

Human-in-the-Loop Knowledge Graph Reasoning System

Components:
- KnowledgeGraph: Core graph with entities and reasoning nodes
- ReasoningEngine: LLM and local reasoning generators
- EmbeddingService: Multilingual semantic search
- SessionManager: Per-user session handling
- DatasetLoader: Generic ontology loading

Usage:
    from src.core import load_knowledge_graph, create_engine, LLMProvider
    
    kg = load_knowledge_graph()
    engine = create_engine(LLMProvider.LOCAL, kg)
    context = engine.build_context("fever and cough")
    for node in engine.generate(context, GenerationConfig()):
        print(node.content)
"""

# Configuration
from .config import (
    AppConfig,
    DatasetConfig,
    EmbeddingConfig,
    LLMConfig,
    get_config,
    load_config,
)

# Knowledge Graph
from .knowledge_graph import (
    KnowledgeGraph,
    ReasoningNode,
    ReasoningEdge,
    Entity,
    NodeType,
    EdgeType,
    EntityCategory,
    NODE_TYPE_INFO,
    create_node_id,
)

# LLM Engine
from .llm_engine import (
    ReasoningEngine,
    OpenAIEngine,
    LocalEngine,
    GraphSynchronizer,
    GenerationConfig,
    ReasoningContext,
    ReasoningChainCache,
    LLMProvider,
    create_engine,
    detect_language,
    LANGUAGE_NAMES,
)

# Dataset Loading
from .dataset_loader import (
    DatasetLoader,
    OntologyTerm,
    load_knowledge_graph,
    build_knowledge_graph,
)

# Embedding Service
from .embedding_service import (
    EmbeddingService,
    SearchResult,
    get_embedding_service,
)

# Session Management
from .session_manager import (
    Session,
    ChatMessage,
    UserInteraction,
    SessionManager,
    get_session_manager,
    generate_session_id,
)


__all__ = [
    # Config
    "AppConfig",
    "DatasetConfig", 
    "EmbeddingConfig",
    "LLMConfig",
    "get_config",
    "load_config",
    
    # Knowledge Graph
    "KnowledgeGraph",
    "ReasoningNode",
    "ReasoningEdge",
    "Entity",
    "NodeType",
    "EdgeType",
    "EntityCategory",
    "NODE_TYPE_INFO",
    "create_node_id",
    
    # LLM Engine
    "ReasoningEngine",
    "OpenAIEngine",
    "LocalEngine",
    "GraphSynchronizer",
    "GenerationConfig",
    "ReasoningContext",
    "ReasoningChainCache",
    "LLMProvider",
    "create_engine",
    "detect_language",
    "LANGUAGE_NAMES",
    
    # Dataset
    "DatasetLoader",
    "OntologyTerm",
    "load_knowledge_graph",
    "build_knowledge_graph",
    
    # Embeddings
    "EmbeddingService",
    "SearchResult",
    "get_embedding_service",
    
    # Sessions
    "Session",
    "ChatMessage",
    "UserInteraction",
    "SessionManager",
    "get_session_manager",
    "generate_session_id",
]