MockAI is a library that allows you to mock AI responses using custom commands suitable for simulating responses during testing without AI inference cost.

To install MockAI, use pip:

pip install mockai

To use MockAI, create a configuration file with your command and the respective full response path.

Example configuration file:

  {
    "/default": "full/path/to/mock_responses/default.json",
    "/mycommand": "full/path/to/mock_responses/mock_responses/my_command.json"
  }

To override the default format for all non-command inputs overwrite it by:

  {
    "_default": "full/path/to/_default.json"
  }

If it's not overwritten it returns the OpenAI JSON response structure.

After creating the configuration files import the mock_completion and set_config from mockai to :

from mockai import mock_completion, set_config

set_config('config.json')
response = mock_completion(model='your-model', messages=[{"role":"user", "content":"/mycommand"}])
print(response)  # output: "Retrieving the output defined in the mock_responses/my_command.json"

MockAI does not support streaming at the moment.

MockAI is licensed under the MIT License.