context-compiler 0.9.0.dev4__tar.gz → 0.9.0.dev6__tar.gz

{context_compiler-0.9.0.dev4 → context_compiler-0.9.0.dev6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: context-compiler
-Version: 0.9.0.dev4
+Version: 0.9.0.dev6
 Summary: Deterministic conversational state engine for LLM applications.
 Project-URL: Homepage, https://github.com/rlippmann/context-compiler
 Project-URL: Repository, https://github.com/rlippmann/context-compiler
@@ -18,14 +18,6 @@ Classifier: Topic :: Software Development :: Libraries
 Requires-Python: >=3.11
 Provides-Extra: demos
 Requires-Dist: litellm>=1.0.0; extra == 'demos'
-Provides-Extra: dev
-Requires-Dist: coverage; extra == 'dev'
-Requires-Dist: hypothesis; extra == 'dev'
-Requires-Dist: mypy<2.0,>=1.12; extra == 'dev'
-Requires-Dist: pre-commit; extra == 'dev'
-Requires-Dist: pytest; extra == 'dev'
-Requires-Dist: pytest-cov; extra == 'dev'
-Requires-Dist: ruff<1.0,>=0.12; extra == 'dev'
 Description-Content-Type: text/markdown
 
 # Context Compiler
@@ -179,7 +171,11 @@ decision = engine.step(user_input)
 if is_error(decision):
     show_to_user(decision["message"])
 elif is_update(decision):
-    messages = build_messages(engine.state, user_input)
+    messages = build_messages(
+        premise=engine.premise,
+        policies=engine.policies,
+        user_input=user_input,
+    )
     render(call_llm(messages))
 else:
     render(call_llm(user_input))
@@ -223,10 +219,9 @@ Preload options load authoritative state:
 - `--initial-state-json` / `--initial-state-file` load saved state
   (via exported state JSON).
 
-REPL commands (controller layer, not engine directives):
+REPL commands (not engine directives):
 
 - `state` shows current saved state.
-- `preview <input>` runs deterministic dry-run without mutating live state.
 - `step <input>` is an explicit alias of normal bare-input step behavior.
 
 Bare REPL input behavior remains unchanged.
@@ -267,7 +262,7 @@ Packaging notes:
 ### Development
 
 ```bash
-uv sync --group dev
+uv sync --dev
 uv run pytest
 ```
 
@@ -276,20 +271,20 @@ uv run pytest
 Each user message produces a `Decision`.
 
 ```python
-class NoDirectiveDecision(TypedDict):
-    kind: Literal["no_directive"]
-
-class UpdateDecision(TypedDict):
-    kind: Literal["update"]
-    state: State
-
-class ErrorDecision(TypedDict):
-    kind: Literal["error"]
-    message: str
-
-Decision = NoDirectiveDecision | UpdateDecision | ErrorDecision
+class DecisionKind(StrEnum):
+    NO_DIRECTIVE = "no_directive"
+    UPDATE = "update"
+    ERROR = "error"
+
+class Decision(TypedDict):
+    kind: DecisionKind
+    message: str | None
 ```
 
+`message` is structurally present on every `Decision`, but it is semantically
+meaningful only for `error`. For `no_directive` and `update`, `message` is
+`None`.
+
 Meaning:
 
 | kind | host behavior |
@@ -299,7 +294,7 @@ Meaning:
 | error | show `message` and do not continue normal downstream processing yet |
 
 For normal app code, prefer the exported decision helpers (`is_error`,
-`is_update`, `is_no_directive`, `get_error_message`, `get_decision_state`)
+`is_update`, `is_no_directive`, `get_error_message`)
 instead of direct key traversal.
 
 See [docs/api-reference.md](docs/api-reference.md) for the full public API
@@ -307,26 +302,12 @@ reference.
 
 Common API entry points:
 
-- engine lifecycle: `create_engine(...)`, `engine.step(...)`, `engine.state`,
+- engine lifecycle: `create_engine()`, `engine.step(...)`,
   `engine.premise`, `engine.policies`, `engine.export_json(...)`,
   `engine.import_json(...)`
 - decision helpers: `is_error(...)`, `is_update(...)`, `is_no_directive(...)`,
-  `get_error_message(...)`, `get_decision_state(...)`
+  `get_error_message(...)`
 - state transport: `engine.export_json(...)`, `engine.import_json(...)`
-- controller API: `step(...)`
-- audit APIs: `preview(...)`, `state_diff(...)`
-
-### Controller And Audit APIs (Reusable Outside REPL)
-
-- `step(engine, user_input)` returns a reusable result envelope around one
-  engine turn
-- `preview(engine, user_input)` performs a deterministic dry run and restores
-  live engine state afterward
-- `state_diff(state_before, state_after)` summarizes structural state changes
-
-For examples and helper accessors such as `get_step_decision(...)`,
-`get_preview_state_after(...)`, `preview_would_mutate(...)`, and
-`diff_has_changes(...)`, see [docs/api-reference.md](docs/api-reference.md).
 
 ---
 
@@ -350,8 +331,8 @@ Identical input sequences always produce identical state.
 For live engine-owned reads, use `engine.premise` and `engine.policies`.
 `engine.policies` returns a caller-owned copy.
 
-`engine.state` remains the public snapshot/serialization boundary when host code
-needs the full authoritative state object.
+Use `engine.export_json()` and `engine.import_json()` for persistence and
+restoration.
 
 ---
 

{context_compiler-0.9.0.dev4 → context_compiler-0.9.0.dev6}/README.md

@@ -149,7 +149,11 @@ decision = engine.step(user_input)
 if is_error(decision):
     show_to_user(decision["message"])
 elif is_update(decision):
-    messages = build_messages(engine.state, user_input)
+    messages = build_messages(
+        premise=engine.premise,
+        policies=engine.policies,
+        user_input=user_input,
+    )
     render(call_llm(messages))
 else:
     render(call_llm(user_input))
@@ -193,10 +197,9 @@ Preload options load authoritative state:
 - `--initial-state-json` / `--initial-state-file` load saved state
   (via exported state JSON).
 
-REPL commands (controller layer, not engine directives):
+REPL commands (not engine directives):
 
 - `state` shows current saved state.
-- `preview <input>` runs deterministic dry-run without mutating live state.
 - `step <input>` is an explicit alias of normal bare-input step behavior.
 
 Bare REPL input behavior remains unchanged.
@@ -237,7 +240,7 @@ Packaging notes:
 ### Development
 
 ```bash
-uv sync --group dev
+uv sync --dev
 uv run pytest
 ```
 
@@ -246,20 +249,20 @@ uv run pytest
 Each user message produces a `Decision`.
 
 ```python
-class NoDirectiveDecision(TypedDict):
-    kind: Literal["no_directive"]
-
-class UpdateDecision(TypedDict):
-    kind: Literal["update"]
-    state: State
-
-class ErrorDecision(TypedDict):
-    kind: Literal["error"]
-    message: str
-
-Decision = NoDirectiveDecision | UpdateDecision | ErrorDecision
+class DecisionKind(StrEnum):
+    NO_DIRECTIVE = "no_directive"
+    UPDATE = "update"
+    ERROR = "error"
+
+class Decision(TypedDict):
+    kind: DecisionKind
+    message: str | None
 ```
 
+`message` is structurally present on every `Decision`, but it is semantically
+meaningful only for `error`. For `no_directive` and `update`, `message` is
+`None`.
+
 Meaning:
 
 | kind | host behavior |
@@ -269,7 +272,7 @@ Meaning:
 | error | show `message` and do not continue normal downstream processing yet |
 
 For normal app code, prefer the exported decision helpers (`is_error`,
-`is_update`, `is_no_directive`, `get_error_message`, `get_decision_state`)
+`is_update`, `is_no_directive`, `get_error_message`)
 instead of direct key traversal.
 
 See [docs/api-reference.md](docs/api-reference.md) for the full public API
@@ -277,26 +280,12 @@ reference.
 
 Common API entry points:
 
-- engine lifecycle: `create_engine(...)`, `engine.step(...)`, `engine.state`,
+- engine lifecycle: `create_engine()`, `engine.step(...)`,
   `engine.premise`, `engine.policies`, `engine.export_json(...)`,
   `engine.import_json(...)`
 - decision helpers: `is_error(...)`, `is_update(...)`, `is_no_directive(...)`,
-  `get_error_message(...)`, `get_decision_state(...)`
+  `get_error_message(...)`
 - state transport: `engine.export_json(...)`, `engine.import_json(...)`
-- controller API: `step(...)`
-- audit APIs: `preview(...)`, `state_diff(...)`
-
-### Controller And Audit APIs (Reusable Outside REPL)
-
-- `step(engine, user_input)` returns a reusable result envelope around one
-  engine turn
-- `preview(engine, user_input)` performs a deterministic dry run and restores
-  live engine state afterward
-- `state_diff(state_before, state_after)` summarizes structural state changes
-
-For examples and helper accessors such as `get_step_decision(...)`,
-`get_preview_state_after(...)`, `preview_would_mutate(...)`, and
-`diff_has_changes(...)`, see [docs/api-reference.md](docs/api-reference.md).
 
 ---
 
@@ -320,8 +309,8 @@ Identical input sequences always produce identical state.
 For live engine-owned reads, use `engine.premise` and `engine.policies`.
 `engine.policies` returns a caller-owned copy.
 
-`engine.state` remains the public snapshot/serialization boundary when host code
-needs the full authoritative state object.
+Use `engine.export_json()` and `engine.import_json()` for persistence and
+restoration.
 
 ---
 

{context_compiler-0.9.0.dev4 → context_compiler-0.9.0.dev6}/demos/01_llm_contradiction_error.py

@@ -7,6 +7,7 @@ from demos.common import (
     build_reinjected_messages,
     compact_user_turns,
     extract_tag_value,
+    observe_engine,
     print_decision,
     print_host_check,
     print_messages,
@@ -14,6 +15,7 @@ from demos.common import (
     print_spec_report,
     print_tag_comparison,
     print_user_inputs,
+    state_observations,
     yes_no,
 )
 from demos.llm_client import complete_messages
@@ -25,9 +27,11 @@ def main() -> None:
     print_user_inputs(user_inputs)
 
     first = engine.step(user_inputs[0])
-    print_decision("turn 1", first, engine.state)
+    premise, policies = observe_engine(engine)
+    print_decision("turn 1", first, premise=premise, policies=policies)
     second = engine.step(user_inputs[1])
-    print_decision("turn 2", second, engine.state)
+    premise, policies = observe_engine(engine)
+    print_decision("turn 2", second, premise=premise, policies=policies)
 
     baseline_messages = build_baseline_messages(
         [
@@ -67,7 +71,12 @@ def main() -> None:
         mediated_output = f"[no call] error required: {second['message']}\nACTION:error"
         print_model_output("Compiler-mediated (full)", mediated_output)
     else:
-        mediated_messages = build_mediated_messages_from_transcript(engine.state, user_inputs)
+        premise, policies = observe_engine(engine)
+        mediated_messages = build_mediated_messages_from_transcript(
+            premise=premise,
+            policies=policies,
+            user_turns=user_inputs,
+        )
         print_messages("compiler-mediated (full)", mediated_messages)
         mediated_output = complete_messages(mediated_messages)
         print_model_output("Compiler-mediated (full)", mediated_output)
@@ -78,7 +87,12 @@ def main() -> None:
         compact_output = f"[no call] error required: {compacted_prompt}\nACTION:error"
         print_model_output("Compiler-mediated + compact", compact_output)
     else:
-        compact_messages = build_mediated_messages_from_transcript(compacted_state, compacted_turns)
+        compacted_premise, compacted_policies = state_observations(compacted_state)
+        compact_messages = build_mediated_messages_from_transcript(
+            premise=compacted_premise,
+            policies=compacted_policies,
+            user_turns=compacted_turns,
+        )
         print_messages("compiler-mediated + compact", compact_messages)
         compact_output = complete_messages(compact_messages)
         print_model_output("Compiler-mediated + compact", compact_output)

{context_compiler-0.9.0.dev4 → context_compiler-0.9.0.dev6}/demos/02_llm_constraint_guardrail.py

@@ -8,12 +8,14 @@ from demos.common import (
     build_mediated_messages_from_transcript,
     build_reinjected_messages,
     compact_user_turns,
+    observe_engine,
     print_decision,
     print_host_check,
     print_messages,
     print_model_output,
     print_spec_report,
     print_user_inputs,
+    state_observations,
     yes_no,
 )
 from demos.llm_client import complete_messages
@@ -115,10 +117,12 @@ def main() -> None:
     print_user_inputs(user_inputs)
 
     first = engine.step(user_inputs[0])
-    print_decision("turn 1", first, engine.state)
+    premise, policies = observe_engine(engine)
+    print_decision("turn 1", first, premise=premise, policies=policies)
 
     second = engine.step(user_inputs[1])
-    print_decision("turn 2", second, engine.state)
+    premise, policies = observe_engine(engine)
+    print_decision("turn 2", second, premise=premise, policies=policies)
 
     baseline_messages = build_baseline_messages(
         [user_inputs[1]],
@@ -154,9 +158,11 @@ def main() -> None:
     reinjected_safe_alternative = safe_alternative_detected(reinjected_output)
     reinjected_violation = recipe_includes_prohibited_item(reinjected_output)
 
+    premise, policies = observe_engine(engine)
     mediated_messages = build_mediated_messages_from_transcript(
-        engine.state,
-        user_inputs,
+        premise=premise,
+        policies=policies,
+        user_turns=user_inputs,
         extra_system_prompt=(
             "If the user requests a prohibited item, refuse the literal request. "
             "State briefly that the request conflicts with compiled policy, then provide "
@@ -178,9 +184,11 @@ def main() -> None:
         compact_refusal = True
         compact_violation = False
     else:
+        compacted_premise, compacted_policies = state_observations(compacted_state)
         compact_messages = build_mediated_messages_from_transcript(
-            compacted_state,
-            compacted_turns,
+            premise=compacted_premise,
+            policies=compacted_policies,
+            user_turns=compacted_turns,
             extra_system_prompt=(
                 "If the user requests a prohibited item, refuse the literal request. "
                 "State briefly that the request conflicts with compiled policy, then provide "

{context_compiler-0.9.0.dev4 → context_compiler-0.9.0.dev6}/demos/03_llm_premise_guardrail.py

@@ -9,6 +9,7 @@ from demos.common import (
     build_reinjected_messages,
     compact_user_turns,
     extract_tag_value,
+    observe_engine,
     print_decision,
     print_host_check,
     print_messages,
@@ -16,6 +17,7 @@ from demos.common import (
     print_spec_report,
     print_tag_comparison,
     print_user_inputs,
+    state_observations,
     yes_no,
 )
 from demos.llm_client import complete_messages
@@ -78,7 +80,13 @@ def main() -> None:
 
     for index, user_input in enumerate(user_inputs, start=1):
         decision = engine.step(user_input)
-        print_decision(f"turn {index}", decision, engine.state)
+        premise, policies = observe_engine(engine)
+        print_decision(
+            f"turn {index}",
+            decision,
+            premise=premise,
+            policies=policies,
+        )
 
     baseline_messages = build_baseline_messages(
         [user_inputs[0], user_inputs[1], user_inputs[2]],
@@ -100,7 +108,12 @@ def main() -> None:
     reinjected_output = complete_messages(reinjected_messages)
     print_model_output("Reinjected-state", reinjected_output)
 
-    mediated_messages = build_mediated_messages_from_transcript(engine.state, user_inputs)
+    premise, policies = observe_engine(engine)
+    mediated_messages = build_mediated_messages_from_transcript(
+        premise=premise,
+        policies=policies,
+        user_turns=user_inputs,
+    )
     print_messages("compiler-mediated (full)", mediated_messages)
     mediated_output = complete_messages(mediated_messages)
     print_model_output("Compiler-mediated (full)", mediated_output)
@@ -111,7 +124,12 @@ def main() -> None:
         compact_output = f"[no call] error required: {compacted_prompt}"
         print_model_output("Compiler-mediated + compact", compact_output)
     else:
-        compact_messages = build_mediated_messages_from_transcript(compacted_state, compacted_turns)
+        compacted_premise, compacted_policies = state_observations(compacted_state)
+        compact_messages = build_mediated_messages_from_transcript(
+            premise=compacted_premise,
+            policies=compacted_policies,
+            user_turns=compacted_turns,
+        )
         print_messages("compiler-mediated + compact", compact_messages)
         compact_output = complete_messages(compact_messages)
         print_model_output("Compiler-mediated + compact", compact_output)

{context_compiler-0.9.0.dev4 → context_compiler-0.9.0.dev6}/demos/04_llm_tool_denylist_guardrail.py

@@ -10,6 +10,7 @@ from demos.common import (
     compact_user_turns,
     extract_tag_value,
     is_verbose,
+    observe_engine,
     print_decision,
     print_host_check,
     print_messages,
@@ -17,6 +18,7 @@ from demos.common import (
     print_spec_report,
     print_tag_comparison,
     print_user_inputs,
+    state_observations,
 )
 from demos.llm_client import complete_messages
 
@@ -69,10 +71,12 @@ def main() -> None:
     print_user_inputs(user_inputs)
 
     first = engine.step(user_inputs[0])
-    print_decision("turn 1", first, engine.state)
+    premise, policies = observe_engine(engine)
+    print_decision("turn 1", first, premise=premise, policies=policies)
 
     second = engine.step(user_inputs[1])
-    print_decision("turn 2", second, engine.state)
+    premise, policies = observe_engine(engine)
+    print_decision("turn 2", second, premise=premise, policies=policies)
 
     baseline_messages = build_baseline_messages(
         [user_inputs[1]],
@@ -107,9 +111,11 @@ def main() -> None:
     reinjected_output = complete_messages(reinjected_messages)
     print_model_output("Reinjected-state", reinjected_output)
 
+    premise, policies = observe_engine(engine)
     mediated_messages = build_mediated_messages_from_transcript(
-        engine.state,
-        user_inputs,
+        premise=premise,
+        policies=policies,
+        user_turns=user_inputs,
         extra_system_prompt=(
             "Only choose tools that are not prohibited."
             + "\nCandidate tools: "
@@ -128,9 +134,11 @@ def main() -> None:
         print_model_output("Compiler-mediated + compact", compact_output)
         compact_tool = None
     else:
+        compacted_premise, compacted_policies = state_observations(compacted_state)
         compact_messages = build_mediated_messages_from_transcript(
-            compacted_state,
-            compacted_turns,
+            premise=compacted_premise,
+            policies=compacted_policies,
+            user_turns=compacted_turns,
             extra_system_prompt=(
                 "Only choose tools that are not prohibited."
                 + "\nCandidate tools: "

{context_compiler-0.9.0.dev4 → context_compiler-0.9.0.dev6}/demos/05_llm_prompt_drift_vs_state.py

@@ -11,6 +11,7 @@ from demos.common import (
     build_reinjected_messages,
     compact_user_turns,
     extract_tag_value,
+    observe_engine,
     print_decision,
     print_host_check,
     print_messages,
@@ -18,6 +19,7 @@ from demos.common import (
     print_spec_report,
     print_tag_comparison,
     print_user_inputs,
+    state_observations,
     yes_no,
 )
 from demos.llm_client import complete_messages
@@ -218,7 +220,13 @@ def _run_demo(turns: int = _DEFAULT_TURNS) -> None:
 
     for index, user_input in enumerate(user_inputs, start=1):
         decision = engine.step(user_input)
-        print_decision(f"turn {index}", decision, engine.state)
+        premise, policies = observe_engine(engine)
+        print_decision(
+            f"turn {index}",
+            decision,
+            premise=premise,
+            policies=policies,
+        )
 
     baseline_messages = build_baseline_messages(
         user_inputs,
@@ -242,9 +250,11 @@ def _run_demo(turns: int = _DEFAULT_TURNS) -> None:
     reinjected_output = complete_messages(reinjected_messages)
     print_model_output("Reinjected-state", reinjected_output)
 
+    premise, policies = observe_engine(engine)
     mediated_messages = build_mediated_messages_from_transcript(
-        engine.state,
-        user_inputs,
+        premise=premise,
+        policies=policies,
+        user_turns=user_inputs,
         extra_system_prompt=_FORMAT_CONTRACT_SYSTEM_PROMPT,
     )
     print_messages("compiler-mediated (full)", mediated_messages)
@@ -257,7 +267,8 @@ def _run_demo(turns: int = _DEFAULT_TURNS) -> None:
         compact_output = f"[no call] error required: {compacted_prompt}"
         print_model_output("Compiler-mediated + compact", compact_output)
     else:
-        premise_value = compacted_state["premise"]
+        compacted_premise, compacted_policies = state_observations(compacted_state)
+        premise_value = compacted_premise
         if (
             premise_value is not None
             and _ORIGINAL_DIRECTIVE not in compacted_turns
@@ -265,8 +276,9 @@ def _run_demo(turns: int = _DEFAULT_TURNS) -> None:
         ):
             compacted_turns = [f"Premise reminder: {premise_value}", *compacted_turns]
         compact_messages = build_mediated_messages_from_transcript(
-            compacted_state,
-            compacted_turns,
+            premise=compacted_premise,
+            policies=compacted_policies,
+            user_turns=compacted_turns,
             extra_system_prompt=_FORMAT_CONTRACT_SYSTEM_PROMPT,
         )
         print_messages("compiler-mediated + compact", compact_messages)

{context_compiler-0.9.0.dev4 → context_compiler-0.9.0.dev6}/demos/06_llm_context_compaction.py

@@ -1,7 +1,7 @@
 """Demo 6: host-side prompt replacement from authoritative step-derived state."""
 
 from context_compiler import DECISION_UPDATE, create_engine
-from demos.common import compact_user_turns, is_verbose, print_info_report
+from demos.common import compact_user_turns, is_verbose, print_info_report, state_observations
 
 DEMO_NAME = "06_context_compaction — superseded directives eliminated"
 FINAL_PREMISE = "chickpea curry"
@@ -148,8 +148,9 @@ def main() -> None:
     baseline_context = "\n".join(f"User: {turn}" for turn in transcript_turns)
     compiled_context = f"- premise: {compiled_premise}"
     compacted_turns, compacted_state, compacted_prompt = compact_user_turns(transcript_turns)
+    compacted_premise, _compacted_policies = state_observations(compacted_state)
     assert compacted_prompt is None
-    assert compacted_state["premise"] == FINAL_PREMISE
+    assert compacted_premise == FINAL_PREMISE
     compacted_context = "\n".join(f"User: {turn}" for turn in compacted_turns)
     baseline_prompt = _build_baseline_prompt(transcript_turns)
     compiled_prompt = _build_compiled_prompt(compiled_premise)

{context_compiler-0.9.0.dev4 → context_compiler-0.9.0.dev6}/demos/07_llm_prompt_vs_state.py

@@ -1,20 +1,24 @@
 """Demo 7: prompt engineering and compiled state are complementary."""
 
 import re
+from collections.abc import Mapping
 
-from context_compiler import State, create_engine
+from context_compiler import create_engine
+from context_compiler.engine import PolicyValue
 from demos.common import (
     build_baseline_messages,
     build_compiled_system_prompt,
     build_reinjected_messages,
     compact_user_turns,
     extract_tag_value,
+    observe_engine,
     print_decision,
     print_host_check,
     print_messages,
     print_model_output,
     print_spec_report,
     print_user_inputs,
+    state_observations,
     yes_no,
 )
 from demos.llm_client import Message, complete_messages
@@ -74,14 +78,22 @@ def build_strong_messages(user_inputs: list[str]) -> list[Message]:
     )
 
 
-def build_compiler_messages(state: State, user_inputs: list[str]) -> list[Message]:
-    compiled_prefix = build_compiled_system_prompt(state)
+def build_compiler_messages(
+    *, premise: str | None, policies: Mapping[str, PolicyValue], user_inputs: list[str]
+) -> list[Message]:
+    compiled_prefix = build_compiled_system_prompt(premise=premise, policies=policies)
     compiler_system_prompt = f"{compiled_prefix}\n{STRONG_PROMPT_ENGINEERING_TEXT}"
     return build_baseline_messages(user_inputs, baseline_system_prompt=compiler_system_prompt)
 
 
-def build_compact_compiler_messages(state: State, compacted_inputs: list[str]) -> list[Message]:
-    return build_compiler_messages(state, compacted_inputs)
+def build_compact_compiler_messages(
+    *, premise: str | None, policies: Mapping[str, PolicyValue], compacted_inputs: list[str]
+) -> list[Message]:
+    return build_compiler_messages(
+        premise=premise,
+        policies=policies,
+        user_inputs=compacted_inputs,
+    )
 
 
 def _actual_summary(*, weak_pass: bool, strong_pass: bool, compiler_pass: bool) -> str:
@@ -108,7 +120,13 @@ def main() -> None:
 
     for index, user_input in enumerate(USER_INPUTS, start=1):
         decision = engine.step(user_input)
-        print_decision(f"turn {index}", decision, engine.state)
+        premise, policies = observe_engine(engine)
+        print_decision(
+            f"turn {index}",
+            decision,
+            premise=premise,
+            policies=policies,
+        )
 
     weak_messages = build_weak_messages(USER_INPUTS)
     print_messages("weak-baseline", weak_messages)
@@ -131,7 +149,12 @@ def main() -> None:
     reinjected_output = complete_messages(reinjected_messages)
     print_model_output("Reinjected-state", reinjected_output)
 
-    compiler_messages = build_compiler_messages(engine.state, USER_INPUTS)
+    premise, policies = observe_engine(engine)
+    compiler_messages = build_compiler_messages(
+        premise=premise,
+        policies=policies,
+        user_inputs=USER_INPUTS,
+    )
     print_messages("compiler-mediated (full)", compiler_messages)
     compiler_output = complete_messages(compiler_messages)
     print_model_output("Compiler-mediated (full)", compiler_output)
@@ -142,7 +165,12 @@ def main() -> None:
         compact_output = f"[no call] error required: {compacted_prompt}"
         print_model_output("Compiler-mediated + compact", compact_output)
     else:
-        compact_messages = build_compact_compiler_messages(compacted_state, compacted_inputs)
+        compacted_premise, compacted_policies = state_observations(compacted_state)
+        compact_messages = build_compact_compiler_messages(
+            premise=compacted_premise,
+            policies=compacted_policies,
+            compacted_inputs=compacted_inputs,
+        )
         print_messages("compiler-mediated + compact", compact_messages)
         compact_output = complete_messages(compact_messages)
         print_model_output("Compiler-mediated + compact", compact_output)
@@ -158,7 +186,7 @@ def main() -> None:
     compiler_pass = premise_matches_expected(compiler_output)
     compact_pass = compacted_prompt is None and premise_matches_expected(compact_output)
 
-    compiled_prefix = build_compiled_system_prompt(engine.state)
+    compiled_prefix = build_compiled_system_prompt(premise=premise, policies=policies)
     shared_prompt_text = compiler_messages[0]["content"].endswith(STRONG_PROMPT_ENGINEERING_TEXT)
     compiler_augmented_only = (
         compiler_messages[1:] == strong_messages[1:]