initial RL training tools and loop

This commit is contained in:
teknium1 2026-02-03 23:41:26 -08:00
parent 51a6b7d2b5
commit f018999da9
5 changed files with 1199 additions and 3 deletions

View file

@ -95,6 +95,23 @@ from .cronjob_tools import (
REMOVE_CRONJOB_SCHEMA
)
# RL Training tools (Tinker-Atropos)
from .rl_training_tool import (
rl_list_environments,
rl_select_environment,
rl_get_current_config,
rl_edit_config,
rl_start_training,
rl_check_status,
rl_stop_training,
rl_get_results,
rl_test_inference,
rl_list_runs,
rl_health_check,
check_rl_api_keys,
get_missing_keys,
)
__all__ = [
# Web tools
'web_search_tool',
@ -152,5 +169,19 @@ __all__ = [
'SCHEDULE_CRONJOB_SCHEMA',
'LIST_CRONJOBS_SCHEMA',
'REMOVE_CRONJOB_SCHEMA',
# RL Training tools
'rl_list_environments',
'rl_select_environment',
'rl_get_current_config',
'rl_edit_config',
'rl_start_training',
'rl_check_status',
'rl_stop_training',
'rl_get_results',
'rl_test_inference',
'rl_list_runs',
'rl_health_check',
'check_rl_api_keys',
'get_missing_keys',
]