From 85820f4401f030165b28c2a7d1af9fd2652a5070 Mon Sep 17 00:00:00 2001
From: Olivier Chafik <ochafik@google.com>
Date: Tue, 9 Apr 2024 21:03:32 +0100
Subject: [PATCH] agent: fix sandbox dockerfile

---
 examples/agent/run_sandboxed_tools.sh | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/examples/agent/run_sandboxed_tools.sh b/examples/agent/run_sandboxed_tools.sh
index 8eddb5d92..5dbe19246 100755
--- a/examples/agent/run_sandboxed_tools.sh
+++ b/examples/agent/run_sandboxed_tools.sh
@@ -52,14 +52,14 @@ echo "
     RUN      mkdir /src /data
 
     # Copy resources in increasing likelihood of change, to keep as much as possible cached
-    COPY     fastify-requirements.txt /root
+    COPY     fastify-requirements.txt /root/
     RUN      pip install -r /root/fastify-requirements.txt
-    COPY     script-requirements.txt  /root
+    COPY     script-requirements.txt  /root/
     RUN      pip install -r /root/script-requirements.txt
-    COPY     fastify.py utils.py      /root
+    COPY     fastify.py utils.py      /root/examples/agent/
 
     WORKDIR  /data
-    ENTRYPOINT PYTHONPATH=/src python /root/fastify.py --port=$PORT '/src/$( basename "$script" )'
+    ENTRYPOINT PYTHONPATH=/src:/root python -m examples.agent.fastify --port=$PORT '/src/$( basename "$script" )'
 " | docker build "$BUILD_DIR" -f - -t "$LLAMA_IMAGE_NAME"
 
 echo "#"