stagehand x mcp-use/app.py  No module named 'agent_wrapper'

I am unable to start this project according to the tutorial. The error message is as follows:

from agent_wrapper import create_streaming_agent
ModuleNotFoundError: No module named 'agent_wrapper'

How can I import this package? Thanks