GPT - Connections

Solving NY Times Connections using GPTs...

Setup

Install dependencies on an environment ideally with pip

pip install -r requirements.txt

connections_prompts.jsonl - The final consolidated dataset
one_shot.py: A one-shot solution, one prompt, one solution
iterative.py: An iterative solution with feedback (3/4 or 4/4)
alpha.py: A flow engineer solver with planning, inspired by AlphaCodium paper
prepare_data.py: Downloads and preprocesses the dataset (it's already there, only run if you want to download the most recent puzzles)

To run on 10 samples:

python one_shot.py --num_samples 10

or

python iterative.py --num_samples 10

For the Alpha model, you can select N number of initial guess solutions to generate

python alpha.py --N 12 --num_samples 10

Note: Depending on your OpenAI plan, you may need to tweak the WEAVE_PARALLELISM environment variable to avoid rate limits. You can do this:

WEAVE_PARALLELISM=5 python alpha.py --num_samples 10