yj dZddlZddlZddlZddlmZddlmZddlm Z ddl m Z m Z m Z dedzfdZdddd ddd d ed e e zd edededzdedzdedzdedeedzdeeeefdzddfdZdS)zr Utility for logging traces from LLM calls. This module provides structured JSONL logging of LLM inputs/outputs. N)Path)Any) BaseModel)ConfiguredModelSettings ModelConfigsettingsreturncPtjrttjSdS)z'Get the traces file path from settings.N)rREASONING_TRACES_FILErF/DATA/AppData/hermes/projects/honcho/src/telemetry/reasoning_traces.pyget_reasoning_traces_file_pathrs#%4H2333 4r F) max_tokensthinking_budget_tokensreasoning_effort json_mode stop_seqsmessages task_type model_configpromptresponserrrrrrc t} | sdS|j} t| tr| } t j||j|j|||||dd|ji| |j |j |j dd} | | | dd<n || dd<t|d r|j r|j | d d <t| d 5} tj| tj| t)j| d ztj| tjddddS#1swxYwYdS)a Log a trace to the configured JSONL file. Args: task_type: Type of task (e.g., "minimal_deriver", "dialectic_chat") model_config: Model configuration used for the call prompt: The full prompt text sent to the LLM (used if messages is None) response: HonchoLLMCallResponse object with the LLM response max_tokens: Max output tokens setting thinking_budget_tokens: Anthropic thinking budget (if used) reasoning_effort: OpenAI reasoning effort (if used) json_mode: Whether JSON mode was enabled stop_seqs: Stop sequences used (if any) messages: Full conversation history for multi-turn/agentic calls N)rrrrrtokens)contentrfinish_reasonsthinking_content) timestamprprovidermodelrinputoutputr"rrtool_calls_mader# tool_callsa )rr isinstancer model_dumptime transportr! input_tokens output_tokensrrhasattrr$openfcntlflockfilenoLOCK_EXwritejsondumpsLOCK_UN)rrrrrrrrrr traces_filer trace_entryfs rlog_reasoning_tracer;s8122K G'9%%'$$&&Y[[ *#$&< 0""    h+ ,&5 ( 9   ##K0+3 GZ(()/ GX&x*++G0HG.6.F Hl+ k3  /1 AHHJJ ...  ;''$./// AHHJJ ...//////////////////sB E33E7:E7)__doc__r0r5r*pathlibrtypingrpydanticr src.configrrrrstrintboollistdictr;r r rrFs  t ")-#'"&,0J/J/J/J/ 77J/ J/ J/ d J/ $JJ/DjJ/J/Cy4J/4S>"T)J/ J/J/J/J/J/J/r