Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
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)
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),
|