)di;dZddlZddlZddlZddlZddlmZmZddlmZddl m Z m Z ddl m Z ddlmZddlmZdd lmZmZej,eZed Zed z d z Zd ZdZGddZy)u GeminiAgent — Persistent Gemini Agent with Session Memory and CTM Hooks ======================================================================== A stateful Gemini agent that: - Maintains conversation history across turns - Persists session state to JSONL files (E: drive) - Extracts and commits [CTM] blocks to the Knowledge Graph - Tracks token usage and compacts history when approaching context limits Uses google.genai (new SDK, NOT deprecated google.generativeai). Author: Genesis Parallel Builder Created: 2026-02-26 N)datetimetimezone)Path)AnyOptional)uuid4)genai)types) ctm_to_kgprocess_ctm_blocksz/mnt/e/genesis-systemdata gcc_sessionsi5 ceZdZdZ ddedededeededeed dfd Zd ed efd Zd e e d e fdZ d e e fdZ de d efdZde d dfdZddZddeded e fdZd efdZd efdZd e fdZd efdZy) GeminiAgenta1 A persistent Gemini agent with session memory and CTM hooks. Conversation history is stored as a list of dicts matching the google.genai Content format: {"role": "user" | "model", "parts": [{"text": "..."}]} Session state is persisted to a JSONL file in data/gcc_sessions/. Nnamemodel system_prompt memory_file max_historyapi_keyreturncv||_||_||_||_t t |_tjtj|_ d|_ g|_ g|_d|_d|_d|_t$j'dd|rt)||_nt$|dz |_|xs#t,j.j1dxsd}t3j4||_t8j;d|j|j|j y ) a[ Initialise a GeminiAgent. Args: name: Short identifier (e.g. "orchestrator", "builder"). model: Gemini model ID (e.g. "gemini-3.1-pro-preview"). system_prompt: System-level instruction injected each chat. memory_file: Path to JSONL session file. Auto-generated if None. max_history: Maximum conversation turns to retain before compaction consideration (soft limit). api_key: Gemini API key. Falls back to GEMINI_API_KEY env var, then the Genesis default key. rTparentsexist_okz_session.jsonlGEMINI_API_KEY'AIzaSyCT_rx0NusUJWoqtT7uxHAKEfHo129SJb8)ru3GeminiAgent[%s] initialised — model=%s session=%sN)rrrrstrr session_idrnowrutc created_at turn_count ctm_bufferhistorytotal_prompt_tokenstotal_completion_tokensestimated_context_tokensGCC_SESSIONS_DIRmkdirrrosenvirongetr Clientclientloggerinfo)selfrrrrrr resolved_keys :/mnt/e/genesis-system/core/gemini_command_centres/agent.py__init__zGeminiAgent.__init__8s ,  *&eg,",,x||4&($& $% '($()% td; #K0D /TF.2IID   9zz~~./ 98  ll<8  A II JJ OO  messagecK|jjdd|igd|j} tj|j |d{}|j|}|jjdd|igd|xjdz c_ |j|t||j }|rE|jj!|tj#d |jt%||j&t(k\r|j+|S7#t $rA}tjd|j||jjd}~wwxYww) a Send a user message and receive a model response. Automatically: - Appends message to history - Calls Gemini API with full history + system prompt - Extracts [CTM] blocks from response and writes to KG - Updates token counters - Triggers history compaction if context is near limit Args: message: User message text. Returns: Model response text. usertextroleparts)contentsNzGeminiAgent[%s] API error: %sr) agent_namez2GeminiAgent[%s] CTM: %d blocks extracted and saved)r&append_build_contentsasyncio to_thread _call_api Exceptionr1errorrpop _extract_textr$_update_token_countsr r%extendr2lenr)COMPACTION_TOKEN_THRESHOLD_compact_history)r3r8r?responseexc response_text ctm_resultss r5chatzGeminiAgent.chatwsQ$ V7H6IJK'') $..!H**84  W8O7PQR 1 !!(+)499M  OO " "; / KKD K    ( (,F F  ! ! #A  LL8$))S I LL      s;2F$D;D9D;CF9D;; F IndexError TypeErrorr)r3rPs r5rJzGeminiAgent._extract_textsp ==     !&&q)1177:?? ? I6 !x=  !s  /AA0/A0cD |j}|xjt|ddxsdz c_|xjt|ddxsdz c_t|ddxsd|_y#t $r)t d|jD}|dz|_YywxYw)z3Update token counters from response usage_metadata.prompt_token_countrcandidates_token_countc3DK|]}t|dddyw)r>rr;N)rM).0ms r5 z3GeminiAgent._update_token_counts..s'/0AgJqM&)*s N)usage_metadatar'getattrr(r)r`sumr&)r3rPusage total_charss r5rKz GeminiAgent._update_token_countss =++E  $ $7KQ(O(TST T $  ( (7;@q  ( 3Q7<1  ) =48LLK-81, 3"E*T12 3 LL> 4K[K[ \ 3 3  LLA499c R s0C#E %(E 3E E E F!E<<Fc|jjs,tjd|j|jy d}t |jdd5}|D]}|j }|s|} ddd|sytj|}|jd|j|_ |jd d |_ |jd g|_ |jd d |_ |jd d |_|jdd |_tjd|j|jt!|jy#1swYxYw#tj"t$f$r+}tj'd|j|Yd}~yd}~wwxYw)a Resume from last saved session state. Reads the most recent JSON object in memory_file and restores history, turn_count, and token counters. Returns: True if session was loaded, False if file does not exist or parse error. u9GeminiAgent[%s] no saved session at %s — starting freshFrrrNr r$rr&r'r(r)u@GeminiAgent[%s] session loaded — %d turns, %d history messagesTu?GeminiAgent[%s] load_session parse error: %s — starting fresh)rexistsr1r2rrstriprloadsr.r r$r&r'r(r)rMJSONDecodeErrorKeyErrorwarning)r3 last_linerlinerrQs r5 load_sessionzGeminiAgent.load_sessionGs&&( KKK     ! Id&&g> )")D::  YYZZ//// #DLL 1#'#;#;'+'C'C(,(E(E",$_a8  r7c d|jd|jd|jdt|jd S)NzGeminiAgent(name=z, model=z, turns=z , history=))rrr$rMr&r^s r5__repr__zGeminiAgent.__repr__sD }HTZZNC__%ZDLL0A/B! E r7)N2N)rN)entity)__name__ __module__ __qualname____doc__rrintr6rTr]dictrrFrCrJrKrOr|boolrrrrr7r5rr-s&*!%; ; ;  ; c] ;  ; #;  ; ~8#8#8t$t*("d" !c !c !=S=T=*" L3#T*d>2d2l 4 4 # r7r)rrDrloggingr,rrpathlibrtypingrruuidrgoogler google.genair rXcore.gemini_command_centres.ctmr r getLoggerrr1 GENESIS_ROOTr*rNrtrrr7r5rsv   ' -I   8 $+, &(>9 %n n r7