mirror of
https://github.com/VectifyAI/PageIndex.git
synced 2026-06-12 19:55:17 +02:00
e9453725b6
Avoid eager optional dependency imports when importing PageIndexFileSystem or filesystem semantic exports.
22 lines
668 B
Python
22 lines
668 B
Python
import os
|
|
|
|
os.environ.setdefault("LITELLM_LOCAL_MODEL_COST_MAP", "true")
|
|
|
|
_OPTIONAL_CORE_IMPORTS = {"litellm", "openai", "PyPDF2", "pymupdf"}
|
|
|
|
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 not in _OPTIONAL_CORE_IMPORTS:
|
|
raise
|
|
|
|
|
|
def __getattr__(name: str):
|
|
if name == "PageIndexFileSystem":
|
|
from .filesystem import PageIndexFileSystem
|
|
|
|
return PageIndexFileSystem
|
|
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
|