PageIndex/pageindex/__init__.py

14 lines
396 B
Python

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