arms 0.0.3

🧬 The adaptive model routing system for exploration and exploitation.
Documentation 
1
2
3
4
5
6
7
8
9
10
11

import random

from amrs.config import ModelName
from amrs.router.router import Router

class RandomRouter(Router):
    def __init__(self, model_list: list[ModelName]):
        super().__init__(model_list)

    def sample(self, _: str) -> ModelName:
        return random.choice(self._model_list)