webnovel-writer/.claude/scripts/data_modules/__init__.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Data Modules - 数据链模块包

用于 webnovel-writer 的数据处理：
- 实体消歧 (entity_linker)
- 状态管理 (state_manager)
- 索引管理 (index_manager)
- RAG 检索 (rag_adapter)
- 风格样本 (style_sampler)
- API 客户端 (api_client) - 只有 Embed + Rerank
"""

from .config import DataModulesConfig, get_config, set_project_root
from .api_client import ModalAPIClient, get_client
from .entity_linker import EntityLinker, DisambiguationResult
from .state_manager import StateManager, EntityState, Relationship, StateChange
from .index_manager import IndexManager, ChapterMeta, SceneMeta
from .rag_adapter import RAGAdapter, SearchResult
from .style_sampler import StyleSampler, StyleSample, SceneType

__all__ = [
    # Config
    "DataModulesConfig",
    "get_config",
    "set_project_root",
    # API Client
    "ModalAPIClient",
    "get_client",
    # Entity Linker
    "EntityLinker",
    "DisambiguationResult",
    # State Manager
    "StateManager",
    "EntityState",
    "Relationship",
    "StateChange",
    # Index Manager
    "IndexManager",
    "ChapterMeta",
    "SceneMeta",
    # RAG Adapter
    "RAGAdapter",
    "SearchResult",
    # Style Sampler
    "StyleSampler",
    "StyleSample",
    "SceneType",
]