prompture 0.0.29.dev8__py3-none-any.whl → 0.0.38.dev2__py3-none-any.whl

prompture/__init__.py

@@ -1,19 +1,161 @@
 """prompture - API package to convert LLM outputs into JSON + test harness."""
 
 from dotenv import load_dotenv
-from .core import ask_for_json, extract_and_jsonify, manual_extract_and_jsonify, Driver, clean_json_text, clean_json_text_with_ai, extract_with_model, stepwise_extract_with_model
+
+from .agent import Agent, AgentIterator, StreamedAgentResult
+from .agent_types import (
+    AgentCallbacks,
+    AgentResult,
+    AgentState,
+    AgentStep,
+    GuardrailError,
+    ModelRetry,
+    RunContext,
+    StepType,
+    StreamEvent,
+    StreamEventType,
+)
+from .async_agent import AsyncAgent, AsyncAgentIterator, AsyncStreamedAgentResult
+from .async_conversation import AsyncConversation
+from .async_driver import AsyncDriver
+from .async_groups import (
+    AsyncLoopGroup,
+    AsyncRouterAgent,
+    AsyncSequentialGroup,
+    ParallelGroup,
+)
+from .cache import (
+    CacheBackend,
+    MemoryCacheBackend,
+    RedisCacheBackend,
+    ResponseCache,
+    SQLiteCacheBackend,
+    configure_cache,
+    get_cache,
+)
+from .callbacks import DriverCallbacks
+from .conversation import Conversation
+from .core import (
+    Driver,
+    ask_for_json,
+    clean_json_text_with_ai,
+    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 .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 .group_types import (
+    AgentError,
+    ErrorPolicy,
+    GroupCallbacks,
+    GroupResult,
+    GroupStep,
+)
+from .groups import (
+    GroupAsAgent,
+    LoopGroup,
+    RouterAgent,
+    SequentialGroup,
+)
+from .image import (
+    ImageContent,
+    ImageInput,
+    image_from_base64,
+    image_from_bytes,
+    image_from_file,
+    image_from_url,
+    make_image,
+)
+from .logging import JSONFormatter, configure_logging
+from .model_rates import get_model_info, get_model_rates, refresh_rates_cache
+from .persistence import ConversationStore
+from .persona import (
+    PERSONAS,
+    Persona,
+    clear_persona_registry,
+    get_persona,
+    get_persona_names,
+    get_persona_registry_snapshot,
+    get_trait,
+    get_trait_names,
+    load_personas_from_directory,
+    register_persona,
+    register_trait,
+    reset_persona_registry,
+    reset_trait_registry,
 )
 from .runner import run_suite_from_spec
+from .serialization import (
+    EXPORT_VERSION,
+    export_conversation,
+    export_usage_session,
+    import_conversation,
+    import_usage_session,
+)
+from .session import UsageSession
+from .settings import settings as _settings
+from .tools import clean_json_text, clean_toon_text
+from .tools_schema import ToolDefinition, ToolRegistry, tool_from_function
 from .validator import validate_against_schema
 
 # 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+
@@ -29,30 +171,129 @@ except Exception:
     __version__ = "0.0.0"
 
 __all__ = [
-    "ask_for_json",
-    "extract_and_jsonify",
-    "manual_extract_and_jsonify",
-    "run_suite_from_spec",
-    "validate_against_schema",
+    "EXPORT_VERSION",
+    "FIELD_DEFINITIONS",
+    "PERSONAS",
+    "Agent",
+    "AgentCallbacks",
+    "AgentError",
+    "AgentIterator",
+    "AgentResult",
+    "AgentState",
+    "AgentStep",
+    "AirLLMDriver",
+    "AsyncAgent",
+    "AsyncAgentIterator",
+    "AsyncConversation",
+    "AsyncDriver",
+    "AsyncLoopGroup",
+    "AsyncRouterAgent",
+    "AsyncSequentialGroup",
+    "AsyncStreamedAgentResult",
+    "AzureDriver",
+    "CacheBackend",
+    "ClaudeDriver",
+    "Conversation",
+    "ConversationStore",
     "Driver",
+    "DriverCallbacks",
+    "ErrorPolicy",
+    "GoogleDriver",
+    "GrokDriver",
+    "GroqDriver",
+    "GroupAsAgent",
+    "GroupCallbacks",
+    "GroupResult",
+    "GroupStep",
+    "GuardrailError",
+    "ImageContent",
+    "ImageInput",
+    "JSONFormatter",
+    "LMStudioDriver",
+    "LocalHTTPDriver",
+    "LoopGroup",
+    "MemoryCacheBackend",
+    "ModelRetry",
+    "OllamaDriver",
+    "OpenAIDriver",
+    "OpenRouterDriver",
+    "ParallelGroup",
+    "Persona",
+    "RedisCacheBackend",
+    "ResponseCache",
+    "RouterAgent",
+    "RunContext",
+    "SQLiteCacheBackend",
+    "SequentialGroup",
+    "StepType",
+    "StreamEvent",
+    "StreamEventType",
+    "StreamedAgentResult",
+    "ToolDefinition",
+    "ToolRegistry",
+    "UsageSession",
+    "add_field_definition",
+    "add_field_definitions",
+    "ask_for_json",
     "clean_json_text",
     "clean_json_text_with_ai",
+    "clean_toon_text",
+    "clear_persona_registry",
+    "clear_registry",
+    "configure_cache",
+    "configure_logging",
+    "export_conversation",
+    "export_usage_session",
+    "extract_and_jsonify",
+    "extract_from_data",
+    "extract_from_pandas",
     "extract_with_model",
-    "stepwise_extract_with_model",
-    # Field Definitions
-    "FIELD_DEFINITIONS",
+    "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_persona",
+    "get_persona_names",
+    "get_persona_registry_snapshot",
     "get_registry_snapshot",
-    "clear_registry",
+    "get_required_fields",
+    "get_trait",
+    "get_trait_names",
+    "image_from_base64",
+    "image_from_bytes",
+    "image_from_file",
+    "image_from_url",
+    "import_conversation",
+    "import_usage_session",
+    "is_async_driver_registered",
+    "is_driver_registered",
+    "list_registered_async_drivers",
+    "list_registered_drivers",
+    "load_entry_point_drivers",
+    "load_personas_from_directory",
+    "make_image",
+    "manual_extract_and_jsonify",
+    "normalize_enum_value",
+    "refresh_rates_cache",
+    "register_async_driver",
+    "register_driver",
+    "register_field",
+    "register_persona",
+    "register_trait",
+    "render_output",
+    "reset_persona_registry",
     "reset_registry",
-    # Enum Field Support
+    "reset_trait_registry",
+    "run_suite_from_spec",
+    "stepwise_extract_with_model",
+    "tool_from_function",
+    "unregister_async_driver",
+    "unregister_driver",
+    "validate_against_schema",
     "validate_enum_value",
-    "normalize_enum_value",
-]
+]

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.38.dev2'
+__version_tuple__ = version_tuple = (0, 0, 38, 'dev2')
+
+__commit_id__ = commit_id = None