fix the context parameter bugs

[imotai]

PR-Codex overview

This PR focuses on adding a new parameter to the arun method in the mock_agent.py file. It also includes changes to the worker method in the agent_server.py file.

Detailed summary

• Added context parameter to the arun method in mock_agent.py
• Updated worker method in agent_server.py to pass context parameter

✨ Ask PR-Codex anything about this PR by commenting with /codex {your question}