@@ -336,6 +336,16 @@ def main() -> None:
336336 RerankerTrainMetricsResponse ,
337337 RerankerTrainDiffRequest ,
338338 RerankerTrainDiffResponse ,
339+ # Domain models - Agent training (ragweld agent)
340+ AgentTrainStartRequest ,
341+ AgentTrainStartResponse ,
342+ AgentTrainRun ,
343+ AgentTrainRunMeta ,
344+ AgentTrainRunsResponse ,
345+ AgentTrainMetricEvent ,
346+ AgentTrainMetricsResponse ,
347+ AgentTrainDiffRequest ,
348+ AgentTrainDiffResponse ,
339349 # Domain models - Learning reranker legacy workflow
340350 RerankerLegacyStatus ,
341351 RerankerLegacyTaskResult ,
@@ -449,6 +459,15 @@ def main() -> None:
449459 RerankerTrainMetricsResponse ,
450460 RerankerTrainDiffRequest ,
451461 RerankerTrainDiffResponse ,
462+ AgentTrainStartRequest ,
463+ AgentTrainStartResponse ,
464+ AgentTrainRun ,
465+ AgentTrainRunMeta ,
466+ AgentTrainRunsResponse ,
467+ AgentTrainMetricEvent ,
468+ AgentTrainMetricsResponse ,
469+ AgentTrainDiffRequest ,
470+ AgentTrainDiffResponse ,
452471 RerankerLegacyStatus ,
453472 RerankerLegacyTaskResult ,
454473 RerankerMineResponse ,
0 commit comments