import requests

res = requests.post("http://127.0.0.1:8000/reset", json={
    "task": {
        "instruction": "test",
        "difficulty": "easy",
        "available_sources": ["gmail"],
        "ground_truth": {}
    }
})

epid = res.json()['observation']['episode_id']

res2 = requests.post("http://127.0.0.1:8000/step", json={
    "action": {
        "action_type": "RETRIEVE",
        "content": "newer_than:90d",
        "source": "gmail",
        "metadata": {"max_results": 2}
    },
    "episode_id": epid
})
print("STEP:", list(res2.json()['observation'].keys()))
print("metadata contents:", res2.json()['observation'].get('metadata'))