Update agent_client.py