diff --git a/python/tools/agbench/src/agbench/res/Dockerfile b/python/tools/agbench/src/agbench/res/Dockerfile index 34ba0d5df..4f4286a2e 100644 --- a/python/tools/agbench/src/agbench/res/Dockerfile +++ b/python/tools/agbench/src/agbench/res/Dockerfile @@ -1,7 +1,8 @@ FROM python:3.11 -MAINTAINER AutoGen +MAINTAINER AGNext # Install packages +# ffmpeg and exiftool are needed for mdconvert RUN apt-get update && apt-get install ffmpeg exiftool -y # Set the image to the Pacific Timezone @@ -10,36 +11,27 @@ RUN ln -snf /usr/share/zoneinfo/US/Pacific /etc/localtime && echo "US/Pacific" > # Upgrade pip RUN pip install --upgrade pip -# Pre-load autogen dependencies, but not autogen itself since we'll often want to install the latest from source -RUN pip install pyautogen[teachable,lmm,graphs,websurfer] -RUN pip uninstall --yes pyautogen +# Pre-load agnext dependencies, but not agnext itself since we'll often want to install the latest from source +RUN pip install openai pillow aiohttp typing-extensions pydantic types-aiofiles grpcio protobuf # Pre-load popular packages as per https://learnpython.com/blog/most-popular-python-packages/ -RUN pip install numpy pandas matplotlib seaborn scikit-learn requests urllib3 nltk pillow pytest +RUN pip install numpy pandas matplotlib seaborn scikit-learn requests urllib3 nltk pytest -# Pre-load packages needed for complex_task file utils -RUN pip install python-docx pdfminer.six requests pillow easyocr python-pptx SpeechRecognition pandas openpyxl pydub mammoth puremagic youtube_transcript_api==0.6.0 +# Pre-load packages needed for mdconvert file utils +RUN pip install python-docx pdfminer.six python-pptx SpeechRecognition openpyxl pydub mammoth puremagic youtube_transcript_api==0.6.0 +# easyocr -# Pre-load Selenium and Playwright -RUN pip install selenium playwright - -# Chromium playwright +# Pre-load Playwright +RUN pip install playwright RUN playwright install --with-deps chromium -RUN playwright install --with-deps firefox - -# Chrome for Selenium (need to run dpkg twice to resolve dependencies) -# RUN wget https://dl.google.com/linux/direct/google-chrome-stable_current_amd64.deb -# RUN dpkg -i google-chrome-stable_current_amd64.deb || : -# RUN apt -f install -y -# RUN dpkg -i google-chrome-stable_current_amd64.deb # Fix an incompatibility with numpy RUN pip uninstall --yes numpy RUN pip install "numpy<2.0" # Pre-load the OCR model -RUN /usr/bin/echo -e "import easyocr\nreader = easyocr.Reader(['en'])" | python +#RUN /usr/bin/echo -e "import easyocr\nreader = easyocr.Reader(['en'])" | python -# Webarena +# Webarena (evaluation code) RUN pip install beartype aiolimiter RUN /usr/bin/echo -e "import nltk\nnltk.download('punkt')" | python