Skip to content

Added TimeWarp and a training data saving class for collecting TimeWarp data #330

Merged
amanjaiswal73892 merged 4 commits intoServiceNow:mainfrom
sparklabutah:main
Mar 17, 2026
Merged

Added TimeWarp and a training data saving class for collecting TimeWarp data #330
amanjaiswal73892 merged 4 commits intoServiceNow:mainfrom
sparklabutah:main

Conversation

@farhanishmam
Copy link
Contributor

Adding TimeWarp so that AgentLab can be used with BrowserGym
The training data saving class is needed to use TimeTraj, which does knowledge distillation on data generated by another agent

elif task_name.startswith("weblinx"):
import weblinx_browsergym
elif task_name.startswith("timewarp"):
import browsergym.timewarp
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It would be great if you can update the readme table to include timewarp.

"""
GenericAgent with training data saving functionality.

This module extends GenericAgent to save training data (system prompt, user prompt, and agent output)
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you for adding a new agent to AgentLab. I would keep this contribution as a separate PR from the Timewarp benchmark. It looks like we can extract step-wise (sys_prompt, user_prompt, and agent_output) from the saved traces using ExpResult. Feel free to write a utility function using that if needed. Currently, this change modifies loop.py (beyond adding timewarp), which we would prefer to avoid if possible.

@farhanishmam
Copy link
Contributor Author

farhanishmam commented Mar 17, 2026

Thanks again @amanjaiswal73892. Reverted to the original loop.py and updated the readme. Please lmk if there's anything else that I need to do on my end.

Copy link
Collaborator

@amanjaiswal73892 amanjaiswal73892 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@amanjaiswal73892 amanjaiswal73892 merged commit cbc35a9 into ServiceNow:main Mar 17, 2026
6 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants