Skip to content

Dhruva is an open-source platform for serving language AI models at scale.

License

Notifications You must be signed in to change notification settings

AI4Bharat/Dhruva-Platform

Repository files navigation

Dhruva

Table of Contents
  1. About The Project
  2. Getting Started
  3. Deployment
  4. Migrations
  5. Seed
  6. Contributing

About The Project

Dhruva is a full-fledged platform for serving AI models at scale.

(back to top)

Built With

  • Next
  • Chakra UI
  • FastApi
  • Celery

(back to top)

Getting Started

Prerequisites

Get the .env file. This can be created from the .env.example, you just need to fill the values.

(back to top)

Installation

Clone both Dhruva-Platform and Dhruva-Seed.

git clone --recurse-submodules https://github.com/AI4Bharat/Dhruva-Platform
git clone https://github.com/AI4Bharat/Dhruva-Seed

Place the .env file in the root directory in Dhruva Platform

(back to top)

Deployment

Development

  • Build the docker images for both the server and the client using the commands below.

    docker build -f Dockerfile -t server .
    
    docker build -f Dockerfile -t client .
    
  • Build the docker image for the seed repository

    docker build -t seed .
    
  • Run docker compose by using the docker-compose-db.yml, docker-compose-metering.yml and docker-compose-monitoring.yml as the compose files first.

    docker compose -f docker-compose-db.yml up -d
    
    docker compose -f docker-compose-metering.yml up -d
    
    docker compose -f docker-compose-monitoring.yml up -d
    
  • Finally run docker compose with the docker-compose-app.yml file as the compose file.

    docker compose -f docker-compose-app.yml up -d
    

You should now be able to view the client at http://localhost:{$FRONTEND_PORT}. The server will be accessible at http://localhost:{$BACKEND_PORT}, with $FRONTEND_PORT and $BACKEND_PORT being the values set in the .env file.

(back to top)

Migrations

We use mongodb-migrations to manage migrations. All migrations are stored in the migrations folder.

To create a new migration, run the following command:

mongodb-migrate-create --description <description>

This command will create a new migration <timestamp>_<description>.py in the migrations folder. A class Migration will be created for you. Implement the upgrade method, and optionally, the downgrade method.

Once implemented, you can run docker compose with docker-compose-app.yml as the compose file to run the migration.

After the migration is run, it will also add a new document in the migrations collection, in the database specified in the .env file, to ensure that each the migration only runs once.

Commit all migrations to the repository and do not delete any migration.

(back to top)

Seed

To add new models and services, please add a JSON fixture in the Dhruva-Seed repository.

(back to top)

Testing

To perform testing of the models hosted on Dhruva, please refer to the Dhruva-Evaluation-Suite repository.

  • Functional testing allows users to easily benchmark their models after setting up on Dhruva against any dataset available on HuggingFace, with any metrics such as WER (for ASR), BleU (for NMT) and others.
  • Users can also benchmark inference speeds of all endpoints and measure the Requests/sec rate using the Performance Testing tool

(back to top)

Contributing

  1. Clone the Project
  2. Create your Feature Branch (git checkout -b feature/AmazingFeature)
  3. Commit your Changes (git commit -m 'Add some AmazingFeature')
  4. Push to the Branch (git push origin feature/AmazingFeature)
  5. Open a Pull Request

(back to top)

About

Dhruva is an open-source platform for serving language AI models at scale.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published