No project description provided
Project description
MockAI
MockAI is a library that allows you to mock AI responses using custom commands suitable for simulating responses during testing without AI inference cost.
Installation
To install MockAI, use pip:
pip install mockai
Usage
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"
Limitations
MockAI does not support streaming at the moment.
License
MockAI is licensed under the MIT License.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.