weakincentives 0.2.0__py3-none-any.whl → 0.3.0__py3-none-any.whl

weakincentives/__init__.py

@@ -10,6 +10,30 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Top-level package for the weakincentives library."""
+"""Curated public surface for :mod:`weakincentives`."""
 
-__all__ = []
+from __future__ import annotations
+
+from .adapters import PromptResponse
+from .prompt import (
+    MarkdownSection,
+    Prompt,
+    SupportsDataclass,
+    Tool,
+    ToolResult,
+    parse_structured_output,
+)
+
+__all__ = [
+    "Prompt",
+    "MarkdownSection",
+    "Tool",
+    "ToolResult",
+    "PromptResponse",
+    "SupportsDataclass",
+    "parse_structured_output",
+]
+
+
+def __dir__() -> list[str]:  # pragma: no cover - convenience for REPL
+    return sorted({*globals().keys(), *__all__})

weakincentives/adapters/__init__.py

@@ -12,19 +12,20 @@
 
 """Integration adapters for optional third-party providers."""
 
+from __future__ import annotations
+
 from .core import (
     PromptEvaluationError,
     PromptResponse,
     ProviderAdapter,
-    ToolCallRecord,
 )
-from .openai import OpenAIAdapter, OpenAIProtocol
 
 __all__ = [
     "ProviderAdapter",
     "PromptResponse",
     "PromptEvaluationError",
-    "ToolCallRecord",
-    "OpenAIAdapter",
-    "OpenAIProtocol",
 ]
+
+
+def __dir__() -> list[str]:  # pragma: no cover - convenience for REPL
+    return sorted({*globals().keys(), *(__all__)})

weakincentives/adapters/core.py

@@ -17,9 +17,9 @@ from __future__ import annotations
 from dataclasses import dataclass
 from typing import Any, Protocol, TypeVar
 
-from ..prompts._types import SupportsDataclass
-from ..prompts.prompt import Prompt
-from ..prompts.tool import ToolResult
+from ..events import EventBus, ToolInvoked
+from ..prompt._types import SupportsDataclass
+from ..prompt.prompt import Prompt
 
 OutputT = TypeVar("OutputT")
 
@@ -32,22 +32,13 @@ class ProviderAdapter(Protocol[OutputT]):
         prompt: Prompt[OutputT],
         *params: SupportsDataclass,
         parse_output: bool = True,
+        bus: EventBus,
     ) -> PromptResponse[OutputT]:
         """Evaluate the prompt and return a structured response."""
 
         ...
 
 
-@dataclass(slots=True)
-class ToolCallRecord[ParamsT, ResultT]:
-    """Record describing a single tool invocation during evaluation."""
-
-    name: str
-    params: ParamsT
-    result: ToolResult[ResultT]
-    call_id: str | None = None
-
-
 @dataclass(slots=True)
 class PromptResponse[OutputT]:
     """Structured result emitted by an adapter evaluation."""
@@ -55,7 +46,7 @@ class PromptResponse[OutputT]:
     prompt_name: str
     text: str | None
     output: OutputT | None
-    tool_results: tuple[ToolCallRecord[Any, Any], ...]
+    tool_results: tuple[ToolInvoked, ...]
     provider_payload: dict[str, Any] | None = None
 
 
@@ -67,13 +58,13 @@ class PromptEvaluationError(RuntimeError):
         message: str,
         *,
         prompt_name: str,
-        stage: str,
+        phase: str,
         provider_payload: dict[str, Any] | None = None,
     ) -> None:
         super().__init__(message)
         self.message = message
         self.prompt_name = prompt_name
-        self.stage = stage
+        self.phase = phase
         self.provider_payload = provider_payload
 
 
@@ -81,5 +72,4 @@ __all__ = [
     "ProviderAdapter",
     "PromptEvaluationError",
     "PromptResponse",
-    "ToolCallRecord",
 ]