߲jdZddlmZddlZddlmZmZmZmZddl m Z d dZ d dZ d dZ dd Zdd lmZy)zTranscript repair for SessionDB batch appends: reconcile in-memory assistant rows with committed SQLite rows (blank-row in-place update, concurrent-winner adoption, watermark-compaction clone lookup) and sync markers after commit.) annotationsN)AnyCallableDictList)_DB_PERSISTED_MARKERc|yt|tr|j St|tr'dj d|Dj Sy)zYTrue when decoded message content is None, whitespace-only, or has no visible text parts.Tc3K|]<}t|ts|jddk(s)|jdd>yw)typetextr N) isinstancedictget).0ps z#is_content_blank..s9qAtATYZY^Y^_eYfjpYp155,qsAAAF)rstrstriplistjoin)contents ris_content_blankr sQ'3==?""'4 77qgqqwwyyy c g}|D]}t|tr|jdnd}d}t|tr"|jdddk(r t |||}||j |nt|d} ||d} | |d<t | r+|jd||jd| |f| |d <|S) aPartition a message batch within an active write transaction. An assistant message carrying an existing integer ``_row_id`` targets its active SQLite row (or the active clone a watermark compaction made of it): a blank row is updated in place; a non-blank one (concurrent winner) has its canonical content adopted without overwrite. Returns the messages that must be inserted as fresh rows._row_idNroleunknown assistantidrzNUPDATE messages SET content = ? WHERE id = ? AND session_id = ? AND active = 1_canonical_content)rrrint_active_assistant_rowappendrexecute) conn session_idmessagesencode_content_fndecode_content_fn inserted_rowsmsgexisting_row_id target_row target_iddecodeds r#resolve_and_repair_transcript_batchr2s+-M00:30E#''),4 os + 0Jk0Y.tZQJ     %  4() #Jy$9:"I G $ LLA"3779#56 :N  )0C$ %%0& rc|jd||fj}||ddk7ryt|dxsddk(r|S|jd||d |d fjS) z_The active assistant row for ``row_id``, or the active clone a watermark compaction made of it.zYSELECT id, role, active, timestamp, content FROM messages WHERE id = ? AND session_id = ?Nrr activerzSELECT id, role, active, timestamp, content FROM messages WHERE session_id = ? AND active = 1 AND role = 'assistant' AND timestamp IS ? AND id != ? ORDER BY id DESC LIMIT 1 timestampr!)r&fetchoner#)r'r(row_idrows rr$r$:s ,, *  hj   {c&k[0 3x= A!# << # S%s4y1   hj rct||D]o\}}d|t<t|jdtr|d|d<t|jdtt fr|d|d<d|vsh|d|d<qy)zZStamp _DB_PERSISTED_MARKER and sync canonical durable fields onto live dicts after commit.Trr6r"rN)ziprrrr#float) batch_msgs batch_rowswrittenr9s rsync_flushed_message_markersr@OsJ 3; (,$% cggi(# .!$YGI  cggk*S%L 9#&{#3GK 3 &!$%9!:GI ;r)Optional)rrreturnbool) r'sqlite3.Connectionr(rr)List[Dict[str, Any]]r*Callable[[Any], Any]r+rFrBrE)r'rDr(rr8r#)r=rEr>rErBNone)__doc__ __future__rsqlite3typingrrrragent.context_compressorrrr2r$r@rArrrNsm#,,9 #,  ,   D* ; r