prompture 0.0.41.dev1__tar.gz → 0.0.42__tar.gz

{prompture-0.0.41.dev1 → prompture-0.0.42}/.env.copy

@@ -53,7 +53,7 @@ GROQ_MODEL=llama2-70b-4096
 # OpenRouter Configuration
 # Required if AI_PROVIDER=openrouter
 OPENROUTER_API_KEY=your-api-key-here
-OPENROUTER_MODEL=openai/gpt-3.5-turbo
+OPENROUTER_MODEL=openai/gpt-4o-mini
 
 # Grok Configuration
 # Required if AI_PROVIDER=grok

{prompture-0.0.41.dev1/prompture.egg-info → prompture-0.0.42}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prompture
-Version: 0.0.41.dev1
+Version: 0.0.42
 Summary: Ask LLMs to return structured JSON and run cross-model tests. API-first.
 Author-email: Juan Denis <juan@vene.co>
 License-Expression: MIT

prompture-0.0.42/VERSION

@@ -0,0 +1 @@
+0.0.42

{prompture-0.0.41.dev1 → prompture-0.0.42}/prompture/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.0.41.dev1'
-__version_tuple__ = version_tuple = (0, 0, 41, 'dev1')
+__version__ = version = '0.0.42'
+__version_tuple__ = version_tuple = (0, 0, 42)
 
 __commit_id__ = commit_id = None

{prompture-0.0.41.dev1 → prompture-0.0.42}/prompture/agent.py

@@ -188,7 +188,7 @@ class Agent(Generic[DepsType]):
             for fn in tools:
                 self._tools.register(fn)
 
-        self._state = AgentState.idle
+        self._lifecycle = AgentState.idle
         self._stop_requested = False
 
     # ------------------------------------------------------------------
@@ -206,7 +206,7 @@ class Agent(Generic[DepsType]):
     @property
     def state(self) -> AgentState:
         """Current lifecycle state of the agent."""
-        return self._state
+        return self._lifecycle
 
     def stop(self) -> None:
         """Request graceful shutdown after the current iteration."""
@@ -265,16 +265,16 @@ class Agent(Generic[DepsType]):
             prompt: The user prompt to send.
             deps: Optional dependencies injected into :class:`RunContext`.
         """
-        self._state = AgentState.running
+        self._lifecycle = AgentState.running
         self._stop_requested = False
         steps: list[AgentStep] = []
 
         try:
             result = self._execute(prompt, steps, deps)
-            self._state = AgentState.idle
+            self._lifecycle = AgentState.idle
             return result
         except Exception:
-            self._state = AgentState.errored
+            self._lifecycle = AgentState.errored
             raise
 
     # ------------------------------------------------------------------
@@ -722,7 +722,7 @@ class Agent(Generic[DepsType]):
 
     def _execute_iter(self, prompt: str, deps: Any) -> Generator[AgentStep, None, AgentResult]:
         """Generator that executes the agent loop and yields each step."""
-        self._state = AgentState.running
+        self._lifecycle = AgentState.running
         self._stop_requested = False
         steps: list[AgentStep] = []
 
@@ -730,10 +730,10 @@ class Agent(Generic[DepsType]):
             result = self._execute(prompt, steps, deps)
             # Yield each step one at a time
             yield from result.steps
-            self._state = AgentState.idle
+            self._lifecycle = AgentState.idle
             return result
         except Exception:
-            self._state = AgentState.errored
+            self._lifecycle = AgentState.errored
             raise
 
     # ------------------------------------------------------------------
@@ -757,7 +757,7 @@ class Agent(Generic[DepsType]):
 
     def _execute_stream(self, prompt: str, deps: Any) -> Generator[StreamEvent, None, AgentResult]:
         """Generator that executes the agent loop and yields stream events."""
-        self._state = AgentState.running
+        self._lifecycle = AgentState.running
         self._stop_requested = False
         steps: list[AgentStep] = []
 
@@ -853,10 +853,10 @@ class Agent(Generic[DepsType]):
                 data=result,
             )
 
-            self._state = AgentState.idle
+            self._lifecycle = AgentState.idle
             return result
         except Exception:
-            self._state = AgentState.errored
+            self._lifecycle = AgentState.errored
             raise
 
 

{prompture-0.0.41.dev1 → prompture-0.0.42}/prompture/async_agent.py

@@ -182,7 +182,7 @@ class AsyncAgent(Generic[DepsType]):
             for fn in tools:
                 self._tools.register(fn)
 
-        self._state = AgentState.idle
+        self._lifecycle = AgentState.idle
         self._stop_requested = False
 
     # ------------------------------------------------------------------
@@ -197,7 +197,7 @@ class AsyncAgent(Generic[DepsType]):
     @property
     def state(self) -> AgentState:
         """Current lifecycle state of the agent."""
-        return self._state
+        return self._lifecycle
 
     def stop(self) -> None:
         """Request graceful shutdown after the current iteration."""
@@ -264,16 +264,16 @@ class AsyncAgent(Generic[DepsType]):
         Creates a fresh conversation, sends the prompt, handles tool calls,
         and optionally parses the final response into ``output_type``.
         """
-        self._state = AgentState.running
+        self._lifecycle = AgentState.running
         self._stop_requested = False
         steps: list[AgentStep] = []
 
         try:
             result = await self._execute(prompt, steps, deps)
-            self._state = AgentState.idle
+            self._lifecycle = AgentState.idle
             return result
         except Exception:
-            self._state = AgentState.errored
+            self._lifecycle = AgentState.errored
             raise
 
     async def iter(self, prompt: str, *, deps: Any = None) -> AsyncAgentIterator:
@@ -714,7 +714,7 @@ class AsyncAgent(Generic[DepsType]):
 
     async def _execute_iter(self, prompt: str, deps: Any) -> AsyncGenerator[AgentStep, None]:
         """Async generator that executes the agent loop and yields each step."""
-        self._state = AgentState.running
+        self._lifecycle = AgentState.running
         self._stop_requested = False
         steps: list[AgentStep] = []
 
@@ -722,11 +722,11 @@ class AsyncAgent(Generic[DepsType]):
             result = await self._execute(prompt, steps, deps)
             for step in result.steps:
                 yield step
-            self._state = AgentState.idle
+            self._lifecycle = AgentState.idle
             # Store result on the generator for retrieval
             self._last_iter_result = result
         except Exception:
-            self._state = AgentState.errored
+            self._lifecycle = AgentState.errored
             raise
 
     # ------------------------------------------------------------------
@@ -735,7 +735,7 @@ class AsyncAgent(Generic[DepsType]):
 
     async def _execute_stream(self, prompt: str, deps: Any) -> AsyncGenerator[StreamEvent, None]:
         """Async generator that executes the agent loop and yields stream events."""
-        self._state = AgentState.running
+        self._lifecycle = AgentState.running
         self._stop_requested = False
         steps: list[AgentStep] = []
 
@@ -803,10 +803,10 @@ class AsyncAgent(Generic[DepsType]):
 
             yield StreamEvent(event_type=StreamEventType.output, data=result)
 
-            self._state = AgentState.idle
+            self._lifecycle = AgentState.idle
             self._last_stream_result = result
         except Exception:
-            self._state = AgentState.errored
+            self._lifecycle = AgentState.errored
             raise
 
 

{prompture-0.0.41.dev1 → prompture-0.0.42}/prompture/async_groups.py

@@ -70,6 +70,27 @@ class ParallelGroup:
         """Request graceful shutdown."""
         self._stop_requested = True
 
+    @property
+    def shared_state(self) -> dict[str, Any]:
+        """Return a copy of the current shared execution state."""
+        return dict(self._state)
+
+    def inject_state(self, state: dict[str, Any], *, recursive: bool = False) -> None:
+        """Merge external key-value pairs into this group's shared state.
+
+        Existing keys are NOT overwritten (uses setdefault semantics).
+
+        Args:
+            state: Key-value pairs to inject.
+            recursive: If True, also inject into nested sub-groups.
+        """
+        for k, v in state.items():
+            self._state.setdefault(k, v)
+        if recursive:
+            for agent, _ in self._agents:
+                if hasattr(agent, "inject_state"):
+                    agent.inject_state(state, recursive=True)
+
     async def run_async(self, prompt: str = "") -> GroupResult:
         """Execute all agents concurrently."""
         self._stop_requested = False
@@ -213,6 +234,27 @@ class AsyncSequentialGroup:
     def stop(self) -> None:
         self._stop_requested = True
 
+    @property
+    def shared_state(self) -> dict[str, Any]:
+        """Return a copy of the current shared execution state."""
+        return dict(self._state)
+
+    def inject_state(self, state: dict[str, Any], *, recursive: bool = False) -> None:
+        """Merge external key-value pairs into this group's shared state.
+
+        Existing keys are NOT overwritten (uses setdefault semantics).
+
+        Args:
+            state: Key-value pairs to inject.
+            recursive: If True, also inject into nested sub-groups.
+        """
+        for k, v in state.items():
+            self._state.setdefault(k, v)
+        if recursive:
+            for agent, _ in self._agents:
+                if hasattr(agent, "inject_state"):
+                    agent.inject_state(state, recursive=True)
+
     async def run(self, prompt: str = "") -> GroupResult:
         """Execute all agents in sequence (async)."""
         self._stop_requested = False
@@ -351,6 +393,27 @@ class AsyncLoopGroup:
     def stop(self) -> None:
         self._stop_requested = True
 
+    @property
+    def shared_state(self) -> dict[str, Any]:
+        """Return a copy of the current shared execution state."""
+        return dict(self._state)
+
+    def inject_state(self, state: dict[str, Any], *, recursive: bool = False) -> None:
+        """Merge external key-value pairs into this group's shared state.
+
+        Existing keys are NOT overwritten (uses setdefault semantics).
+
+        Args:
+            state: Key-value pairs to inject.
+            recursive: If True, also inject into nested sub-groups.
+        """
+        for k, v in state.items():
+            self._state.setdefault(k, v)
+        if recursive:
+            for agent, _ in self._agents:
+                if hasattr(agent, "inject_state"):
+                    agent.inject_state(state, recursive=True)
+
     async def run(self, prompt: str = "") -> GroupResult:
         """Execute the loop (async)."""
         self._stop_requested = False

{prompture-0.0.41.dev1 → prompture-0.0.42}/prompture/cost_mixin.py

@@ -2,9 +2,34 @@
 
 from __future__ import annotations
 
+import copy
 from typing import Any
 
 
+def prepare_strict_schema(schema: dict[str, Any]) -> dict[str, Any]:
+    """Prepare a JSON schema for OpenAI strict structured-output mode.
+
+    OpenAI's ``strict: true`` requires every object to have
+    ``"additionalProperties": false`` and a ``"required"`` array listing
+    all property keys.  This function recursively patches a schema copy
+    so callers don't need to worry about these constraints.
+    """
+    schema = copy.deepcopy(schema)
+    _patch_strict(schema)
+    return schema
+
+
+def _patch_strict(node: dict[str, Any]) -> None:
+    """Recursively add strict-mode constraints to an object schema node."""
+    if node.get("type") == "object" and "properties" in node:
+        node.setdefault("additionalProperties", False)
+        node.setdefault("required", list(node["properties"].keys()))
+        for prop in node["properties"].values():
+            _patch_strict(prop)
+    elif node.get("type") == "array" and isinstance(node.get("items"), dict):
+        _patch_strict(node["items"])
+
+
 class CostMixin:
     """Mixin that provides ``_calculate_cost`` to sync and async drivers.
 

{prompture-0.0.41.dev1 → prompture-0.0.42}/prompture/drivers/async_azure_driver.py

@@ -11,7 +11,7 @@ except Exception:
     AsyncAzureOpenAI = None
 
 from ..async_driver import AsyncDriver
-from ..cost_mixin import CostMixin
+from ..cost_mixin import CostMixin, prepare_strict_schema
 from .azure_driver import AzureDriver
 
 
@@ -89,12 +89,13 @@ class AsyncAzureDriver(CostMixin, AsyncDriver):
         if options.get("json_mode"):
             json_schema = options.get("json_schema")
             if json_schema:
+                schema_copy = prepare_strict_schema(json_schema)
                 kwargs["response_format"] = {
                     "type": "json_schema",
                     "json_schema": {
                         "name": "extraction",
                         "strict": True,
-                        "schema": json_schema,
+                        "schema": schema_copy,
                     },
                 }
             else:

{prompture-0.0.41.dev1 → prompture-0.0.42}/prompture/drivers/async_moonshot_driver.py

@@ -17,7 +17,7 @@ from typing import Any
 import httpx
 
 from ..async_driver import AsyncDriver
-from ..cost_mixin import CostMixin
+from ..cost_mixin import CostMixin, prepare_strict_schema
 from .moonshot_driver import MoonshotDriver
 
 
@@ -88,12 +88,13 @@ class AsyncMoonshotDriver(CostMixin, AsyncDriver):
         if options.get("json_mode"):
             json_schema = options.get("json_schema")
             if json_schema:
+                schema_copy = prepare_strict_schema(json_schema)
                 data["response_format"] = {
                     "type": "json_schema",
                     "json_schema": {
                         "name": "extraction",
                         "strict": True,
-                        "schema": json_schema,
+                        "schema": schema_copy,
                     },
                 }
             else:

{prompture-0.0.41.dev1 → prompture-0.0.42}/prompture/drivers/async_openai_driver.py

@@ -13,7 +13,7 @@ except Exception:
     AsyncOpenAI = None
 
 from ..async_driver import AsyncDriver
-from ..cost_mixin import CostMixin
+from ..cost_mixin import CostMixin, prepare_strict_schema
 from .openai_driver import OpenAIDriver
 
 
@@ -80,12 +80,13 @@ class AsyncOpenAIDriver(CostMixin, AsyncDriver):
         if options.get("json_mode"):
             json_schema = options.get("json_schema")
             if json_schema:
+                schema_copy = prepare_strict_schema(json_schema)
                 kwargs["response_format"] = {
                     "type": "json_schema",
                     "json_schema": {
                         "name": "extraction",
                         "strict": True,
-                        "schema": json_schema,
+                        "schema": schema_copy,
                     },
                 }
             else:

{prompture-0.0.41.dev1 → prompture-0.0.42}/prompture/drivers/async_openrouter_driver.py

@@ -10,7 +10,7 @@ from typing import Any
 import httpx
 
 from ..async_driver import AsyncDriver
-from ..cost_mixin import CostMixin
+from ..cost_mixin import CostMixin, prepare_strict_schema
 from .openrouter_driver import OpenRouterDriver
 
 
@@ -78,12 +78,13 @@ class AsyncOpenRouterDriver(CostMixin, AsyncDriver):
         if options.get("json_mode"):
             json_schema = options.get("json_schema")
             if json_schema:
+                schema_copy = prepare_strict_schema(json_schema)
                 data["response_format"] = {
                     "type": "json_schema",
                     "json_schema": {
                         "name": "extraction",
                         "strict": True,
-                        "schema": json_schema,
+                        "schema": schema_copy,
                     },
                 }
             else:

{prompture-0.0.41.dev1 → prompture-0.0.42}/prompture/drivers/async_zai_driver.py

@@ -13,7 +13,7 @@ from typing import Any
 import httpx
 
 from ..async_driver import AsyncDriver
-from ..cost_mixin import CostMixin
+from ..cost_mixin import CostMixin, prepare_strict_schema
 from .zai_driver import ZaiDriver
 
 
@@ -83,12 +83,13 @@ class AsyncZaiDriver(CostMixin, AsyncDriver):
         if options.get("json_mode"):
             json_schema = options.get("json_schema")
             if json_schema:
+                schema_copy = prepare_strict_schema(json_schema)
                 data["response_format"] = {
                     "type": "json_schema",
                     "json_schema": {
                         "name": "extraction",
                         "strict": True,
-                        "schema": json_schema,
+                        "schema": schema_copy,
                     },
                 }
             else:

{prompture-0.0.41.dev1 → prompture-0.0.42}/prompture/drivers/azure_driver.py

@@ -10,7 +10,7 @@ try:
 except Exception:
     AzureOpenAI = None
 
-from ..cost_mixin import CostMixin
+from ..cost_mixin import CostMixin, prepare_strict_schema
 from ..driver import Driver
 
 
@@ -128,12 +128,13 @@ class AzureDriver(CostMixin, Driver):
         if options.get("json_mode"):
             json_schema = options.get("json_schema")
             if json_schema:
+                schema_copy = prepare_strict_schema(json_schema)
                 kwargs["response_format"] = {
                     "type": "json_schema",
                     "json_schema": {
                         "name": "extraction",
                         "strict": True,
-                        "schema": json_schema,
+                        "schema": schema_copy,
                     },
                 }
             else:

{prompture-0.0.41.dev1 → prompture-0.0.42}/prompture/drivers/moonshot_driver.py

@@ -16,7 +16,7 @@ from typing import Any
 
 import requests
 
-from ..cost_mixin import CostMixin
+from ..cost_mixin import CostMixin, prepare_strict_schema
 from ..driver import Driver
 
 
@@ -116,12 +116,13 @@ class MoonshotDriver(CostMixin, Driver):
         if options.get("json_mode"):
             json_schema = options.get("json_schema")
             if json_schema:
+                schema_copy = prepare_strict_schema(json_schema)
                 data["response_format"] = {
                     "type": "json_schema",
                     "json_schema": {
                         "name": "extraction",
                         "strict": True,
-                        "schema": json_schema,
+                        "schema": schema_copy,
                     },
                 }
             else:

{prompture-0.0.41.dev1 → prompture-0.0.42}/prompture/drivers/openai_driver.py

@@ -12,7 +12,7 @@ try:
 except Exception:
     OpenAI = None
 
-from ..cost_mixin import CostMixin
+from ..cost_mixin import CostMixin, prepare_strict_schema
 from ..driver import Driver
 
 
@@ -125,12 +125,13 @@ class OpenAIDriver(CostMixin, Driver):
         if options.get("json_mode"):
             json_schema = options.get("json_schema")
             if json_schema:
+                schema_copy = prepare_strict_schema(json_schema)
                 kwargs["response_format"] = {
                     "type": "json_schema",
                     "json_schema": {
                         "name": "extraction",
                         "strict": True,
-                        "schema": json_schema,
+                        "schema": schema_copy,
                     },
                 }
             else:

{prompture-0.0.41.dev1 → prompture-0.0.42}/prompture/drivers/openrouter_driver.py

@@ -2,6 +2,7 @@
 Requires the `requests` package. Uses OPENROUTER_API_KEY env var.
 """
 
+import contextlib
 import json
 import os
 from collections.abc import Iterator
@@ -9,7 +10,7 @@ from typing import Any
 
 import requests
 
-from ..cost_mixin import CostMixin
+from ..cost_mixin import CostMixin, prepare_strict_schema
 from ..driver import Driver
 
 
@@ -128,12 +129,13 @@ class OpenRouterDriver(CostMixin, Driver):
         if options.get("json_mode"):
             json_schema = options.get("json_schema")
             if json_schema:
+                schema_copy = prepare_strict_schema(json_schema)
                 data["response_format"] = {
                     "type": "json_schema",
                     "json_schema": {
                         "name": "extraction",
                         "strict": True,
-                        "schema": json_schema,
+                        "schema": schema_copy,
                     },
                 }
             else:
@@ -149,7 +151,13 @@ class OpenRouterDriver(CostMixin, Driver):
             response.raise_for_status()
             resp = response.json()
         except requests.exceptions.HTTPError as e:
+            body = ""
+            if e.response is not None:
+                with contextlib.suppress(Exception):
+                    body = e.response.text
             error_msg = f"OpenRouter API request failed: {e!s}"
+            if body:
+                error_msg += f"\nResponse: {body}"
             raise RuntimeError(error_msg) from e
         except requests.exceptions.RequestException as e:
             raise RuntimeError(f"OpenRouter API request failed: {e!s}") from e

{prompture-0.0.41.dev1 → prompture-0.0.42}/prompture/drivers/zai_driver.py

@@ -12,7 +12,7 @@ from typing import Any
 
 import requests
 
-from ..cost_mixin import CostMixin
+from ..cost_mixin import CostMixin, prepare_strict_schema
 from ..driver import Driver
 
 
@@ -96,12 +96,13 @@ class ZaiDriver(CostMixin, Driver):
         if options.get("json_mode"):
             json_schema = options.get("json_schema")
             if json_schema:
+                schema_copy = prepare_strict_schema(json_schema)
                 data["response_format"] = {
                     "type": "json_schema",
                     "json_schema": {
                         "name": "extraction",
                         "strict": True,
-                        "schema": json_schema,
+                        "schema": schema_copy,
                     },
                 }
             else:

{prompture-0.0.41.dev1 → prompture-0.0.42}/prompture/groups.py

@@ -114,6 +114,27 @@ class SequentialGroup:
         """Request graceful shutdown after the current agent finishes."""
         self._stop_requested = True
 
+    @property
+    def shared_state(self) -> dict[str, Any]:
+        """Return a copy of the current shared execution state."""
+        return dict(self._state)
+
+    def inject_state(self, state: dict[str, Any], *, recursive: bool = False) -> None:
+        """Merge external key-value pairs into this group's shared state.
+
+        Existing keys are NOT overwritten (uses setdefault semantics).
+
+        Args:
+            state: Key-value pairs to inject.
+            recursive: If True, also inject into nested sub-groups.
+        """
+        for k, v in state.items():
+            self._state.setdefault(k, v)
+        if recursive:
+            for agent, _ in self._agents:
+                if hasattr(agent, "inject_state"):
+                    agent.inject_state(state, recursive=True)
+
     def save(self, path: str) -> None:
         """Run and save result to file. Convenience wrapper."""
         result = self.run()
@@ -267,6 +288,27 @@ class LoopGroup:
         """Request graceful shutdown."""
         self._stop_requested = True
 
+    @property
+    def shared_state(self) -> dict[str, Any]:
+        """Return a copy of the current shared execution state."""
+        return dict(self._state)
+
+    def inject_state(self, state: dict[str, Any], *, recursive: bool = False) -> None:
+        """Merge external key-value pairs into this group's shared state.
+
+        Existing keys are NOT overwritten (uses setdefault semantics).
+
+        Args:
+            state: Key-value pairs to inject.
+            recursive: If True, also inject into nested sub-groups.
+        """
+        for k, v in state.items():
+            self._state.setdefault(k, v)
+        if recursive:
+            for agent, _ in self._agents:
+                if hasattr(agent, "inject_state"):
+                    agent.inject_state(state, recursive=True)
+
     def run(self, prompt: str = "") -> GroupResult:
         """Execute the loop."""
         self._stop_requested = False

{prompture-0.0.41.dev1 → prompture-0.0.42}/prompture/settings.py

@@ -45,7 +45,7 @@ class Settings(BaseSettings):
 
     # OpenRouter
     openrouter_api_key: Optional[str] = None
-    openrouter_model: str = "openai/gpt-3.5-turbo"
+    openrouter_model: str = "openai/gpt-4o-mini"
 
     # Grok
     grok_api_key: Optional[str] = None

{prompture-0.0.41.dev1 → prompture-0.0.42/prompture.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prompture
-Version: 0.0.41.dev1
+Version: 0.0.42
 Summary: Ask LLMs to return structured JSON and run cross-model tests. API-first.
 Author-email: Juan Denis <juan@vene.co>
 License-Expression: MIT

{prompture-0.0.41.dev1 → prompture-0.0.42}/prompture.egg-info/SOURCES.txt

@@ -4,6 +4,7 @@ LICENSE
 MANIFEST.in
 README.md
 ROADMAP.md
+VERSION
 pyproject.toml
 requirements.txt
 test.py