GuillaumeSalouHF HF Staff commited on
Commit
9f90f1f
·
1 Parent(s): 384bc5e

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

Browse files

- fix(session-uploader): forward user_id and total_cost_usd to dataset row (8d4a07e1f801865d8243b84f55fb16aeafe77ba1)

Files changed (1) hide show
  1. agent/core/session_uploader.py +2 -0
agent/core/session_uploader.py CHANGED
@@ -90,9 +90,11 @@ def upload_session_as_file(
90
  # across sessions with different tool rosters.
91
  session_row = {
92
  "session_id": data["session_id"],
 
93
  "session_start_time": data["session_start_time"],
94
  "session_end_time": data["session_end_time"],
95
  "model_name": data["model_name"],
 
96
  "messages": json.dumps(scrubbed_messages),
97
  "events": json.dumps(scrubbed_events),
98
  "tools": json.dumps(scrubbed_tools),
 
90
  # across sessions with different tool rosters.
91
  session_row = {
92
  "session_id": data["session_id"],
93
+ "user_id": data.get("user_id"),
94
  "session_start_time": data["session_start_time"],
95
  "session_end_time": data["session_end_time"],
96
  "model_name": data["model_name"],
97
+ "total_cost_usd": data.get("total_cost_usd"),
98
  "messages": json.dumps(scrubbed_messages),
99
  "events": json.dumps(scrubbed_events),
100
  "tools": json.dumps(scrubbed_tools),