FROM vincentwang25/self-reasoning-base

# Set up task
COPY agent.py /home/agent/agent.py
COPY config.json /home/agent/config.json

# sample random KV pairs
COPY wordlist.txt /wordlist.txt
COPY setup_task.sh /setup_task.sh
RUN chmod +x /setup_task.sh

RUN touch /key_values.txt
RUN touch /query_key.txt
RUN touch /home/agent/result.txt

# Give model agent permissions over everything in the home folder
RUN chown agent:agent -R /home/agent

ENV HOME=/home/agent
WORKDIR /home/agent
