Dante (diverse-audiences-non-toxic-engagements) -- a prosocial algorithm

A submission to the Prosocial Ranking Challenge

Note — the main project files for each of the components are located in app. This directory structure is a relic of starting from the template code which we can fix at some point.

Helpful Docs — Poetry, Celery, Redis, JSONPath

Quick Reference

These variables are specified in docker-compose.yml (To be updated). Copied here for reference.

Component Locations

Component	Location
ranker	0.0.0.0:5001
postgres	postgres://postgres:postgres@database:5432
redis	redis://redis:6379
redis-celery-broker	redis://redis-celery-broker:6380

Environment Variables

Variable	Description	Docker default
DANTE_CACHE_DIR	Location for data / model cache	/app/cache/
DANTE_CONFIG_PATH	Path to configuration file	/app/config.ini
DANTE_LOG_DIR	Location for logs	`/app/logs/

Setup

1. Make sure you have docker and docker-compose installed.
2. Make sure you have celery, redis-py, and pytest installed.
3. Add models artifacts: Download all the contents of this Google Drive folder into your user cache dir (see on the platformdirs doc for where to find it on mac os, windows, and linux). There is no $version, while the $appname is dante. So for example on my Linux laptop the user cache dir is located at /home/<username>/.cache/dante/.
4. Add a config.ini to the root of the repository. This is meant to supply the configuration without needing to rebuild the image, which will be helpful when deploying in production. But for now you can copy this file from our example config: cp dante/osomerank/config.ini.sample config.ini
5. Start the docker daemon.
6. Run the component(s) in 2 ways:

• 6.1. Build and run all components with a single command: make run
• 6.2. Build and run a single component (use a bind mount to give it access to the models artifacts needed to run dante)

export DANTE_CACHE_DIR=/path/to/files # The path where you store the Google Drive contents
docker build -f docker/Dockerfile.ranker -t ranker .
docker run --mount type=bind,src=${DANTE_CACHE_DIR},dst=/app/cache ranker

Run test on the ranker

• Ensure you have all the dependencies installed by using poetry install from root directory
• poetry shell
• Once you have the container running, test the ranker using this command: python tests/latency.py

Clearing Space

Your personal machine might fill up when repeatedly building all these docker images and volumes. To clear space, run:

1. sudo docker system prune -a
2. sudo docker volume prune -a

Setting up development environment

When developing, an optional dependency group for development is provided in pyproject file. This can be installed with:

poetry install --with=dev