ATRIA Models Manager

Introduction

atria-models-manager is a ATRIA library responsible for managing all the models that we can use in the atria-model-gw model. The objective is for the atria.model-gw to only have to import the models from this component, avoiding defining the logic of the models in the atria-model-gw. For the first version, it will only feature Perplexity’s Sonar model, in the future, we will be able to migrate the atria-model-gw models here.

It contains all the logic necessary to call an LLM. Prompt construction, query tokenization, etc. It also contains all the definitions of the exceptions that an LLM can throw.

How to use it

To use the atria-models-manager, you need to install it in your Python environment and import the model into the code.