promptev-context-engine 0.0.1__py3-none-any.whl

context_engine/__init__.py

@@ -0,0 +1,147 @@
+from context_engine.actions import compute, get_document_text, list_documents, query_structured
+from context_engine.config import (
+    ContextEngineConfig,
+    EmbeddingConfig,
+    ExtractionConfig,
+    FusionConfig,
+    GraphConfig,
+    LLMConfig,
+    RerankerConfig,
+    StorageConfig,
+)
+from context_engine.crypto import decrypt_dict, encrypt_dict, get_secret_key
+from context_engine.engine import TRUSTED, UNSET, ContextEngine
+from context_engine.errors import EngineActionError
+from context_engine.extraction import Extracted, extract
+from context_engine.fusion import rrf_fuse
+from context_engine.hooks import Hooks, emit_error, emit_progress, emit_usage
+from context_engine.mcp_server import create_mcp_app
+from context_engine.providers import (
+    Embedder,
+    LLMClient,
+    build_embedder,
+    build_llm_client,
+    call_llm,
+    rerank,
+)
+from context_engine.redaction import RedactionPolicy, RedactionRule, apply_redaction
+from context_engine.search import GraphLegUnavailable, Hit, SearchResult, run_search
+from context_engine.storage import ChunkRow, PostgresBackend, SearchScope, StorageBackend
+from context_engine.structured import (
+    ExtractionResult,
+    extract_structured_data,
+    resolve_fields,
+    upsert_registry,
+)
+from context_engine.tools.approval import (
+    ApprovalExpired,
+    ApprovalNotPending,
+    ApprovalRecord,
+    resolve_approval,
+)
+from context_engine.tools.config import ToolConfig, ToolKind, config_schema
+from context_engine.usage import (
+    DocumentReport,
+    IngestReport,
+    UsageEvent,
+    graph_units,
+    units_for_file,
+)
+from context_engine.version import __version__
+
+__all__ = [
+    "TRUSTED",
+    "UNSET",
+    "ApprovalExpired",
+    "ApprovalNotPending",
+    "ApprovalRecord",
+    "ChunkRow",
+    "ContextEngine",
+    "ContextEngineConfig",
+    "DocumentReport",
+    "Embedder",
+    "EmbeddingConfig",
+    "EngineActionError",
+    "Extracted",
+    "ExtractionConfig",
+    "ExtractionResult",
+    "FusionConfig",
+    "GraphConfig",
+    "GraphLegUnavailable",
+    "Hit",
+    "Hooks",
+    "IngestReport",
+    "LLMClient",
+    "LLMConfig",
+    "PostgresBackend",
+    "RedactionPolicy",
+    "RedactionRule",
+    "RerankerConfig",
+    "SearchResult",
+    "SearchScope",
+    "StorageBackend",
+    "StorageConfig",
+    "ToolConfig",
+    "ToolKind",
+    "UsageEvent",
+    "__version__",
+    "apply_redaction",
+    "build_embedder",
+    "build_llm_client",
+    "call_llm",
+    "compute",
+    "config_schema",
+    "create_django_urlpatterns",
+    "create_flask_blueprint",
+    "create_mcp_app",
+    "create_router",
+    "create_tools_router",
+    "decrypt_dict",
+    "emit_error",
+    "emit_progress",
+    "emit_usage",
+    "encrypt_dict",
+    "extract",
+    "extract_structured_data",
+    "get_document_text",
+    "get_secret_key",
+    "graph_units",
+    "list_documents",
+    "presidio_detector",
+    "presidio_detectors",
+    "query_structured",
+    "rerank",
+    "resolve_approval",
+    "resolve_fields",
+    "rrf_fuse",
+    "run_search",
+    "units_for_file",
+    "upsert_registry",
+]
+
+# The web-framework router factories are loaded lazily so that `import
+# context_engine` — and `context-engine migrate`, and any Flask/Django/CLI
+# consumer of the core engine — never imports FastAPI/Flask/Django. Each is
+# pulled in only when the attribute is actually accessed; the underlying
+# module's `_require_*()` guard raises an actionable ImportError if the
+# matching optional extra isn't installed.
+_LAZY_FACTORIES = {
+    "create_router": ("context_engine.router", "create_router"),
+    "create_tools_router": ("context_engine.tools.router", "create_tools_router"),
+    "create_flask_blueprint": ("context_engine.flask_router", "create_flask_blueprint"),
+    "create_django_urlpatterns": ("context_engine.django_router", "create_django_urlpatterns"),
+    # Presidio adapter — lazy for the same reason: importing it must not pull
+    # `presidio_analyzer` (or its spaCy model) into a process that never uses it.
+    "presidio_detector": ("context_engine.redaction_presidio", "presidio_detector"),
+    "presidio_detectors": ("context_engine.redaction_presidio", "presidio_detectors"),
+}
+
+
+def __getattr__(name: str):
+    target = _LAZY_FACTORIES.get(name)
+    if target is None:
+        raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+    import importlib
+
+    module, attr = target
+    return getattr(importlib.import_module(module), attr)