prompture 0.0.35__py3-none-any.whl → 0.0.40.dev1__py3-none-any.whl

prompture/__init__.py

@@ -2,8 +2,28 @@
 
 from dotenv import load_dotenv
 
+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,
@@ -68,9 +88,61 @@ from .field_definitions import (
     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 .ledger import ModelUsageLedger, get_recently_used_models
 from .logging import JSONFormatter, configure_logging
-from .model_rates import get_model_info, get_model_rates, refresh_rates_cache
+from .model_rates import (
+    ModelCapabilities,
+    get_model_capabilities,
+    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
@@ -106,29 +178,66 @@ except Exception:
     __version__ = "0.0.0"
 
 __all__ = [
+    "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",
+    "ModelCapabilities",
+    "ModelRetry",
+    "ModelUsageLedger",
     "OllamaDriver",
     "OpenAIDriver",
     "OpenRouterDriver",
+    "ParallelGroup",
+    "Persona",
     "RedisCacheBackend",
     "ResponseCache",
+    "RouterAgent",
+    "RunContext",
     "SQLiteCacheBackend",
+    "SequentialGroup",
+    "StepType",
+    "StreamEvent",
+    "StreamEventType",
+    "StreamedAgentResult",
     "ToolDefinition",
     "ToolRegistry",
     "UsageSession",
@@ -138,9 +247,12 @@ __all__ = [
     "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",
@@ -152,25 +264,42 @@ __all__ = [
     "get_driver_for_model",
     "get_field_definition",
     "get_field_names",
+    "get_model_capabilities",
     "get_model_info",
     "get_model_rates",
+    "get_persona",
+    "get_persona_names",
+    "get_persona_registry_snapshot",
+    "get_recently_used_models",
     "get_registry_snapshot",
     "get_required_fields",
-    # Plugin registration API
+    "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",
-    # Other exports
+    "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",
+    "reset_trait_registry",
     "run_suite_from_spec",
     "stepwise_extract_with_model",
     "tool_from_function",

prompture/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.0.35'
-__version_tuple__ = version_tuple = (0, 0, 35)
+__version__ = version = '0.0.40.dev1'
+__version_tuple__ = version_tuple = (0, 0, 40, 'dev1')
 
 __commit_id__ = commit_id = None