feat: export OCR public API surface
This commit is contained in:
@@ -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",
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user