fix(session-uploader): forward user_id + total_cost_usd to dataset row

#23
by GuillaumeSalouHF HF Staff - opened
smolagents org

Follow-up to discussion #22.

PR #22 added user_id + total_cost_usd to Session.get_trajectory(), but the upload path in agent/core/session_uploader.py builds session_row with hardcoded fields and silently drops anything extra. Result: rows in smolagents/ml-intern-sessions still show user_id=null even though the trajectory dict carries the values.

Confirmed by kubectl exec on the running pod: session.py is patched, but uploaded sessions still have null fields.

Change

Add user_id and total_cost_usd to the session_row dict using data.get() so older sessions (pre-#22) that lack them still upload cleanly with null.

Diff: 1 file, +2 lines.

GuillaumeSalouHF changed pull request status to open
GuillaumeSalouHF changed pull request status to merged

Sign up or log in to comment