__init__.py

"""GraphAnything — turn anything into a knowledge graph.

Public API for the unified graph-construction core. Three drivers consume it:

  - CLI / REPL   (human-driven)        → `graphanything` console-script
  - Skill / MCP  (Claude-driven)       → /graphanything
  - Agent SDK    (LLM-driven, headless) → register your own driver

All three go through the same Session state machine + extractor registry +
viz module. Adding a new source = registering an Extractor; adding a new
shell = wrapping the 5 verbs (propose / refine / sample / review / run).

LLM calls go through GraphAnything's built-in OpenAI-compatible client at
`GraphAnything.llm_client`, which talks to any `chat.completions`-shaped
endpoint (vLLM serve, llama.cpp, Ollama, LM Studio, OpenAI, …). 8 built-in
extractors, 10 schema presets, 9 viz formats, 17 MCP tools, 19 CLI
sub-commands.
"""
from __future__ import annotations

__all__ = [
    "open_session",
    "load_session",
    "Session",
    "Schema",
    "register_extractor",
    "list_extractors",
    "run_extractor",
    "list_presets",
    "load_preset",
    "render",
    "ProvenanceStamp",
    "attach_provenance",
]


def __getattr__(name):
    # Lazy imports so heavy deps (networkx, prompt_toolkit) are only loaded
    # when actually used. Keeps `graphanything --help` snappy.
    import importlib

    _map = {
        "open_session": (".session", "open_session"),
        "load_session": (".session", "load_session"),
        "Session": (".session", "Session"),
        "Schema": (".schema", "Schema"),
        "register_extractor": (".extractors", "register_extractor"),
        "list_extractors": (".extractors", "list_extractors"),
        "run_extractor": (".extractors", "run_extractor"),
        "list_presets": (".schema", "list_presets"),
        "load_preset": (".schema", "load_preset"),
        "render": (".viz", "render"),
        "ProvenanceStamp": (".provenance", "ProvenanceStamp"),
        "attach_provenance": (".provenance", "attach_provenance"),
    }
    if name in _map:
        mod_name, attr = _map[name]
        mod = importlib.import_module(mod_name, package=__name__)
        return getattr(mod, attr)
    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")