TrainingRunner
public class TrainingRunner
io/agentscope/core/training/runner/TrainingRunner.java:83–292Training 运行器,对智能体行为进行评测、追踪与可观测性分析。位于 io.agentscope.core.training.runner 包。核心成员包括 start、stop、commit、isRunning、getConfig、builder。
Training Runner - Training Process Controller
Responsible for:
-
Intercepting Agent requests and determining if they should be used for training
-
Replacing ChatClient to Trinity backend
-
Collecting trace data
-
Calculating rewards and calling feedback API
-
Periodically calling commit API to trigger training
Usage example:
`// Method 1: Using sampling rate strategy
TrainingRunner runner = TrainingRunner.builder()
.trinityEndpoint("http://localhost:8080")
.modelName("/path/to/model")
.selectionStrategy(SamplingRateStrategy.of(0.1)) // 10% sampling rate
.rewardCalculator(agent -> 0.0) // Implement custom reward logic
.commitIntervalSeconds(300) // Commit every 5 minutes
.build();
// Method 2: Using explicit marking strategy
TrainingRunner runner = TrainingRunner.builder()
.trinityEndpoint("http://localhost:8080")
.modelName("/path/to/model")
.selectionStrategy(ExplicitMarkingStrategy.create()) // User manually marks
.rewardCalculator(agent -> {
// Custom reward calculation logic
return 1.0;
})
.build();
// Method 3: Using custom RewardCalculator class
TrainingRunner runner = TrainingRunner.builder()
.trinityEndpoint("http://localhost:8080")
.modelName("/path/to/model")
.selectionStrategy(SamplingRateStrategy.of(0.1))
.rewardCalculator(new CustomRewardCalculator()) // Implement RewardCalculator interface
.build();
// Start training runner
runner.start();
// Users use Agent normally (transparent)
agent.call(msg).block();
// Stop training
runner.stop();
`
嵌套类摘要Nested Class Summary
| 类Class | 类型Kind | 说明Description |
|---|---|---|
Builder | 类class | TrainingRunner的构建器。 Builder for TrainingRunner |
方法摘要Method Summary
| 方法Method | 说明Description |
|---|---|
public synchronized void start()synchronized | 执行启动相关操作。 Start training runner |
public synchronized void stop()synchronized | 执行停止相关操作。 Stop training runner |
public Mono<Void> commit() | 返回提交。 Manually trigger commit |
public boolean isRunning() | 判断是否运行中。 Returns whether the running. |
public TrainingConfig getConfig() | 获取配置。 Returns the config. |
public static Builder builder()static | 返回构建器实例。 Returns the builder. |
方法详情Method Detail
start
public synchronized void start()
执行启动相关操作。
Start training runner
io/agentscope/core/training/runner/TrainingRunner.java:115–134stop
public synchronized void stop()
执行停止相关操作。
Stop training runner
io/agentscope/core/training/runner/TrainingRunner.java:139–169commit
public Mono<Void> commit()
返回提交。
Manually trigger commit
Mono<Void>io/agentscope/core/training/runner/TrainingRunner.java:174–180isRunning
public boolean isRunning()
判断是否运行中。
Returns whether the running.
booleanio/agentscope/core/training/runner/TrainingRunner.java:216–218getConfig
public TrainingConfig getConfig()
获取配置。
Returns the config.
TrainingConfigio/agentscope/core/training/runner/TrainingRunner.java:220–222builder
public static Builder builder()
返回构建器实例。
Returns the builder.
Builderio/agentscope/core/training/runner/TrainingRunner.java:224–226嵌套类详情Nested Class Detail
Builder
public static class BuilderTrainingRunner的构建器。位于 io.agentscope.core.training.runner 包。核心成员包括 config、trinityEndpoint、modelName、selectionStrategy、rewardCalculator、commitIntervalSeconds。
Builder for TrainingRunner