feat: export OCR public API surface

This commit is contained in:
2026-07-21 23:02:34 -04:00
parent 89e86448b0
commit 26e0e72926
2 changed files with 24 additions and 2 deletions
+15 -2
View File
@@ -6,7 +6,7 @@
""" """
from polygateway.client import GatewayClient, gather_bounded from polygateway.client import GatewayClient, gather_bounded
from polygateway.config import EmbeddingSettings, GatewaySettings from polygateway.config import EmbeddingSettings, GatewaySettings, OcrSettings
from polygateway.embedding import EmbeddingClient from polygateway.embedding import EmbeddingClient
from polygateway.errors import ( from polygateway.errors import (
AllSourcesExhausted, AllSourcesExhausted,
@@ -19,9 +19,17 @@ from polygateway.errors import (
SourceDeadError, SourceDeadError,
TransientError, TransientError,
) )
from polygateway.ocr import OcrClient
from polygateway.pricing import ModelPrice, PricingTable from polygateway.pricing import ModelPrice, PricingTable
from polygateway.providers import DEFAULT_PROFILES, ProviderProfile, register_provider from polygateway.providers import DEFAULT_PROFILES, ProviderProfile, register_provider
from polygateway.types import EmbeddingResponse, LLMResponse, SourceConfig from polygateway.types import (
EmbeddingResponse,
LLMResponse,
OcrLayoutElement,
OcrLayoutResult,
OcrTextResult,
SourceConfig,
)
__version__ = "0.1.0" __version__ = "0.1.0"
@@ -38,6 +46,11 @@ __all__ = [
"GovernanceBackendError", "GovernanceBackendError",
"LLMResponse", "LLMResponse",
"ModelPrice", "ModelPrice",
"OcrClient",
"OcrLayoutElement",
"OcrLayoutResult",
"OcrSettings",
"OcrTextResult",
"PolyGatewayError", "PolyGatewayError",
"PricingTable", "PricingTable",
"ProviderProfile", "ProviderProfile",
+9
View File
@@ -5,3 +5,12 @@ import polygateway
def test_package_importable_with_version() -> None: def test_package_importable_with_version() -> None:
assert polygateway.__version__ == "0.1.0" assert polygateway.__version__ == "0.1.0"
def test_ocr_public_surface_exported():
"""M3 OCR 公共 API 面(设计 §3;transport 结果与 Protocol 不出顶层)。"""
import polygateway
for name in ("OcrClient", "OcrSettings", "OcrTextResult", "OcrLayoutResult", "OcrLayoutElement"):
assert hasattr(polygateway, name), name
assert name in polygateway.__all__, name