i dZddlZddlZddlmZddlmZmZmZeje Z de de fdZ de de fdZ dd eee efd e d e d e fd ZdS)zTrajectory saving utilities and static helpers. _convert_to_trajectory_format stays as an AIAgent method (batch_runner.py calls agent._convert_to_trajectory_format). Only the static helpers and the file-write logic live here. N)datetime)AnyDictListcontentreturncf|rd|vr|S|ddddS)z4Convert tags to tags.zz)replacers 5/home/ubuntu/.hermes/hermes-agent/agent/trajectory.pyconvert_scratchpad_to_thinkrsA .g== ??3Y ? ? G GHacm n nnc|sdSd|vod|vS)zMCheck if content has an opening without a closing tag.Fr r r s rhas_incomplete_scratchpadrs& u #w . [3LT[3[[r trajectorymodel completedfilenamec||rdnd}|tj||d} t|dd5}|t j|d d zdddn #1swxYwYtd |dS#t$r&}t d |Yd}~dSd}~wwxYw) ayAppend a trajectory entry to a JSONL file. Args: trajectory: The ShareGPT-format conversation list. model: Model name for metadata. completed: Whether the conversation completed successfully. filename: Override output filename. Defaults to trajectory_samples.jsonl or failed_trajectories.jsonl based on ``completed``. Nztrajectory_samples.jsonlzfailed_trajectories.jsonl) conversations timestamprrazutf-8)encodingF) ensure_ascii zTrajectory saved to %szFailed to save trajectory: %s) rnow isoformatopenwritejsondumpsloggerinfo Exceptionwarning)rrrrentryfes rsave_trajectoryr,sJ1:[--@[$\^^--//   E; (C' 2 2 2 Ba GGDJu5999D@ A A A B B B B B B B B B B B B B B B ,h77777 ;;;6:::::::::;s;B'-A?3 B'?BB'BB'' C1CC)N)__doc__r#loggingrtypingrrr getLogger__name__r%strrboolrr,rrrr4s """"""""""  8 $ $oooooo\s\t\\\\6:;;T#s(^ 4;S;#;/2;;;;;;r