import sys
from client import TraceEnvClient
from models import TraceAction

with TraceEnvClient(base_url="http://127.0.0.1:8000").sync() as client:
    res = client.reset()
    act = TraceAction(action_type="RETRIEVE", content="newer_than:90d", source="gmail", metadata={"max_results": 2})
    step_res = client.step(act)
    print(step_res.observation.metadata.keys())