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

weakincentives/__init__.py

@@ -35,5 +35,5 @@ __all__ = [
 ]
 
 
-def __dir__() -> list[str]:  # pragma: no cover - convenience for REPL
+def __dir__() -> list[str]:
     return sorted({*globals().keys(), *__all__})

weakincentives/adapters/__init__.py

@@ -27,5 +27,6 @@ __all__ = [
 ]
 
 
-def __dir__() -> list[str]:  # pragma: no cover - convenience for REPL
-    return sorted({*globals().keys(), *(__all__)})
+def __dir__() -> list[str]:
+    extra_exports = {"LiteLLMAdapter", "OpenAIAdapter"}
+    return sorted({*globals().keys(), *(__all__), *extra_exports})

weakincentives/examples/code_review_prompt.py

@@ -19,7 +19,7 @@ from dataclasses import dataclass, field
 
 from ..prompt import MarkdownSection, Prompt
 from ..session import Session
-from ..tools import PlanningToolsSection, VfsToolsSection
+from ..tools import AstevalSection, PlanningToolsSection, VfsToolsSection
 from .code_review_tools import build_tools
 
 
@@ -72,10 +72,12 @@ def build_code_review_prompt(session: Session) -> Prompt[ReviewResponse]:
             - `vfs_list_directory` lists directories and files staged in the virtual
               filesystem snapshot.
             - `vfs_read_file` reads staged file contents.
-            - `vfs_write_file` stages ASCII edits before applying them to the host
+            - `vfs_write_file` stages UTF-8 edits before applying them to the host
               workspace.
             - `vfs_delete_entry` removes staged files or directories that are no
               longer needed.
+            - `evaluate_python` runs Python snippets inside a sandbox with
+              optional staged file reads and writes.
             If the task requires information beyond these capabilities, ask the
             user for clarification rather than guessing.
 
@@ -97,6 +99,7 @@ def build_code_review_prompt(session: Session) -> Prompt[ReviewResponse]:
     )
     planning_section = PlanningToolsSection(session=session)
     vfs_section = VfsToolsSection(session=session)
+    asteval_section = AstevalSection(session=session)
     user_turn_section = MarkdownSection[ReviewTurnParams](
         title="Review Request",
         template="${request}",
@@ -106,7 +109,13 @@ def build_code_review_prompt(session: Session) -> Prompt[ReviewResponse]:
         ns="examples/code-review",
         key="code-review-session",
         name="code_review_agent",
-        sections=[guidance_section, planning_section, vfs_section, user_turn_section],
+        sections=[
+            guidance_section,
+            planning_section,
+            vfs_section,
+            asteval_section,
+            user_turn_section,
+        ],
     )
 
 

weakincentives/examples/code_review_session.py

@@ -95,7 +95,7 @@ class CodeReviewSession:
                 indent=2,
             )
         if response.text:
-            return response.text
+            return response.text  # pragma: no cover - convenience path for plain text
         return "(no response from assistant)"
 
     def render_tool_history(self) -> str:
@@ -126,11 +126,15 @@ class CodeReviewSession:
             f"       → {event.result.message}"
         )
         if payload:
-            print(f"       payload: {payload}")
+            print(
+                f"       payload: {payload}"
+            )  # pragma: no cover - console output only
         latest = select_latest(self._session, ToolCallLog)
         if latest is not None:
             count = len(select_all(self._session, ToolCallLog))
-            print(f"       (session recorded this call as #{count})")
+            print(  # pragma: no cover - console output only
+                f"       (session recorded this call as #{count})"
+            )
 
     def _register_tool_history(self) -> None:
         for result_type in (

weakincentives/prompt/markdown.py

@@ -59,7 +59,7 @@ class MarkdownSection[ParamsT: SupportsDataclass](Section[ParamsT]):
         try:
             normalized_params = self._normalize_params(params)
             rendered_body = template.substitute(normalized_params)
-        except KeyError as error:  # pragma: no cover - handled at prompt level
+        except KeyError as error:
             missing = error.args[0]
             raise PromptRenderError(
                 "Missing placeholder during render.",

weakincentives/prompt/prompt.py

@@ -46,7 +46,7 @@ class RenderedPrompt[OutputT]:
         default=_EMPTY_TOOL_PARAM_DESCRIPTIONS
     )
 
-    def __str__(self) -> str:  # pragma: no cover - convenience for logging
+    def __str__(self) -> str:
         return self.text
 
     @property
@@ -71,11 +71,11 @@ def _clone_dataclass(instance: SupportsDataclass) -> SupportsDataclass:
 
 
 def _format_specialization_argument(argument: object | None) -> str:
-    if argument is None:  # pragma: no cover - defensive formatting
+    if argument is None:
         return "?"
     if isinstance(argument, type):
         return argument.__name__
-    return repr(argument)  # pragma: no cover - fallback for debugging
+    return repr(argument)
 
 
 @dataclass(frozen=True, slots=True)
@@ -314,7 +314,7 @@ class Prompt[OutputT]:
         if candidate is None or container is None:
             return None, None, None
 
-        if not isinstance(candidate, type):  # pragma: no cover - defensive guard
+        if not isinstance(candidate, type):
             candidate_type = cast(type[Any], type(candidate))
             raise PromptValidationError(
                 "Prompt output type must be a dataclass.",
@@ -333,7 +333,7 @@ class Prompt[OutputT]:
 
     def _build_response_format_params(self) -> ResponseFormatParams:
         container = self._output_container
-        if container is None:  # pragma: no cover - defensive guard
+        if container is None:
             raise RuntimeError(
                 "Output container missing during response format construction."
             )
@@ -458,7 +458,7 @@ class Prompt[OutputT]:
                 dataclass_type=params_type,
                 placeholder=error.placeholder,
             ) from error
-        except Exception as error:  # pragma: no cover - defensive guard
+        except Exception as error:
             raise PromptRenderError(
                 "Section rendering failed.",
                 section_path=node.path,
@@ -519,7 +519,7 @@ class Prompt[OutputT]:
             else:
                 try:
                     enabled = node.section.is_enabled(section_params)
-                except Exception as error:  # pragma: no cover - defensive guard
+                except Exception as error:
                     raise PromptRenderError(
                         "Section enabled predicate failed.",
                         section_path=node.path,

weakincentives/prompt/structured_output.py

@@ -110,7 +110,7 @@ def parse_structured_output[PayloadT](
             parsed_items.append(parsed_item)
         return cast(PayloadT, parsed_items)
 
-    raise OutputParseError(  # pragma: no cover - defensive guard
+    raise OutputParseError(
         "Unknown output container declared.",
         dataclass_type=dataclass_type,
     )
@@ -141,7 +141,7 @@ def _extract_json_payload(text: str, dataclass_type: type[Any]) -> object:
             continue
         try:
             payload, _ = decoder.raw_decode(text, index)
-        except json.JSONDecodeError:  # pragma: no cover - defensive fallback
+        except json.JSONDecodeError:
             continue
         return payload
 

weakincentives/prompt/tool.py

@@ -51,7 +51,7 @@ class Tool[ParamsT: SupportsDataclass, ResultT: SupportsDataclass]:
         )
         if params_type is None or result_type is None:
             origin = getattr(self, "__orig_class__", None)
-            if origin is not None:  # pragma: no cover - defensive fallback
+            if origin is not None:  # pragma: no cover - interpreter-specific path
                 args = get_args(origin)
                 if len(args) == 2 and all(isinstance(arg, type) for arg in args):
                     params_type = cast(type[SupportsDataclass], args[0])
@@ -161,7 +161,7 @@ class Tool[ParamsT: SupportsDataclass, ResultT: SupportsDataclass]:
 
         try:
             hints = get_type_hints(handler, include_extras=True)
-        except Exception:  # pragma: no cover - fallback for invalid hints
+        except Exception:
             hints = {}
 
         annotation = hints.get(parameter.name, parameter.annotation)

weakincentives/serde/dataclass_serde.py

@@ -12,6 +12,7 @@
 
 from __future__ import annotations
 
+# pyright: reportUnknownArgumentType=false, reportUnknownVariableType=false, reportUnknownMemberType=false, reportUnknownParameterType=false, reportUnnecessaryIsInstance=false, reportCallIssue=false, reportArgumentType=false, reportPossiblyUnboundVariable=false
 import dataclasses
 import re
 from collections.abc import Callable, Iterable, Mapping, Sequence, Sized
@@ -57,10 +58,7 @@ def _ordered_values(values: Iterable[object]) -> list[object]:
 
     items = list(values)
     if isinstance(values, (set, frozenset)):
-        try:
-            return sorted(items)
-        except TypeError:
-            return sorted(items, key=repr)
+        return sorted(items, key=repr)
     return items
 
 
@@ -101,7 +99,7 @@ def _merge_annotated_meta(
         base = args[0]
         for extra in args[1:]:
             if isinstance(extra, Mapping):
-                merged.update(extra)
+                merged.update(cast(Mapping[str, object], extra))
     return base, merged
 
 
@@ -191,14 +189,16 @@ def _apply_constraints(value: object, meta: Mapping[str, object], path: str) ->
 
     members = meta.get("in") or meta.get("enum")
     if isinstance(members, Iterable) and not isinstance(members, (str, bytes)):
-        options = _ordered_values(members)
+        options_iter = cast(Iterable[object], members)
+        options = _ordered_values(options_iter)
         normalized_options = [_normalize_option(option) for option in options]
         if result not in normalized_options:
             _fail(f"must be one of {normalized_options}")
 
     not_members = meta.get("not_in")
     if isinstance(not_members, Iterable) and not isinstance(not_members, (str, bytes)):
-        forbidden = _ordered_values(not_members)
+        forbidden_iter = cast(Iterable[object], not_members)
+        forbidden = _ordered_values(forbidden_iter)
         normalized_forbidden = [_normalize_option(option) for option in forbidden]
         if result in normalized_forbidden:
             _fail(f"may not be one of {normalized_forbidden}")
@@ -222,8 +222,9 @@ def _apply_constraints(value: object, meta: Mapping[str, object], path: str) ->
 
     converter = meta.get("convert", meta.get("transform"))
     if converter:
+        converter_fn = cast(Callable[[object], object], converter)
         try:
-            result = converter(result)
+            result = converter_fn(result)
         except (TypeError, ValueError) as error:
             raise type(error)(f"{path}: {error}") from error
         except Exception as error:  # pragma: no cover - defensive
@@ -243,7 +244,7 @@ def _coerce_to_type(
     origin = get_origin(base_type)
     type_name = getattr(base_type, "__name__", type(base_type).__name__)
 
-    if base_type in {object, _AnyType}:
+    if base_type is object or base_type is _AnyType:
         return _apply_constraints(value, merged_meta, path)
 
     if origin is Union:
@@ -290,7 +291,7 @@ def _coerce_to_type(
             if value == literal:
                 return _apply_constraints(literal, merged_meta, path)
             if config.coerce:
-                literal_type = type(literal)
+                literal_type = cast(type[object], type(literal))
                 try:
                     if isinstance(literal, bool) and isinstance(value, str):
                         coerced_literal = _bool_from_str(value)
@@ -313,7 +314,7 @@ def _coerce_to_type(
             return _apply_constraints(value, merged_meta, path)
         if not isinstance(value, Mapping):
             type_name = getattr(
-                dataclass_type, "__name__", dataclass_type.__class__.__name__
+                dataclass_type, "__name__", type(dataclass_type).__name__
             )
             raise TypeError(f"{path}: expected mapping for dataclass {type_name}")
         try:
@@ -353,7 +354,7 @@ def _coerce_to_type(
                 if isinstance(value, str):
                     value = [value]
                 elif isinstance(value, Iterable):
-                    value = list(value)
+                    value = list(cast(Iterable[object], value))
                 else:
                     raise TypeError(f"{path}: expected set")
             else:
@@ -367,7 +368,7 @@ def _coerce_to_type(
         if isinstance(value, str):  # pragma: no cover - handled by earlier coercion
             items = [value]
         elif isinstance(value, Iterable):
-            items = list(value)
+            items = list(cast(Iterable[object], value))
         else:  # pragma: no cover - defensive guard
             raise TypeError(f"{path}: expected iterable")
         args = get_args(base_type)
@@ -402,8 +403,9 @@ def _coerce_to_type(
         key_type, value_type = (
             get_args(base_type) if get_args(base_type) else (object, object)
         )
+        mapping_value = cast(Mapping[object, object], value)
         result_dict: dict[object, object] = {}
-        for key, item in value.items():
+        for key, item in mapping_value.items():
             coerced_key = _coerce_to_type(key, key_type, None, f"{path} keys", config)
             coerced_value = _coerce_to_type(
                 item, value_type, None, f"{path}[{coerced_key}]", config

weakincentives/session/session.py

@@ -95,6 +95,11 @@ class Session:
 
         return cast(tuple[S, ...], self._state.get(slice_type, ()))
 
+    def seed_slice[S](self, slice_type: type[S], values: Iterable[S]) -> None:
+        """Initialize or replace the stored tuple for the provided type."""
+
+        self._state[slice_type] = tuple(values)
+
     def _on_tool_invoked(self, event: object) -> None:
         if isinstance(event, ToolInvoked):
             self._handle_tool_invoked(event)

weakincentives/tools/__init__.py

@@ -14,6 +14,13 @@
 
 from __future__ import annotations
 
+from .asteval import (
+    AstevalSection,
+    EvalFileRead,
+    EvalFileWrite,
+    EvalParams,
+    EvalResult,
+)
 from .errors import ToolValidationError
 from .planning import (
     AddStep,
@@ -44,6 +51,11 @@ from .vfs import (
 
 __all__ = [
     "ToolValidationError",
+    "AstevalSection",
+    "EvalParams",
+    "EvalResult",
+    "EvalFileRead",
+    "EvalFileWrite",
     "Plan",
     "PlanStep",
     "PlanStatus",