A simple, powerful, minimal codebase to generate synthetic data using OpenAI or MistralAI
pip install openai mistralai numpy
- Change
PROVIDER
underparams.py
mistral
oropenai
- Change
GPT_MODEL
,OPENAI_API_KEY
andOUTPUT_FILE_PATH
underparams.py
- Change
MISTRALAI_MODEL
,MISTRALAI_API_KEY
andOUTPUT_FILE_PATH
underparams.py
- Run with
python main.py
- Change the topics in
topics.py
- Change the system contexts in
system_messages.py
- Change the number of workers in
params.py