>i z d Z ddlZddlZddlmZmZ ddlmZ ddlmZm Z ddl
m Z ed Z dZ
G d d
e Zy) u
Genesis Persistent Context Architecture — JIT Hydration Interceptor
Story 2.06 — Track B
JITHydrationInterceptor — wires zero-amnesia memory injection into the
interceptor chain.
Priority 10 — runs after telemetry (0), before business logic (50+).
N)datetimetimezone)Path)BaseInterceptorInterceptorMetadata)interceptor_jit_hydrationz(/mnt/e/genesis-system/data/observabilityzHHydration failedc x e Zd ZU dZ edd Zeed< dedefdZd ededd
fd Z d e
dedefd
Z dedefdZ y
)JITHydrationInterceptoruC
Injects ZERO_AMNESIA_STATE XML envelope into every task payload.
Priority 10 — runs after telemetry (0), before business logic (50+).
Failure in hydration NEVER blocks execution: an error envelope is
inserted instead, guaranteeing the downstream LLM call always receives
a system_injection key.
jit_hydration
)nameprioritymetadata task_payloadreturnc j K t | d{ S 7 # t $ r t |d< |cY S w xY ww)a
Calls interceptor_jit_hydration, attaches system_injection to payload.
On any hydration failure, injects an ERROR envelope so execution is
never blocked by memory unavailability.
Args:
task_payload: Dict with task data. May contain "task_id", "prompt",
"description", "task", "file", "code", "context".
Returns:
The same dict with "system_injection" key set to the XML envelope.
Nsystem_injectionr Exception_ERROR_ENVELOPE)selfr s >/mnt/e/genesis-system/core/memory/jit_hydration_interceptor.py pre_executez#JITHydrationInterceptor.pre_execute( s; 2<@@ @@ />L+ , s( 3 3 0303resultNc K |j dd } t j dd t j t
j j dt | |j dd d}t t d z d
5 }|j t j | d z d d d y # 1 sw Y y xY w# t $ r Y y w xY ww)
u8
Logs system_injection length to observability events.
Side-effects only — does NOT mutate result or task_payload.
Args:
result: The execution result dict (not mutated).
task_payload: The enriched task dict (used for task_id + injection length).
r T)parentsexist_okjit_hydration_completetask_idunknown) timestamp
event_typeinjection_lengthr z events.jsonla
N)get
EVENTS_DIRmkdirr nowr utc isoformatlenopenwritejsondumpsr )r r r injectioneventfs r post_executez$JITHydrationInterceptor.post_execute< s !$$%7< TD 9 %\\(,, 7 A A C 6$' N'++IyA E j>137
21
5)D01
2
2
2 sM CA9C (C 7C ?C C C C C CCCCerrorc 0 K t | |t dS w)a7
Returns a correction payload with error envelope in system_injection.
Args:
error: The exception that caused the failure.
task_payload: The original task dict.
Returns:
Dict with "error", "task_payload", and "system_injection" keys.
)r6 r r )strr )r r6 r s r on_errorz JITHydrationInterceptor.on_errorT s Z ( /
s correction_payloadc K d|v r d|d |d< t | d{ S 7 # t $ r t |d< |cY S w xY ww)a
Prepends CORRECTION: prefix to the prompt and re-hydrates.
If no "prompt" key is present the payload is still safely re-hydrated
without mutation. On hydration failure, inserts an error envelope.
Args:
correction_payload: Task dict for the correction pass.
Returns:
Re-hydrated payload with updated system_injection.
promptz CORRECTION: Nr r )r r: s r
on_correctionz%JITHydrationInterceptor.on_correctione sa ) )-9:LX:V9W+X x ( &23EFF FF &5D 1 2% % &s, A( &( A( ?A?A)
__name__
__module__ __qualname____doc__r r __annotations__dictr r5 r r9 r= r r
r
s{ %8
%H!
d t ( T d 0
I
T
d
"&d &t &rE r
)rA asyncior0 r r pathlibr "core.interceptors.base_interceptorr r core.memory.jit_hydrationr r( r r
rD rE r rJ sB ' S ?
<
=
O
_&o _&rE