- Create a new virtual environment
python -m venv venv
- Activate the virtual environment:
source venv/bin/activate
- Install the dependencies:
pip install -r requirements.txt
- If you want to run the sql tests, create the env file by copying the sample file:
cp .env.sample .env
Then update the new .env file with your Snowflake credentials.
- Navigate to the
mitofolder. - To run the
chattests, run the command:
python -m evals.main --test_type=chat
- To run the
inline_code_completiontests, run the command:
python -m evals.main --test_type=inline_code_completion
- To run the
smart-debuggertests, run the command:
python -m evals.main --test_type=smart_debug
- To run the
sqltests, run the command:
python -m evals.main --test_type=sql
To specify which tests to run, set some of the following flags:
--test_type--test--prompt--tags--model
For example, to run all tests for the single_shot_prompt prompt, run:
python -m evals.main --test_type=chat --prompt=single_shot_prompt