prompture 0.0.33.dev1__py3-none-any.whl → 0.0.34__py3-none-any.whl

prompture/__init__.py

@@ -1,33 +1,95 @@
 """prompture - API package to convert LLM outputs into JSON + test harness."""
 
 from dotenv import load_dotenv
+
+from .async_conversation import AsyncConversation
+from .async_driver import AsyncDriver
+from .cache import (
+    CacheBackend,
+    MemoryCacheBackend,
+    RedisCacheBackend,
+    ResponseCache,
+    SQLiteCacheBackend,
+    configure_cache,
+    get_cache,
+)
+from .callbacks import DriverCallbacks
+from .conversation import Conversation
 from .core import (
-    ask_for_json,
-    extract_and_jsonify,
-    manual_extract_and_jsonify,
     Driver,
+    ask_for_json,
     clean_json_text_with_ai,
-    extract_with_model,
-    stepwise_extract_with_model,
+    extract_and_jsonify,
     extract_from_data,
     extract_from_pandas,
+    extract_with_model,
+    manual_extract_and_jsonify,
     render_output,
+    stepwise_extract_with_model,
+)
+from .discovery import get_available_models
+from .drivers import (
+    AirLLMDriver,
+    AzureDriver,
+    ClaudeDriver,
+    GoogleDriver,
+    GrokDriver,
+    GroqDriver,
+    LMStudioDriver,
+    LocalHTTPDriver,
+    OllamaDriver,
+    OpenAIDriver,
+    OpenRouterDriver,
+    get_driver,
+    get_driver_for_model,
+    # Plugin registration API
+    is_async_driver_registered,
+    is_driver_registered,
+    list_registered_async_drivers,
+    list_registered_drivers,
+    load_entry_point_drivers,
+    register_async_driver,
+    register_driver,
+    unregister_async_driver,
+    unregister_driver,
 )
-from .drivers import get_driver, get_driver_for_model, OpenAIDriver, LocalHTTPDriver, OllamaDriver, ClaudeDriver, LMStudioDriver, AzureDriver, GoogleDriver, GroqDriver, OpenRouterDriver, GrokDriver, AirLLMDriver
-from .tools import clean_json_text, clean_toon_text
 from .field_definitions import (
-    FIELD_DEFINITIONS, get_field_definition, get_required_fields, get_field_names,
-    field_from_registry, register_field, add_field_definition, add_field_definitions,
-    get_registry_snapshot, clear_registry, reset_registry, validate_enum_value,
-    normalize_enum_value
+    FIELD_DEFINITIONS,
+    add_field_definition,
+    add_field_definitions,
+    clear_registry,
+    field_from_registry,
+    get_field_definition,
+    get_field_names,
+    get_registry_snapshot,
+    get_required_fields,
+    normalize_enum_value,
+    register_field,
+    reset_registry,
+    validate_enum_value,
 )
+from .logging import JSONFormatter, configure_logging
+from .model_rates import get_model_info, get_model_rates, refresh_rates_cache
 from .runner import run_suite_from_spec
+from .session import UsageSession
+from .settings import settings as _settings
+from .tools import clean_json_text, clean_toon_text
 from .validator import validate_against_schema
-from .discovery import get_available_models
 
 # Load environment variables from .env file
 load_dotenv()
 
+# Auto-configure cache from settings if enabled
+if _settings.cache_enabled:
+    configure_cache(
+        backend=_settings.cache_backend,
+        enabled=True,
+        ttl=_settings.cache_ttl_seconds,
+        maxsize=_settings.cache_memory_maxsize,
+        db_path=_settings.cache_sqlite_path,
+        redis_url=_settings.cache_redis_url,
+    )
+
 # runtime package version (from installed metadata)
 try:
     # Python 3.8+
@@ -43,51 +105,73 @@ except Exception:
     __version__ = "0.0.0"
 
 __all__ = [
-    "ask_for_json",
-    "extract_and_jsonify",
-    "manual_extract_and_jsonify",
-    "run_suite_from_spec",
-    "validate_against_schema",
+    "FIELD_DEFINITIONS",
+    "AirLLMDriver",
+    "AsyncConversation",
+    "AsyncDriver",
+    "AzureDriver",
+    "CacheBackend",
+    "ClaudeDriver",
+    "Conversation",
     "Driver",
+    "DriverCallbacks",
+    "GoogleDriver",
+    "GrokDriver",
+    "GroqDriver",
+    "JSONFormatter",
+    "LMStudioDriver",
+    "LocalHTTPDriver",
+    "MemoryCacheBackend",
+    "OllamaDriver",
+    "OpenAIDriver",
+    "OpenRouterDriver",
+    "RedisCacheBackend",
+    "ResponseCache",
+    "SQLiteCacheBackend",
+    "UsageSession",
+    "add_field_definition",
+    "add_field_definitions",
+    "ask_for_json",
     "clean_json_text",
-    "clean_toon_text",
     "clean_json_text_with_ai",
-    "extract_with_model",
-    "stepwise_extract_with_model",
-    # TOON Data Extraction Functions
+    "clean_toon_text",
+    "clear_registry",
+    "configure_cache",
+    "configure_logging",
+    "extract_and_jsonify",
     "extract_from_data",
     "extract_from_pandas",
-    "render_output",
-    # Field Definitions
-    "FIELD_DEFINITIONS",
+    "extract_with_model",
+    "field_from_registry",
+    "get_available_models",
+    "get_cache",
+    "get_driver",
+    "get_driver_for_model",
     "get_field_definition",
-    "get_required_fields",
     "get_field_names",
-    # New Field Registry API
-    "field_from_registry",
-    "register_field",
-    "add_field_definition",
-    "add_field_definitions",
+    "get_model_info",
+    "get_model_rates",
     "get_registry_snapshot",
-    "clear_registry",
+    "get_required_fields",
+    # Plugin registration API
+    "is_async_driver_registered",
+    "is_driver_registered",
+    "list_registered_async_drivers",
+    "list_registered_drivers",
+    "load_entry_point_drivers",
+    # Other exports
+    "manual_extract_and_jsonify",
+    "normalize_enum_value",
+    "refresh_rates_cache",
+    "register_async_driver",
+    "register_driver",
+    "register_field",
+    "render_output",
     "reset_registry",
-    # Enum Field Support
+    "run_suite_from_spec",
+    "stepwise_extract_with_model",
+    "unregister_async_driver",
+    "unregister_driver",
+    "validate_against_schema",
     "validate_enum_value",
-    "normalize_enum_value",
-    # Drivers
-    "get_driver",
-    "get_driver_for_model",
-    "OpenAIDriver",
-    "LocalHTTPDriver",
-    "OllamaDriver",
-    "ClaudeDriver",
-    "LMStudioDriver",
-    "AzureDriver",
-    "GoogleDriver",
-    "GroqDriver",
-    "OpenRouterDriver",
-    "GrokDriver",
-    "AirLLMDriver",
-    # Discovery
-    "get_available_models",
 ]

prompture/_version.py

@@ -0,0 +1,34 @@
+# file generated by setuptools-scm
+# don't change, don't track in version control
+
+__all__ = [
+    "__version__",
+    "__version_tuple__",
+    "version",
+    "version_tuple",
+    "__commit_id__",
+    "commit_id",
+]
+
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from typing import Tuple
+    from typing import Union
+
+    VERSION_TUPLE = Tuple[Union[int, str], ...]
+    COMMIT_ID = Union[str, None]
+else:
+    VERSION_TUPLE = object
+    COMMIT_ID = object
+
+version: str
+__version__: str
+__version_tuple__: VERSION_TUPLE
+version_tuple: VERSION_TUPLE
+commit_id: COMMIT_ID
+__commit_id__: COMMIT_ID
+
+__version__ = version = '0.0.34'
+__version_tuple__ = version_tuple = (0, 0, 34)
+
+__commit_id__ = commit_id = None

prompture/aio/__init__.py

@@ -0,0 +1,74 @@
+"""prompture.aio — Async public API for Prompture.
+
+Usage::
+
+    from prompture.aio import extract_with_model, AsyncDriver
+
+All functions mirror their synchronous counterparts in ``prompture.core``
+but are ``async def`` and must be ``await``-ed.
+"""
+
+from ..async_conversation import AsyncConversation
+from ..async_core import (
+    ask_for_json,
+    clean_json_text_with_ai,
+    extract_and_jsonify,
+    extract_from_data,
+    extract_from_pandas,
+    extract_with_model,
+    gather_extract,
+    manual_extract_and_jsonify,
+    render_output,
+    stepwise_extract_with_model,
+)
+from ..async_driver import AsyncDriver
+from ..drivers import (
+    AsyncAirLLMDriver,
+    AsyncAzureDriver,
+    AsyncClaudeDriver,
+    AsyncGoogleDriver,
+    AsyncGrokDriver,
+    AsyncGroqDriver,
+    AsyncHuggingFaceDriver,
+    AsyncLMStudioDriver,
+    AsyncLocalHTTPDriver,
+    AsyncOllamaDriver,
+    AsyncOpenAIDriver,
+    AsyncOpenRouterDriver,
+    get_async_driver,
+    get_async_driver_for_model,
+)
+
+__all__ = [
+    # Async driver classes
+    "AsyncAirLLMDriver",
+    "AsyncAzureDriver",
+    "AsyncClaudeDriver",
+    # Async conversation
+    "AsyncConversation",
+    # Async base class
+    "AsyncDriver",
+    "AsyncGoogleDriver",
+    "AsyncGrokDriver",
+    "AsyncGroqDriver",
+    "AsyncHuggingFaceDriver",
+    "AsyncLMStudioDriver",
+    "AsyncLocalHTTPDriver",
+    "AsyncOllamaDriver",
+    "AsyncOpenAIDriver",
+    "AsyncOpenRouterDriver",
+    # Async core functions
+    "ask_for_json",
+    "clean_json_text_with_ai",
+    "extract_and_jsonify",
+    "extract_from_data",
+    "extract_from_pandas",
+    "extract_with_model",
+    "gather_extract",
+    # Async driver factories
+    "get_async_driver",
+    "get_async_driver_for_model",
+    "manual_extract_and_jsonify",
+    "render_output",
+    "stepwise_extract_with_model",
+]