о а ты делал уже? надо глянуть

Да Я правда забросил коммиты пока, с работой никак не успеваю Вот так оно было у меня: https://github.com/Raumberg/myllm/blob/main/myllm/rewards/init.py

Ну и там можно еще имплементации ревардов глянуть разных

https://docs.nvidia.com/nemo/rl/latest/guides/environments.html

Дать юзеру окошко, в котором можно будет метод написать

У меня в либе можно было класс отзаимствовать И написать метод с расчетом реварда, что-то такое я делал, вроде Как вариант)