PageIndex/pageindex/__init__.py

import os

os.environ.setdefault("LITELLM_LOCAL_MODEL_COST_MAP", "true")

try:
    from .page_index import *
    from .page_index_md import md_to_tree
    from .retrieve import get_document, get_document_structure, get_page_content
    from .client import PageIndexClient
except ModuleNotFoundError as exc:
    if exc.name != "litellm":
        raise

from .filesystem import PageIndexFileSystem