A nifty sandbox for experimenting with robots in MuJoCo using text prompts
- Install the required packages:
$ pip install -r requirements.txt
-
Set up your OpenAI API key in
llm/prompt_handler.py
. -
Run the simulation:
$ python main.py
Enter natural language commands to control the robot. For example:
- "Move forward"
- "Turn left"
- "Go to the right"
Type 'quit' to end the simulation.
environments/
: Contains the MuJoCo environment setupmodels/
: Contains the XML model file for the robotcontrollers/
: Implements the robot control logicllm/
: Handles natural language processing and LLM integrationutils/
: Contains helper functionsmain.py
: The main script to run the simulation
Modify the XML model, environment parameters, and control logic to suit your specific robot and task requirements.