`gj8jdZddlmZddlZddlZddlZddlZddlmZmZddl m Z m Z m Z m Z mZejeZdZdZdZGd d eZdd Zdd Zdd ZddZdddd ddZddZdZedd d dZ d! d"dZddddeddd d#dZ y)$uUpload a Hermes session transcript to Hugging Face as an agent trace. Hermes stores sessions in its own SQLite store (``hermes_state.SessionDB``), so we reconstruct the conversation and emit it in the **Claude Code JSONL** shape — one of the three formats the Hugging Face Agent Trace Viewer auto-detects (Claude Code / Codex / Pi). No dataset-side preprocessing is needed; the Hub tags the dataset ``agent-traces`` and opens it in the viewer. Docs: https://huggingface.co/docs/hub/agent-traces Design notes ------------ * **Zero LLM turn.** This is a deterministic export — it never spends a model call. The ``hermes trace upload`` subcommand calls :func:`upload_session_trace` directly. * **Private by default.** Traces can contain prompts, tool output, local paths, and secrets. The dataset is created private and every text body is passed through Hermes' secret redactor (``force=True``) unless the caller explicitly opts out with ``redact=False``. * **Never raises.** Returns a user-facing status string so command handlers can echo it straight back to the user. Programmatic callers that need the URL can use :func:`build_trace_jsonl` + :func:`_do_upload` directly. ) annotationsN)datetimetimezone)AnyDictListOptionalTuplez hermes-tracesz hermes-agentzTrace upload blocked: secret redaction failed, so the transcript may still contain credentials or other sensitive data. Fix the redactor or rerun with --no-redact only after manually reviewing the transcript.ceZdZdZy)TraceRedactionErrorzinput)rr0r1r2r stripr4loadsJSONDecodeError ValueErrorr)r5r#r$r r%r3uuiduuid4hex) tool_callsr7r8tcfnr>raw_argsparseds r_tool_calls_to_blocksrSds#%F j$ '  "d#  VVJ  %2vvf~99666+& h $ ,191AH-r$ 'FF  FGDJJv,>$GH  &&,BF4::<+;+;CR+@*A"B   + 6 M%((*5 , (+ ,((*5 Fij)*DEE Fs 'E  2E0 E-,E-0>F.r-T)modelcwdr7cXg}dtd ddl}rA|jgdddd}|jdk(r|jj dfd }|D]} | jd } | d k(rttj} | d k(rt| jd |} | jt| jd || sdddg} || } d | d<d |xsd| d| d<|jtj | d| | dk(r| jdxs| jdxsd}t#t%| jd tr| jd n#tj | jd |}|| } d| d<dd||dgd| d<|jtj | d| | jd }t%|tr t#||}n t||}|| } d| d<d|d| d<|jtj | d| dj'||rdzSdzS#t $rdYJwxYw)uTRender Hermes conversation messages as Claude Code JSONL text. Each non-system message becomes one JSONL line in the Claude Code transcript shape the HF Agent Trace Viewer auto-detects: * ``user`` / ``tool`` -> ``{"type": "user", "message": {...}}`` * ``assistant`` -> ``{"type": "assistant", "message": {...}}`` with ``content`` blocks (text + ``tool_use``). Tool results are emitted as user turns carrying a ``tool_result`` block keyed by ``tool_call_id`` — the same way Claude Code records them. Turns are linked via ``uuid`` / ``parentUuid``. Nr-r)gitz rev-parsez --abbrev-refHEADT)capture_outputr&timeoutrUc Pddxstjt|d S)NFexternal) parentUuid isSidechainuserTyperU sessionIdversion gitBranchrK timestamp)osgetcwd_HERMES_VERSION) turn_uuidbase_tsrU git_branchparent session_ids r_commonz"build_trace_jsonl.._commons3  "%"))+#&#   rrolesystem assistantr6rNr&r+r,unknown)rnrTr6messageF) ensure_asciir? tool_call_id tool_nameuser tool_result)r, tool_use_idr6)rnr6 )rhr returnzDict[str, Any])r subprocessrun returncodestdoutrGr"r2r rKrLr;extendrSr3r4r5r)rjoin)messagesrlrTrUr7linesr{rrmmsgrnrhr8entryrxresult_contentr6message_contentrirjrks ` ` @@@rbuild_trace_jsonlrs*E FjGJ  <#$sA||q XX^^-    5wwv 8   % ; ' (:FCF MM/ 0EvN O#)267I&E'E&M#+)! E)  LLE> ?F  6>''.1SSWW[5ISVK$&01CS&I "ZZ 23N I&E"E&M)#.- E)  LLE> ?F '')$ gs ##*7F#;O0&AO "f $*Gi TZZE:;k5n 99U ut 55" 55O  sAJ J)(J)cdD]<}tj|}|s|js,|jcSy)z=Return the user's Hugging Face token from the usual env vars.)HF_TOKENHUGGINGFACE_HUB_TOKENHUGGING_FACE_HUB_TOKENHUGGINGFACE_TOKENN)regetenvrG)varvals r_resolve_hf_tokenrs:ciin 399;99;  ru]Can't upload — no Hugging Face token is available. To set it up: 1. Create a token with WRITE access at https://huggingface.co/settings/tokens (New token -> type "Write" -> copy it). 2. Add it to your environment as HF_TOKEN (e.g. in ~/.hermes/.env): HF_TOKEN=hf_xxxxxxxxxxxxxxxxxxxx 3. Run /upload-trace again (or `hermes trace upload`).) dataset_nameprivatec ddlm}|jdd ddlm}||} |j}t|tr|jd nd } | sy | d|} |j| d|dd|d} |j|jd| | dd|d| d| d| S#t$rYwxYw#t $rYywxYw#t$r } tjd | Yd } ~ y d } ~ wwxYw#t$r)} tjd| | d| d| cYd } ~ Sd } ~ wwxYw#t$r&} tjd| | d| cYd } ~ Sd } ~ wwxYw)zCreate (idempotently) the private dataset and push the trace file. Returns a user-facing status string. Never raises. r) lazy_depsztool.trace_uploadF)prompt)HfApizXHugging Face upload needs the `huggingface_hub` package (`pip install huggingface_hub`).)tokenr>NzHF whoami failed: %szfYour Hugging Face token was rejected (whoami failed). Make sure it has WRITE access and isn't expired.z https://huggingface.co/datasets/z /blob/main/z; View in the trace viewer: https://huggingface.co/datasets/)toolsrensurer"huggingface_hubr ImportErrorwhoamirr1r2r#r$ create_repo upload_fileencode) jsonlrrlrrrrapiwhorverrs r _do_uploadr s #,U; 4) e CDjjl",S$"7swwvT Ma ~&GA y'D  zl&1L 5 !LL1%/ |<  ;7);|n]IIP  STW    434 D-q1CD A97AF1'"QC@@A 597AF0445suB9C3C*D(D89 CC CC D C;;D D5 D0*D50D58 E'E"E'"E'cddlm}|r ||n|}|j|xs|}|j|xsi}|j |}||fS)zLoad a session's conversation + metadata from the SQLite store. Returns ``(messages, meta)``. ``meta`` is ``{}`` when the session row is missing (messages may still be present for a live, untitled session). r) SessionDBdb_path) hermes_staterresolve_session_id get_sessionget_messages_as_conversation)rlrrdbresolvedmetars rload_session_messagesrIsY''.7 #IKB$$Z0>JH >>( # )rD..x8H T>r)rTrUr7rrrrc|sy|xs t}|stS t||\}} |sy|xs| j dxsd } t||| || } | jsy t| |||| S#t$r)} tj d|| d|d| cYd} ~ Sd} ~ wwxYw#t$r tcYSwxYw) zTop-level entry point used by the CLI/gateway/subcommand. Loads the session, converts it to Claude Code JSONL, and uploads it to the user's private ``{user}/hermes-traces`` dataset. Returns a user-facing status string and never raises. zNo active session to upload.rz.Failed to load session %s for trace upload: %szCould not load session rNz-No transcript to upload for this session yet.rTr-)rlrTrUr7z1No transcript content to upload for this session.)rrlrr) r_NO_TOKEN_MESSAGErr"r#r$r2rr r%rGr) rlrTrUr7rrrrrrrresolved_modelrs rupload_session_tracerYs" -  (&(E   ;.z7K$ >5dhhw/52N *! !    ;;=B  !  + ;GUVW( Bqc::; *))*s/A9B.9 B+B& B+&B+.C?C)rzr )r&rr'boolrzr)r6rr7rrzList[Dict[str, Any]])rNrr7rrzr) rrrlr rTr rUr r7rrzr )rz Optional[str]) rr rr rlr rr rrrzr )N)rlr rzz+Tuple[List[Dict[str, Any]], Dict[str, Any]])rlr rTr rUr r7rrrrr rrrzr )!r __future__rr4loggingrerKrrtypingrrrr r getLoggerr r#DEFAULT_DATASET_NAMErgr% RuntimeErrorr rr)r;rSrrrrrrrrrrs2#  '33   8 $& KG,GRG"L2 N i6"i6i6  i6 i6  i6 i6`= - ;T ;T ;T ;T  ;T  ;T ;T~"  0 &, 55 5 5  5  55 5 5r